Fix: Require bf16 for spdot_weights

ashvardanian · Oct 29, 2024 · 3e868ee · 3e868ee
1 parent dc8111e
commit 3e868ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -148,6 +148,7 @@
         "simd",
         "simsimd",
         "Skylake",
+        "spdot",
         "sqeuclidean",
         "SVMs",
         "tanimoto",

diff --git a/include/simsimd/simsimd.h b/include/simsimd/simsimd.h
@@ -1091,7 +1091,9 @@ SIMSIMD_INTERNAL void _simsimd_find_metric_punned_u16(simsimd_capability_t v, si
     if (v & simsimd_cap_sve2_k) switch (k) {
         case simsimd_metric_intersect_k: *m = (m_t)&simsimd_intersect_u16_sve2, *c = simsimd_cap_sve2_k; return;
         case simsimd_metric_spdot_counts_k: *m = (m_t)&simsimd_spdot_counts_u16_sve2, *c = simsimd_cap_sve2_k; return;
+#if SIMSIMD_TARGET_SVE_BF16 //! We also need `bf16` support for weights
         case simsimd_metric_spdot_weights_k: *m = (m_t)&simsimd_spdot_weights_u16_sve2, *c = simsimd_cap_sve2_k; return;
+#endif
         default: break;
         }
 #endif