usearch 2.21.3 → 2.25.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +24 -24
- package/binding.gyp +15 -37
- package/c/CMakeLists.txt +4 -5
- package/c/README.md +7 -7
- package/c/lib.cpp +32 -14
- package/c/test.c +83 -1
- package/c/usearch.h +24 -3
- package/cpp/CMakeLists.txt +4 -0
- package/cpp/README.md +5 -5
- package/cpp/bench.cpp +97 -116
- package/cpp/test.cpp +117 -20
- package/include/usearch/index.hpp +235 -52
- package/include/usearch/index_dense.hpp +105 -40
- package/include/usearch/index_plugins.hpp +1387 -204
- package/javascript/dist/cjs/usearch.d.ts +8 -3
- package/javascript/dist/cjs/usearch.js +12 -4
- package/javascript/dist/esm/usearch.d.ts +8 -3
- package/javascript/dist/esm/usearch.js +12 -4
- package/javascript/lib.cpp +17 -0
- package/javascript/usearch.ts +14 -6
- package/numkong/.clang-tidy +30 -0
- package/{simsimd → numkong}/.cmake-format.py +18 -0
- package/numkong/.editorconfig +12 -0
- package/numkong/.git-blame-ignore-revs +123 -0
- package/numkong/.mailmap +15 -0
- package/numkong/.spi.yml +11 -0
- package/numkong/.swift-format +14 -0
- package/numkong/CITATION.cff +81 -0
- package/numkong/CONTRIBUTING.md +500 -0
- package/numkong/Cargo.toml +54 -0
- package/numkong/MANIFEST.in +8 -0
- package/numkong/Package.swift +84 -0
- package/numkong/README.md +569 -0
- package/numkong/bench/README.md +217 -0
- package/numkong/bench/bench-browser.html +323 -0
- package/numkong/bench/bench.cpp +344 -0
- package/numkong/bench/bench.hpp +816 -0
- package/numkong/bench/bench.js +148 -0
- package/numkong/bench/bench.mjs +616 -0
- package/numkong/bench/bench_attention.cpp +113 -0
- package/numkong/bench/bench_cast.cpp +264 -0
- package/numkong/bench/bench_cross_amx.cpp +62 -0
- package/numkong/bench/bench_cross_arm.cpp +245 -0
- package/numkong/bench/bench_cross_blas.cpp +309 -0
- package/numkong/bench/bench_cross_loongarch.cpp +106 -0
- package/numkong/bench/bench_cross_power.cpp +84 -0
- package/numkong/bench/bench_cross_rvv.cpp +121 -0
- package/numkong/bench/bench_cross_serial.cpp +151 -0
- package/numkong/bench/bench_cross_sme.cpp +163 -0
- package/numkong/bench/bench_cross_wasm.cpp +142 -0
- package/numkong/bench/bench_cross_x86.cpp +455 -0
- package/numkong/bench/bench_curved.cpp +210 -0
- package/numkong/bench/bench_dot.cpp +315 -0
- package/numkong/bench/bench_each.cpp +503 -0
- package/numkong/bench/bench_geospatial.cpp +112 -0
- package/numkong/bench/bench_maxsim.cpp +154 -0
- package/numkong/bench/bench_mesh.cpp +148 -0
- package/numkong/bench/bench_probability.cpp +60 -0
- package/numkong/bench/bench_reduce.cpp +371 -0
- package/numkong/bench/bench_set.cpp +73 -0
- package/numkong/bench/bench_sparse.cpp +198 -0
- package/numkong/bench/bench_spatial.cpp +347 -0
- package/numkong/bench/bench_trigonometry.cpp +137 -0
- package/numkong/bench/swift/Bench.swift +506 -0
- package/numkong/binding.gyp +169 -0
- package/numkong/build.rs +484 -0
- package/numkong/c/dispatch.h +512 -0
- package/numkong/c/dispatch_bf16.c +447 -0
- package/numkong/c/dispatch_bf16c.c +52 -0
- package/numkong/c/dispatch_e2m3.c +296 -0
- package/numkong/c/dispatch_e3m2.c +284 -0
- package/numkong/c/dispatch_e4m3.c +318 -0
- package/numkong/c/dispatch_e5m2.c +316 -0
- package/numkong/c/dispatch_f16.c +452 -0
- package/numkong/c/dispatch_f16c.c +58 -0
- package/numkong/c/dispatch_f32.c +434 -0
- package/numkong/c/dispatch_f32c.c +99 -0
- package/numkong/c/dispatch_f64.c +348 -0
- package/numkong/c/dispatch_f64c.c +98 -0
- package/numkong/c/dispatch_i16.c +96 -0
- package/numkong/c/dispatch_i32.c +89 -0
- package/numkong/c/dispatch_i4.c +153 -0
- package/numkong/c/dispatch_i64.c +86 -0
- package/numkong/c/dispatch_i8.c +348 -0
- package/numkong/c/dispatch_other.c +348 -0
- package/numkong/c/dispatch_u1.c +193 -0
- package/numkong/c/dispatch_u16.c +124 -0
- package/numkong/c/dispatch_u32.c +118 -0
- package/numkong/c/dispatch_u4.c +153 -0
- package/numkong/c/dispatch_u64.c +102 -0
- package/numkong/c/dispatch_u8.c +364 -0
- package/numkong/c/numkong.c +940 -0
- package/numkong/cmake/nk_arm_isa_probes.cmake +61 -0
- package/numkong/cmake/nk_isa_probe.cmake +90 -0
- package/numkong/cmake/nk_loongarch_isa_probes.cmake +15 -0
- package/numkong/cmake/nk_math_lib.cmake +46 -0
- package/numkong/cmake/nk_power_isa_probes.cmake +15 -0
- package/numkong/cmake/nk_riscv_isa_probes.cmake +19 -0
- package/numkong/cmake/nk_x86_isa_probes.cmake +35 -0
- package/numkong/cmake/toolchain-aarch64-gnu.cmake +86 -0
- package/numkong/cmake/toolchain-android-arm64.cmake +25 -0
- package/numkong/cmake/toolchain-android-armv7.cmake +25 -0
- package/numkong/cmake/toolchain-loongarch64-gnu.cmake +68 -0
- package/numkong/cmake/toolchain-ppc64le-gnu.cmake +74 -0
- package/numkong/cmake/toolchain-riscv64-gnu.cmake +107 -0
- package/numkong/cmake/toolchain-riscv64-llvm.cmake +114 -0
- package/numkong/cmake/toolchain-wasi.cmake +77 -0
- package/numkong/cmake/toolchain-wasm.cmake +70 -0
- package/numkong/cmake/toolchain-wasm64.cmake +62 -0
- package/numkong/cmake/toolchain-x86_64-llvm.cmake +47 -0
- package/numkong/include/README.md +643 -0
- package/numkong/include/module.modulemap +129 -0
- package/numkong/include/numkong/attention/sapphireamx.h +1363 -0
- package/numkong/include/numkong/attention/sme.h +2075 -0
- package/numkong/include/numkong/attention.h +49 -0
- package/numkong/include/numkong/capabilities.h +930 -0
- package/numkong/include/numkong/cast/README.md +262 -0
- package/numkong/include/numkong/cast/diamond.h +64 -0
- package/numkong/include/numkong/cast/haswell.h +823 -0
- package/numkong/include/numkong/cast/icelake.h +475 -0
- package/numkong/include/numkong/cast/loongsonasx.h +252 -0
- package/numkong/include/numkong/cast/neon.h +1174 -0
- package/numkong/include/numkong/cast/powervsx.h +449 -0
- package/numkong/include/numkong/cast/rvv.h +970 -0
- package/numkong/include/numkong/cast/sapphire.h +262 -0
- package/numkong/include/numkong/cast/serial.h +2336 -0
- package/numkong/include/numkong/cast/skylake.h +915 -0
- package/numkong/include/numkong/cast/v128relaxed.h +537 -0
- package/numkong/include/numkong/cast.h +254 -0
- package/numkong/include/numkong/cast.hpp +44 -0
- package/numkong/include/numkong/curved/README.md +223 -0
- package/numkong/include/numkong/curved/genoa.h +182 -0
- package/numkong/include/numkong/curved/haswell.h +276 -0
- package/numkong/include/numkong/curved/neon.h +329 -0
- package/numkong/include/numkong/curved/neonbfdot.h +211 -0
- package/numkong/include/numkong/curved/rvv.h +305 -0
- package/numkong/include/numkong/curved/serial.h +207 -0
- package/numkong/include/numkong/curved/skylake.h +457 -0
- package/numkong/include/numkong/curved/smef64.h +530 -0
- package/numkong/include/numkong/curved.h +513 -0
- package/numkong/include/numkong/curved.hpp +144 -0
- package/numkong/include/numkong/dot/README.md +442 -0
- package/numkong/include/numkong/dot/alder.h +563 -0
- package/numkong/include/numkong/dot/diamond.h +158 -0
- package/numkong/include/numkong/dot/genoa.h +289 -0
- package/numkong/include/numkong/dot/haswell.h +1723 -0
- package/numkong/include/numkong/dot/icelake.h +997 -0
- package/numkong/include/numkong/dot/loongsonasx.h +671 -0
- package/numkong/include/numkong/dot/neon.h +869 -0
- package/numkong/include/numkong/dot/neonbfdot.h +243 -0
- package/numkong/include/numkong/dot/neonfhm.h +358 -0
- package/numkong/include/numkong/dot/neonfp8.h +323 -0
- package/numkong/include/numkong/dot/neonsdot.h +622 -0
- package/numkong/include/numkong/dot/powervsx.h +752 -0
- package/numkong/include/numkong/dot/rvv.h +722 -0
- package/numkong/include/numkong/dot/rvvbb.h +72 -0
- package/numkong/include/numkong/dot/rvvbf16.h +123 -0
- package/numkong/include/numkong/dot/rvvhalf.h +129 -0
- package/numkong/include/numkong/dot/sapphire.h +141 -0
- package/numkong/include/numkong/dot/serial.h +874 -0
- package/numkong/include/numkong/dot/sierra.h +405 -0
- package/numkong/include/numkong/dot/skylake.h +1116 -0
- package/numkong/include/numkong/dot/sve.h +420 -0
- package/numkong/include/numkong/dot/svebfdot.h +75 -0
- package/numkong/include/numkong/dot/svehalf.h +168 -0
- package/numkong/include/numkong/dot/svesdot.h +90 -0
- package/numkong/include/numkong/dot/v128relaxed.h +1293 -0
- package/numkong/include/numkong/dot.h +1136 -0
- package/numkong/include/numkong/dot.hpp +94 -0
- package/numkong/include/numkong/dots/README.md +496 -0
- package/numkong/include/numkong/dots/alder.h +117 -0
- package/numkong/include/numkong/dots/diamond.h +86 -0
- package/numkong/include/numkong/dots/genoa.h +100 -0
- package/numkong/include/numkong/dots/graniteamx.h +733 -0
- package/numkong/include/numkong/dots/haswell.h +310 -0
- package/numkong/include/numkong/dots/icelake.h +180 -0
- package/numkong/include/numkong/dots/loongsonasx.h +176 -0
- package/numkong/include/numkong/dots/neon.h +123 -0
- package/numkong/include/numkong/dots/neonbfdot.h +59 -0
- package/numkong/include/numkong/dots/neonfhm.h +96 -0
- package/numkong/include/numkong/dots/neonfp8.h +99 -0
- package/numkong/include/numkong/dots/neonsdot.h +144 -0
- package/numkong/include/numkong/dots/powervsx.h +194 -0
- package/numkong/include/numkong/dots/rvv.h +2593 -0
- package/numkong/include/numkong/dots/sapphireamx.h +4017 -0
- package/numkong/include/numkong/dots/serial.h +2867 -0
- package/numkong/include/numkong/dots/sierra.h +100 -0
- package/numkong/include/numkong/dots/skylake.h +207 -0
- package/numkong/include/numkong/dots/sme.h +5041 -0
- package/numkong/include/numkong/dots/smebi32.h +477 -0
- package/numkong/include/numkong/dots/smef64.h +1328 -0
- package/numkong/include/numkong/dots/smehalf.h +47 -0
- package/numkong/include/numkong/dots/v128relaxed.h +317 -0
- package/numkong/include/numkong/dots.h +3110 -0
- package/numkong/include/numkong/dots.hpp +641 -0
- package/numkong/include/numkong/each/README.md +465 -0
- package/numkong/include/numkong/each/haswell.h +1658 -0
- package/numkong/include/numkong/each/icelake.h +272 -0
- package/numkong/include/numkong/each/neon.h +1138 -0
- package/numkong/include/numkong/each/neonbfdot.h +211 -0
- package/numkong/include/numkong/each/neonhalf.h +318 -0
- package/numkong/include/numkong/each/rvv.h +1121 -0
- package/numkong/include/numkong/each/sapphire.h +343 -0
- package/numkong/include/numkong/each/serial.h +282 -0
- package/numkong/include/numkong/each/skylake.h +1562 -0
- package/numkong/include/numkong/each/v128relaxed.h +562 -0
- package/numkong/include/numkong/each.h +2232 -0
- package/numkong/include/numkong/each.hpp +434 -0
- package/numkong/include/numkong/geospatial/README.md +147 -0
- package/numkong/include/numkong/geospatial/haswell.h +633 -0
- package/numkong/include/numkong/geospatial/neon.h +615 -0
- package/numkong/include/numkong/geospatial/rvv.h +701 -0
- package/numkong/include/numkong/geospatial/serial.h +309 -0
- package/numkong/include/numkong/geospatial/skylake.h +613 -0
- package/numkong/include/numkong/geospatial/v128relaxed.h +652 -0
- package/numkong/include/numkong/geospatial.h +453 -0
- package/numkong/include/numkong/geospatial.hpp +235 -0
- package/numkong/include/numkong/matrix.hpp +336 -0
- package/numkong/include/numkong/maxsim/README.md +187 -0
- package/numkong/include/numkong/maxsim/alder.h +511 -0
- package/numkong/include/numkong/maxsim/genoa.h +115 -0
- package/numkong/include/numkong/maxsim/haswell.h +553 -0
- package/numkong/include/numkong/maxsim/icelake.h +480 -0
- package/numkong/include/numkong/maxsim/neonsdot.h +394 -0
- package/numkong/include/numkong/maxsim/sapphireamx.h +877 -0
- package/numkong/include/numkong/maxsim/serial.h +490 -0
- package/numkong/include/numkong/maxsim/sme.h +993 -0
- package/numkong/include/numkong/maxsim/v128relaxed.h +280 -0
- package/numkong/include/numkong/maxsim.h +571 -0
- package/numkong/include/numkong/maxsim.hpp +133 -0
- package/numkong/include/numkong/mesh/README.md +219 -0
- package/numkong/include/numkong/mesh/haswell.h +2013 -0
- package/numkong/include/numkong/mesh/neon.h +1821 -0
- package/numkong/include/numkong/mesh/neonbfdot.h +771 -0
- package/numkong/include/numkong/mesh/rvv.h +936 -0
- package/numkong/include/numkong/mesh/serial.h +721 -0
- package/numkong/include/numkong/mesh/skylake.h +2344 -0
- package/numkong/include/numkong/mesh/v128relaxed.h +1081 -0
- package/numkong/include/numkong/mesh.h +666 -0
- package/numkong/include/numkong/mesh.hpp +701 -0
- package/numkong/include/numkong/numkong.h +78 -0
- package/numkong/include/numkong/numkong.hpp +58 -0
- package/numkong/include/numkong/probability/README.md +177 -0
- package/numkong/include/numkong/probability/haswell.h +267 -0
- package/numkong/include/numkong/probability/neon.h +255 -0
- package/numkong/include/numkong/probability/rvv.h +414 -0
- package/numkong/include/numkong/probability/serial.h +181 -0
- package/numkong/include/numkong/probability/skylake.h +324 -0
- package/numkong/include/numkong/probability.h +380 -0
- package/numkong/include/numkong/probability.hpp +120 -0
- package/numkong/include/numkong/random.h +50 -0
- package/numkong/include/numkong/random.hpp +285 -0
- package/numkong/include/numkong/reduce/README.md +552 -0
- package/numkong/include/numkong/reduce/alder.h +636 -0
- package/numkong/include/numkong/reduce/genoa.h +201 -0
- package/numkong/include/numkong/reduce/haswell.h +3785 -0
- package/numkong/include/numkong/reduce/icelake.h +549 -0
- package/numkong/include/numkong/reduce/neon.h +3980 -0
- package/numkong/include/numkong/reduce/neonbfdot.h +142 -0
- package/numkong/include/numkong/reduce/neonfhm.h +293 -0
- package/numkong/include/numkong/reduce/neonsdot.h +357 -0
- package/numkong/include/numkong/reduce/rvv.h +3533 -0
- package/numkong/include/numkong/reduce/serial.h +753 -0
- package/numkong/include/numkong/reduce/sierra.h +338 -0
- package/numkong/include/numkong/reduce/skylake.h +3794 -0
- package/numkong/include/numkong/reduce/sve.h +52 -0
- package/numkong/include/numkong/reduce/v128relaxed.h +2302 -0
- package/numkong/include/numkong/reduce.h +1582 -0
- package/numkong/include/numkong/reduce.hpp +778 -0
- package/numkong/include/numkong/scalar/README.md +89 -0
- package/numkong/include/numkong/scalar/haswell.h +122 -0
- package/numkong/include/numkong/scalar/loongsonasx.h +74 -0
- package/numkong/include/numkong/scalar/neon.h +122 -0
- package/numkong/include/numkong/scalar/neonhalf.h +70 -0
- package/numkong/include/numkong/scalar/powervsx.h +96 -0
- package/numkong/include/numkong/scalar/rvv.h +211 -0
- package/numkong/include/numkong/scalar/sapphire.h +74 -0
- package/numkong/include/numkong/scalar/serial.h +332 -0
- package/numkong/include/numkong/scalar/v128relaxed.h +56 -0
- package/numkong/include/numkong/scalar.h +696 -0
- package/numkong/include/numkong/set/README.md +179 -0
- package/numkong/include/numkong/set/haswell.h +334 -0
- package/numkong/include/numkong/set/icelake.h +485 -0
- package/numkong/include/numkong/set/loongsonasx.h +181 -0
- package/numkong/include/numkong/set/neon.h +363 -0
- package/numkong/include/numkong/set/powervsx.h +326 -0
- package/numkong/include/numkong/set/rvv.h +226 -0
- package/numkong/include/numkong/set/rvvbb.h +117 -0
- package/numkong/include/numkong/set/serial.h +174 -0
- package/numkong/include/numkong/set/sve.h +187 -0
- package/numkong/include/numkong/set/v128relaxed.h +240 -0
- package/numkong/include/numkong/set.h +471 -0
- package/numkong/include/numkong/set.hpp +114 -0
- package/numkong/include/numkong/sets/README.md +149 -0
- package/numkong/include/numkong/sets/haswell.h +63 -0
- package/numkong/include/numkong/sets/icelake.h +66 -0
- package/numkong/include/numkong/sets/loongsonasx.h +52 -0
- package/numkong/include/numkong/sets/neon.h +61 -0
- package/numkong/include/numkong/sets/powervsx.h +65 -0
- package/numkong/include/numkong/sets/serial.h +43 -0
- package/numkong/include/numkong/sets/smebi32.h +1135 -0
- package/numkong/include/numkong/sets/v128relaxed.h +58 -0
- package/numkong/include/numkong/sets.h +382 -0
- package/numkong/include/numkong/sparse/README.md +156 -0
- package/numkong/include/numkong/sparse/icelake.h +463 -0
- package/numkong/include/numkong/sparse/neon.h +288 -0
- package/numkong/include/numkong/sparse/serial.h +117 -0
- package/numkong/include/numkong/sparse/sve2.h +504 -0
- package/numkong/include/numkong/sparse/turin.h +322 -0
- package/numkong/include/numkong/sparse.h +363 -0
- package/numkong/include/numkong/sparse.hpp +113 -0
- package/numkong/include/numkong/spatial/README.md +439 -0
- package/numkong/include/numkong/spatial/alder.h +604 -0
- package/numkong/include/numkong/spatial/diamond.h +240 -0
- package/numkong/include/numkong/spatial/genoa.h +222 -0
- package/numkong/include/numkong/spatial/haswell.h +979 -0
- package/numkong/include/numkong/spatial/icelake.h +1067 -0
- package/numkong/include/numkong/spatial/loongsonasx.h +483 -0
- package/numkong/include/numkong/spatial/neon.h +846 -0
- package/numkong/include/numkong/spatial/neonbfdot.h +164 -0
- package/numkong/include/numkong/spatial/neonfp8.h +258 -0
- package/numkong/include/numkong/spatial/neonsdot.h +429 -0
- package/numkong/include/numkong/spatial/powervsx.h +738 -0
- package/numkong/include/numkong/spatial/rvv.h +991 -0
- package/numkong/include/numkong/spatial/rvvbf16.h +128 -0
- package/numkong/include/numkong/spatial/rvvhalf.h +120 -0
- package/numkong/include/numkong/spatial/serial.h +347 -0
- package/numkong/include/numkong/spatial/sierra.h +516 -0
- package/numkong/include/numkong/spatial/skylake.h +605 -0
- package/numkong/include/numkong/spatial/sve.h +246 -0
- package/numkong/include/numkong/spatial/svebfdot.h +125 -0
- package/numkong/include/numkong/spatial/svehalf.h +136 -0
- package/numkong/include/numkong/spatial/svesdot.h +143 -0
- package/numkong/include/numkong/spatial/v128relaxed.h +969 -0
- package/numkong/include/numkong/spatial.h +1681 -0
- package/numkong/include/numkong/spatial.hpp +183 -0
- package/numkong/include/numkong/spatials/README.md +580 -0
- package/numkong/include/numkong/spatials/alder.h +94 -0
- package/numkong/include/numkong/spatials/diamond.h +82 -0
- package/numkong/include/numkong/spatials/genoa.h +94 -0
- package/numkong/include/numkong/spatials/graniteamx.h +173 -0
- package/numkong/include/numkong/spatials/haswell.h +219 -0
- package/numkong/include/numkong/spatials/icelake.h +113 -0
- package/numkong/include/numkong/spatials/loongsonasx.h +153 -0
- package/numkong/include/numkong/spatials/neon.h +109 -0
- package/numkong/include/numkong/spatials/neonbfdot.h +60 -0
- package/numkong/include/numkong/spatials/neonfhm.h +92 -0
- package/numkong/include/numkong/spatials/neonfp8.h +111 -0
- package/numkong/include/numkong/spatials/neonsdot.h +143 -0
- package/numkong/include/numkong/spatials/powervsx.h +153 -0
- package/numkong/include/numkong/spatials/rvv.h +1976 -0
- package/numkong/include/numkong/spatials/sapphireamx.h +1149 -0
- package/numkong/include/numkong/spatials/serial.h +248 -0
- package/numkong/include/numkong/spatials/sierra.h +96 -0
- package/numkong/include/numkong/spatials/skylake.h +184 -0
- package/numkong/include/numkong/spatials/sme.h +1927 -0
- package/numkong/include/numkong/spatials/smef64.h +483 -0
- package/numkong/include/numkong/spatials/v128relaxed.h +240 -0
- package/numkong/include/numkong/spatials.h +3374 -0
- package/numkong/include/numkong/spatials.hpp +509 -0
- package/numkong/include/numkong/tensor.hpp +1981 -0
- package/numkong/include/numkong/trigonometry/README.md +184 -0
- package/numkong/include/numkong/trigonometry/haswell.h +653 -0
- package/numkong/include/numkong/trigonometry/neon.h +638 -0
- package/numkong/include/numkong/trigonometry/rvv.h +700 -0
- package/numkong/include/numkong/trigonometry/serial.h +703 -0
- package/numkong/include/numkong/trigonometry/skylake.h +725 -0
- package/numkong/include/numkong/trigonometry/v128relaxed.h +667 -0
- package/numkong/include/numkong/trigonometry.h +467 -0
- package/numkong/include/numkong/trigonometry.hpp +166 -0
- package/numkong/include/numkong/types.h +1695 -0
- package/numkong/include/numkong/types.hpp +6027 -0
- package/numkong/include/numkong/vector.hpp +766 -0
- package/numkong/javascript/@numkong-darwin-arm64/package.json +26 -0
- package/numkong/javascript/@numkong-darwin-x64/package.json +26 -0
- package/numkong/javascript/@numkong-linux-arm64/package.json +25 -0
- package/numkong/javascript/@numkong-linux-x64/package.json +25 -0
- package/numkong/javascript/@numkong-win32-arm64/package.json +25 -0
- package/numkong/javascript/@numkong-win32-x64/package.json +25 -0
- package/numkong/javascript/README.md +254 -0
- package/numkong/javascript/node-gyp-build.d.ts +4 -0
- package/numkong/javascript/numkong-browser.ts +40 -0
- package/numkong/javascript/numkong-wasm.ts +750 -0
- package/numkong/javascript/numkong.c +712 -0
- package/numkong/javascript/numkong.ts +613 -0
- package/{simsimd → numkong}/javascript/tsconfig-base.json +6 -3
- package/numkong/javascript/tsconfig-cjs.json +13 -0
- package/{simsimd → numkong}/javascript/tsconfig-esm.json +1 -1
- package/numkong/javascript/types.ts +784 -0
- package/numkong/numkong.gypi +104 -0
- package/numkong/package.json +108 -0
- package/numkong/probes/arm_neon.c +8 -0
- package/numkong/probes/arm_neon_bfdot.c +9 -0
- package/numkong/probes/arm_neon_fhm.c +9 -0
- package/numkong/probes/arm_neon_half.c +8 -0
- package/numkong/probes/arm_neon_sdot.c +9 -0
- package/numkong/probes/arm_neonfp8.c +9 -0
- package/numkong/probes/arm_sme.c +16 -0
- package/numkong/probes/arm_sme2.c +16 -0
- package/numkong/probes/arm_sme2p1.c +16 -0
- package/numkong/probes/arm_sme_bf16.c +16 -0
- package/numkong/probes/arm_sme_bi32.c +16 -0
- package/numkong/probes/arm_sme_f64.c +16 -0
- package/numkong/probes/arm_sme_fa64.c +14 -0
- package/numkong/probes/arm_sme_half.c +16 -0
- package/numkong/probes/arm_sme_lut2.c +15 -0
- package/numkong/probes/arm_sve.c +18 -0
- package/numkong/probes/arm_sve2.c +20 -0
- package/numkong/probes/arm_sve2p1.c +18 -0
- package/numkong/probes/arm_sve_bfdot.c +20 -0
- package/numkong/probes/arm_sve_half.c +18 -0
- package/numkong/probes/arm_sve_sdot.c +21 -0
- package/numkong/probes/loongarch_lasx.c +12 -0
- package/numkong/probes/power_vsx.c +12 -0
- package/numkong/probes/probe.js +127 -0
- package/numkong/probes/riscv_rvv.c +14 -0
- package/numkong/probes/riscv_rvv_bb.c +15 -0
- package/numkong/probes/riscv_rvv_bf16.c +17 -0
- package/numkong/probes/riscv_rvv_half.c +14 -0
- package/numkong/probes/wasm_v128relaxed.c +11 -0
- package/numkong/probes/x86_alder.c +17 -0
- package/numkong/probes/x86_diamond.c +17 -0
- package/numkong/probes/x86_genoa.c +17 -0
- package/numkong/probes/x86_graniteamx.c +19 -0
- package/numkong/probes/x86_haswell.c +11 -0
- package/numkong/probes/x86_icelake.c +17 -0
- package/numkong/probes/x86_sapphire.c +16 -0
- package/numkong/probes/x86_sapphireamx.c +18 -0
- package/numkong/probes/x86_sierra.c +17 -0
- package/numkong/probes/x86_skylake.c +15 -0
- package/numkong/probes/x86_turin.c +17 -0
- package/numkong/rust/README.md +698 -0
- package/numkong/rust/capabilities.rs +96 -0
- package/numkong/rust/cast.rs +314 -0
- package/numkong/rust/curved.rs +391 -0
- package/numkong/rust/each.rs +2721 -0
- package/numkong/rust/geospatial.rs +343 -0
- package/numkong/rust/matrix.rs +5424 -0
- package/numkong/rust/maxsim.rs +540 -0
- package/numkong/rust/mesh.rs +837 -0
- package/numkong/rust/numkong.rs +409 -0
- package/numkong/rust/probability.rs +272 -0
- package/numkong/rust/reduce.rs +1249 -0
- package/numkong/rust/set.rs +194 -0
- package/numkong/rust/sparse.rs +513 -0
- package/numkong/rust/spatial.rs +1815 -0
- package/numkong/rust/tensor.rs +7565 -0
- package/numkong/rust/types.rs +3666 -0
- package/numkong/rust/vector.rs +1899 -0
- package/numkong/rustfmt.toml +2 -0
- package/numkong/swift/Matrix.swift +1388 -0
- package/numkong/swift/MaxSim.swift +151 -0
- package/numkong/swift/NumKong.swift +283 -0
- package/numkong/swift/Numerics.swift +329 -0
- package/numkong/swift/README.md +433 -0
- package/numkong/swift/Scalar.swift +741 -0
- package/numkong/swift/Tensor.swift +209 -0
- package/numkong/test/README.md +315 -0
- package/numkong/test/playwright.config.ts +16 -0
- package/numkong/test/swift/Test.swift +534 -0
- package/numkong/test/test-browser-bundle-runner.mjs +36 -0
- package/numkong/test/test-browser-bundle.html +188 -0
- package/numkong/test/test-browser-runner.mjs +43 -0
- package/numkong/test/test-browser.html +305 -0
- package/numkong/test/test-dtypes.mjs +453 -0
- package/numkong/test/test-wasm.mjs +267 -0
- package/numkong/test/test.cpp +442 -0
- package/numkong/test/test.hpp +580 -0
- package/numkong/test/test.mjs +363 -0
- package/numkong/test/test_base.py +1041 -0
- package/numkong/test/test_cast.cpp +242 -0
- package/numkong/test/test_cross.hpp +551 -0
- package/numkong/test/test_cross.py +467 -0
- package/numkong/test/test_cross_amx.cpp +50 -0
- package/numkong/test/test_cross_arm.cpp +225 -0
- package/numkong/test/test_cross_blas.cpp +354 -0
- package/numkong/test/test_cross_loongarch.cpp +90 -0
- package/numkong/test/test_cross_power.cpp +73 -0
- package/numkong/test/test_cross_rvv.cpp +109 -0
- package/numkong/test/test_cross_serial.cpp +181 -0
- package/numkong/test/test_cross_sme.cpp +148 -0
- package/numkong/test/test_cross_wasm.cpp +141 -0
- package/numkong/test/test_cross_x86.cpp +442 -0
- package/numkong/test/test_curved.cpp +182 -0
- package/numkong/test/test_curved.py +207 -0
- package/numkong/test/test_dot.cpp +353 -0
- package/numkong/test/test_dot.py +296 -0
- package/numkong/test/test_each.cpp +437 -0
- package/numkong/test/test_each.py +756 -0
- package/numkong/test/test_geospatial.cpp +155 -0
- package/numkong/test/test_geospatial.py +241 -0
- package/numkong/test/test_maxsim.cpp +138 -0
- package/numkong/test/test_maxsim.py +209 -0
- package/numkong/test/test_mesh.cpp +203 -0
- package/numkong/test/test_mesh.py +394 -0
- package/numkong/test/test_probability.cpp +136 -0
- package/numkong/test/test_probability.py +166 -0
- package/numkong/test/test_reduce.cpp +563 -0
- package/numkong/test/test_reduce.py +781 -0
- package/numkong/test/test_set.cpp +151 -0
- package/numkong/test/test_set.py +120 -0
- package/numkong/test/test_similarities.py +579 -0
- package/numkong/test/test_sparse.cpp +121 -0
- package/numkong/test/test_sparse.py +118 -0
- package/numkong/test/test_spatial.cpp +518 -0
- package/numkong/test/test_spatial.py +269 -0
- package/numkong/test/test_tensor.cpp +884 -0
- package/numkong/test/test_tensor.py +1417 -0
- package/numkong/test/test_trigonometry.cpp +156 -0
- package/numkong/test/test_trigonometry.py +180 -0
- package/package.json +7 -8
- package/prebuilds/darwin-arm64+x64/usearch.node +0 -0
- package/prebuilds/linux-arm64/usearch.node +0 -0
- package/prebuilds/linux-x64/usearch.node +0 -0
- package/fp16/CMakeLists.txt +0 -176
- package/fp16/LICENSE +0 -11
- package/fp16/README.md +0 -20
- package/fp16/bench/alt-element.cc +0 -89
- package/fp16/bench/from-alt-array.cc +0 -175
- package/fp16/bench/from-ieee-array.cc +0 -399
- package/fp16/bench/ieee-element.cc +0 -272
- package/fp16/bench/to-alt-array.cc +0 -91
- package/fp16/bench/to-ieee-array.cc +0 -303
- package/fp16/cmake/DownloadGoogleBenchmark.cmake +0 -15
- package/fp16/cmake/DownloadGoogleTest.cmake +0 -15
- package/fp16/cmake/DownloadPSimd.cmake +0 -15
- package/fp16/configure.py +0 -65
- package/fp16/confu.yaml +0 -8
- package/fp16/include/fp16/__init__.py +0 -0
- package/fp16/include/fp16/avx.py +0 -50
- package/fp16/include/fp16/avx2.py +0 -53
- package/fp16/include/fp16/bitcasts.h +0 -92
- package/fp16/include/fp16/fp16.h +0 -451
- package/fp16/include/fp16/psimd.h +0 -131
- package/fp16/include/fp16.h +0 -11
- package/fp16/jni/Android.mk +0 -6
- package/fp16/jni/Application.mk +0 -4
- package/fp16/test/alt-from-fp32-value.cc +0 -498
- package/fp16/test/alt-to-fp32-bits.cc +0 -263
- package/fp16/test/alt-to-fp32-psimd.cc +0 -145
- package/fp16/test/alt-to-fp32-value.cc +0 -348
- package/fp16/test/alt-to-fp32x2-psimd.cc +0 -245
- package/fp16/test/bitcasts.cc +0 -56
- package/fp16/test/ieee-from-fp32-value.cc +0 -523
- package/fp16/test/ieee-to-fp32-bits.cc +0 -333
- package/fp16/test/ieee-to-fp32-psimd.cc +0 -312
- package/fp16/test/ieee-to-fp32-value.cc +0 -420
- package/fp16/test/ieee-to-fp32x2-psimd.cc +0 -541
- package/fp16/test/peachpy/alt-xmm-to-fp32-xmm-avx.cc +0 -96
- package/fp16/test/peachpy/alt-xmm-to-fp32-ymm-avx2.cc +0 -96
- package/fp16/test/peachpy/stubs.py +0 -38
- package/fp16/test/tables.cc +0 -530
- package/fp16/test/tables.h +0 -13
- package/fp16/third-party/THHalf.h +0 -92
- package/fp16/third-party/eigen-half.h +0 -170
- package/fp16/third-party/float16-compressor.h +0 -80
- package/fp16/third-party/half.hpp +0 -2916
- package/fp16/third-party/npy-halffloat.h +0 -328
- package/simsimd/.git-blame-ignore-revs +0 -2
- package/simsimd/CONTRIBUTING.md +0 -321
- package/simsimd/Cargo.toml +0 -49
- package/simsimd/MANIFEST.in +0 -4
- package/simsimd/Package.swift +0 -28
- package/simsimd/README.md +0 -1493
- package/simsimd/binding.gyp +0 -35
- package/simsimd/build.rs +0 -70
- package/simsimd/c/lib.c +0 -374
- package/simsimd/include/module.modulemap +0 -4
- package/simsimd/include/simsimd/binary.h +0 -486
- package/simsimd/include/simsimd/curved.h +0 -1541
- package/simsimd/include/simsimd/dot.h +0 -1853
- package/simsimd/include/simsimd/elementwise.h +0 -2465
- package/simsimd/include/simsimd/geospatial.h +0 -43
- package/simsimd/include/simsimd/mesh.h +0 -69
- package/simsimd/include/simsimd/probability.h +0 -606
- package/simsimd/include/simsimd/simsimd.h +0 -2567
- package/simsimd/include/simsimd/sparse.h +0 -1384
- package/simsimd/include/simsimd/spatial.h +0 -2335
- package/simsimd/include/simsimd/types.h +0 -666
- package/simsimd/javascript/fallback.ts +0 -233
- package/simsimd/javascript/lib.c +0 -125
- package/simsimd/javascript/node-gyp-build.d.ts +0 -1
- package/simsimd/javascript/simsimd.ts +0 -179
- package/simsimd/javascript/tsconfig-cjs.json +0 -8
- package/simsimd/package-lock.json +0 -1761
- package/simsimd/package.json +0 -61
- package/simsimd/rust/lib.rs +0 -1821
- package/simsimd/scripts/bench.cxx +0 -1170
- package/simsimd/scripts/bench.ipynb +0 -187
- package/simsimd/scripts/bench.js +0 -127
- package/simsimd/scripts/bench.rs +0 -876
- package/simsimd/scripts/bench_vectors.py +0 -734
- package/simsimd/scripts/bench_vectors_live.py +0 -213
- package/simsimd/scripts/test.c +0 -187
- package/simsimd/scripts/test.mjs +0 -220
- package/simsimd/scripts/test.py +0 -1505
- package/simsimd/swift/SimSIMD.swift +0 -149
- package/simsimd/swift/Test.swift +0 -128
- /package/{simsimd → numkong}/CODE_OF_CONDUCT.md +0 -0
- /package/{simsimd → numkong}/LICENSE +0 -0
- /package/{simsimd → numkong}/javascript/dist-package-cjs.json +0 -0
- /package/{simsimd → numkong}/javascript/dist-package-esm.json +0 -0
|
@@ -1,486 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @file binary.h
|
|
3
|
-
* @brief SIMD-accelerated Binary Similarity Measures.
|
|
4
|
-
* @author Ash Vardanian
|
|
5
|
-
* @date July 1, 2023
|
|
6
|
-
*
|
|
7
|
-
* Contains:
|
|
8
|
-
* - Bit-level Hamming distance
|
|
9
|
-
* - Bit-level Jaccard distance (Tanimoto coefficient)
|
|
10
|
-
* - TODO: Hamming distance for integer vectors - `u16`, `u32`
|
|
11
|
-
*
|
|
12
|
-
* For hardware architectures:
|
|
13
|
-
* - Arm: NEON, SVE
|
|
14
|
-
* - x86: Haswell, Ice Lake
|
|
15
|
-
*
|
|
16
|
-
* The hardest part of optimizing binary similarity measures is the population count operation.
|
|
17
|
-
* It's natively supported by almost every instruction set, but the throughput and latency can
|
|
18
|
-
* be suboptimal. There are several ways to optimize this operation:
|
|
19
|
-
*
|
|
20
|
-
* - Lookup tables, mostly using nibbles (4-bit lookups)
|
|
21
|
-
* - Harley-Seal population counts: https://arxiv.org/pdf/1611.07612
|
|
22
|
-
*
|
|
23
|
-
* On binary vectors, when computing Jaccard distance we can clearly see how the CPU struggles
|
|
24
|
-
* to compute that many population counts. There are several instructions we should keep in mind
|
|
25
|
-
* for future optimizations:
|
|
26
|
-
*
|
|
27
|
-
* - `_mm512_popcnt_epi64` maps to `VPOPCNTQ (ZMM, K, ZMM)`:
|
|
28
|
-
* - On Ice Lake: 3 cycles latency, ports: 1*p5
|
|
29
|
-
* - On Genoa: 2 cycles latency, ports: 1*FP01
|
|
30
|
-
* - `_mm512_shuffle_epi8` maps to `VPSHUFB (ZMM, ZMM, ZMM)`:
|
|
31
|
-
* - On Ice Lake: 1 cycles latency, ports: 1*p5
|
|
32
|
-
* - On Genoa: 2 cycles latency, ports: 1*FP12
|
|
33
|
-
* - `_mm512_sad_epu8` maps to `VPSADBW (ZMM, ZMM, ZMM)`:
|
|
34
|
-
* - On Ice Lake: 3 cycles latency, ports: 1*p5
|
|
35
|
-
* - On Zen4: 3 cycles latency, ports: 1*FP01
|
|
36
|
-
* - `_mm512_tertiarylogic_epi64` maps to `VPTERNLOGQ (ZMM, ZMM, ZMM, I8)`:
|
|
37
|
-
* - On Ice Lake: 1 cycles latency, ports: 1*p05
|
|
38
|
-
* - On Zen4: 1 cycles latency, ports: 1*FP0123
|
|
39
|
-
* - `_mm512_gf2p8mul_epi8` maps to `VPGF2P8AFFINEQB (ZMM, ZMM, ZMM)`:
|
|
40
|
-
* - On Ice Lake: 5 cycles latency, ports: 1*p0
|
|
41
|
-
* - On Zen4: 3 cycles latency, ports: 1*FP01
|
|
42
|
-
*
|
|
43
|
-
* x86 intrinsics: https://www.intel.com/content/www/us/en/docs/intrinsics-guide/
|
|
44
|
-
* Arm intrinsics: https://developer.arm.com/architectures/instruction-sets/intrinsics/
|
|
45
|
-
* SSE POPCOUNT experiments by Wojciech Muła: https://github.com/WojciechMula/sse-popcount
|
|
46
|
-
* R&D progress tracker: https://github.com/ashvardanian/SimSIMD/pull/138
|
|
47
|
-
*/
|
|
48
|
-
#ifndef SIMSIMD_BINARY_H
|
|
49
|
-
#define SIMSIMD_BINARY_H
|
|
50
|
-
|
|
51
|
-
#include "types.h"
|
|
52
|
-
|
|
53
|
-
#ifdef __cplusplus
|
|
54
|
-
extern "C" {
|
|
55
|
-
#endif
|
|
56
|
-
|
|
57
|
-
// clang-format off
|
|
58
|
-
|
|
59
|
-
/* Serial backends for bitsets and integers. */
|
|
60
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_serial(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
61
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_serial(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
62
|
-
|
|
63
|
-
/* Arm NEON backend for bitsets and integers. */
|
|
64
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_neon(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
65
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_neon(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
66
|
-
|
|
67
|
-
/* Arm SVE backend for bitsets and integers. */
|
|
68
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_sve(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
69
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_sve(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
70
|
-
|
|
71
|
-
/* x86 AVX2 backend for bitsets and integers for Intel Haswell CPUs and newer, needs only POPCNT extensions. */
|
|
72
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_haswell(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
73
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_haswell(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
74
|
-
|
|
75
|
-
/* x86 AVX512 backend for bitsets and integers for Intel Ice Lake CPUs and newer, using VPOPCNTDQ extensions. */
|
|
76
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_ice(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
77
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_ice(simsimd_b8_t const* a, simsimd_b8_t const* b, simsimd_size_t n_words, simsimd_distance_t* distance);
|
|
78
|
-
// clang-format on
|
|
79
|
-
|
|
80
|
-
SIMSIMD_PUBLIC unsigned char simsimd_popcount_b8(simsimd_b8_t x) {
|
|
81
|
-
static unsigned char lookup_table[] = {
|
|
82
|
-
0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4, 1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, //
|
|
83
|
-
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
|
|
84
|
-
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
|
|
85
|
-
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
|
|
86
|
-
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5, 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
|
|
87
|
-
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
|
|
88
|
-
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6, 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
|
|
89
|
-
3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7, 4, 5, 5, 6, 5, 6, 6, 7, 5, 6, 6, 7, 6, 7, 7, 8};
|
|
90
|
-
return lookup_table[x];
|
|
91
|
-
}
|
|
92
|
-
|
|
93
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_serial(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
94
|
-
simsimd_distance_t *result) {
|
|
95
|
-
simsimd_u32_t differences = 0;
|
|
96
|
-
for (simsimd_size_t i = 0; i != n_words; ++i) differences += simsimd_popcount_b8(a[i] ^ b[i]);
|
|
97
|
-
*result = differences;
|
|
98
|
-
}
|
|
99
|
-
|
|
100
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_serial(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
101
|
-
simsimd_distance_t *result) {
|
|
102
|
-
simsimd_u32_t intersection = 0, union_ = 0;
|
|
103
|
-
for (simsimd_size_t i = 0; i != n_words; ++i)
|
|
104
|
-
intersection += simsimd_popcount_b8(a[i] & b[i]), union_ += simsimd_popcount_b8(a[i] | b[i]);
|
|
105
|
-
*result = (union_ != 0) ? 1 - (simsimd_f64_t)intersection / (simsimd_f64_t)union_ : 1;
|
|
106
|
-
}
|
|
107
|
-
|
|
108
|
-
#if _SIMSIMD_TARGET_ARM
|
|
109
|
-
#if SIMSIMD_TARGET_NEON
|
|
110
|
-
#pragma GCC push_options
|
|
111
|
-
#pragma GCC target("arch=armv8-a+simd")
|
|
112
|
-
#pragma clang attribute push(__attribute__((target("arch=armv8-a+simd"))), apply_to = function)
|
|
113
|
-
|
|
114
|
-
SIMSIMD_INTERNAL simsimd_u32_t _simsimd_reduce_u8x16_neon(uint8x16_t vec) {
|
|
115
|
-
// Split the vector into two halves and widen to `uint16x8_t`
|
|
116
|
-
uint16x8_t low_half = vmovl_u8(vget_low_u8(vec)); // widen lower 8 elements
|
|
117
|
-
uint16x8_t high_half = vmovl_u8(vget_high_u8(vec)); // widen upper 8 elements
|
|
118
|
-
|
|
119
|
-
// Sum the widened halves
|
|
120
|
-
uint16x8_t sum16 = vaddq_u16(low_half, high_half);
|
|
121
|
-
|
|
122
|
-
// Now reduce the `uint16x8_t` to a single `simsimd_u32_t`
|
|
123
|
-
uint32x4_t sum32 = vpaddlq_u16(sum16); // pairwise add into 32-bit integers
|
|
124
|
-
uint64x2_t sum64 = vpaddlq_u32(sum32); // pairwise add into 64-bit integers
|
|
125
|
-
simsimd_u32_t final_sum = vaddvq_u64(sum64); // final horizontal add to 32-bit result
|
|
126
|
-
return final_sum;
|
|
127
|
-
}
|
|
128
|
-
|
|
129
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_neon(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
130
|
-
simsimd_distance_t *result) {
|
|
131
|
-
simsimd_i32_t differences = 0;
|
|
132
|
-
simsimd_size_t i = 0;
|
|
133
|
-
// In each 8-bit word we may have up to 8 differences.
|
|
134
|
-
// So for up-to 31 cycles (31 * 16 = 496 word-dimensions = 3968 bits)
|
|
135
|
-
// we can aggregate the differences into a `uint8x16_t` vector,
|
|
136
|
-
// where each component will be up-to 255.
|
|
137
|
-
while (i + 16 <= n_words) {
|
|
138
|
-
uint8x16_t differences_cycle_vec = vdupq_n_u8(0);
|
|
139
|
-
for (simsimd_size_t cycle = 0; cycle < 31 && i + 16 <= n_words; ++cycle, i += 16) {
|
|
140
|
-
uint8x16_t a_vec = vld1q_u8(a + i);
|
|
141
|
-
uint8x16_t b_vec = vld1q_u8(b + i);
|
|
142
|
-
uint8x16_t xor_count_vec = vcntq_u8(veorq_u8(a_vec, b_vec));
|
|
143
|
-
differences_cycle_vec = vaddq_u8(differences_cycle_vec, xor_count_vec);
|
|
144
|
-
}
|
|
145
|
-
differences += _simsimd_reduce_u8x16_neon(differences_cycle_vec);
|
|
146
|
-
}
|
|
147
|
-
// Handle the tail
|
|
148
|
-
for (; i != n_words; ++i) differences += simsimd_popcount_b8(a[i] ^ b[i]);
|
|
149
|
-
*result = differences;
|
|
150
|
-
}
|
|
151
|
-
|
|
152
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_neon(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
153
|
-
simsimd_distance_t *result) {
|
|
154
|
-
simsimd_i32_t intersection = 0, union_ = 0;
|
|
155
|
-
simsimd_size_t i = 0;
|
|
156
|
-
// In each 8-bit word we may have up to 8 intersections/unions.
|
|
157
|
-
// So for up-to 31 cycles (31 * 16 = 496 word-dimensions = 3968 bits)
|
|
158
|
-
// we can aggregate the intersections/unions into a `uint8x16_t` vector,
|
|
159
|
-
// where each component will be up-to 255.
|
|
160
|
-
while (i + 16 <= n_words) {
|
|
161
|
-
uint8x16_t intersections_cycle_vec = vdupq_n_u8(0);
|
|
162
|
-
uint8x16_t unions_cycle_vec = vdupq_n_u8(0);
|
|
163
|
-
for (simsimd_size_t cycle = 0; cycle < 31 && i + 16 <= n_words; ++cycle, i += 16) {
|
|
164
|
-
uint8x16_t a_vec = vld1q_u8(a + i);
|
|
165
|
-
uint8x16_t b_vec = vld1q_u8(b + i);
|
|
166
|
-
uint8x16_t and_count_vec = vcntq_u8(vandq_u8(a_vec, b_vec));
|
|
167
|
-
uint8x16_t or_count_vec = vcntq_u8(vorrq_u8(a_vec, b_vec));
|
|
168
|
-
intersections_cycle_vec = vaddq_u8(intersections_cycle_vec, and_count_vec);
|
|
169
|
-
unions_cycle_vec = vaddq_u8(unions_cycle_vec, or_count_vec);
|
|
170
|
-
}
|
|
171
|
-
intersection += _simsimd_reduce_u8x16_neon(intersections_cycle_vec);
|
|
172
|
-
union_ += _simsimd_reduce_u8x16_neon(unions_cycle_vec);
|
|
173
|
-
}
|
|
174
|
-
// Handle the tail
|
|
175
|
-
for (; i != n_words; ++i)
|
|
176
|
-
intersection += simsimd_popcount_b8(a[i] & b[i]), union_ += simsimd_popcount_b8(a[i] | b[i]);
|
|
177
|
-
*result = (union_ != 0) ? 1 - (simsimd_f64_t)intersection / (simsimd_f64_t)union_ : 1;
|
|
178
|
-
}
|
|
179
|
-
|
|
180
|
-
#pragma clang attribute pop
|
|
181
|
-
#pragma GCC pop_options
|
|
182
|
-
#endif // SIMSIMD_TARGET_NEON
|
|
183
|
-
|
|
184
|
-
#if SIMSIMD_TARGET_SVE
|
|
185
|
-
#pragma GCC push_options
|
|
186
|
-
#pragma GCC target("arch=armv8.2-a+sve")
|
|
187
|
-
#pragma clang attribute push(__attribute__((target("arch=armv8.2-a+sve"))), apply_to = function)
|
|
188
|
-
|
|
189
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_sve(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
190
|
-
simsimd_distance_t *result) {
|
|
191
|
-
|
|
192
|
-
// On very small register sizes, NEON is at least as fast as SVE.
|
|
193
|
-
simsimd_size_t const words_per_register = svcntb();
|
|
194
|
-
if (words_per_register <= 32) {
|
|
195
|
-
simsimd_hamming_b8_neon(a, b, n_words, result);
|
|
196
|
-
return;
|
|
197
|
-
}
|
|
198
|
-
|
|
199
|
-
// On larger register sizes, SVE is faster.
|
|
200
|
-
simsimd_size_t i = 0, cycle = 0;
|
|
201
|
-
simsimd_i32_t differences = 0;
|
|
202
|
-
svuint8_t differences_cycle_vec = svdup_n_u8(0);
|
|
203
|
-
svbool_t const all_vec = svptrue_b8();
|
|
204
|
-
while (i < n_words) {
|
|
205
|
-
do {
|
|
206
|
-
svbool_t pg_vec = svwhilelt_b8((unsigned int)i, (unsigned int)n_words);
|
|
207
|
-
svuint8_t a_vec = svld1_u8(pg_vec, a + i);
|
|
208
|
-
svuint8_t b_vec = svld1_u8(pg_vec, b + i);
|
|
209
|
-
differences_cycle_vec =
|
|
210
|
-
svadd_u8_z(all_vec, differences_cycle_vec, svcnt_u8_x(all_vec, sveor_u8_m(all_vec, a_vec, b_vec)));
|
|
211
|
-
i += words_per_register;
|
|
212
|
-
++cycle;
|
|
213
|
-
} while (i < n_words && cycle < 31);
|
|
214
|
-
differences += svaddv_u8(all_vec, differences_cycle_vec);
|
|
215
|
-
differences_cycle_vec = svdup_n_u8(0);
|
|
216
|
-
cycle = 0; // Reset the cycle counter.
|
|
217
|
-
}
|
|
218
|
-
|
|
219
|
-
*result = differences;
|
|
220
|
-
}
|
|
221
|
-
|
|
222
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_sve(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
223
|
-
simsimd_distance_t *result) {
|
|
224
|
-
|
|
225
|
-
// On very small register sizes, NEON is at least as fast as SVE.
|
|
226
|
-
simsimd_size_t const words_per_register = svcntb();
|
|
227
|
-
if (words_per_register <= 32) {
|
|
228
|
-
simsimd_jaccard_b8_neon(a, b, n_words, result);
|
|
229
|
-
return;
|
|
230
|
-
}
|
|
231
|
-
|
|
232
|
-
// On larger register sizes, SVE is faster.
|
|
233
|
-
simsimd_size_t i = 0, cycle = 0;
|
|
234
|
-
simsimd_i32_t intersection = 0, union_ = 0;
|
|
235
|
-
svuint8_t intersection_cycle_vec = svdup_n_u8(0);
|
|
236
|
-
svuint8_t union_cycle_vec = svdup_n_u8(0);
|
|
237
|
-
svbool_t const all_vec = svptrue_b8();
|
|
238
|
-
while (i < n_words) {
|
|
239
|
-
do {
|
|
240
|
-
svbool_t pg_vec = svwhilelt_b8((unsigned int)i, (unsigned int)n_words);
|
|
241
|
-
svuint8_t a_vec = svld1_u8(pg_vec, a + i);
|
|
242
|
-
svuint8_t b_vec = svld1_u8(pg_vec, b + i);
|
|
243
|
-
intersection_cycle_vec =
|
|
244
|
-
svadd_u8_z(all_vec, intersection_cycle_vec, svcnt_u8_x(all_vec, svand_u8_m(all_vec, a_vec, b_vec)));
|
|
245
|
-
union_cycle_vec =
|
|
246
|
-
svadd_u8_z(all_vec, union_cycle_vec, svcnt_u8_x(all_vec, svorr_u8_m(all_vec, a_vec, b_vec)));
|
|
247
|
-
i += words_per_register;
|
|
248
|
-
++cycle;
|
|
249
|
-
} while (i < n_words && cycle < 31);
|
|
250
|
-
intersection += svaddv_u8(all_vec, intersection_cycle_vec);
|
|
251
|
-
intersection_cycle_vec = svdup_n_u8(0);
|
|
252
|
-
union_ += svaddv_u8(all_vec, union_cycle_vec);
|
|
253
|
-
union_cycle_vec = svdup_n_u8(0);
|
|
254
|
-
cycle = 0; // Reset the cycle counter.
|
|
255
|
-
}
|
|
256
|
-
|
|
257
|
-
*result = (union_ != 0) ? 1 - (simsimd_f64_t)intersection / (simsimd_f64_t)union_ : 1;
|
|
258
|
-
}
|
|
259
|
-
|
|
260
|
-
#pragma clang attribute pop
|
|
261
|
-
#pragma GCC pop_options
|
|
262
|
-
#endif // SIMSIMD_TARGET_SVE
|
|
263
|
-
#endif // _SIMSIMD_TARGET_ARM
|
|
264
|
-
|
|
265
|
-
#if _SIMSIMD_TARGET_X86
|
|
266
|
-
#if SIMSIMD_TARGET_ICE
|
|
267
|
-
#pragma GCC push_options
|
|
268
|
-
#pragma GCC target("avx2", "avx512f", "avx512vl", "bmi2", "avx512bw", "avx512vpopcntdq")
|
|
269
|
-
#pragma clang attribute push(__attribute__((target("avx2,avx512f,avx512vl,bmi2,avx512bw,avx512vpopcntdq"))), \
|
|
270
|
-
apply_to = function)
|
|
271
|
-
|
|
272
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_ice(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
273
|
-
simsimd_distance_t *result) {
|
|
274
|
-
|
|
275
|
-
simsimd_size_t xor_count;
|
|
276
|
-
// It's harder to squeeze out performance from tiny representations, so we unroll the loops for binary metrics.
|
|
277
|
-
if (n_words <= 64) { // Up to 512 bits.
|
|
278
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words);
|
|
279
|
-
__m512i a_vec = _mm512_maskz_loadu_epi8(mask, a);
|
|
280
|
-
__m512i b_vec = _mm512_maskz_loadu_epi8(mask, b);
|
|
281
|
-
__m512i xor_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a_vec, b_vec));
|
|
282
|
-
xor_count = _mm512_reduce_add_epi64(xor_count_vec);
|
|
283
|
-
}
|
|
284
|
-
else if (n_words <= 128) { // Up to 1024 bits.
|
|
285
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words - 64);
|
|
286
|
-
__m512i a1_vec = _mm512_loadu_epi8(a);
|
|
287
|
-
__m512i b1_vec = _mm512_loadu_epi8(b);
|
|
288
|
-
__m512i a2_vec = _mm512_maskz_loadu_epi8(mask, a + 64);
|
|
289
|
-
__m512i b2_vec = _mm512_maskz_loadu_epi8(mask, b + 64);
|
|
290
|
-
__m512i xor1_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a1_vec, b1_vec));
|
|
291
|
-
__m512i xor2_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a2_vec, b2_vec));
|
|
292
|
-
xor_count = _mm512_reduce_add_epi64(_mm512_add_epi64(xor2_count_vec, xor1_count_vec));
|
|
293
|
-
}
|
|
294
|
-
else if (n_words <= 196) { // Up to 1568 bits.
|
|
295
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words - 128);
|
|
296
|
-
__m512i a1_vec = _mm512_loadu_epi8(a);
|
|
297
|
-
__m512i b1_vec = _mm512_loadu_epi8(b);
|
|
298
|
-
__m512i a2_vec = _mm512_loadu_epi8(a + 64);
|
|
299
|
-
__m512i b2_vec = _mm512_loadu_epi8(b + 64);
|
|
300
|
-
__m512i a3_vec = _mm512_maskz_loadu_epi8(mask, a + 128);
|
|
301
|
-
__m512i b3_vec = _mm512_maskz_loadu_epi8(mask, b + 128);
|
|
302
|
-
__m512i xor1_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a1_vec, b1_vec));
|
|
303
|
-
__m512i xor2_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a2_vec, b2_vec));
|
|
304
|
-
__m512i xor3_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a3_vec, b3_vec));
|
|
305
|
-
xor_count =
|
|
306
|
-
_mm512_reduce_add_epi64(_mm512_add_epi64(xor3_count_vec, _mm512_add_epi64(xor2_count_vec, xor1_count_vec)));
|
|
307
|
-
}
|
|
308
|
-
else if (n_words <= 256) { // Up to 2048 bits.
|
|
309
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words - 192);
|
|
310
|
-
__m512i a1_vec = _mm512_loadu_epi8(a);
|
|
311
|
-
__m512i b1_vec = _mm512_loadu_epi8(b);
|
|
312
|
-
__m512i a2_vec = _mm512_loadu_epi8(a + 64);
|
|
313
|
-
__m512i b2_vec = _mm512_loadu_epi8(b + 64);
|
|
314
|
-
__m512i a3_vec = _mm512_loadu_epi8(a + 128);
|
|
315
|
-
__m512i b3_vec = _mm512_loadu_epi8(b + 128);
|
|
316
|
-
__m512i a4_vec = _mm512_maskz_loadu_epi8(mask, a + 192);
|
|
317
|
-
__m512i b4_vec = _mm512_maskz_loadu_epi8(mask, b + 192);
|
|
318
|
-
__m512i xor1_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a1_vec, b1_vec));
|
|
319
|
-
__m512i xor2_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a2_vec, b2_vec));
|
|
320
|
-
__m512i xor3_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a3_vec, b3_vec));
|
|
321
|
-
__m512i xor4_count_vec = _mm512_popcnt_epi64(_mm512_xor_si512(a4_vec, b4_vec));
|
|
322
|
-
xor_count = _mm512_reduce_add_epi64(_mm512_add_epi64(_mm512_add_epi64(xor4_count_vec, xor3_count_vec),
|
|
323
|
-
_mm512_add_epi64(xor2_count_vec, xor1_count_vec)));
|
|
324
|
-
}
|
|
325
|
-
else {
|
|
326
|
-
__m512i xor_count_vec = _mm512_setzero_si512();
|
|
327
|
-
__m512i a_vec, b_vec;
|
|
328
|
-
|
|
329
|
-
simsimd_hamming_b8_ice_cycle:
|
|
330
|
-
if (n_words < 64) {
|
|
331
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words);
|
|
332
|
-
a_vec = _mm512_maskz_loadu_epi8(mask, a);
|
|
333
|
-
b_vec = _mm512_maskz_loadu_epi8(mask, b);
|
|
334
|
-
n_words = 0;
|
|
335
|
-
}
|
|
336
|
-
else {
|
|
337
|
-
a_vec = _mm512_loadu_epi8(a);
|
|
338
|
-
b_vec = _mm512_loadu_epi8(b);
|
|
339
|
-
a += 64, b += 64, n_words -= 64;
|
|
340
|
-
}
|
|
341
|
-
__m512i xor_vec = _mm512_xor_si512(a_vec, b_vec);
|
|
342
|
-
xor_count_vec = _mm512_add_epi64(xor_count_vec, _mm512_popcnt_epi64(xor_vec));
|
|
343
|
-
if (n_words) goto simsimd_hamming_b8_ice_cycle;
|
|
344
|
-
|
|
345
|
-
xor_count = _mm512_reduce_add_epi64(xor_count_vec);
|
|
346
|
-
}
|
|
347
|
-
*result = xor_count;
|
|
348
|
-
}
|
|
349
|
-
|
|
350
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_ice(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
351
|
-
simsimd_distance_t *result) {
|
|
352
|
-
|
|
353
|
-
simsimd_size_t intersection = 0, union_ = 0;
|
|
354
|
-
// It's harder to squeeze out performance from tiny representations, so we unroll the loops for binary metrics.
|
|
355
|
-
if (n_words <= 64) { // Up to 512 bits.
|
|
356
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words);
|
|
357
|
-
__m512i a_vec = _mm512_maskz_loadu_epi8(mask, a);
|
|
358
|
-
__m512i b_vec = _mm512_maskz_loadu_epi8(mask, b);
|
|
359
|
-
__m512i and_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a_vec, b_vec));
|
|
360
|
-
__m512i or_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a_vec, b_vec));
|
|
361
|
-
intersection = _mm512_reduce_add_epi64(and_count_vec);
|
|
362
|
-
union_ = _mm512_reduce_add_epi64(or_count_vec);
|
|
363
|
-
}
|
|
364
|
-
else if (n_words <= 128) { // Up to 1024 bits.
|
|
365
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words - 64);
|
|
366
|
-
__m512i a1_vec = _mm512_loadu_epi8(a);
|
|
367
|
-
__m512i b1_vec = _mm512_loadu_epi8(b);
|
|
368
|
-
__m512i a2_vec = _mm512_maskz_loadu_epi8(mask, a + 64);
|
|
369
|
-
__m512i b2_vec = _mm512_maskz_loadu_epi8(mask, b + 64);
|
|
370
|
-
__m512i and1_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a1_vec, b1_vec));
|
|
371
|
-
__m512i or1_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a1_vec, b1_vec));
|
|
372
|
-
__m512i and2_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a2_vec, b2_vec));
|
|
373
|
-
__m512i or2_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a2_vec, b2_vec));
|
|
374
|
-
intersection = _mm512_reduce_add_epi64(_mm512_add_epi64(and2_count_vec, and1_count_vec));
|
|
375
|
-
union_ = _mm512_reduce_add_epi64(_mm512_add_epi64(or2_count_vec, or1_count_vec));
|
|
376
|
-
}
|
|
377
|
-
else if (n_words <= 196) { // Up to 1568 bits.
|
|
378
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words - 128);
|
|
379
|
-
__m512i a1_vec = _mm512_loadu_epi8(a);
|
|
380
|
-
__m512i b1_vec = _mm512_loadu_epi8(b);
|
|
381
|
-
__m512i a2_vec = _mm512_loadu_epi8(a + 64);
|
|
382
|
-
__m512i b2_vec = _mm512_loadu_epi8(b + 64);
|
|
383
|
-
__m512i a3_vec = _mm512_maskz_loadu_epi8(mask, a + 128);
|
|
384
|
-
__m512i b3_vec = _mm512_maskz_loadu_epi8(mask, b + 128);
|
|
385
|
-
__m512i and1_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a1_vec, b1_vec));
|
|
386
|
-
__m512i or1_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a1_vec, b1_vec));
|
|
387
|
-
__m512i and2_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a2_vec, b2_vec));
|
|
388
|
-
__m512i or2_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a2_vec, b2_vec));
|
|
389
|
-
__m512i and3_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a3_vec, b3_vec));
|
|
390
|
-
__m512i or3_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a3_vec, b3_vec));
|
|
391
|
-
intersection = _mm512_reduce_add_epi64( //
|
|
392
|
-
_mm512_add_epi64(and3_count_vec, _mm512_add_epi64(and2_count_vec, and1_count_vec)));
|
|
393
|
-
union_ = _mm512_reduce_add_epi64( //
|
|
394
|
-
_mm512_add_epi64(or3_count_vec, _mm512_add_epi64(or2_count_vec, or1_count_vec)));
|
|
395
|
-
}
|
|
396
|
-
else if (n_words <= 256) { // Up to 2048 bits.
|
|
397
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words - 192);
|
|
398
|
-
__m512i a1_vec = _mm512_loadu_epi8(a);
|
|
399
|
-
__m512i b1_vec = _mm512_loadu_epi8(b);
|
|
400
|
-
__m512i a2_vec = _mm512_loadu_epi8(a + 64);
|
|
401
|
-
__m512i b2_vec = _mm512_loadu_epi8(b + 64);
|
|
402
|
-
__m512i a3_vec = _mm512_loadu_epi8(a + 128);
|
|
403
|
-
__m512i b3_vec = _mm512_loadu_epi8(b + 128);
|
|
404
|
-
__m512i a4_vec = _mm512_maskz_loadu_epi8(mask, a + 192);
|
|
405
|
-
__m512i b4_vec = _mm512_maskz_loadu_epi8(mask, b + 192);
|
|
406
|
-
__m512i and1_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a1_vec, b1_vec));
|
|
407
|
-
__m512i or1_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a1_vec, b1_vec));
|
|
408
|
-
__m512i and2_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a2_vec, b2_vec));
|
|
409
|
-
__m512i or2_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a2_vec, b2_vec));
|
|
410
|
-
__m512i and3_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a3_vec, b3_vec));
|
|
411
|
-
__m512i or3_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a3_vec, b3_vec));
|
|
412
|
-
__m512i and4_count_vec = _mm512_popcnt_epi64(_mm512_and_si512(a4_vec, b4_vec));
|
|
413
|
-
__m512i or4_count_vec = _mm512_popcnt_epi64(_mm512_or_si512(a4_vec, b4_vec));
|
|
414
|
-
intersection = _mm512_reduce_add_epi64(_mm512_add_epi64(_mm512_add_epi64(and4_count_vec, and3_count_vec),
|
|
415
|
-
_mm512_add_epi64(and2_count_vec, and1_count_vec)));
|
|
416
|
-
union_ = _mm512_reduce_add_epi64(_mm512_add_epi64(_mm512_add_epi64(or4_count_vec, or3_count_vec),
|
|
417
|
-
_mm512_add_epi64(or2_count_vec, or1_count_vec)));
|
|
418
|
-
}
|
|
419
|
-
else {
|
|
420
|
-
__m512i and_count_vec = _mm512_setzero_si512(), or_count_vec = _mm512_setzero_si512();
|
|
421
|
-
__m512i a_vec, b_vec;
|
|
422
|
-
|
|
423
|
-
simsimd_jaccard_b8_ice_cycle:
|
|
424
|
-
if (n_words < 64) {
|
|
425
|
-
__mmask64 mask = (__mmask64)_bzhi_u64(0xFFFFFFFFFFFFFFFF, n_words);
|
|
426
|
-
a_vec = _mm512_maskz_loadu_epi8(mask, a);
|
|
427
|
-
b_vec = _mm512_maskz_loadu_epi8(mask, b);
|
|
428
|
-
n_words = 0;
|
|
429
|
-
}
|
|
430
|
-
else {
|
|
431
|
-
a_vec = _mm512_loadu_epi8(a);
|
|
432
|
-
b_vec = _mm512_loadu_epi8(b);
|
|
433
|
-
a += 64, b += 64, n_words -= 64;
|
|
434
|
-
}
|
|
435
|
-
__m512i and_vec = _mm512_and_si512(a_vec, b_vec);
|
|
436
|
-
__m512i or_vec = _mm512_or_si512(a_vec, b_vec);
|
|
437
|
-
and_count_vec = _mm512_add_epi64(and_count_vec, _mm512_popcnt_epi64(and_vec));
|
|
438
|
-
or_count_vec = _mm512_add_epi64(or_count_vec, _mm512_popcnt_epi64(or_vec));
|
|
439
|
-
if (n_words) goto simsimd_jaccard_b8_ice_cycle;
|
|
440
|
-
|
|
441
|
-
intersection = _mm512_reduce_add_epi64(and_count_vec);
|
|
442
|
-
union_ = _mm512_reduce_add_epi64(or_count_vec);
|
|
443
|
-
}
|
|
444
|
-
*result = (union_ != 0) ? 1 - (simsimd_f64_t)intersection / (simsimd_f64_t)union_ : 1;
|
|
445
|
-
}
|
|
446
|
-
|
|
447
|
-
#pragma clang attribute pop
|
|
448
|
-
#pragma GCC pop_options
|
|
449
|
-
#endif // SIMSIMD_TARGET_ICE
|
|
450
|
-
|
|
451
|
-
#if SIMSIMD_TARGET_HASWELL
|
|
452
|
-
#pragma GCC push_options
|
|
453
|
-
#pragma GCC target("popcnt")
|
|
454
|
-
#pragma clang attribute push(__attribute__((target("popcnt"))), apply_to = function)
|
|
455
|
-
|
|
456
|
-
SIMSIMD_PUBLIC void simsimd_hamming_b8_haswell(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
457
|
-
simsimd_distance_t *result) {
|
|
458
|
-
// x86 supports unaligned loads and works just fine with the scalar version for small vectors.
|
|
459
|
-
simsimd_size_t differences = 0;
|
|
460
|
-
for (; n_words >= 8; n_words -= 8, a += 8, b += 8)
|
|
461
|
-
differences += _mm_popcnt_u64(*(simsimd_u64_t const *)a ^ *(simsimd_u64_t const *)b);
|
|
462
|
-
for (; n_words; --n_words, ++a, ++b) differences += _mm_popcnt_u32(*a ^ *b);
|
|
463
|
-
*result = differences;
|
|
464
|
-
}
|
|
465
|
-
|
|
466
|
-
SIMSIMD_PUBLIC void simsimd_jaccard_b8_haswell(simsimd_b8_t const *a, simsimd_b8_t const *b, simsimd_size_t n_words,
|
|
467
|
-
simsimd_distance_t *result) {
|
|
468
|
-
// x86 supports unaligned loads and works just fine with the scalar version for small vectors.
|
|
469
|
-
simsimd_size_t intersection = 0, union_ = 0;
|
|
470
|
-
for (; n_words >= 8; n_words -= 8, a += 8, b += 8)
|
|
471
|
-
intersection += _mm_popcnt_u64(*(simsimd_u64_t const *)a & *(simsimd_u64_t const *)b),
|
|
472
|
-
union_ += _mm_popcnt_u64(*(simsimd_u64_t const *)a | *(simsimd_u64_t const *)b);
|
|
473
|
-
for (; n_words; --n_words, ++a, ++b) intersection += _mm_popcnt_u32(*a & *b), union_ += _mm_popcnt_u32(*a | *b);
|
|
474
|
-
*result = (union_ != 0) ? 1 - (simsimd_f64_t)intersection / (simsimd_f64_t)union_ : 1;
|
|
475
|
-
}
|
|
476
|
-
|
|
477
|
-
#pragma clang attribute pop
|
|
478
|
-
#pragma GCC pop_options
|
|
479
|
-
#endif // SIMSIMD_TARGET_HASWELL
|
|
480
|
-
#endif // _SIMSIMD_TARGET_X86
|
|
481
|
-
|
|
482
|
-
#ifdef __cplusplus
|
|
483
|
-
}
|
|
484
|
-
#endif
|
|
485
|
-
|
|
486
|
-
#endif
|