Skip to content

Commit b76b94e

Browse files
authored
Merge pull request #6 from VectorDB-NTU/rota-debug
using complete flip data in FHT rotator
2 parents f6fcf46 + 85d5aa8 commit b76b94e

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

rabitqlib/utils/rotator.hpp

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -244,15 +244,19 @@ class FhtKacRotator : public Rotator<float> {
244244
fht_float_(rotated_vec);
245245
vec_rescale(rotated_vec, trunc_dim_, fac_);
246246

247-
flip_sign(flip_.data(), rotated_vec, padded_dim_);
247+
flip_sign(flip_.data() + (padded_dim_ / kByteLen), rotated_vec, padded_dim_);
248248
fht_float_(rotated_vec);
249249
vec_rescale(rotated_vec, trunc_dim_, fac_);
250250

251-
flip_sign(flip_.data(), rotated_vec, padded_dim_);
251+
flip_sign(
252+
flip_.data() + (2 * padded_dim_ / kByteLen), rotated_vec, padded_dim_
253+
);
252254
fht_float_(rotated_vec);
253255
vec_rescale(rotated_vec, trunc_dim_, fac_);
254256

255-
flip_sign(flip_.data(), rotated_vec, padded_dim_);
257+
flip_sign(
258+
flip_.data() + (3 * padded_dim_ / kByteLen), rotated_vec, padded_dim_
259+
);
256260
fht_float_(rotated_vec);
257261
vec_rescale(rotated_vec, trunc_dim_, fac_);
258262

0 commit comments

Comments
 (0)