Skip to content

Commit b11e6b6

Browse files
committed
Refactor to use the SIMD dynamic dispatch.
Signed-off-by: Mulugeta Mammo <[email protected]>
1 parent 60769ab commit b11e6b6

4 files changed

Lines changed: 349 additions & 417 deletions

File tree

faiss/CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ set(FAISS_SIMD_AVX2_SRC
1212
utils/simd_impl/distances_avx2.cpp
1313
)
1414
set(FAISS_SIMD_AVX512_SRC
15+
impl/PolysemousTraining_avx512.cpp
1516
utils/simd_impl/distances_avx512.cpp
1617
)
1718
set(FAISS_SIMD_NEON_SRC
@@ -92,6 +93,7 @@ set(FAISS_SRC
9293
impl/HNSW.cpp
9394
impl/NSG.cpp
9495
impl/PolysemousTraining.cpp
96+
impl/PolysemousTraining_avx512.cpp
9597
impl/ProductQuantizer.cpp
9698
impl/AdditiveQuantizer.cpp
9799
impl/RaBitQuantizer.cpp
@@ -226,6 +228,7 @@ set(FAISS_HEADERS
226228
impl/Panorama.h
227229
impl/PanoramaStats.h
228230
impl/PolysemousTraining.h
231+
impl/PolysemousTraining_avx512.h
229232
impl/ProductQuantizer-inl.h
230233
impl/ProductQuantizer.h
231234
impl/Quantizer.h

0 commit comments

Comments
 (0)