Optimize test_simd (#1096)

Signed-off-by: CaiYudong <[email protected]>
zilliztech · Feb 24, 2025 · b9aa5a7 · b9aa5a7
1 parent d9651b6
commit b9aa5a7
Show file tree

Hide file tree

Showing 2 changed files with 224 additions and 169 deletions.
diff --git a/src/simd/hook.cc b/src/simd/hook.cc
@@ -188,7 +188,7 @@ fvec_hook(std::string& simd_type) {
         fvec_Linf = fvec_Linf_avx512;
 
         fvec_norm_L2sqr = fvec_norm_L2sqr_avx512;
-        fvec_L2sqr_ny = fvec_L2sqr_ny_sse;
+        fvec_L2sqr_ny = fvec_L2sqr_ny_avx;
         fvec_inner_products_ny = fvec_inner_products_ny_sse;
         fvec_madd = fvec_madd_avx512;
         fvec_madd_and_argmin = fvec_madd_and_argmin_sse;
@@ -227,7 +227,7 @@ fvec_hook(std::string& simd_type) {
         fvec_Linf = fvec_Linf_avx;
 
         fvec_norm_L2sqr = fvec_norm_L2sqr_avx;
-        fvec_L2sqr_ny = fvec_L2sqr_ny_sse;
+        fvec_L2sqr_ny = fvec_L2sqr_ny_avx;
         fvec_inner_products_ny = fvec_inner_products_ny_sse;
         fvec_madd = fvec_madd_avx;
         fvec_madd_and_argmin = fvec_madd_and_argmin_sse;