default to avx in inner product (#18)

Co-authored-by: blaise-muhirwa <[email protected]>
BlaiseMuhirwa · Dec 6, 2023 · 2e9f808 · 2e9f808
1 parent 9b75e70
commit 2e9f808
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flatnav/distances/InnerProductDistance.h b/flatnav/distances/InnerProductDistance.h
@@ -92,7 +92,11 @@ class InnerProductDistance : public DistanceInterface<InnerProductDistance> {
 #endif
     if (!(_dimension % 16 == 0)) {
       if (_dimension % 4 == 0) {
+#if defined(USE_AVX)
+        _distance_computer = distanceImplInnerProductSIMD4ExtAVX;
+#else
         _distance_computer = distanceImplInnerProductSIMD4ExtSSE;
+#endif
       } else if (_dimension > 16) {
         _distance_computer = distanceImplInnerProductSIMD16ExtResiduals;
       } else if (_dimension > 4) {