usearch 2.21.4 → 2.25.1

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.
Files changed (604) hide show
  1. package/README.md +65 -69
  2. package/binding.gyp +15 -37
  3. package/c/CMakeLists.txt +4 -5
  4. package/c/README.md +8 -8
  5. package/c/lib.cpp +32 -14
  6. package/c/test.c +124 -1
  7. package/c/usearch.h +24 -3
  8. package/cpp/CMakeLists.txt +4 -0
  9. package/cpp/README.md +5 -5
  10. package/cpp/bench.cpp +97 -116
  11. package/cpp/test.cpp +117 -20
  12. package/include/usearch/index.hpp +224 -48
  13. package/include/usearch/index_dense.hpp +105 -40
  14. package/include/usearch/index_plugins.hpp +1387 -204
  15. package/javascript/README.md +1 -1
  16. package/javascript/dist/cjs/usearch.d.ts +8 -3
  17. package/javascript/dist/cjs/usearch.js +12 -4
  18. package/javascript/dist/esm/usearch.d.ts +8 -3
  19. package/javascript/dist/esm/usearch.js +12 -4
  20. package/javascript/lib.cpp +17 -0
  21. package/javascript/usearch.test.js +117 -94
  22. package/javascript/usearch.ts +14 -6
  23. package/numkong/.clang-tidy +30 -0
  24. package/{simsimd → numkong}/.cmake-format.py +18 -0
  25. package/numkong/.editorconfig +12 -0
  26. package/numkong/.git-blame-ignore-revs +123 -0
  27. package/numkong/.mailmap +15 -0
  28. package/numkong/.spi.yml +11 -0
  29. package/numkong/.swift-format +14 -0
  30. package/numkong/CITATION.cff +81 -0
  31. package/numkong/CONTRIBUTING.md +500 -0
  32. package/numkong/Cargo.toml +54 -0
  33. package/numkong/MANIFEST.in +8 -0
  34. package/numkong/Package.swift +84 -0
  35. package/numkong/README.md +569 -0
  36. package/numkong/bench/README.md +217 -0
  37. package/numkong/bench/bench-browser.html +323 -0
  38. package/numkong/bench/bench.cpp +344 -0
  39. package/numkong/bench/bench.hpp +816 -0
  40. package/numkong/bench/bench.js +148 -0
  41. package/numkong/bench/bench.mjs +616 -0
  42. package/numkong/bench/bench_attention.cpp +113 -0
  43. package/numkong/bench/bench_cast.cpp +264 -0
  44. package/numkong/bench/bench_cross_amx.cpp +62 -0
  45. package/numkong/bench/bench_cross_arm.cpp +245 -0
  46. package/numkong/bench/bench_cross_blas.cpp +309 -0
  47. package/numkong/bench/bench_cross_loongarch.cpp +106 -0
  48. package/numkong/bench/bench_cross_power.cpp +84 -0
  49. package/numkong/bench/bench_cross_rvv.cpp +121 -0
  50. package/numkong/bench/bench_cross_serial.cpp +151 -0
  51. package/numkong/bench/bench_cross_sme.cpp +163 -0
  52. package/numkong/bench/bench_cross_wasm.cpp +142 -0
  53. package/numkong/bench/bench_cross_x86.cpp +455 -0
  54. package/numkong/bench/bench_curved.cpp +210 -0
  55. package/numkong/bench/bench_dot.cpp +315 -0
  56. package/numkong/bench/bench_each.cpp +503 -0
  57. package/numkong/bench/bench_geospatial.cpp +112 -0
  58. package/numkong/bench/bench_maxsim.cpp +154 -0
  59. package/numkong/bench/bench_mesh.cpp +148 -0
  60. package/numkong/bench/bench_probability.cpp +60 -0
  61. package/numkong/bench/bench_reduce.cpp +371 -0
  62. package/numkong/bench/bench_set.cpp +73 -0
  63. package/numkong/bench/bench_sparse.cpp +198 -0
  64. package/numkong/bench/bench_spatial.cpp +347 -0
  65. package/numkong/bench/bench_trigonometry.cpp +137 -0
  66. package/numkong/bench/swift/Bench.swift +506 -0
  67. package/numkong/binding.gyp +169 -0
  68. package/numkong/build.rs +484 -0
  69. package/numkong/c/dispatch.h +512 -0
  70. package/numkong/c/dispatch_bf16.c +447 -0
  71. package/numkong/c/dispatch_bf16c.c +52 -0
  72. package/numkong/c/dispatch_e2m3.c +296 -0
  73. package/numkong/c/dispatch_e3m2.c +284 -0
  74. package/numkong/c/dispatch_e4m3.c +318 -0
  75. package/numkong/c/dispatch_e5m2.c +316 -0
  76. package/numkong/c/dispatch_f16.c +452 -0
  77. package/numkong/c/dispatch_f16c.c +58 -0
  78. package/numkong/c/dispatch_f32.c +434 -0
  79. package/numkong/c/dispatch_f32c.c +99 -0
  80. package/numkong/c/dispatch_f64.c +348 -0
  81. package/numkong/c/dispatch_f64c.c +98 -0
  82. package/numkong/c/dispatch_i16.c +96 -0
  83. package/numkong/c/dispatch_i32.c +89 -0
  84. package/numkong/c/dispatch_i4.c +153 -0
  85. package/numkong/c/dispatch_i64.c +86 -0
  86. package/numkong/c/dispatch_i8.c +348 -0
  87. package/numkong/c/dispatch_other.c +348 -0
  88. package/numkong/c/dispatch_u1.c +193 -0
  89. package/numkong/c/dispatch_u16.c +124 -0
  90. package/numkong/c/dispatch_u32.c +118 -0
  91. package/numkong/c/dispatch_u4.c +153 -0
  92. package/numkong/c/dispatch_u64.c +102 -0
  93. package/numkong/c/dispatch_u8.c +364 -0
  94. package/numkong/c/numkong.c +940 -0
  95. package/numkong/cmake/nk_arm_isa_probes.cmake +61 -0
  96. package/numkong/cmake/nk_isa_probe.cmake +90 -0
  97. package/numkong/cmake/nk_loongarch_isa_probes.cmake +15 -0
  98. package/numkong/cmake/nk_math_lib.cmake +46 -0
  99. package/numkong/cmake/nk_power_isa_probes.cmake +15 -0
  100. package/numkong/cmake/nk_riscv_isa_probes.cmake +19 -0
  101. package/numkong/cmake/nk_x86_isa_probes.cmake +35 -0
  102. package/numkong/cmake/toolchain-aarch64-gnu.cmake +86 -0
  103. package/numkong/cmake/toolchain-android-arm64.cmake +25 -0
  104. package/numkong/cmake/toolchain-android-armv7.cmake +25 -0
  105. package/numkong/cmake/toolchain-loongarch64-gnu.cmake +68 -0
  106. package/numkong/cmake/toolchain-ppc64le-gnu.cmake +74 -0
  107. package/numkong/cmake/toolchain-riscv64-gnu.cmake +107 -0
  108. package/numkong/cmake/toolchain-riscv64-llvm.cmake +114 -0
  109. package/numkong/cmake/toolchain-wasi.cmake +77 -0
  110. package/numkong/cmake/toolchain-wasm.cmake +70 -0
  111. package/numkong/cmake/toolchain-wasm64.cmake +62 -0
  112. package/numkong/cmake/toolchain-x86_64-llvm.cmake +47 -0
  113. package/numkong/include/README.md +643 -0
  114. package/numkong/include/module.modulemap +129 -0
  115. package/numkong/include/numkong/attention/sapphireamx.h +1363 -0
  116. package/numkong/include/numkong/attention/sme.h +2075 -0
  117. package/numkong/include/numkong/attention.h +49 -0
  118. package/numkong/include/numkong/capabilities.h +930 -0
  119. package/numkong/include/numkong/cast/README.md +262 -0
  120. package/numkong/include/numkong/cast/diamond.h +64 -0
  121. package/numkong/include/numkong/cast/haswell.h +823 -0
  122. package/numkong/include/numkong/cast/icelake.h +475 -0
  123. package/numkong/include/numkong/cast/loongsonasx.h +252 -0
  124. package/numkong/include/numkong/cast/neon.h +1174 -0
  125. package/numkong/include/numkong/cast/powervsx.h +449 -0
  126. package/numkong/include/numkong/cast/rvv.h +970 -0
  127. package/numkong/include/numkong/cast/sapphire.h +262 -0
  128. package/numkong/include/numkong/cast/serial.h +2336 -0
  129. package/numkong/include/numkong/cast/skylake.h +915 -0
  130. package/numkong/include/numkong/cast/v128relaxed.h +537 -0
  131. package/numkong/include/numkong/cast.h +254 -0
  132. package/numkong/include/numkong/cast.hpp +44 -0
  133. package/numkong/include/numkong/curved/README.md +223 -0
  134. package/numkong/include/numkong/curved/genoa.h +182 -0
  135. package/numkong/include/numkong/curved/haswell.h +276 -0
  136. package/numkong/include/numkong/curved/neon.h +329 -0
  137. package/numkong/include/numkong/curved/neonbfdot.h +211 -0
  138. package/numkong/include/numkong/curved/rvv.h +305 -0
  139. package/numkong/include/numkong/curved/serial.h +207 -0
  140. package/numkong/include/numkong/curved/skylake.h +457 -0
  141. package/numkong/include/numkong/curved/smef64.h +530 -0
  142. package/numkong/include/numkong/curved.h +513 -0
  143. package/numkong/include/numkong/curved.hpp +144 -0
  144. package/numkong/include/numkong/dot/README.md +442 -0
  145. package/numkong/include/numkong/dot/alder.h +563 -0
  146. package/numkong/include/numkong/dot/diamond.h +158 -0
  147. package/numkong/include/numkong/dot/genoa.h +289 -0
  148. package/numkong/include/numkong/dot/haswell.h +1723 -0
  149. package/numkong/include/numkong/dot/icelake.h +997 -0
  150. package/numkong/include/numkong/dot/loongsonasx.h +671 -0
  151. package/numkong/include/numkong/dot/neon.h +869 -0
  152. package/numkong/include/numkong/dot/neonbfdot.h +243 -0
  153. package/numkong/include/numkong/dot/neonfhm.h +358 -0
  154. package/numkong/include/numkong/dot/neonfp8.h +323 -0
  155. package/numkong/include/numkong/dot/neonsdot.h +622 -0
  156. package/numkong/include/numkong/dot/powervsx.h +752 -0
  157. package/numkong/include/numkong/dot/rvv.h +722 -0
  158. package/numkong/include/numkong/dot/rvvbb.h +72 -0
  159. package/numkong/include/numkong/dot/rvvbf16.h +123 -0
  160. package/numkong/include/numkong/dot/rvvhalf.h +129 -0
  161. package/numkong/include/numkong/dot/sapphire.h +141 -0
  162. package/numkong/include/numkong/dot/serial.h +874 -0
  163. package/numkong/include/numkong/dot/sierra.h +405 -0
  164. package/numkong/include/numkong/dot/skylake.h +1116 -0
  165. package/numkong/include/numkong/dot/sve.h +420 -0
  166. package/numkong/include/numkong/dot/svebfdot.h +75 -0
  167. package/numkong/include/numkong/dot/svehalf.h +168 -0
  168. package/numkong/include/numkong/dot/svesdot.h +90 -0
  169. package/numkong/include/numkong/dot/v128relaxed.h +1293 -0
  170. package/numkong/include/numkong/dot.h +1136 -0
  171. package/numkong/include/numkong/dot.hpp +94 -0
  172. package/numkong/include/numkong/dots/README.md +496 -0
  173. package/numkong/include/numkong/dots/alder.h +117 -0
  174. package/numkong/include/numkong/dots/diamond.h +86 -0
  175. package/numkong/include/numkong/dots/genoa.h +100 -0
  176. package/numkong/include/numkong/dots/graniteamx.h +733 -0
  177. package/numkong/include/numkong/dots/haswell.h +310 -0
  178. package/numkong/include/numkong/dots/icelake.h +180 -0
  179. package/numkong/include/numkong/dots/loongsonasx.h +176 -0
  180. package/numkong/include/numkong/dots/neon.h +123 -0
  181. package/numkong/include/numkong/dots/neonbfdot.h +59 -0
  182. package/numkong/include/numkong/dots/neonfhm.h +96 -0
  183. package/numkong/include/numkong/dots/neonfp8.h +99 -0
  184. package/numkong/include/numkong/dots/neonsdot.h +144 -0
  185. package/numkong/include/numkong/dots/powervsx.h +194 -0
  186. package/numkong/include/numkong/dots/rvv.h +2593 -0
  187. package/numkong/include/numkong/dots/sapphireamx.h +4017 -0
  188. package/numkong/include/numkong/dots/serial.h +2867 -0
  189. package/numkong/include/numkong/dots/sierra.h +100 -0
  190. package/numkong/include/numkong/dots/skylake.h +207 -0
  191. package/numkong/include/numkong/dots/sme.h +5041 -0
  192. package/numkong/include/numkong/dots/smebi32.h +477 -0
  193. package/numkong/include/numkong/dots/smef64.h +1328 -0
  194. package/numkong/include/numkong/dots/smehalf.h +47 -0
  195. package/numkong/include/numkong/dots/v128relaxed.h +317 -0
  196. package/numkong/include/numkong/dots.h +3110 -0
  197. package/numkong/include/numkong/dots.hpp +641 -0
  198. package/numkong/include/numkong/each/README.md +465 -0
  199. package/numkong/include/numkong/each/haswell.h +1658 -0
  200. package/numkong/include/numkong/each/icelake.h +272 -0
  201. package/numkong/include/numkong/each/neon.h +1138 -0
  202. package/numkong/include/numkong/each/neonbfdot.h +211 -0
  203. package/numkong/include/numkong/each/neonhalf.h +318 -0
  204. package/numkong/include/numkong/each/rvv.h +1121 -0
  205. package/numkong/include/numkong/each/sapphire.h +343 -0
  206. package/numkong/include/numkong/each/serial.h +282 -0
  207. package/numkong/include/numkong/each/skylake.h +1562 -0
  208. package/numkong/include/numkong/each/v128relaxed.h +562 -0
  209. package/numkong/include/numkong/each.h +2232 -0
  210. package/numkong/include/numkong/each.hpp +434 -0
  211. package/numkong/include/numkong/geospatial/README.md +147 -0
  212. package/numkong/include/numkong/geospatial/haswell.h +633 -0
  213. package/numkong/include/numkong/geospatial/neon.h +615 -0
  214. package/numkong/include/numkong/geospatial/rvv.h +701 -0
  215. package/numkong/include/numkong/geospatial/serial.h +309 -0
  216. package/numkong/include/numkong/geospatial/skylake.h +613 -0
  217. package/numkong/include/numkong/geospatial/v128relaxed.h +652 -0
  218. package/numkong/include/numkong/geospatial.h +453 -0
  219. package/numkong/include/numkong/geospatial.hpp +235 -0
  220. package/numkong/include/numkong/matrix.hpp +336 -0
  221. package/numkong/include/numkong/maxsim/README.md +187 -0
  222. package/numkong/include/numkong/maxsim/alder.h +511 -0
  223. package/numkong/include/numkong/maxsim/genoa.h +115 -0
  224. package/numkong/include/numkong/maxsim/haswell.h +553 -0
  225. package/numkong/include/numkong/maxsim/icelake.h +480 -0
  226. package/numkong/include/numkong/maxsim/neonsdot.h +394 -0
  227. package/numkong/include/numkong/maxsim/sapphireamx.h +877 -0
  228. package/numkong/include/numkong/maxsim/serial.h +490 -0
  229. package/numkong/include/numkong/maxsim/sme.h +993 -0
  230. package/numkong/include/numkong/maxsim/v128relaxed.h +280 -0
  231. package/numkong/include/numkong/maxsim.h +571 -0
  232. package/numkong/include/numkong/maxsim.hpp +133 -0
  233. package/numkong/include/numkong/mesh/README.md +219 -0
  234. package/numkong/include/numkong/mesh/haswell.h +2013 -0
  235. package/numkong/include/numkong/mesh/neon.h +1821 -0
  236. package/numkong/include/numkong/mesh/neonbfdot.h +771 -0
  237. package/numkong/include/numkong/mesh/rvv.h +936 -0
  238. package/numkong/include/numkong/mesh/serial.h +721 -0
  239. package/numkong/include/numkong/mesh/skylake.h +2344 -0
  240. package/numkong/include/numkong/mesh/v128relaxed.h +1081 -0
  241. package/numkong/include/numkong/mesh.h +666 -0
  242. package/numkong/include/numkong/mesh.hpp +701 -0
  243. package/numkong/include/numkong/numkong.h +78 -0
  244. package/numkong/include/numkong/numkong.hpp +58 -0
  245. package/numkong/include/numkong/probability/README.md +177 -0
  246. package/numkong/include/numkong/probability/haswell.h +267 -0
  247. package/numkong/include/numkong/probability/neon.h +255 -0
  248. package/numkong/include/numkong/probability/rvv.h +414 -0
  249. package/numkong/include/numkong/probability/serial.h +181 -0
  250. package/numkong/include/numkong/probability/skylake.h +324 -0
  251. package/numkong/include/numkong/probability.h +380 -0
  252. package/numkong/include/numkong/probability.hpp +120 -0
  253. package/numkong/include/numkong/random.h +50 -0
  254. package/numkong/include/numkong/random.hpp +285 -0
  255. package/numkong/include/numkong/reduce/README.md +552 -0
  256. package/numkong/include/numkong/reduce/alder.h +636 -0
  257. package/numkong/include/numkong/reduce/genoa.h +201 -0
  258. package/numkong/include/numkong/reduce/haswell.h +3785 -0
  259. package/numkong/include/numkong/reduce/icelake.h +549 -0
  260. package/numkong/include/numkong/reduce/neon.h +3980 -0
  261. package/numkong/include/numkong/reduce/neonbfdot.h +142 -0
  262. package/numkong/include/numkong/reduce/neonfhm.h +293 -0
  263. package/numkong/include/numkong/reduce/neonsdot.h +357 -0
  264. package/numkong/include/numkong/reduce/rvv.h +3533 -0
  265. package/numkong/include/numkong/reduce/serial.h +753 -0
  266. package/numkong/include/numkong/reduce/sierra.h +338 -0
  267. package/numkong/include/numkong/reduce/skylake.h +3794 -0
  268. package/numkong/include/numkong/reduce/sve.h +52 -0
  269. package/numkong/include/numkong/reduce/v128relaxed.h +2302 -0
  270. package/numkong/include/numkong/reduce.h +1582 -0
  271. package/numkong/include/numkong/reduce.hpp +778 -0
  272. package/numkong/include/numkong/scalar/README.md +89 -0
  273. package/numkong/include/numkong/scalar/haswell.h +122 -0
  274. package/numkong/include/numkong/scalar/loongsonasx.h +74 -0
  275. package/numkong/include/numkong/scalar/neon.h +122 -0
  276. package/numkong/include/numkong/scalar/neonhalf.h +70 -0
  277. package/numkong/include/numkong/scalar/powervsx.h +96 -0
  278. package/numkong/include/numkong/scalar/rvv.h +211 -0
  279. package/numkong/include/numkong/scalar/sapphire.h +74 -0
  280. package/numkong/include/numkong/scalar/serial.h +332 -0
  281. package/numkong/include/numkong/scalar/v128relaxed.h +56 -0
  282. package/numkong/include/numkong/scalar.h +696 -0
  283. package/numkong/include/numkong/set/README.md +179 -0
  284. package/numkong/include/numkong/set/haswell.h +334 -0
  285. package/numkong/include/numkong/set/icelake.h +485 -0
  286. package/numkong/include/numkong/set/loongsonasx.h +181 -0
  287. package/numkong/include/numkong/set/neon.h +363 -0
  288. package/numkong/include/numkong/set/powervsx.h +326 -0
  289. package/numkong/include/numkong/set/rvv.h +226 -0
  290. package/numkong/include/numkong/set/rvvbb.h +117 -0
  291. package/numkong/include/numkong/set/serial.h +174 -0
  292. package/numkong/include/numkong/set/sve.h +187 -0
  293. package/numkong/include/numkong/set/v128relaxed.h +240 -0
  294. package/numkong/include/numkong/set.h +471 -0
  295. package/numkong/include/numkong/set.hpp +114 -0
  296. package/numkong/include/numkong/sets/README.md +149 -0
  297. package/numkong/include/numkong/sets/haswell.h +63 -0
  298. package/numkong/include/numkong/sets/icelake.h +66 -0
  299. package/numkong/include/numkong/sets/loongsonasx.h +52 -0
  300. package/numkong/include/numkong/sets/neon.h +61 -0
  301. package/numkong/include/numkong/sets/powervsx.h +65 -0
  302. package/numkong/include/numkong/sets/serial.h +43 -0
  303. package/numkong/include/numkong/sets/smebi32.h +1135 -0
  304. package/numkong/include/numkong/sets/v128relaxed.h +58 -0
  305. package/numkong/include/numkong/sets.h +382 -0
  306. package/numkong/include/numkong/sparse/README.md +156 -0
  307. package/numkong/include/numkong/sparse/icelake.h +463 -0
  308. package/numkong/include/numkong/sparse/neon.h +288 -0
  309. package/numkong/include/numkong/sparse/serial.h +117 -0
  310. package/numkong/include/numkong/sparse/sve2.h +504 -0
  311. package/numkong/include/numkong/sparse/turin.h +322 -0
  312. package/numkong/include/numkong/sparse.h +363 -0
  313. package/numkong/include/numkong/sparse.hpp +113 -0
  314. package/numkong/include/numkong/spatial/README.md +439 -0
  315. package/numkong/include/numkong/spatial/alder.h +604 -0
  316. package/numkong/include/numkong/spatial/diamond.h +240 -0
  317. package/numkong/include/numkong/spatial/genoa.h +222 -0
  318. package/numkong/include/numkong/spatial/haswell.h +979 -0
  319. package/numkong/include/numkong/spatial/icelake.h +1067 -0
  320. package/numkong/include/numkong/spatial/loongsonasx.h +483 -0
  321. package/numkong/include/numkong/spatial/neon.h +846 -0
  322. package/numkong/include/numkong/spatial/neonbfdot.h +164 -0
  323. package/numkong/include/numkong/spatial/neonfp8.h +258 -0
  324. package/numkong/include/numkong/spatial/neonsdot.h +429 -0
  325. package/numkong/include/numkong/spatial/powervsx.h +738 -0
  326. package/numkong/include/numkong/spatial/rvv.h +991 -0
  327. package/numkong/include/numkong/spatial/rvvbf16.h +128 -0
  328. package/numkong/include/numkong/spatial/rvvhalf.h +120 -0
  329. package/numkong/include/numkong/spatial/serial.h +347 -0
  330. package/numkong/include/numkong/spatial/sierra.h +516 -0
  331. package/numkong/include/numkong/spatial/skylake.h +605 -0
  332. package/numkong/include/numkong/spatial/sve.h +246 -0
  333. package/numkong/include/numkong/spatial/svebfdot.h +125 -0
  334. package/numkong/include/numkong/spatial/svehalf.h +136 -0
  335. package/numkong/include/numkong/spatial/svesdot.h +143 -0
  336. package/numkong/include/numkong/spatial/v128relaxed.h +969 -0
  337. package/numkong/include/numkong/spatial.h +1681 -0
  338. package/numkong/include/numkong/spatial.hpp +183 -0
  339. package/numkong/include/numkong/spatials/README.md +580 -0
  340. package/numkong/include/numkong/spatials/alder.h +94 -0
  341. package/numkong/include/numkong/spatials/diamond.h +82 -0
  342. package/numkong/include/numkong/spatials/genoa.h +94 -0
  343. package/numkong/include/numkong/spatials/graniteamx.h +173 -0
  344. package/numkong/include/numkong/spatials/haswell.h +219 -0
  345. package/numkong/include/numkong/spatials/icelake.h +113 -0
  346. package/numkong/include/numkong/spatials/loongsonasx.h +153 -0
  347. package/numkong/include/numkong/spatials/neon.h +109 -0
  348. package/numkong/include/numkong/spatials/neonbfdot.h +60 -0
  349. package/numkong/include/numkong/spatials/neonfhm.h +92 -0
  350. package/numkong/include/numkong/spatials/neonfp8.h +111 -0
  351. package/numkong/include/numkong/spatials/neonsdot.h +143 -0
  352. package/numkong/include/numkong/spatials/powervsx.h +153 -0
  353. package/numkong/include/numkong/spatials/rvv.h +1976 -0
  354. package/numkong/include/numkong/spatials/sapphireamx.h +1149 -0
  355. package/numkong/include/numkong/spatials/serial.h +248 -0
  356. package/numkong/include/numkong/spatials/sierra.h +96 -0
  357. package/numkong/include/numkong/spatials/skylake.h +184 -0
  358. package/numkong/include/numkong/spatials/sme.h +1927 -0
  359. package/numkong/include/numkong/spatials/smef64.h +483 -0
  360. package/numkong/include/numkong/spatials/v128relaxed.h +240 -0
  361. package/numkong/include/numkong/spatials.h +3374 -0
  362. package/numkong/include/numkong/spatials.hpp +509 -0
  363. package/numkong/include/numkong/tensor.hpp +1981 -0
  364. package/numkong/include/numkong/trigonometry/README.md +184 -0
  365. package/numkong/include/numkong/trigonometry/haswell.h +653 -0
  366. package/numkong/include/numkong/trigonometry/neon.h +638 -0
  367. package/numkong/include/numkong/trigonometry/rvv.h +700 -0
  368. package/numkong/include/numkong/trigonometry/serial.h +703 -0
  369. package/numkong/include/numkong/trigonometry/skylake.h +725 -0
  370. package/numkong/include/numkong/trigonometry/v128relaxed.h +667 -0
  371. package/numkong/include/numkong/trigonometry.h +467 -0
  372. package/numkong/include/numkong/trigonometry.hpp +166 -0
  373. package/numkong/include/numkong/types.h +1695 -0
  374. package/numkong/include/numkong/types.hpp +6027 -0
  375. package/numkong/include/numkong/vector.hpp +766 -0
  376. package/numkong/javascript/@numkong-darwin-arm64/package.json +26 -0
  377. package/numkong/javascript/@numkong-darwin-x64/package.json +26 -0
  378. package/numkong/javascript/@numkong-linux-arm64/package.json +25 -0
  379. package/numkong/javascript/@numkong-linux-x64/package.json +25 -0
  380. package/numkong/javascript/@numkong-win32-arm64/package.json +25 -0
  381. package/numkong/javascript/@numkong-win32-x64/package.json +25 -0
  382. package/numkong/javascript/README.md +254 -0
  383. package/numkong/javascript/node-gyp-build.d.ts +4 -0
  384. package/numkong/javascript/numkong-browser.ts +40 -0
  385. package/numkong/javascript/numkong-wasm.ts +750 -0
  386. package/numkong/javascript/numkong.c +712 -0
  387. package/numkong/javascript/numkong.ts +613 -0
  388. package/{simsimd → numkong}/javascript/tsconfig-base.json +6 -3
  389. package/numkong/javascript/tsconfig-cjs.json +13 -0
  390. package/{simsimd → numkong}/javascript/tsconfig-esm.json +1 -1
  391. package/numkong/javascript/types.ts +784 -0
  392. package/numkong/numkong.gypi +104 -0
  393. package/numkong/package.json +108 -0
  394. package/numkong/probes/arm_neon.c +8 -0
  395. package/numkong/probes/arm_neon_bfdot.c +9 -0
  396. package/numkong/probes/arm_neon_fhm.c +9 -0
  397. package/numkong/probes/arm_neon_half.c +8 -0
  398. package/numkong/probes/arm_neon_sdot.c +9 -0
  399. package/numkong/probes/arm_neonfp8.c +9 -0
  400. package/numkong/probes/arm_sme.c +16 -0
  401. package/numkong/probes/arm_sme2.c +16 -0
  402. package/numkong/probes/arm_sme2p1.c +16 -0
  403. package/numkong/probes/arm_sme_bf16.c +16 -0
  404. package/numkong/probes/arm_sme_bi32.c +16 -0
  405. package/numkong/probes/arm_sme_f64.c +16 -0
  406. package/numkong/probes/arm_sme_fa64.c +14 -0
  407. package/numkong/probes/arm_sme_half.c +16 -0
  408. package/numkong/probes/arm_sme_lut2.c +15 -0
  409. package/numkong/probes/arm_sve.c +18 -0
  410. package/numkong/probes/arm_sve2.c +20 -0
  411. package/numkong/probes/arm_sve2p1.c +18 -0
  412. package/numkong/probes/arm_sve_bfdot.c +20 -0
  413. package/numkong/probes/arm_sve_half.c +18 -0
  414. package/numkong/probes/arm_sve_sdot.c +21 -0
  415. package/numkong/probes/loongarch_lasx.c +12 -0
  416. package/numkong/probes/power_vsx.c +12 -0
  417. package/numkong/probes/probe.js +127 -0
  418. package/numkong/probes/riscv_rvv.c +14 -0
  419. package/numkong/probes/riscv_rvv_bb.c +15 -0
  420. package/numkong/probes/riscv_rvv_bf16.c +17 -0
  421. package/numkong/probes/riscv_rvv_half.c +14 -0
  422. package/numkong/probes/wasm_v128relaxed.c +11 -0
  423. package/numkong/probes/x86_alder.c +17 -0
  424. package/numkong/probes/x86_diamond.c +17 -0
  425. package/numkong/probes/x86_genoa.c +17 -0
  426. package/numkong/probes/x86_graniteamx.c +19 -0
  427. package/numkong/probes/x86_haswell.c +11 -0
  428. package/numkong/probes/x86_icelake.c +17 -0
  429. package/numkong/probes/x86_sapphire.c +16 -0
  430. package/numkong/probes/x86_sapphireamx.c +18 -0
  431. package/numkong/probes/x86_sierra.c +17 -0
  432. package/numkong/probes/x86_skylake.c +15 -0
  433. package/numkong/probes/x86_turin.c +17 -0
  434. package/numkong/rust/README.md +698 -0
  435. package/numkong/rust/capabilities.rs +96 -0
  436. package/numkong/rust/cast.rs +314 -0
  437. package/numkong/rust/curved.rs +391 -0
  438. package/numkong/rust/each.rs +2721 -0
  439. package/numkong/rust/geospatial.rs +343 -0
  440. package/numkong/rust/matrix.rs +5424 -0
  441. package/numkong/rust/maxsim.rs +540 -0
  442. package/numkong/rust/mesh.rs +837 -0
  443. package/numkong/rust/numkong.rs +409 -0
  444. package/numkong/rust/probability.rs +272 -0
  445. package/numkong/rust/reduce.rs +1249 -0
  446. package/numkong/rust/set.rs +194 -0
  447. package/numkong/rust/sparse.rs +513 -0
  448. package/numkong/rust/spatial.rs +1815 -0
  449. package/numkong/rust/tensor.rs +7565 -0
  450. package/numkong/rust/types.rs +3666 -0
  451. package/numkong/rust/vector.rs +1899 -0
  452. package/numkong/rustfmt.toml +2 -0
  453. package/numkong/swift/Matrix.swift +1388 -0
  454. package/numkong/swift/MaxSim.swift +151 -0
  455. package/numkong/swift/NumKong.swift +283 -0
  456. package/numkong/swift/Numerics.swift +329 -0
  457. package/numkong/swift/README.md +433 -0
  458. package/numkong/swift/Scalar.swift +741 -0
  459. package/numkong/swift/Tensor.swift +209 -0
  460. package/numkong/test/README.md +315 -0
  461. package/numkong/test/playwright.config.ts +16 -0
  462. package/numkong/test/swift/Test.swift +534 -0
  463. package/numkong/test/test-browser-bundle-runner.mjs +36 -0
  464. package/numkong/test/test-browser-bundle.html +188 -0
  465. package/numkong/test/test-browser-runner.mjs +43 -0
  466. package/numkong/test/test-browser.html +305 -0
  467. package/numkong/test/test-dtypes.mjs +453 -0
  468. package/numkong/test/test-wasm.mjs +267 -0
  469. package/numkong/test/test.cpp +442 -0
  470. package/numkong/test/test.hpp +580 -0
  471. package/numkong/test/test.mjs +363 -0
  472. package/numkong/test/test_base.py +1041 -0
  473. package/numkong/test/test_cast.cpp +242 -0
  474. package/numkong/test/test_cross.hpp +551 -0
  475. package/numkong/test/test_cross.py +467 -0
  476. package/numkong/test/test_cross_amx.cpp +50 -0
  477. package/numkong/test/test_cross_arm.cpp +225 -0
  478. package/numkong/test/test_cross_blas.cpp +354 -0
  479. package/numkong/test/test_cross_loongarch.cpp +90 -0
  480. package/numkong/test/test_cross_power.cpp +73 -0
  481. package/numkong/test/test_cross_rvv.cpp +109 -0
  482. package/numkong/test/test_cross_serial.cpp +181 -0
  483. package/numkong/test/test_cross_sme.cpp +148 -0
  484. package/numkong/test/test_cross_wasm.cpp +141 -0
  485. package/numkong/test/test_cross_x86.cpp +442 -0
  486. package/numkong/test/test_curved.cpp +182 -0
  487. package/numkong/test/test_curved.py +207 -0
  488. package/numkong/test/test_dot.cpp +353 -0
  489. package/numkong/test/test_dot.py +296 -0
  490. package/numkong/test/test_each.cpp +437 -0
  491. package/numkong/test/test_each.py +756 -0
  492. package/numkong/test/test_geospatial.cpp +155 -0
  493. package/numkong/test/test_geospatial.py +241 -0
  494. package/numkong/test/test_maxsim.cpp +138 -0
  495. package/numkong/test/test_maxsim.py +209 -0
  496. package/numkong/test/test_mesh.cpp +203 -0
  497. package/numkong/test/test_mesh.py +394 -0
  498. package/numkong/test/test_probability.cpp +136 -0
  499. package/numkong/test/test_probability.py +166 -0
  500. package/numkong/test/test_reduce.cpp +563 -0
  501. package/numkong/test/test_reduce.py +781 -0
  502. package/numkong/test/test_set.cpp +151 -0
  503. package/numkong/test/test_set.py +120 -0
  504. package/numkong/test/test_similarities.py +579 -0
  505. package/numkong/test/test_sparse.cpp +121 -0
  506. package/numkong/test/test_sparse.py +118 -0
  507. package/numkong/test/test_spatial.cpp +518 -0
  508. package/numkong/test/test_spatial.py +269 -0
  509. package/numkong/test/test_tensor.cpp +884 -0
  510. package/numkong/test/test_tensor.py +1417 -0
  511. package/numkong/test/test_trigonometry.cpp +156 -0
  512. package/numkong/test/test_trigonometry.py +180 -0
  513. package/package.json +7 -8
  514. package/prebuilds/darwin-arm64+x64/usearch.node +0 -0
  515. package/prebuilds/linux-arm64/usearch.node +0 -0
  516. package/prebuilds/linux-x64/usearch.node +0 -0
  517. package/fp16/CMakeLists.txt +0 -176
  518. package/fp16/LICENSE +0 -11
  519. package/fp16/README.md +0 -20
  520. package/fp16/bench/alt-element.cc +0 -89
  521. package/fp16/bench/from-alt-array.cc +0 -175
  522. package/fp16/bench/from-ieee-array.cc +0 -399
  523. package/fp16/bench/ieee-element.cc +0 -272
  524. package/fp16/bench/to-alt-array.cc +0 -91
  525. package/fp16/bench/to-ieee-array.cc +0 -303
  526. package/fp16/cmake/DownloadGoogleBenchmark.cmake +0 -15
  527. package/fp16/cmake/DownloadGoogleTest.cmake +0 -15
  528. package/fp16/cmake/DownloadPSimd.cmake +0 -15
  529. package/fp16/configure.py +0 -65
  530. package/fp16/confu.yaml +0 -8
  531. package/fp16/include/fp16/__init__.py +0 -0
  532. package/fp16/include/fp16/avx.py +0 -50
  533. package/fp16/include/fp16/avx2.py +0 -53
  534. package/fp16/include/fp16/bitcasts.h +0 -92
  535. package/fp16/include/fp16/fp16.h +0 -451
  536. package/fp16/include/fp16/psimd.h +0 -131
  537. package/fp16/include/fp16.h +0 -11
  538. package/fp16/jni/Android.mk +0 -6
  539. package/fp16/jni/Application.mk +0 -4
  540. package/fp16/test/alt-from-fp32-value.cc +0 -498
  541. package/fp16/test/alt-to-fp32-bits.cc +0 -263
  542. package/fp16/test/alt-to-fp32-psimd.cc +0 -145
  543. package/fp16/test/alt-to-fp32-value.cc +0 -348
  544. package/fp16/test/alt-to-fp32x2-psimd.cc +0 -245
  545. package/fp16/test/bitcasts.cc +0 -56
  546. package/fp16/test/ieee-from-fp32-value.cc +0 -523
  547. package/fp16/test/ieee-to-fp32-bits.cc +0 -333
  548. package/fp16/test/ieee-to-fp32-psimd.cc +0 -312
  549. package/fp16/test/ieee-to-fp32-value.cc +0 -420
  550. package/fp16/test/ieee-to-fp32x2-psimd.cc +0 -541
  551. package/fp16/test/peachpy/alt-xmm-to-fp32-xmm-avx.cc +0 -96
  552. package/fp16/test/peachpy/alt-xmm-to-fp32-ymm-avx2.cc +0 -96
  553. package/fp16/test/peachpy/stubs.py +0 -38
  554. package/fp16/test/tables.cc +0 -530
  555. package/fp16/test/tables.h +0 -13
  556. package/fp16/third-party/THHalf.h +0 -92
  557. package/fp16/third-party/eigen-half.h +0 -170
  558. package/fp16/third-party/float16-compressor.h +0 -80
  559. package/fp16/third-party/half.hpp +0 -2916
  560. package/fp16/third-party/npy-halffloat.h +0 -328
  561. package/simsimd/.git-blame-ignore-revs +0 -2
  562. package/simsimd/CONTRIBUTING.md +0 -321
  563. package/simsimd/Cargo.toml +0 -49
  564. package/simsimd/MANIFEST.in +0 -4
  565. package/simsimd/Package.swift +0 -28
  566. package/simsimd/README.md +0 -1493
  567. package/simsimd/binding.gyp +0 -35
  568. package/simsimd/build.rs +0 -70
  569. package/simsimd/c/lib.c +0 -374
  570. package/simsimd/include/module.modulemap +0 -4
  571. package/simsimd/include/simsimd/binary.h +0 -486
  572. package/simsimd/include/simsimd/curved.h +0 -1541
  573. package/simsimd/include/simsimd/dot.h +0 -1853
  574. package/simsimd/include/simsimd/elementwise.h +0 -2465
  575. package/simsimd/include/simsimd/geospatial.h +0 -43
  576. package/simsimd/include/simsimd/mesh.h +0 -69
  577. package/simsimd/include/simsimd/probability.h +0 -606
  578. package/simsimd/include/simsimd/simsimd.h +0 -2567
  579. package/simsimd/include/simsimd/sparse.h +0 -1384
  580. package/simsimd/include/simsimd/spatial.h +0 -2335
  581. package/simsimd/include/simsimd/types.h +0 -666
  582. package/simsimd/javascript/fallback.ts +0 -233
  583. package/simsimd/javascript/lib.c +0 -125
  584. package/simsimd/javascript/node-gyp-build.d.ts +0 -1
  585. package/simsimd/javascript/simsimd.ts +0 -179
  586. package/simsimd/javascript/tsconfig-cjs.json +0 -8
  587. package/simsimd/package-lock.json +0 -1761
  588. package/simsimd/package.json +0 -61
  589. package/simsimd/rust/lib.rs +0 -1821
  590. package/simsimd/scripts/bench.cxx +0 -1170
  591. package/simsimd/scripts/bench.ipynb +0 -187
  592. package/simsimd/scripts/bench.js +0 -127
  593. package/simsimd/scripts/bench.rs +0 -876
  594. package/simsimd/scripts/bench_vectors.py +0 -734
  595. package/simsimd/scripts/bench_vectors_live.py +0 -213
  596. package/simsimd/scripts/test.c +0 -187
  597. package/simsimd/scripts/test.mjs +0 -220
  598. package/simsimd/scripts/test.py +0 -1505
  599. package/simsimd/swift/SimSIMD.swift +0 -149
  600. package/simsimd/swift/Test.swift +0 -128
  601. /package/{simsimd → numkong}/CODE_OF_CONDUCT.md +0 -0
  602. /package/{simsimd → numkong}/LICENSE +0 -0
  603. /package/{simsimd → numkong}/javascript/dist-package-cjs.json +0 -0
  604. /package/{simsimd → numkong}/javascript/dist-package-esm.json +0 -0
@@ -1,876 +0,0 @@
1
- //! Unified SimSIMD Benchmark Suite
2
- //!
3
- //! Compares SimSIMD vs native Rust implementations using Criterion.
4
- //! Reports both performance and accuracy metrics.
5
- //!
6
- //! Run with:
7
- //! ```bash
8
- //! cargo bench --bench bench -- --quiet --noplot
9
- //!
10
- //! # Or with custom dimensions:
11
- //! SIMSIMD_BENCH_DENSE_DIMENSIONS=2048 cargo bench --bench bench -- --quiet --noplot
12
- //! ```
13
- use std::hint::black_box;
14
- use std::mem::size_of;
15
- use std::time::Duration;
16
-
17
- use criterion::{criterion_group, criterion_main, Criterion, Throughput};
18
- use rand::rngs::StdRng;
19
- use rand::Rng;
20
- use rand::SeedableRng;
21
-
22
- use simsimd::SpatialSimilarity as SimSIMD;
23
-
24
- const BOLD: &str = "\x1b[1m";
25
- const RESET: &str = "\x1b[0m";
26
-
27
- fn get_dense_dimensions() -> usize {
28
- std::env::var("SIMSIMD_BENCH_DENSE_DIMENSIONS")
29
- .ok()
30
- .and_then(|s| s.parse().ok())
31
- .unwrap_or(1536)
32
- }
33
-
34
- #[allow(dead_code)]
35
- fn get_curved_dimensions() -> usize {
36
- std::env::var("SIMSIMD_BENCH_CURVED_DIMENSIONS")
37
- .ok()
38
- .and_then(|s| s.parse().ok())
39
- .unwrap_or(8)
40
- }
41
-
42
- fn calculate_errors(baseline: &[f64], contender: &[f64]) -> (f64, f64) {
43
- let mut mean_delta = 0.0f64;
44
- let mut mean_relative_error = 0.0f64;
45
-
46
- for (&b, &c) in baseline.iter().zip(contender.iter()) {
47
- let abs_delta = (b - c).abs();
48
- mean_delta += abs_delta;
49
- if abs_delta != 0.0 && b != 0.0 {
50
- mean_relative_error += abs_delta / b.abs();
51
- }
52
- }
53
-
54
- mean_delta /= baseline.len() as f64;
55
- mean_relative_error /= baseline.len() as f64;
56
- (mean_delta, mean_relative_error)
57
- }
58
-
59
- fn print_error(abs_err: f64, rel_err: f64) {
60
- println!(
61
- " abs error: {}{:.6e}{}",
62
- BOLD, abs_err, RESET
63
- );
64
- println!(
65
- " rel error: {}{:.6e}{}",
66
- BOLD, rel_err, RESET
67
- );
68
- }
69
-
70
- // region: Accurate Baseline Implementations (f64)
71
-
72
- fn accurate_cos(a: &[f32], b: &[f32]) -> f64 {
73
- let mut dot = 0.0f64;
74
- let mut norm_a = 0.0f64;
75
- let mut norm_b = 0.0f64;
76
- for i in 0..a.len() {
77
- let (av, bv) = (a[i] as f64, b[i] as f64);
78
- dot += av * bv;
79
- norm_a += av * av;
80
- norm_b += bv * bv;
81
- }
82
- 1.0 - dot / (norm_a * norm_b).sqrt()
83
- }
84
-
85
- fn accurate_l2sq(a: &[f32], b: &[f32]) -> f64 {
86
- let mut sum = 0.0f64;
87
- for i in 0..a.len() {
88
- let diff = (a[i] - b[i]) as f64;
89
- sum += diff * diff;
90
- }
91
- sum
92
- }
93
-
94
- fn accurate_dot(a: &[f32], b: &[f32]) -> f64 {
95
- let mut sum = 0.0f64;
96
- for i in 0..a.len() {
97
- sum += (a[i] as f64) * (b[i] as f64);
98
- }
99
- -sum
100
- }
101
-
102
- fn accurate_l2(a: &[f32], b: &[f32]) -> f64 {
103
- accurate_l2sq(a, b).sqrt()
104
- }
105
-
106
- fn accurate_i8_cos(a: &[i8], b: &[i8]) -> f64 {
107
- let mut dot = 0i32;
108
- let mut norm_a = 0i32;
109
- let mut norm_b = 0i32;
110
- for i in 0..a.len() {
111
- let (av, bv) = (a[i] as i32, b[i] as i32);
112
- dot += av * bv;
113
- norm_a += av * av;
114
- norm_b += bv * bv;
115
- }
116
- let dot_f = dot as f32;
117
- let norm_a_f = norm_a as f32;
118
- let norm_b_f = norm_b as f32;
119
- (1.0 - dot_f / (norm_a_f * norm_b_f).sqrt()) as f64
120
- }
121
-
122
- fn accurate_i8_dot(a: &[i8], b: &[i8]) -> f64 {
123
- let mut sum = 0i32;
124
- for i in 0..a.len() {
125
- sum += (a[i] as i32) * (b[i] as i32);
126
- }
127
- -(sum as f64)
128
- }
129
-
130
- fn accurate_i8_l2sq(a: &[i8], b: &[i8]) -> f64 {
131
- let mut sum = 0i32;
132
- for i in 0..a.len() {
133
- let diff = (a[i] - b[i]) as i32;
134
- sum += diff * diff;
135
- }
136
- sum as f64
137
- }
138
-
139
- fn accurate_i8_l2(a: &[i8], b: &[i8]) -> f64 {
140
- accurate_i8_l2sq(a, b).sqrt()
141
- }
142
-
143
- // endregion: Accurate Baseline Implementations (f64)
144
-
145
- // region: Baseline Implementations - Cosine Similarity
146
-
147
- fn baseline_cos_procedural(a: &[f32], b: &[f32]) -> f64 {
148
- let mut dot = 0.0f32;
149
- let mut norm_a = 0.0f32;
150
- let mut norm_b = 0.0f32;
151
- for i in 0..a.len() {
152
- let (av, bv) = (a[i], b[i]);
153
- dot += av * bv;
154
- norm_a += av * av;
155
- norm_b += bv * bv;
156
- }
157
- (1.0 - dot / (norm_a * norm_b).sqrt()) as f64
158
- }
159
-
160
- fn baseline_cos_functional(a: &[f32], b: &[f32]) -> f64 {
161
- let (dot, norm_a, norm_b) = a
162
- .iter()
163
- .zip(b)
164
- .map(|(a, b)| (*a * *b, *a * *a, *b * *b))
165
- .fold((0.0f32, 0.0f32, 0.0f32), |acc, x| {
166
- (acc.0 + x.0, acc.1 + x.1, acc.2 + x.2)
167
- });
168
- (1.0 - (dot / (norm_a.sqrt() * norm_b.sqrt()))) as f64
169
- }
170
-
171
- fn baseline_cos_unrolled(a: &[f32], b: &[f32]) -> f64 {
172
- let mut i = 0;
173
- let remainder = a.len() % 8;
174
- let mut acc = [0.0f32; 8];
175
- let mut norm_a = [0.0f32; 8];
176
- let mut norm_b = [0.0f32; 8];
177
-
178
- unsafe {
179
- while i < a.len() - remainder {
180
- let [a1, a2, a3, a4, a5, a6, a7, a8] = [
181
- *a.get_unchecked(i),
182
- *a.get_unchecked(i + 1),
183
- *a.get_unchecked(i + 2),
184
- *a.get_unchecked(i + 3),
185
- *a.get_unchecked(i + 4),
186
- *a.get_unchecked(i + 5),
187
- *a.get_unchecked(i + 6),
188
- *a.get_unchecked(i + 7),
189
- ];
190
- let [b1, b2, b3, b4, b5, b6, b7, b8] = [
191
- *b.get_unchecked(i),
192
- *b.get_unchecked(i + 1),
193
- *b.get_unchecked(i + 2),
194
- *b.get_unchecked(i + 3),
195
- *b.get_unchecked(i + 4),
196
- *b.get_unchecked(i + 5),
197
- *b.get_unchecked(i + 6),
198
- *b.get_unchecked(i + 7),
199
- ];
200
-
201
- acc[0] += a1 * b1;
202
- acc[1] += a2 * b2;
203
- acc[2] += a3 * b3;
204
- acc[3] += a4 * b4;
205
- acc[4] += a5 * b5;
206
- acc[5] += a6 * b6;
207
- acc[6] += a7 * b7;
208
- acc[7] += a8 * b8;
209
- norm_a[0] += a1 * a1;
210
- norm_a[1] += a2 * a2;
211
- norm_a[2] += a3 * a3;
212
- norm_a[3] += a4 * a4;
213
- norm_a[4] += a5 * a5;
214
- norm_a[5] += a6 * a6;
215
- norm_a[6] += a7 * a7;
216
- norm_a[7] += a8 * a8;
217
- norm_b[0] += b1 * b1;
218
- norm_b[1] += b2 * b2;
219
- norm_b[2] += b3 * b3;
220
- norm_b[3] += b4 * b4;
221
- norm_b[4] += b5 * b5;
222
- norm_b[5] += b6 * b6;
223
- norm_b[6] += b7 * b7;
224
- norm_b[7] += b8 * b8;
225
- i += 8;
226
- }
227
- while i < a.len() {
228
- let (av, bv) = (*a.get_unchecked(i), *b.get_unchecked(i));
229
- acc[0] += av * bv;
230
- norm_a[0] += av * av;
231
- norm_b[0] += bv * bv;
232
- i += 1;
233
- }
234
- }
235
-
236
- let dot: f32 = acc.iter().sum();
237
- let na: f32 = norm_a.iter().sum();
238
- let nb: f32 = norm_b.iter().sum();
239
- (1.0 - (dot / (na.sqrt() * nb.sqrt()))) as f64
240
- }
241
-
242
- // endregion: Baseline Implementations - Cosine Similarity
243
-
244
- // region: Baseline Implementations - Squared Euclidean Distance
245
-
246
- fn baseline_l2sq_procedural(a: &[f32], b: &[f32]) -> f64 {
247
- let mut sum = 0.0f32;
248
- for i in 0..a.len() {
249
- let diff = a[i] - b[i];
250
- sum += diff * diff;
251
- }
252
- sum as f64
253
- }
254
-
255
- fn baseline_l2sq_functional(a: &[f32], b: &[f32]) -> f64 {
256
- a.iter()
257
- .zip(b)
258
- .map(|(x, y)| {
259
- let diff = *x - *y;
260
- diff * diff
261
- })
262
- .sum::<f32>() as f64
263
- }
264
-
265
- fn baseline_l2sq_unrolled(a: &[f32], b: &[f32]) -> f64 {
266
- let mut i = 0;
267
- let remainder = a.len() % 8;
268
- let mut acc = [0.0f32; 8];
269
-
270
- unsafe {
271
- while i < a.len() - remainder {
272
- let [a1, a2, a3, a4, a5, a6, a7, a8] = [
273
- *a.get_unchecked(i),
274
- *a.get_unchecked(i + 1),
275
- *a.get_unchecked(i + 2),
276
- *a.get_unchecked(i + 3),
277
- *a.get_unchecked(i + 4),
278
- *a.get_unchecked(i + 5),
279
- *a.get_unchecked(i + 6),
280
- *a.get_unchecked(i + 7),
281
- ];
282
- let [b1, b2, b3, b4, b5, b6, b7, b8] = [
283
- *b.get_unchecked(i),
284
- *b.get_unchecked(i + 1),
285
- *b.get_unchecked(i + 2),
286
- *b.get_unchecked(i + 3),
287
- *b.get_unchecked(i + 4),
288
- *b.get_unchecked(i + 5),
289
- *b.get_unchecked(i + 6),
290
- *b.get_unchecked(i + 7),
291
- ];
292
-
293
- let [d1, d2, d3, d4, d5, d6, d7, d8] = [
294
- a1 - b1,
295
- a2 - b2,
296
- a3 - b3,
297
- a4 - b4,
298
- a5 - b5,
299
- a6 - b6,
300
- a7 - b7,
301
- a8 - b8,
302
- ];
303
- acc[0] += d1 * d1;
304
- acc[1] += d2 * d2;
305
- acc[2] += d3 * d3;
306
- acc[3] += d4 * d4;
307
- acc[4] += d5 * d5;
308
- acc[5] += d6 * d6;
309
- acc[6] += d7 * d7;
310
- acc[7] += d8 * d8;
311
- i += 8;
312
- }
313
- while i < a.len() {
314
- let diff = *a.get_unchecked(i) - *b.get_unchecked(i);
315
- acc[0] += diff * diff;
316
- i += 1;
317
- }
318
- }
319
-
320
- acc.iter().sum::<f32>() as f64
321
- }
322
-
323
- // endregion: Baseline Implementations - Squared Euclidean Distance
324
-
325
- // region: Baseline Implementations - Dot Product
326
-
327
- fn baseline_dot_procedural(a: &[f32], b: &[f32]) -> f64 {
328
- let mut sum = 0.0f32;
329
- for i in 0..a.len() {
330
- sum += a[i] * b[i];
331
- }
332
- -(sum as f64)
333
- }
334
-
335
- fn baseline_dot_functional(a: &[f32], b: &[f32]) -> f64 {
336
- -(a.iter().zip(b).map(|(x, y)| *x * *y).sum::<f32>() as f64)
337
- }
338
-
339
- fn baseline_dot_unrolled(a: &[f32], b: &[f32]) -> f64 {
340
- let mut i = 0;
341
- let remainder = a.len() % 8;
342
- let mut acc = [0.0f32; 8];
343
-
344
- unsafe {
345
- while i < a.len() - remainder {
346
- let [a1, a2, a3, a4, a5, a6, a7, a8] = [
347
- *a.get_unchecked(i),
348
- *a.get_unchecked(i + 1),
349
- *a.get_unchecked(i + 2),
350
- *a.get_unchecked(i + 3),
351
- *a.get_unchecked(i + 4),
352
- *a.get_unchecked(i + 5),
353
- *a.get_unchecked(i + 6),
354
- *a.get_unchecked(i + 7),
355
- ];
356
- let [b1, b2, b3, b4, b5, b6, b7, b8] = [
357
- *b.get_unchecked(i),
358
- *b.get_unchecked(i + 1),
359
- *b.get_unchecked(i + 2),
360
- *b.get_unchecked(i + 3),
361
- *b.get_unchecked(i + 4),
362
- *b.get_unchecked(i + 5),
363
- *b.get_unchecked(i + 6),
364
- *b.get_unchecked(i + 7),
365
- ];
366
-
367
- acc[0] += a1 * b1;
368
- acc[1] += a2 * b2;
369
- acc[2] += a3 * b3;
370
- acc[3] += a4 * b4;
371
- acc[4] += a5 * b5;
372
- acc[5] += a6 * b6;
373
- acc[6] += a7 * b7;
374
- acc[7] += a8 * b8;
375
- i += 8;
376
- }
377
- while i < a.len() {
378
- acc[0] += *a.get_unchecked(i) * *b.get_unchecked(i);
379
- i += 1;
380
- }
381
- }
382
-
383
- -(acc.iter().sum::<f32>() as f64)
384
- }
385
-
386
- // endregion: Baseline Implementations - Dot Product
387
-
388
- // region: Baseline Implementations - Euclidean Distance
389
-
390
- fn baseline_l2_procedural(a: &[f32], b: &[f32]) -> f64 {
391
- baseline_l2sq_procedural(a, b).sqrt()
392
- }
393
-
394
- fn baseline_l2_functional(a: &[f32], b: &[f32]) -> f64 {
395
- baseline_l2sq_functional(a, b).sqrt()
396
- }
397
-
398
- fn baseline_l2_unrolled(a: &[f32], b: &[f32]) -> f64 {
399
- baseline_l2sq_unrolled(a, b).sqrt()
400
- }
401
-
402
- // endregion: Baseline Implementations - Euclidean Distance
403
-
404
- // region: Baseline Implementations - i8 (i8 -> f64)
405
-
406
- fn baseline_i8_cos_procedural(a: &[i8], b: &[i8]) -> f64 {
407
- let mut dot = 0i32;
408
- let mut norm_a = 0i32;
409
- let mut norm_b = 0i32;
410
- for i in 0..a.len() {
411
- let (av, bv) = (a[i] as i32, b[i] as i32);
412
- dot += av * bv;
413
- norm_a += av * av;
414
- norm_b += bv * bv;
415
- }
416
- let dot_f = dot as f32;
417
- let norm_a_f = norm_a as f32;
418
- let norm_b_f = norm_b as f32;
419
- (1.0 - dot_f / (norm_a_f * norm_b_f).sqrt()) as f64
420
- }
421
-
422
- fn baseline_i8_dot_procedural(a: &[i8], b: &[i8]) -> f64 {
423
- let mut sum = 0i32;
424
- for i in 0..a.len() {
425
- sum += (a[i] as i32) * (b[i] as i32);
426
- }
427
- -(sum as f64)
428
- }
429
-
430
- fn baseline_i8_l2sq_procedural(a: &[i8], b: &[i8]) -> f64 {
431
- let mut sum = 0i32;
432
- for i in 0..a.len() {
433
- let diff = (a[i] - b[i]) as i32;
434
- sum += diff * diff;
435
- }
436
- sum as f64
437
- }
438
-
439
- fn baseline_i8_l2_procedural(a: &[i8], b: &[i8]) -> f64 {
440
- baseline_i8_l2sq_procedural(a, b).sqrt()
441
- }
442
-
443
- // endregion: Baseline Implementations - i8 (i8 -> f64)
444
-
445
- // region: Benchmark Functions
446
-
447
- pub fn f32_cosine_benchmark(c: &mut Criterion) {
448
- let dimensions = get_dense_dimensions();
449
- let bench_pairs_count = (128 * 1024 * 1024) / (dimensions * size_of::<f32>() * 2);
450
- let bench_pairs: Vec<_> = (0..bench_pairs_count)
451
- .map(|i| {
452
- let mut rng = StdRng::seed_from_u64(1000 + i as u64);
453
- let a: Vec<f32> = (0..dimensions).map(|_| rng.random()).collect();
454
- let b: Vec<f32> = (0..dimensions).map(|_| rng.random()).collect();
455
- (a, b)
456
- })
457
- .collect();
458
-
459
- let accurate: Vec<f64> = bench_pairs
460
- .iter()
461
- .map(|(a, b)| accurate_cos(a, b))
462
- .collect();
463
- let procedural: Vec<f64> = bench_pairs
464
- .iter()
465
- .map(|(a, b)| baseline_cos_procedural(a, b))
466
- .collect();
467
- let functional: Vec<f64> = bench_pairs
468
- .iter()
469
- .map(|(a, b)| baseline_cos_functional(a, b))
470
- .collect();
471
- let unrolled: Vec<f64> = bench_pairs
472
- .iter()
473
- .map(|(a, b)| baseline_cos_unrolled(a, b))
474
- .collect();
475
- let simsimd: Vec<f64> = bench_pairs
476
- .iter()
477
- .map(|(a, b)| SimSIMD::cosine(a, b).unwrap())
478
- .collect();
479
-
480
- let simsimd_err = calculate_errors(&accurate, &simsimd);
481
- let procedural_err = calculate_errors(&accurate, &procedural);
482
- let functional_err = calculate_errors(&accurate, &functional);
483
- let unrolled_err = calculate_errors(&accurate, &unrolled);
484
-
485
- let mut group = c.benchmark_group(format!("f32/{}d/cosine", dimensions));
486
- let bytes_per_op = (2 * dimensions * size_of::<f32>() * bench_pairs_count) as u64;
487
- group.throughput(Throughput::Bytes(bytes_per_op));
488
- group
489
- .sample_size(50)
490
- .measurement_time(Duration::from_secs(2))
491
- .noise_threshold(0.05);
492
-
493
- group.bench_function("simsimd", |b| {
494
- b.iter(|| {
495
- for (a, b) in &bench_pairs {
496
- black_box(SimSIMD::cosine(a, b));
497
- }
498
- })
499
- });
500
- print_error(simsimd_err.0, simsimd_err.1);
501
-
502
- group.bench_function("procedural", |b| {
503
- b.iter(|| {
504
- for (a, b) in &bench_pairs {
505
- black_box(baseline_cos_procedural(a, b));
506
- }
507
- })
508
- });
509
- print_error(procedural_err.0, procedural_err.1);
510
-
511
- group.bench_function("functional", |b| {
512
- b.iter(|| {
513
- for (a, b) in &bench_pairs {
514
- black_box(baseline_cos_functional(a, b));
515
- }
516
- })
517
- });
518
- print_error(functional_err.0, functional_err.1);
519
-
520
- group.bench_function("unrolled", |b| {
521
- b.iter(|| {
522
- for (a, b) in &bench_pairs {
523
- black_box(baseline_cos_unrolled(a, b));
524
- }
525
- })
526
- });
527
- print_error(unrolled_err.0, unrolled_err.1);
528
- group.finish();
529
- }
530
-
531
- pub fn f32_dot_benchmark(c: &mut Criterion) {
532
- let dimensions = get_dense_dimensions();
533
- let bench_pairs_count = (128 * 1024 * 1024) / (dimensions * size_of::<f32>() * 2);
534
- let bench_pairs: Vec<_> = (0..bench_pairs_count)
535
- .map(|i| {
536
- let mut rng = StdRng::seed_from_u64(1000 + i as u64);
537
- let a: Vec<f32> = (0..dimensions).map(|_| rng.random()).collect();
538
- let b: Vec<f32> = (0..dimensions).map(|_| rng.random()).collect();
539
- (a, b)
540
- })
541
- .collect();
542
-
543
- let accurate: Vec<f64> = bench_pairs
544
- .iter()
545
- .map(|(a, b)| accurate_dot(a, b))
546
- .collect();
547
- let procedural: Vec<f64> = bench_pairs
548
- .iter()
549
- .map(|(a, b)| baseline_dot_procedural(a, b))
550
- .collect();
551
- let functional: Vec<f64> = bench_pairs
552
- .iter()
553
- .map(|(a, b)| baseline_dot_functional(a, b))
554
- .collect();
555
- let unrolled: Vec<f64> = bench_pairs
556
- .iter()
557
- .map(|(a, b)| baseline_dot_unrolled(a, b))
558
- .collect();
559
- let simsimd: Vec<f64> = bench_pairs
560
- .iter()
561
- .map(|(a, b)| SimSIMD::dot(a, b).unwrap())
562
- .collect();
563
-
564
- let simsimd_err = calculate_errors(&accurate, &simsimd);
565
- let procedural_err = calculate_errors(&accurate, &procedural);
566
- let functional_err = calculate_errors(&accurate, &functional);
567
- let unrolled_err = calculate_errors(&accurate, &unrolled);
568
-
569
- let mut group = c.benchmark_group(format!("f32/{}d/dot", dimensions));
570
- let bytes_per_op = (2 * dimensions * size_of::<f32>() * bench_pairs_count) as u64;
571
- group.throughput(Throughput::Bytes(bytes_per_op));
572
- group
573
- .sample_size(50)
574
- .measurement_time(Duration::from_secs(2))
575
- .noise_threshold(0.05);
576
-
577
- group.bench_function("simsimd", |b| {
578
- b.iter(|| {
579
- for (a, b) in &bench_pairs {
580
- black_box(SimSIMD::dot(a, b));
581
- }
582
- })
583
- });
584
- print_error(simsimd_err.0, simsimd_err.1);
585
-
586
- group.bench_function("procedural", |b| {
587
- b.iter(|| {
588
- for (a, b) in &bench_pairs {
589
- black_box(baseline_dot_procedural(a, b));
590
- }
591
- })
592
- });
593
- print_error(procedural_err.0, procedural_err.1);
594
-
595
- group.bench_function("functional", |b| {
596
- b.iter(|| {
597
- for (a, b) in &bench_pairs {
598
- black_box(baseline_dot_functional(a, b));
599
- }
600
- })
601
- });
602
- print_error(functional_err.0, functional_err.1);
603
-
604
- group.bench_function("unrolled", |b| {
605
- b.iter(|| {
606
- for (a, b) in &bench_pairs {
607
- black_box(baseline_dot_unrolled(a, b));
608
- }
609
- })
610
- });
611
- print_error(unrolled_err.0, unrolled_err.1);
612
- group.finish();
613
- }
614
-
615
- pub fn f32_l2_benchmark(c: &mut Criterion) {
616
- let dimensions = get_dense_dimensions();
617
- let bench_pairs_count = (128 * 1024 * 1024) / (dimensions * size_of::<f32>() * 2);
618
- let bench_pairs: Vec<_> = (0..bench_pairs_count)
619
- .map(|i| {
620
- let mut rng = StdRng::seed_from_u64(1000 + i as u64);
621
- let a: Vec<f32> = (0..dimensions).map(|_| rng.random()).collect();
622
- let b: Vec<f32> = (0..dimensions).map(|_| rng.random()).collect();
623
- (a, b)
624
- })
625
- .collect();
626
-
627
- let accurate: Vec<f64> = bench_pairs.iter().map(|(a, b)| accurate_l2(a, b)).collect();
628
- let procedural: Vec<f64> = bench_pairs
629
- .iter()
630
- .map(|(a, b)| baseline_l2_procedural(a, b))
631
- .collect();
632
- let functional: Vec<f64> = bench_pairs
633
- .iter()
634
- .map(|(a, b)| baseline_l2_functional(a, b))
635
- .collect();
636
- let unrolled: Vec<f64> = bench_pairs
637
- .iter()
638
- .map(|(a, b)| baseline_l2_unrolled(a, b))
639
- .collect();
640
- let simsimd: Vec<f64> = bench_pairs
641
- .iter()
642
- .map(|(a, b)| SimSIMD::euclidean(a, b).unwrap())
643
- .collect();
644
-
645
- let simsimd_err = calculate_errors(&accurate, &simsimd);
646
- let procedural_err = calculate_errors(&accurate, &procedural);
647
- let functional_err = calculate_errors(&accurate, &functional);
648
- let unrolled_err = calculate_errors(&accurate, &unrolled);
649
-
650
- let mut group = c.benchmark_group(format!("f32/{}d/l2", dimensions));
651
- let bytes_per_op = (2 * dimensions * size_of::<f32>() * bench_pairs_count) as u64;
652
- group.throughput(Throughput::Bytes(bytes_per_op));
653
- group
654
- .sample_size(50)
655
- .measurement_time(Duration::from_secs(2))
656
- .noise_threshold(0.05);
657
-
658
- group.bench_function("simsimd", |b| {
659
- b.iter(|| {
660
- for (a, b) in &bench_pairs {
661
- black_box(SimSIMD::euclidean(a, b));
662
- }
663
- })
664
- });
665
- print_error(simsimd_err.0, simsimd_err.1);
666
-
667
- group.bench_function("procedural", |b| {
668
- b.iter(|| {
669
- for (a, b) in &bench_pairs {
670
- black_box(baseline_l2_procedural(a, b));
671
- }
672
- })
673
- });
674
- print_error(procedural_err.0, procedural_err.1);
675
-
676
- group.bench_function("functional", |b| {
677
- b.iter(|| {
678
- for (a, b) in &bench_pairs {
679
- black_box(baseline_l2_functional(a, b));
680
- }
681
- })
682
- });
683
- print_error(functional_err.0, functional_err.1);
684
-
685
- group.bench_function("unrolled", |b| {
686
- b.iter(|| {
687
- for (a, b) in &bench_pairs {
688
- black_box(baseline_l2_unrolled(a, b));
689
- }
690
- })
691
- });
692
- print_error(unrolled_err.0, unrolled_err.1);
693
- group.finish();
694
- }
695
-
696
- pub fn i8_cosine_benchmark(c: &mut Criterion) {
697
- let dimensions = get_dense_dimensions();
698
- let bench_pairs_count = (128 * 1024 * 1024) / (dimensions * size_of::<i8>() * 2);
699
- let bench_pairs: Vec<_> = (0..bench_pairs_count)
700
- .map(|i| {
701
- let mut rng = StdRng::seed_from_u64(1000 + i as u64);
702
- let a: Vec<i8> = (0..dimensions).map(|_| rng.random()).collect();
703
- let b: Vec<i8> = (0..dimensions).map(|_| rng.random()).collect();
704
- (a, b)
705
- })
706
- .collect();
707
-
708
- let accurate: Vec<f64> = bench_pairs
709
- .iter()
710
- .map(|(a, b)| accurate_i8_cos(a, b))
711
- .collect();
712
- let procedural: Vec<f64> = bench_pairs
713
- .iter()
714
- .map(|(a, b)| baseline_i8_cos_procedural(a, b))
715
- .collect();
716
- let simsimd: Vec<f64> = bench_pairs
717
- .iter()
718
- .map(|(a, b)| SimSIMD::cosine(a, b).unwrap())
719
- .collect();
720
-
721
- let simsimd_err = calculate_errors(&accurate, &simsimd);
722
- let procedural_err = calculate_errors(&accurate, &procedural);
723
-
724
- let mut group = c.benchmark_group(format!("i8/{}d/cosine", dimensions));
725
- let bytes_per_op = (2 * dimensions * size_of::<i8>() * bench_pairs_count) as u64;
726
- group.throughput(Throughput::Bytes(bytes_per_op));
727
- group
728
- .sample_size(50)
729
- .measurement_time(Duration::from_secs(2))
730
- .noise_threshold(0.05);
731
-
732
- group.bench_function("simsimd", |b| {
733
- b.iter(|| {
734
- for (a, b) in &bench_pairs {
735
- black_box(SimSIMD::cosine(a, b));
736
- }
737
- })
738
- });
739
- print_error(simsimd_err.0, simsimd_err.1);
740
-
741
- group.bench_function("procedural", |b| {
742
- b.iter(|| {
743
- for (a, b) in &bench_pairs {
744
- black_box(baseline_i8_cos_procedural(a, b));
745
- }
746
- })
747
- });
748
- print_error(procedural_err.0, procedural_err.1);
749
- group.finish();
750
- }
751
-
752
- pub fn i8_dot_benchmark(c: &mut Criterion) {
753
- let dimensions = get_dense_dimensions();
754
- let bench_pairs_count = (128 * 1024 * 1024) / (dimensions * size_of::<i8>() * 2);
755
- let bench_pairs: Vec<_> = (0..bench_pairs_count)
756
- .map(|i| {
757
- let mut rng = StdRng::seed_from_u64(1000 + i as u64);
758
- let a: Vec<i8> = (0..dimensions).map(|_| rng.random()).collect();
759
- let b: Vec<i8> = (0..dimensions).map(|_| rng.random()).collect();
760
- (a, b)
761
- })
762
- .collect();
763
-
764
- let accurate: Vec<f64> = bench_pairs
765
- .iter()
766
- .map(|(a, b)| accurate_i8_dot(a, b))
767
- .collect();
768
- let procedural: Vec<f64> = bench_pairs
769
- .iter()
770
- .map(|(a, b)| baseline_i8_dot_procedural(a, b))
771
- .collect();
772
- let simsimd: Vec<f64> = bench_pairs
773
- .iter()
774
- .map(|(a, b)| SimSIMD::dot(a, b).unwrap())
775
- .collect();
776
-
777
- let simsimd_err = calculate_errors(&accurate, &simsimd);
778
- let procedural_err = calculate_errors(&accurate, &procedural);
779
-
780
- let mut group = c.benchmark_group(format!("i8/{}d/dot", dimensions));
781
- let bytes_per_op = (2 * dimensions * size_of::<i8>() * bench_pairs_count) as u64;
782
- group.throughput(Throughput::Bytes(bytes_per_op));
783
- group
784
- .sample_size(50)
785
- .measurement_time(Duration::from_secs(2))
786
- .noise_threshold(0.05);
787
-
788
- group.bench_function("simsimd", |b| {
789
- b.iter(|| {
790
- for (a, b) in &bench_pairs {
791
- black_box(SimSIMD::dot(a, b));
792
- }
793
- })
794
- });
795
- print_error(simsimd_err.0, simsimd_err.1);
796
-
797
- group.bench_function("procedural", |b| {
798
- b.iter(|| {
799
- for (a, b) in &bench_pairs {
800
- black_box(baseline_i8_dot_procedural(a, b));
801
- }
802
- })
803
- });
804
- print_error(procedural_err.0, procedural_err.1);
805
- group.finish();
806
- }
807
-
808
- pub fn i8_l2_benchmark(c: &mut Criterion) {
809
- let dimensions = get_dense_dimensions();
810
- let bench_pairs_count = (128 * 1024 * 1024) / (dimensions * size_of::<i8>() * 2);
811
- let bench_pairs: Vec<_> = (0..bench_pairs_count)
812
- .map(|i| {
813
- let mut rng = StdRng::seed_from_u64(1000 + i as u64);
814
- let a: Vec<i8> = (0..dimensions).map(|_| rng.random()).collect();
815
- let b: Vec<i8> = (0..dimensions).map(|_| rng.random()).collect();
816
- (a, b)
817
- })
818
- .collect();
819
-
820
- let accurate: Vec<f64> = bench_pairs
821
- .iter()
822
- .map(|(a, b)| accurate_i8_l2(a, b))
823
- .collect();
824
- let procedural: Vec<f64> = bench_pairs
825
- .iter()
826
- .map(|(a, b)| baseline_i8_l2_procedural(a, b))
827
- .collect();
828
- let simsimd: Vec<f64> = bench_pairs
829
- .iter()
830
- .map(|(a, b)| SimSIMD::euclidean(a, b).unwrap())
831
- .collect();
832
-
833
- let simsimd_err = calculate_errors(&accurate, &simsimd);
834
- let procedural_err = calculate_errors(&accurate, &procedural);
835
-
836
- let mut group = c.benchmark_group(format!("i8/{}d/l2", dimensions));
837
- let bytes_per_op = (2 * dimensions * size_of::<i8>() * bench_pairs_count) as u64;
838
- group.throughput(Throughput::Bytes(bytes_per_op));
839
- group
840
- .sample_size(50)
841
- .measurement_time(Duration::from_secs(2))
842
- .noise_threshold(0.05);
843
-
844
- group.bench_function("simsimd", |b| {
845
- b.iter(|| {
846
- for (a, b) in &bench_pairs {
847
- black_box(SimSIMD::euclidean(a, b));
848
- }
849
- })
850
- });
851
- print_error(simsimd_err.0, simsimd_err.1);
852
-
853
- group.bench_function("procedural", |b| {
854
- b.iter(|| {
855
- for (a, b) in &bench_pairs {
856
- black_box(baseline_i8_l2_procedural(a, b));
857
- }
858
- })
859
- });
860
- print_error(procedural_err.0, procedural_err.1);
861
- group.finish();
862
- }
863
-
864
- // endregion: Benchmark Functions
865
-
866
- fn custom_criterion() -> Criterion {
867
- Criterion::default().without_plots().noise_threshold(0.05)
868
- }
869
-
870
- criterion_group! {
871
- name = benches;
872
- config = custom_criterion();
873
- targets = f32_cosine_benchmark, f32_dot_benchmark, f32_l2_benchmark,
874
- i8_cosine_benchmark, i8_dot_benchmark, i8_l2_benchmark
875
- }
876
- criterion_main!(benches);