Skip to content

Commit 71ab2e5

Browse files
wx257osn2facebook-github-bot
authored andcommitted
Fix simdlib (#2448)
Summary: - fix type of an argument of the `simd8uint32` constructor on `simdlib_emulated` - fix `blendv` on aarch64, which returns completely inverted results - remove unused `#include <faiss/utils/simdlib.h>` Pull Request resolved: #2448 Reviewed By: alexanderguzhva Differential Revision: D39259465 Pulled By: mdouze fbshipit-source-id: 646ee490269f9d71ff48a22e33c1250e115c02c2
1 parent 8f5adc8 commit 71ab2e5

File tree

4 files changed

+3
-5
lines changed

4 files changed

+3
-5
lines changed

faiss/IndexIVFFastScan.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@
2626
#include <faiss/utils/distances.h>
2727
#include <faiss/utils/hamming.h>
2828
#include <faiss/utils/quantize_lut.h>
29-
#include <faiss/utils/simdlib.h>
3029
#include <faiss/utils/utils.h>
3130

3231
namespace faiss {

faiss/impl/ResidualQuantizer.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
#include <faiss/utils/Heap.h>
2222
#include <faiss/utils/distances.h>
2323
#include <faiss/utils/hamming.h>
24-
#include <faiss/utils/simdlib.h>
2524
#include <faiss/utils/utils.h>
2625

2726
extern "C" {

faiss/utils/simdlib_emulated.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,7 @@ struct simd8uint32 : simd256bit {
438438

439439
explicit simd8uint32(const simd256bit& x) : simd256bit(x) {}
440440

441-
explicit simd8uint32(const uint8_t* x) : simd256bit((const void*)x) {}
441+
explicit simd8uint32(const uint32_t* x) : simd256bit((const void*)x) {}
442442

443443
std::string elements_to_string(const char* fmt) const {
444444
char res[1000], *ptr = res;

faiss/utils/simdlib_neon.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -646,8 +646,8 @@ inline simd32uint8 blendv(
646646
const uint8x16x2_t msb_mask = {
647647
vtstq_u8(mask.data.val[0], msb), vtstq_u8(mask.data.val[1], msb)};
648648
const uint8x16x2_t selected = {
649-
vbslq_u8(msb_mask.val[0], a.data.val[0], b.data.val[0]),
650-
vbslq_u8(msb_mask.val[1], a.data.val[1], b.data.val[1])};
649+
vbslq_u8(msb_mask.val[0], b.data.val[0], a.data.val[0]),
650+
vbslq_u8(msb_mask.val[1], b.data.val[1], a.data.val[1])};
651651
return simd32uint8{selected};
652652
}
653653

0 commit comments

Comments
 (0)