usearch 0.18.3 → 0.18.4

package/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "usearch"
-version = "0.18.3"
+version = "0.18.4"
 authors = ["Ashot Vardanian <1983160+ashvardanian@users.noreply.github.com>"]
 description = "Smaller & Faster Single-File Vector Search Engine from Unum"
 edition = "2021"

package/conanfile.py

@@ -4,7 +4,7 @@ from conans import ConanFile
 class USearchConan(ConanFile):
 
     name = 'USearch'
-    version = '0.18.3'
+    version = '0.18.4'
     license = 'Apache 2.0'
     url = 'https://github.com/unum-cloud/usearch'
     description = 'Smaller & Faster Single-File Vector Search Engine from Unum'

package/include/usearch/index.hpp

@@ -269,7 +269,7 @@ template <typename scalar_at = float, typename result_at = scalar_at> struct cos
             ab += result_t(a[i]) * result_t(b[i]), //
                 a2 += square<result_t>(a[i]),      //
                 b2 += square<result_t>(b[i]);
-        return ab ? (1 - ab / (std::sqrt(a2) * std::sqrt(b2))) : 0;
+        return (ab != 0) ? (1 - ab / (std::sqrt(a2) * std::sqrt(b2))) : 0;
     }
 };
 

package/include/usearch/index_punned_dense.hpp

@@ -36,7 +36,7 @@ struct cos_f8_t {
             a2 += square(ai);
             b2 += square(bi);
         }
-        return 1.f - ab / (std::sqrt(a2) * std::sqrt(b2));
+        return (ab != 0) ? (1.f - ab / (std::sqrt(a2) * std::sqrt(b2))) : 0;
     }
 };
 
@@ -505,30 +505,27 @@ class index_punned_dense_gt {
     static metric_and_meta_t ip_metric_f32_(std::size_t dimensions) {
         (void)dimensions;
 #if USEARCH_USE_SIMSIMD
-#if defined(USEARCH_DEFINED_X86)
-        if (dimensions % 4 == 0)
+        if (hardware_supports(isa_t::avx2_k) && dimensions % 4 == 0)
             return {
                 pun_metric_<simsimd_f32_t>([=](simsimd_f32_t const* a, simsimd_f32_t const* b) {
                     return 1.f - simsimd_dot_f32x4avx2(a, b, dimensions);
                 }),
                 isa_t::avx2_k,
             };
-#elif defined(USEARCH_DEFINED_ARM)
-        if (supports_arm_sve())
+        if (hardware_supports(isa_t::sve_k))
             return {
                 pun_metric_<simsimd_f32_t>([=](simsimd_f32_t const* a, simsimd_f32_t const* b) {
                     return 1.f - simsimd_dot_f32sve(a, b, dimensions);
                 }),
                 isa_t::sve_k,
             };
-        if (dimensions % 4 == 0)
+        if (hardware_supports(isa_t::neon_k) && dimensions % 4 == 0)
             return {
                 pun_metric_<simsimd_f32_t>([=](simsimd_f32_t const* a, simsimd_f32_t const* b) {
                     return 1.f - simsimd_dot_f32x4neon(a, b, dimensions);
                 }),
                 isa_t::neon_k,
             };
-#endif
 #endif
         return pun_metric_<f32_t>(ip_gt<f32_t>{});
     }
@@ -536,23 +533,20 @@ class index_punned_dense_gt {
     static metric_and_meta_t cos_metric_f16_(std::size_t dimensions) {
         (void)dimensions;
 #if USEARCH_USE_SIMSIMD
-#if defined(USEARCH_DEFINED_X86)
-        if (dimensions % 16 == 0)
+        if (hardware_supports(isa_t::avx512_k) && dimensions % 16 == 0)
             return {
                 pun_metric_<simsimd_f16_t>([=](simsimd_f16_t const* a, simsimd_f16_t const* b) {
                     return 1.f - simsimd_cos_f16x16avx512(a, b, dimensions);
                 }),
                 isa_t::avx512_k,
             };
-#elif defined(USEARCH_DEFINED_ARM)
-        if (dimensions % 4 == 0)
+        if (hardware_supports(isa_t::neon_k) && dimensions % 4 == 0)
             return {
                 pun_metric_<simsimd_f16_t>([=](simsimd_f16_t const* a, simsimd_f16_t const* b) {
                     return 1.f - simsimd_cos_f16x4neon(a, b, dimensions);
                 }),
                 isa_t::neon_k,
             };
-#endif
 #endif
         return pun_metric_<f16_t>(cos_gt<f16_t, f32_t>{});
     }
@@ -560,14 +554,12 @@ class index_punned_dense_gt {
     static metric_and_meta_t cos_metric_f8_(std::size_t dimensions) {
         (void)dimensions;
 #if USEARCH_USE_SIMSIMD
-#if defined(USEARCH_DEFINED_ARM)
-        if (dimensions % 16 == 0)
+        if (hardware_supports(isa_t::neon_k) && dimensions % 16 == 0)
             return {
                 pun_metric_<int8_t>(
                     [=](int8_t const* a, int8_t const* b) { return 1.f - simsimd_cos_i8x16neon(a, b, dimensions); }),
                 isa_t::neon_k,
             };
-#endif
 #endif
         return pun_metric_<f8_bits_t>(cos_f8_t{dimensions});
     }

package/include/usearch/index_punned_helpers.hpp

@@ -76,13 +76,23 @@ inline char const* isa_name(isa_t isa) noexcept {
     }
 }
 
-inline bool supports_arm_sve() noexcept {
-#if defined(USEARCH_DEFINED_ARM)
-#if defined(USEARCH_DEFINED_LINUX)
+inline bool hardware_supports(isa_t isa) noexcept {
+#if defined(USEARCH_DEFINED_ARM) && defined(USEARCH_DEFINED_LINUX)
     unsigned long capabilities = getauxval(AT_HWCAP);
-    if (capabilities & HWCAP_SVE)
-        return true;
+    switch (isa) {
+    case isa_t::neon_k: return capabilities & HWCAP_NEON;
+    case isa_t::sve_k: return capabilities & HWCAP_SVE;
+    default: return false;
+    }
 #endif
+
+#if defined(USEARCH_DEFINED_X86) && defined(USEARCH_DEFINED_GCC)
+    __builtin_cpu_init();
+    switch (isa) {
+    case isa_t::avx2_k: return __builtin_cpu_supports("avx2");
+    case isa_t::avx512_k: return __builtin_cpu_supports("avx512f");
+    default: return false;
+    }
 #endif
     return false;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "usearch",
-    "version": "0.18.3",
+    "version": "0.18.4",
     "description": "Smaller & Faster Single-File Vector Search Engine from Unum",
     "author": "Ashot Vardanian",
     "license": "Apache 2.0",

package/python/lib.cpp

@@ -13,6 +13,7 @@
  *  @copyright Copyright (c) 2023
  */
 #define PY_SSIZE_T_CLEAN
+#define NOMINMAX // Some of our dependencies call `std::max(x, y)`, which crashes Windows builds
 #include <thread>
 
 #include <pybind11/numpy.h>

package/setup.py

@@ -9,7 +9,7 @@ compile_args = ['-std=c++11', '-O3']
 link_args = []
 macros_args = [
     ('USEARCH_USE_NATIVE_F16', '0'),
-    ('USEARCH_USE_SIMSIMD', '1'),
+    ('USEARCH_USE_SIMSIMD', '0'),
 ]
 
 if sys.platform == 'linux':