Improve: Turin static dispatch

ashvardanian · Oct 29, 2024 · b855580 · b855580
1 parent c0977d8
commit b855580
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/include/simsimd/simsimd.h b/include/simsimd/simsimd.h
@@ -2027,6 +2027,8 @@ SIMSIMD_PUBLIC void simsimd_intersect_u16(simsimd_u16_t const *a, simsimd_u16_t
     simsimd_intersect_u16_sve2(a, b, a_length, b_length, d);
 #elif SIMSIMD_TARGET_NEON
     simsimd_intersect_u16_neon(a, b, a_length, b_length, d);
+#elif SIMSIMD_TARGET_TURIN
+    simsimd_intersect_u16_turin(a, b, a_length, b_length, d);
 #elif SIMSIMD_TARGET_ICE
     simsimd_intersect_u16_ice(a, b, a_length, b_length, d);
 #else
@@ -2040,6 +2042,8 @@ SIMSIMD_PUBLIC void simsimd_intersect_u32(simsimd_u32_t const *a, simsimd_u32_t
     simsimd_intersect_u32_sve2(a, b, a_length, b_length, d);
 #elif SIMSIMD_TARGET_NEON
     simsimd_intersect_u32_neon(a, b, a_length, b_length, d);
+#elif SIMSIMD_TARGET_TURIN
+    simsimd_intersect_u32_turin(a, b, a_length, b_length, d);
 #elif SIMSIMD_TARGET_ICE
     simsimd_intersect_u32_ice(a, b, a_length, b_length, d);
 #else

diff --git a/scripts/bench.cxx b/scripts/bench.cxx
@@ -965,6 +965,11 @@ int main(int argc, char **argv) {
     sparse_<u32_k>("intersect_u32_ice", simsimd_intersect_u32_ice, simsimd_intersect_u32_accurate);
 #endif
 
+#if SIMSIMD_TARGET_TURIN
+    sparse_<u16_k>("intersect_u16_turin", simsimd_intersect_u16_turin, simsimd_intersect_u16_accurate);
+    sparse_<u32_k>("intersect_u32_turin", simsimd_intersect_u32_turin, simsimd_intersect_u32_accurate);
+#endif
+
 #if SIMSIMD_TARGET_SKYLAKE
     dense_<f32_k>("dot_f32_skylake", simsimd_dot_f32_skylake, simsimd_dot_f32_accurate);
     dense_<f32_k>("cos_f32_skylake", simsimd_cos_f32_skylake, simsimd_cos_f32_accurate);