numkong 7.0.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.
Files changed (294) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +495 -0
  3. package/binding.gyp +540 -0
  4. package/c/dispatch.h +512 -0
  5. package/c/dispatch_bf16.c +389 -0
  6. package/c/dispatch_bf16c.c +52 -0
  7. package/c/dispatch_e2m3.c +263 -0
  8. package/c/dispatch_e3m2.c +243 -0
  9. package/c/dispatch_e4m3.c +276 -0
  10. package/c/dispatch_e5m2.c +272 -0
  11. package/c/dispatch_f16.c +376 -0
  12. package/c/dispatch_f16c.c +58 -0
  13. package/c/dispatch_f32.c +378 -0
  14. package/c/dispatch_f32c.c +99 -0
  15. package/c/dispatch_f64.c +296 -0
  16. package/c/dispatch_f64c.c +98 -0
  17. package/c/dispatch_i16.c +96 -0
  18. package/c/dispatch_i32.c +89 -0
  19. package/c/dispatch_i4.c +150 -0
  20. package/c/dispatch_i64.c +86 -0
  21. package/c/dispatch_i8.c +289 -0
  22. package/c/dispatch_other.c +330 -0
  23. package/c/dispatch_u1.c +148 -0
  24. package/c/dispatch_u16.c +124 -0
  25. package/c/dispatch_u32.c +118 -0
  26. package/c/dispatch_u4.c +150 -0
  27. package/c/dispatch_u64.c +102 -0
  28. package/c/dispatch_u8.c +303 -0
  29. package/c/numkong.c +950 -0
  30. package/include/README.md +573 -0
  31. package/include/module.modulemap +129 -0
  32. package/include/numkong/attention/sapphireamx.h +1361 -0
  33. package/include/numkong/attention/sme.h +2066 -0
  34. package/include/numkong/attention.h +49 -0
  35. package/include/numkong/capabilities.h +748 -0
  36. package/include/numkong/cast/README.md +262 -0
  37. package/include/numkong/cast/haswell.h +975 -0
  38. package/include/numkong/cast/icelake.h +470 -0
  39. package/include/numkong/cast/neon.h +1192 -0
  40. package/include/numkong/cast/rvv.h +1021 -0
  41. package/include/numkong/cast/sapphire.h +262 -0
  42. package/include/numkong/cast/serial.h +2262 -0
  43. package/include/numkong/cast/skylake.h +856 -0
  44. package/include/numkong/cast/v128relaxed.h +180 -0
  45. package/include/numkong/cast.h +230 -0
  46. package/include/numkong/curved/README.md +223 -0
  47. package/include/numkong/curved/genoa.h +182 -0
  48. package/include/numkong/curved/haswell.h +276 -0
  49. package/include/numkong/curved/neon.h +205 -0
  50. package/include/numkong/curved/neonbfdot.h +212 -0
  51. package/include/numkong/curved/neonhalf.h +212 -0
  52. package/include/numkong/curved/rvv.h +305 -0
  53. package/include/numkong/curved/serial.h +207 -0
  54. package/include/numkong/curved/skylake.h +457 -0
  55. package/include/numkong/curved/smef64.h +506 -0
  56. package/include/numkong/curved.h +517 -0
  57. package/include/numkong/curved.hpp +144 -0
  58. package/include/numkong/dot/README.md +425 -0
  59. package/include/numkong/dot/alder.h +563 -0
  60. package/include/numkong/dot/genoa.h +315 -0
  61. package/include/numkong/dot/haswell.h +1688 -0
  62. package/include/numkong/dot/icelake.h +883 -0
  63. package/include/numkong/dot/neon.h +818 -0
  64. package/include/numkong/dot/neonbfdot.h +244 -0
  65. package/include/numkong/dot/neonfhm.h +360 -0
  66. package/include/numkong/dot/neonhalf.h +198 -0
  67. package/include/numkong/dot/neonsdot.h +508 -0
  68. package/include/numkong/dot/rvv.h +714 -0
  69. package/include/numkong/dot/rvvbb.h +72 -0
  70. package/include/numkong/dot/rvvbf16.h +123 -0
  71. package/include/numkong/dot/rvvhalf.h +129 -0
  72. package/include/numkong/dot/sapphire.h +141 -0
  73. package/include/numkong/dot/serial.h +838 -0
  74. package/include/numkong/dot/sierra.h +405 -0
  75. package/include/numkong/dot/skylake.h +1084 -0
  76. package/include/numkong/dot/sve.h +379 -0
  77. package/include/numkong/dot/svebfdot.h +74 -0
  78. package/include/numkong/dot/svehalf.h +123 -0
  79. package/include/numkong/dot/v128relaxed.h +1258 -0
  80. package/include/numkong/dot.h +1070 -0
  81. package/include/numkong/dot.hpp +94 -0
  82. package/include/numkong/dots/README.md +496 -0
  83. package/include/numkong/dots/alder.h +114 -0
  84. package/include/numkong/dots/genoa.h +94 -0
  85. package/include/numkong/dots/haswell.h +295 -0
  86. package/include/numkong/dots/icelake.h +171 -0
  87. package/include/numkong/dots/neon.h +120 -0
  88. package/include/numkong/dots/neonbfdot.h +58 -0
  89. package/include/numkong/dots/neonfhm.h +94 -0
  90. package/include/numkong/dots/neonhalf.h +57 -0
  91. package/include/numkong/dots/neonsdot.h +108 -0
  92. package/include/numkong/dots/rvv.h +2486 -0
  93. package/include/numkong/dots/sapphireamx.h +3973 -0
  94. package/include/numkong/dots/serial.h +2844 -0
  95. package/include/numkong/dots/sierra.h +97 -0
  96. package/include/numkong/dots/skylake.h +196 -0
  97. package/include/numkong/dots/sme.h +5372 -0
  98. package/include/numkong/dots/smebi32.h +461 -0
  99. package/include/numkong/dots/smef64.h +1318 -0
  100. package/include/numkong/dots/smehalf.h +47 -0
  101. package/include/numkong/dots/v128relaxed.h +294 -0
  102. package/include/numkong/dots.h +2804 -0
  103. package/include/numkong/dots.hpp +639 -0
  104. package/include/numkong/each/README.md +469 -0
  105. package/include/numkong/each/haswell.h +1658 -0
  106. package/include/numkong/each/icelake.h +272 -0
  107. package/include/numkong/each/neon.h +1104 -0
  108. package/include/numkong/each/neonbfdot.h +212 -0
  109. package/include/numkong/each/neonhalf.h +410 -0
  110. package/include/numkong/each/rvv.h +1121 -0
  111. package/include/numkong/each/sapphire.h +477 -0
  112. package/include/numkong/each/serial.h +260 -0
  113. package/include/numkong/each/skylake.h +1562 -0
  114. package/include/numkong/each.h +2146 -0
  115. package/include/numkong/each.hpp +434 -0
  116. package/include/numkong/geospatial/README.md +147 -0
  117. package/include/numkong/geospatial/haswell.h +593 -0
  118. package/include/numkong/geospatial/neon.h +571 -0
  119. package/include/numkong/geospatial/rvv.h +701 -0
  120. package/include/numkong/geospatial/serial.h +309 -0
  121. package/include/numkong/geospatial/skylake.h +577 -0
  122. package/include/numkong/geospatial/v128relaxed.h +613 -0
  123. package/include/numkong/geospatial.h +453 -0
  124. package/include/numkong/geospatial.hpp +235 -0
  125. package/include/numkong/matrix.hpp +336 -0
  126. package/include/numkong/maxsim/README.md +187 -0
  127. package/include/numkong/maxsim/alder.h +511 -0
  128. package/include/numkong/maxsim/genoa.h +115 -0
  129. package/include/numkong/maxsim/haswell.h +553 -0
  130. package/include/numkong/maxsim/icelake.h +480 -0
  131. package/include/numkong/maxsim/neonsdot.h +394 -0
  132. package/include/numkong/maxsim/sapphireamx.h +877 -0
  133. package/include/numkong/maxsim/serial.h +490 -0
  134. package/include/numkong/maxsim/sme.h +929 -0
  135. package/include/numkong/maxsim/v128relaxed.h +280 -0
  136. package/include/numkong/maxsim.h +571 -0
  137. package/include/numkong/maxsim.hpp +133 -0
  138. package/include/numkong/mesh/README.md +227 -0
  139. package/include/numkong/mesh/haswell.h +2235 -0
  140. package/include/numkong/mesh/neon.h +1329 -0
  141. package/include/numkong/mesh/neonbfdot.h +842 -0
  142. package/include/numkong/mesh/neonhalf.h +616 -0
  143. package/include/numkong/mesh/rvv.h +916 -0
  144. package/include/numkong/mesh/serial.h +742 -0
  145. package/include/numkong/mesh/skylake.h +1135 -0
  146. package/include/numkong/mesh/v128relaxed.h +1052 -0
  147. package/include/numkong/mesh.h +652 -0
  148. package/include/numkong/mesh.hpp +762 -0
  149. package/include/numkong/numkong.h +78 -0
  150. package/include/numkong/numkong.hpp +57 -0
  151. package/include/numkong/probability/README.md +173 -0
  152. package/include/numkong/probability/haswell.h +267 -0
  153. package/include/numkong/probability/neon.h +225 -0
  154. package/include/numkong/probability/rvv.h +409 -0
  155. package/include/numkong/probability/serial.h +169 -0
  156. package/include/numkong/probability/skylake.h +324 -0
  157. package/include/numkong/probability.h +383 -0
  158. package/include/numkong/probability.hpp +120 -0
  159. package/include/numkong/random.h +50 -0
  160. package/include/numkong/random.hpp +285 -0
  161. package/include/numkong/reduce/README.md +547 -0
  162. package/include/numkong/reduce/alder.h +632 -0
  163. package/include/numkong/reduce/genoa.h +201 -0
  164. package/include/numkong/reduce/haswell.h +3783 -0
  165. package/include/numkong/reduce/icelake.h +549 -0
  166. package/include/numkong/reduce/neon.h +3841 -0
  167. package/include/numkong/reduce/neonbfdot.h +353 -0
  168. package/include/numkong/reduce/neonfhm.h +665 -0
  169. package/include/numkong/reduce/neonhalf.h +157 -0
  170. package/include/numkong/reduce/neonsdot.h +357 -0
  171. package/include/numkong/reduce/rvv.h +3407 -0
  172. package/include/numkong/reduce/serial.h +757 -0
  173. package/include/numkong/reduce/sierra.h +338 -0
  174. package/include/numkong/reduce/skylake.h +3792 -0
  175. package/include/numkong/reduce/v128relaxed.h +2302 -0
  176. package/include/numkong/reduce.h +1597 -0
  177. package/include/numkong/reduce.hpp +633 -0
  178. package/include/numkong/scalar/README.md +89 -0
  179. package/include/numkong/scalar/haswell.h +113 -0
  180. package/include/numkong/scalar/neon.h +122 -0
  181. package/include/numkong/scalar/neonhalf.h +70 -0
  182. package/include/numkong/scalar/rvv.h +211 -0
  183. package/include/numkong/scalar/sapphire.h +63 -0
  184. package/include/numkong/scalar/serial.h +332 -0
  185. package/include/numkong/scalar/v128relaxed.h +56 -0
  186. package/include/numkong/scalar.h +683 -0
  187. package/include/numkong/set/README.md +179 -0
  188. package/include/numkong/set/haswell.h +334 -0
  189. package/include/numkong/set/icelake.h +485 -0
  190. package/include/numkong/set/neon.h +364 -0
  191. package/include/numkong/set/rvv.h +226 -0
  192. package/include/numkong/set/rvvbb.h +117 -0
  193. package/include/numkong/set/serial.h +174 -0
  194. package/include/numkong/set/sve.h +185 -0
  195. package/include/numkong/set/v128relaxed.h +240 -0
  196. package/include/numkong/set.h +457 -0
  197. package/include/numkong/set.hpp +114 -0
  198. package/include/numkong/sets/README.md +149 -0
  199. package/include/numkong/sets/haswell.h +63 -0
  200. package/include/numkong/sets/icelake.h +66 -0
  201. package/include/numkong/sets/neon.h +61 -0
  202. package/include/numkong/sets/serial.h +43 -0
  203. package/include/numkong/sets/smebi32.h +1099 -0
  204. package/include/numkong/sets/v128relaxed.h +58 -0
  205. package/include/numkong/sets.h +339 -0
  206. package/include/numkong/sparse/README.md +156 -0
  207. package/include/numkong/sparse/icelake.h +463 -0
  208. package/include/numkong/sparse/neon.h +288 -0
  209. package/include/numkong/sparse/serial.h +117 -0
  210. package/include/numkong/sparse/sve2.h +507 -0
  211. package/include/numkong/sparse/turin.h +322 -0
  212. package/include/numkong/sparse.h +363 -0
  213. package/include/numkong/sparse.hpp +113 -0
  214. package/include/numkong/spatial/README.md +435 -0
  215. package/include/numkong/spatial/alder.h +607 -0
  216. package/include/numkong/spatial/genoa.h +290 -0
  217. package/include/numkong/spatial/haswell.h +960 -0
  218. package/include/numkong/spatial/icelake.h +586 -0
  219. package/include/numkong/spatial/neon.h +773 -0
  220. package/include/numkong/spatial/neonbfdot.h +165 -0
  221. package/include/numkong/spatial/neonhalf.h +118 -0
  222. package/include/numkong/spatial/neonsdot.h +261 -0
  223. package/include/numkong/spatial/rvv.h +984 -0
  224. package/include/numkong/spatial/rvvbf16.h +123 -0
  225. package/include/numkong/spatial/rvvhalf.h +117 -0
  226. package/include/numkong/spatial/sapphire.h +343 -0
  227. package/include/numkong/spatial/serial.h +346 -0
  228. package/include/numkong/spatial/sierra.h +323 -0
  229. package/include/numkong/spatial/skylake.h +606 -0
  230. package/include/numkong/spatial/sve.h +224 -0
  231. package/include/numkong/spatial/svebfdot.h +122 -0
  232. package/include/numkong/spatial/svehalf.h +109 -0
  233. package/include/numkong/spatial/v128relaxed.h +717 -0
  234. package/include/numkong/spatial.h +1425 -0
  235. package/include/numkong/spatial.hpp +183 -0
  236. package/include/numkong/spatials/README.md +580 -0
  237. package/include/numkong/spatials/alder.h +94 -0
  238. package/include/numkong/spatials/genoa.h +94 -0
  239. package/include/numkong/spatials/haswell.h +219 -0
  240. package/include/numkong/spatials/icelake.h +113 -0
  241. package/include/numkong/spatials/neon.h +109 -0
  242. package/include/numkong/spatials/neonbfdot.h +60 -0
  243. package/include/numkong/spatials/neonfhm.h +92 -0
  244. package/include/numkong/spatials/neonhalf.h +58 -0
  245. package/include/numkong/spatials/neonsdot.h +109 -0
  246. package/include/numkong/spatials/rvv.h +1960 -0
  247. package/include/numkong/spatials/sapphireamx.h +1149 -0
  248. package/include/numkong/spatials/serial.h +226 -0
  249. package/include/numkong/spatials/sierra.h +96 -0
  250. package/include/numkong/spatials/skylake.h +184 -0
  251. package/include/numkong/spatials/sme.h +1901 -0
  252. package/include/numkong/spatials/smef64.h +465 -0
  253. package/include/numkong/spatials/v128relaxed.h +240 -0
  254. package/include/numkong/spatials.h +3021 -0
  255. package/include/numkong/spatials.hpp +508 -0
  256. package/include/numkong/tensor.hpp +1592 -0
  257. package/include/numkong/trigonometry/README.md +184 -0
  258. package/include/numkong/trigonometry/haswell.h +652 -0
  259. package/include/numkong/trigonometry/neon.h +639 -0
  260. package/include/numkong/trigonometry/rvv.h +699 -0
  261. package/include/numkong/trigonometry/serial.h +703 -0
  262. package/include/numkong/trigonometry/skylake.h +721 -0
  263. package/include/numkong/trigonometry/v128relaxed.h +666 -0
  264. package/include/numkong/trigonometry.h +467 -0
  265. package/include/numkong/trigonometry.hpp +166 -0
  266. package/include/numkong/types.h +1384 -0
  267. package/include/numkong/types.hpp +5603 -0
  268. package/include/numkong/vector.hpp +698 -0
  269. package/javascript/README.md +246 -0
  270. package/javascript/dist/cjs/numkong-wasm.d.ts +166 -0
  271. package/javascript/dist/cjs/numkong-wasm.js +617 -0
  272. package/javascript/dist/cjs/numkong.d.ts +343 -0
  273. package/javascript/dist/cjs/numkong.js +523 -0
  274. package/javascript/dist/cjs/package.json +3 -0
  275. package/javascript/dist/cjs/types.d.ts +284 -0
  276. package/javascript/dist/cjs/types.js +653 -0
  277. package/javascript/dist/esm/numkong-wasm.d.ts +166 -0
  278. package/javascript/dist/esm/numkong-wasm.js +595 -0
  279. package/javascript/dist/esm/numkong.d.ts +343 -0
  280. package/javascript/dist/esm/numkong.js +452 -0
  281. package/javascript/dist/esm/package.json +3 -0
  282. package/javascript/dist/esm/types.d.ts +284 -0
  283. package/javascript/dist/esm/types.js +630 -0
  284. package/javascript/dist-package-cjs.json +3 -0
  285. package/javascript/dist-package-esm.json +3 -0
  286. package/javascript/node-gyp-build.d.ts +1 -0
  287. package/javascript/numkong-wasm.ts +756 -0
  288. package/javascript/numkong.c +689 -0
  289. package/javascript/numkong.ts +575 -0
  290. package/javascript/tsconfig-base.json +39 -0
  291. package/javascript/tsconfig-cjs.json +8 -0
  292. package/javascript/tsconfig-esm.json +8 -0
  293. package/javascript/types.ts +674 -0
  294. package/package.json +87 -0
@@ -0,0 +1,114 @@
1
+ /**
2
+ * @brief SIMD-accelerated Batched Dot Products for Alder Lake.
3
+ * @file include/numkong/dots/alder.h
4
+ * @author Ash Vardanian
5
+ * @date March 4, 2026
6
+ *
7
+ * @sa include/numkong/dots.h
8
+ *
9
+ * Uses AVX-VNNI (256-bit) for integer GEMM via DPBUSD with algebraic
10
+ * sign transformations for signed*signed and unsigned*unsigned cases.
11
+ */
12
+ #ifndef NK_DOTS_ALDER_H
13
+ #define NK_DOTS_ALDER_H
14
+
15
+ #if NK_TARGET_X86_
16
+ #if NK_TARGET_ALDER
17
+
18
+ #include "numkong/dot/alder.h" // Alder-specific dot product helpers
19
+ #include "numkong/dot/haswell.h" // Haswell partial load functions
20
+ #include "numkong/dots/serial.h" // GEMM macro definitions
21
+
22
+ #if defined(__cplusplus)
23
+ extern "C" {
24
+ #endif
25
+
26
+ #if defined(__clang__)
27
+ #pragma clang attribute push(__attribute__((target("avx2,f16c,fma,bmi,bmi2,avxvnni"))), apply_to = function)
28
+ #elif defined(__GNUC__)
29
+ #pragma GCC push_options
30
+ #pragma GCC target("avx2", "f16c", "fma", "bmi", "bmi2", "avxvnni")
31
+ #endif
32
+
33
+ /* I8 GEMM: depth_simd_dimensions=32 — compensated (B sums precomputed in pack) */
34
+ nk_define_cross_compensated_pack_size_(dots, i8, alder, i8, i8,
35
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32,
36
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
37
+ nk_define_cross_compensated_pack_(dots, i8, alder, i8, i8, nk_assign_from_to_,
38
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32, nk_dots_reduce_moments_i8_,
39
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
40
+ nk_define_cross_compensated_symmetric_(dots, i8, alder, i8, i32,
41
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32, nk_b256_vec_t,
42
+ nk_dot_i8x32_state_alder_t, nk_b128_vec_t, nk_dot_i8x32_init_alder,
43
+ nk_load_b256_haswell_, nk_partial_load_b8x32_serial_, nk_dot_i8x32_update_alder,
44
+ nk_dot_i8x32_finalize_alder, nk_store_b128_haswell_,
45
+ nk_partial_store_b32x4_haswell_, nk_load_b128_haswell_,
46
+ nk_partial_load_b32x4_haswell_, nk_sum_i8x32_state_alder_t,
47
+ nk_sum_i8x32_init_alder, nk_sum_i8x32_update_alder, nk_sum_i8x32_finalize_alder,
48
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
49
+ nk_define_cross_compensated_packed_(dots, i8, alder, i8, i8, i32,
50
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32, nk_b256_vec_t,
51
+ nk_dot_i8x32_state_alder_t, nk_b128_vec_t, nk_dot_i8x32_init_alder,
52
+ nk_load_b256_haswell_, nk_partial_load_b8x32_serial_, nk_load_b256_haswell_,
53
+ nk_partial_load_b8x32_serial_, nk_dot_i8x32_update_alder,
54
+ nk_dot_i8x32_finalize_alder, nk_store_b128_haswell_,
55
+ nk_partial_store_b32x4_haswell_, nk_load_b128_haswell_,
56
+ nk_partial_load_b32x4_haswell_, nk_dots_reduce_sum_i8_stub_,
57
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
58
+
59
+ /* U8 GEMM: depth_simd_dimensions=32 — compensated (operand swap, B sums precomputed) */
60
+ nk_define_cross_compensated_pack_size_(dots, u8, alder, u8, u8,
61
+ /*sum_value_type=*/u32, /*norm_value_type=*/u32,
62
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
63
+ nk_define_cross_compensated_pack_(dots, u8, alder, u8, u8, nk_assign_from_to_,
64
+ /*sum_value_type=*/u32, /*norm_value_type=*/u32, nk_dots_reduce_moments_u8_,
65
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
66
+ nk_define_cross_compensated_symmetric_(dots, u8, alder, u8, u32,
67
+ /*sum_value_type=*/u32, /*norm_value_type=*/u32, nk_b256_vec_t,
68
+ nk_dot_u8x32_state_alder_t, nk_b128_vec_t, nk_dot_u8x32_init_alder,
69
+ nk_load_b256_haswell_, nk_partial_load_b8x32_serial_, nk_dot_u8x32_update_alder,
70
+ nk_dot_u8x32_finalize_alder, nk_store_b128_haswell_,
71
+ nk_partial_store_b32x4_haswell_, nk_load_b128_haswell_,
72
+ nk_partial_load_b32x4_haswell_, nk_sum_u8x32_state_alder_t,
73
+ nk_sum_u8x32_init_alder, nk_sum_u8x32_update_alder, nk_sum_u8x32_finalize_alder,
74
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
75
+ nk_define_cross_compensated_packed_(dots, u8, alder, u8, u8, u32,
76
+ /*sum_value_type=*/u32, /*norm_value_type=*/u32, nk_b256_vec_t,
77
+ nk_dot_u8x32_state_alder_t, nk_b128_vec_t, nk_dot_u8x32_init_alder,
78
+ nk_load_b256_haswell_, nk_partial_load_b8x32_serial_, nk_load_b256_haswell_,
79
+ nk_partial_load_b8x32_serial_, nk_dot_u8x32_update_alder,
80
+ nk_dot_u8x32_finalize_alder, nk_store_b128_haswell_,
81
+ nk_partial_store_b32x4_haswell_, nk_load_b128_haswell_,
82
+ nk_partial_load_b32x4_haswell_, nk_dots_reduce_sum_u8_stub_,
83
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
84
+
85
+ /* E2M3 GEMM via DPBUSD integer path: depth_simd_dimensions=32 (32 e2m3s = 32 bytes = AVX2 register width) */
86
+ nk_define_cross_pack_size_(dots, e2m3, alder, e2m3, e2m3, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/32,
87
+ /*dimensions_per_value=*/1)
88
+ nk_define_cross_pack_(dots, e2m3, alder, e2m3, e2m3, nk_assign_from_to_, /*norm_value_type=*/f32,
89
+ nk_dots_reduce_sumsq_e2m3_, /*depth_simd_dimensions=*/32,
90
+ /*dimensions_per_value=*/1)
91
+ nk_define_cross_symmetric_(dots, e2m3, alder, e2m3, f32, nk_b256_vec_t, nk_dot_e2m3x32_state_alder_t, nk_b128_vec_t,
92
+ nk_dot_e2m3x32_init_alder, nk_load_b256_haswell_, nk_partial_load_b8x32_serial_,
93
+ nk_dot_e2m3x32_update_alder, nk_dot_e2m3x32_finalize_alder, nk_store_b128_haswell_,
94
+ nk_partial_store_b32x4_haswell_,
95
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
96
+ nk_define_cross_packed_(dots, e2m3, alder, e2m3, e2m3, f32, nk_b256_vec_t, nk_dot_e2m3x32_state_alder_t, nk_b128_vec_t,
97
+ nk_dot_e2m3x32_init_alder, nk_load_b256_haswell_, nk_partial_load_b8x32_serial_,
98
+ nk_load_b256_haswell_, nk_partial_load_b8x32_serial_, nk_dot_e2m3x32_update_alder,
99
+ nk_dot_e2m3x32_finalize_alder, nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
100
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
101
+
102
+ #if defined(__clang__)
103
+ #pragma clang attribute pop
104
+ #elif defined(__GNUC__)
105
+ #pragma GCC pop_options
106
+ #endif
107
+
108
+ #if defined(__cplusplus)
109
+ } // extern "C"
110
+ #endif
111
+
112
+ #endif // NK_TARGET_ALDER
113
+ #endif // NK_TARGET_X86_
114
+ #endif // NK_DOTS_ALDER_H
@@ -0,0 +1,94 @@
1
+ /**
2
+ * @brief SIMD-accelerated Batched Dot Products for Genoa.
3
+ * @file include/numkong/dots/genoa.h
4
+ * @author Ash Vardanian
5
+ * @date December 27, 2025
6
+ *
7
+ * @sa include/numkong/dots.h
8
+ */
9
+ #ifndef NK_DOTS_GENOA_H
10
+ #define NK_DOTS_GENOA_H
11
+
12
+ #if NK_TARGET_X86_
13
+ #if NK_TARGET_GENOA
14
+
15
+ #include "numkong/dot/genoa.h"
16
+ #include "numkong/cast/icelake.h" // `nk_load_e4m3x32_to_bf16x32_icelake_`
17
+
18
+ #if defined(__cplusplus)
19
+ extern "C" {
20
+ #endif
21
+
22
+ #if defined(__clang__)
23
+ #pragma clang attribute push( \
24
+ __attribute__((target("avx2,avx512f,avx512vl,avx512bw,avx512dq,avx512bf16,f16c,fma,bmi,bmi2"))), \
25
+ apply_to = function)
26
+ #elif defined(__GNUC__)
27
+ #pragma GCC push_options
28
+ #pragma GCC target("avx2", "avx512f", "avx512vl", "avx512bw", "avx512dq", "avx512bf16", "f16c", "fma", "bmi", "bmi2")
29
+ #endif
30
+
31
+ /* BF16 GEMM: depth_simd_dimensions=32 (32 bf16s = 64 bytes = 1 cache line) */
32
+ nk_define_cross_pack_size_(dots, bf16, genoa, bf16, bf16, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/32,
33
+ /*dimensions_per_value=*/1)
34
+ nk_define_cross_pack_(dots, bf16, genoa, bf16, bf16, nk_assign_from_to_, /*norm_value_type=*/f32,
35
+ nk_dots_reduce_sumsq_bf16_, /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
36
+ nk_define_cross_symmetric_(dots, bf16, genoa, bf16, f32, nk_b512_vec_t, nk_dot_through_bf16_state_genoa_t_,
37
+ nk_b128_vec_t, nk_dot_through_bf16_init_genoa_, nk_load_b512_skylake_,
38
+ nk_partial_load_b16x32_skylake_, nk_dot_through_bf16_update_genoa_,
39
+ nk_dot_through_bf16_finalize_genoa_, nk_store_b128_haswell_, nk_partial_store_b32x4_skylake_,
40
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
41
+ nk_define_cross_packed_(dots, bf16, genoa, bf16, bf16, f32, nk_b512_vec_t, nk_dot_through_bf16_state_genoa_t_,
42
+ nk_b128_vec_t, nk_dot_through_bf16_init_genoa_, nk_load_b512_skylake_,
43
+ nk_partial_load_b16x32_skylake_, nk_load_b512_skylake_, nk_partial_load_b16x32_skylake_,
44
+ nk_dot_through_bf16_update_genoa_, nk_dot_through_bf16_finalize_genoa_, nk_store_b128_haswell_,
45
+ nk_partial_store_b32x4_skylake_,
46
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
47
+
48
+ /* E4M3 GEMM: depth_simd_dimensions=32 (32 e4m3s = 32 bytes = half cache line), F32 accumulator */
49
+ nk_define_cross_pack_size_(dots, e4m3, genoa, e4m3, bf16, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/32,
50
+ /*dimensions_per_value=*/1)
51
+ nk_define_cross_pack_(dots, e4m3, genoa, e4m3, bf16, nk_e4m3_to_bf16, /*norm_value_type=*/f32,
52
+ nk_dots_reduce_sumsq_e4m3_, /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
53
+ nk_define_cross_symmetric_(dots, e4m3, genoa, e4m3, f32, nk_b512_vec_t, nk_dot_through_bf16_state_genoa_t_,
54
+ nk_b128_vec_t, nk_dot_through_bf16_init_genoa_, nk_load_e4m3x32_to_bf16x32_icelake_,
55
+ nk_partial_load_e4m3x32_to_bf16x32_icelake_, nk_dot_through_bf16_update_genoa_,
56
+ nk_dot_through_bf16_finalize_genoa_, nk_store_b128_haswell_, nk_partial_store_b32x4_skylake_,
57
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
58
+ nk_define_cross_packed_(dots, e4m3, genoa, e4m3, bf16, f32, nk_b512_vec_t, nk_dot_through_bf16_state_genoa_t_,
59
+ nk_b128_vec_t, nk_dot_through_bf16_init_genoa_, nk_load_e4m3x32_to_bf16x32_icelake_,
60
+ nk_partial_load_e4m3x32_to_bf16x32_icelake_, nk_load_b512_skylake_,
61
+ nk_partial_load_b16x32_skylake_, nk_dot_through_bf16_update_genoa_,
62
+ nk_dot_through_bf16_finalize_genoa_, nk_store_b128_haswell_, nk_partial_store_b32x4_skylake_,
63
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
64
+
65
+ /* E5M2 GEMM: depth_simd_dimensions=32 (32 e5m2s = 32 bytes = half cache line), F32 accumulator */
66
+ nk_define_cross_pack_size_(dots, e5m2, genoa, e5m2, bf16, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/32,
67
+ /*dimensions_per_value=*/1)
68
+ nk_define_cross_pack_(dots, e5m2, genoa, e5m2, bf16, nk_e5m2_to_bf16, /*norm_value_type=*/f32,
69
+ nk_dots_reduce_sumsq_e5m2_, /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
70
+ nk_define_cross_symmetric_(dots, e5m2, genoa, e5m2, f32, nk_b512_vec_t, nk_dot_through_bf16_state_genoa_t_,
71
+ nk_b128_vec_t, nk_dot_through_bf16_init_genoa_, nk_load_e5m2x32_to_bf16x32_icelake_,
72
+ nk_partial_load_e5m2x32_to_bf16x32_icelake_, nk_dot_through_bf16_update_genoa_,
73
+ nk_dot_through_bf16_finalize_genoa_, nk_store_b128_haswell_, nk_partial_store_b32x4_skylake_,
74
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
75
+ nk_define_cross_packed_(dots, e5m2, genoa, e5m2, bf16, f32, nk_b512_vec_t, nk_dot_through_bf16_state_genoa_t_,
76
+ nk_b128_vec_t, nk_dot_through_bf16_init_genoa_, nk_load_e5m2x32_to_bf16x32_icelake_,
77
+ nk_partial_load_e5m2x32_to_bf16x32_icelake_, nk_load_b512_skylake_,
78
+ nk_partial_load_b16x32_skylake_, nk_dot_through_bf16_update_genoa_,
79
+ nk_dot_through_bf16_finalize_genoa_, nk_store_b128_haswell_, nk_partial_store_b32x4_skylake_,
80
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
81
+
82
+ #if defined(__clang__)
83
+ #pragma clang attribute pop
84
+ #elif defined(__GNUC__)
85
+ #pragma GCC pop_options
86
+ #endif
87
+
88
+ #if defined(__cplusplus)
89
+ } // extern "C"
90
+ #endif
91
+
92
+ #endif // NK_TARGET_GENOA
93
+ #endif // NK_TARGET_X86_
94
+ #endif // NK_DOTS_GENOA_H
@@ -0,0 +1,295 @@
1
+ /**
2
+ * @brief SIMD-accelerated Batched Dot Products for Haswell.
3
+ * @file include/numkong/dots/haswell.h
4
+ * @author Ash Vardanian
5
+ * @date December 27, 2025
6
+ *
7
+ * @sa include/numkong/dots.h
8
+ *
9
+ * @section haswell_dots_instructions Key AVX2/FMA GEMM Instructions
10
+ *
11
+ * Intrinsic Instruction Latency Throughput Ports
12
+ * _mm256_fmadd_ps/pd VFMADD (YMM, YMM, YMM) 5cy 0.5/cy p01
13
+ * _mm256_mul_ps VMULPS (YMM, YMM, YMM) 5cy 0.5/cy p01
14
+ * _mm256_add_ps VADDPS (YMM, YMM, YMM) 3cy 1/cy p01
15
+ * _mm256_cvtph_ps VCVTPH2PS (YMM, XMM) 5cy 1/cy p01
16
+ * _mm256_madd_epi16 VPMADDWD (YMM, YMM, YMM) 5cy 1/cy p0
17
+ *
18
+ * GEMM kernels use tiled dot products with 4-way parallel accumulation to hide FMA latency.
19
+ * Type-specific tile sizes: f32/f64 use depth_simd_dimensions=4, f16/bf16 use depth_simd_dimensions=8,
20
+ * i8/u8/fp8 use depth_simd_dimensions=16. Integer dot products use VPMADDWD for efficient i16 pair
21
+ * multiplication with i32 accumulation.
22
+ */
23
+ #ifndef NK_DOTS_HASWELL_H
24
+ #define NK_DOTS_HASWELL_H
25
+
26
+ #if NK_TARGET_X86_
27
+ #if NK_TARGET_HASWELL
28
+
29
+ #include "numkong/dot/haswell.h"
30
+ #include "numkong/cast.h" // `nk_f16_to_f32`
31
+
32
+ #if defined(__cplusplus)
33
+ extern "C" {
34
+ #endif
35
+
36
+ #if defined(__clang__)
37
+ #pragma clang attribute push(__attribute__((target("avx2,f16c,fma,bmi,bmi2"))), apply_to = function)
38
+ #elif defined(__GNUC__)
39
+ #pragma GCC push_options
40
+ #pragma GCC target("avx2", "f16c", "fma", "bmi", "bmi2")
41
+ #endif
42
+
43
+ /* F32 GEMM: depth_simd_dimensions=4 (4 f32s = 16 bytes for f32->f64 upcast accumulation) */
44
+ nk_define_cross_pack_size_(dots, f32, haswell, f32, f32, /*norm_value_type=*/f64, /*depth_simd_dimensions=*/4,
45
+ /*dimensions_per_value=*/1)
46
+ nk_define_cross_pack_(dots, f32, haswell, f32, f32, nk_assign_from_to_, /*norm_value_type=*/f64,
47
+ nk_dots_reduce_sumsq_f32_,
48
+ /*depth_simd_dimensions=*/4, /*dimensions_per_value=*/1)
49
+ nk_define_cross_symmetric_(dots, f32, haswell, f32, f64, nk_b128_vec_t, nk_dot_f32x4_state_haswell_t, nk_b256_vec_t,
50
+ nk_dot_f32x4_init_haswell, nk_load_b128_haswell_, nk_partial_load_b32x4_haswell_,
51
+ nk_dot_f32x4_update_haswell, nk_dot_f32x4_finalize_haswell, nk_store_b256_haswell_,
52
+ nk_partial_store_b64x4_haswell_,
53
+ /*depth_simd_dimensions=*/4, /*dimensions_per_value=*/1)
54
+ nk_define_cross_packed_(dots, f32, haswell, f32, f32, f64, nk_b128_vec_t, nk_dot_f32x4_state_haswell_t, nk_b256_vec_t,
55
+ nk_dot_f32x4_init_haswell, nk_load_b128_haswell_, nk_partial_load_b32x4_haswell_,
56
+ nk_load_b128_haswell_, nk_partial_load_b32x4_haswell_, nk_dot_f32x4_update_haswell,
57
+ nk_dot_f32x4_finalize_haswell, nk_store_b256_haswell_, nk_partial_store_b64x4_haswell_,
58
+ /*depth_simd_dimensions=*/4, /*dimensions_per_value=*/1)
59
+
60
+ /* F64 GEMM: depth_simd_dimensions=4 (4 f64s = 32 bytes = AVX2 register width) */
61
+ nk_define_cross_pack_size_(dots, f64, haswell, f64, f64, /*norm_value_type=*/f64, /*depth_simd_dimensions=*/4,
62
+ /*dimensions_per_value=*/1)
63
+ nk_define_cross_pack_(dots, f64, haswell, f64, f64, nk_assign_from_to_, /*norm_value_type=*/f64,
64
+ nk_dots_reduce_sumsq_f64_,
65
+ /*depth_simd_dimensions=*/4, /*dimensions_per_value=*/1)
66
+ nk_define_cross_symmetric_(dots, f64, haswell, f64, f64, nk_b256_vec_t, nk_dot_f64x4_state_haswell_t, nk_b256_vec_t,
67
+ nk_dot_f64x4_init_haswell, nk_load_b256_haswell_, nk_partial_load_b64x4_haswell_,
68
+ nk_dot_f64x4_update_haswell, nk_dot_f64x4_finalize_haswell, nk_store_b256_haswell_,
69
+ nk_partial_store_b64x4_haswell_,
70
+ /*depth_simd_dimensions=*/4, /*dimensions_per_value=*/1)
71
+ nk_define_cross_packed_(dots, f64, haswell, f64, f64, f64, nk_b256_vec_t, nk_dot_f64x4_state_haswell_t, nk_b256_vec_t,
72
+ nk_dot_f64x4_init_haswell, nk_load_b256_haswell_, nk_partial_load_b64x4_haswell_,
73
+ nk_load_b256_haswell_, nk_partial_load_b64x4_haswell_, nk_dot_f64x4_update_haswell,
74
+ nk_dot_f64x4_finalize_haswell, nk_store_b256_haswell_, nk_partial_store_b64x4_haswell_,
75
+ /*depth_simd_dimensions=*/4, /*dimensions_per_value=*/1)
76
+
77
+ /* F16 GEMM: depth_simd_dimensions=8 (8 f16s = 16 bytes = 128-bit input) → upcasted to 8×f32 (256-bit) */
78
+ nk_define_cross_pack_size_(dots, f16, haswell, f16, f32, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/8,
79
+ /*dimensions_per_value=*/1)
80
+ nk_define_cross_pack_(dots, f16, haswell, f16, f32, nk_f16_to_f32, /*norm_value_type=*/f32, nk_dots_reduce_sumsq_f16_,
81
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1) // Store as F32
82
+ nk_define_cross_symmetric_(dots, f16, haswell, f16, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
83
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_f16x8_to_f32x8_haswell_,
84
+ nk_partial_load_f16x8_to_f32x8_haswell_, nk_dot_through_f32_update_haswell_,
85
+ nk_dot_through_f32_finalize_haswell_, nk_store_b128_haswell_,
86
+ nk_partial_store_b32x4_haswell_,
87
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
88
+ nk_define_cross_packed_(dots, f16, haswell, f16, f32, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
89
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_f16x8_to_f32x8_haswell_,
90
+ nk_partial_load_f16x8_to_f32x8_haswell_, nk_load_b256_haswell_, nk_partial_load_b32x8_serial_,
91
+ nk_dot_through_f32_update_haswell_, nk_dot_through_f32_finalize_haswell_,
92
+ nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
93
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
94
+
95
+ /* BF16 GEMM: depth_simd_dimensions=8 (8 bf16s = 16 bytes = 128-bit input) → upcasted to 8×f32 (256-bit) */
96
+ nk_define_cross_pack_size_(dots, bf16, haswell, bf16, f32, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/8,
97
+ /*dimensions_per_value=*/1)
98
+ nk_define_cross_pack_(dots, bf16, haswell, bf16, f32, nk_bf16_to_f32, /*norm_value_type=*/f32,
99
+ nk_dots_reduce_sumsq_bf16_,
100
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1) // Store as F32
101
+ nk_define_cross_symmetric_(dots, bf16, haswell, bf16, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
102
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_bf16x8_to_f32x8_haswell_,
103
+ nk_partial_load_bf16x8_to_f32x8_haswell_, nk_dot_through_f32_update_haswell_,
104
+ nk_dot_through_f32_finalize_haswell_, nk_store_b128_haswell_,
105
+ nk_partial_store_b32x4_haswell_,
106
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
107
+ nk_define_cross_packed_(dots, bf16, haswell, bf16, f32, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
108
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_bf16x8_to_f32x8_haswell_,
109
+ nk_partial_load_bf16x8_to_f32x8_haswell_, nk_load_b256_haswell_, nk_partial_load_b32x8_serial_,
110
+ nk_dot_through_f32_update_haswell_, nk_dot_through_f32_finalize_haswell_,
111
+ nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
112
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
113
+
114
+ /* E4M3 GEMM: depth_simd_dimensions=8 (8 e4m3s = 8 bytes) → upcasted to 8×f32 (256-bit) */
115
+ nk_define_cross_pack_size_(dots, e4m3, haswell, e4m3, f32, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/8,
116
+ /*dimensions_per_value=*/1)
117
+ nk_define_cross_pack_(dots, e4m3, haswell, e4m3, f32, nk_e4m3_to_f32, /*norm_value_type=*/f32,
118
+ nk_dots_reduce_sumsq_e4m3_,
119
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
120
+ nk_define_cross_symmetric_(dots, e4m3, haswell, e4m3, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
121
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_e4m3x8_to_f32x8_haswell_,
122
+ nk_partial_load_e4m3x8_to_f32x8_haswell_, nk_dot_through_f32_update_haswell_,
123
+ nk_dot_through_f32_finalize_haswell_, nk_store_b128_haswell_,
124
+ nk_partial_store_b32x4_haswell_,
125
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
126
+ nk_define_cross_packed_(dots, e4m3, haswell, e4m3, f32, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
127
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_e4m3x8_to_f32x8_haswell_,
128
+ nk_partial_load_e4m3x8_to_f32x8_haswell_, nk_load_b256_haswell_, nk_partial_load_b32x8_serial_,
129
+ nk_dot_through_f32_update_haswell_, nk_dot_through_f32_finalize_haswell_,
130
+ nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
131
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
132
+
133
+ /* E5M2 GEMM: depth_simd_dimensions=8 (8 e5m2s = 8 bytes) → upcasted to 8×f32 (256-bit) */
134
+ nk_define_cross_pack_size_(dots, e5m2, haswell, e5m2, f32, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/8,
135
+ /*dimensions_per_value=*/1)
136
+ nk_define_cross_pack_(dots, e5m2, haswell, e5m2, f32, nk_e5m2_to_f32, /*norm_value_type=*/f32,
137
+ nk_dots_reduce_sumsq_e5m2_,
138
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
139
+ nk_define_cross_symmetric_(dots, e5m2, haswell, e5m2, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
140
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_e5m2x8_to_f32x8_haswell_,
141
+ nk_partial_load_e5m2x8_to_f32x8_haswell_, nk_dot_through_f32_update_haswell_,
142
+ nk_dot_through_f32_finalize_haswell_, nk_store_b128_haswell_,
143
+ nk_partial_store_b32x4_haswell_,
144
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
145
+ nk_define_cross_packed_(dots, e5m2, haswell, e5m2, f32, f32, nk_b256_vec_t, nk_dot_through_f32_state_haswell_t_,
146
+ nk_b128_vec_t, nk_dot_through_f32_init_haswell_, nk_load_e5m2x8_to_f32x8_haswell_,
147
+ nk_partial_load_e5m2x8_to_f32x8_haswell_, nk_load_b256_haswell_, nk_partial_load_b32x8_serial_,
148
+ nk_dot_through_f32_update_haswell_, nk_dot_through_f32_finalize_haswell_,
149
+ nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
150
+ /*depth_simd_dimensions=*/8, /*dimensions_per_value=*/1)
151
+
152
+ /* E2M3 GEMM: integer LUT path, depth_simd_dimensions=32 (32 e2m3s = 32 bytes = AVX2 register width) */
153
+ nk_define_cross_pack_size_(dots, e2m3, haswell, e2m3, e2m3, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/32,
154
+ /*dimensions_per_value=*/1)
155
+ nk_define_cross_pack_(dots, e2m3, haswell, e2m3, e2m3, nk_assign_from_to_, /*norm_value_type=*/f32,
156
+ nk_dots_reduce_sumsq_e2m3_,
157
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
158
+ nk_define_cross_symmetric_(dots, e2m3, haswell, e2m3, f32, nk_b256_vec_t, nk_dot_e2m3x32_state_haswell_t, nk_b128_vec_t,
159
+ nk_dot_e2m3x32_init_haswell, nk_load_b256_haswell_, nk_partial_load_b8x32_serial_,
160
+ nk_dot_e2m3x32_update_haswell, nk_dot_e2m3x32_finalize_haswell, nk_store_b128_haswell_,
161
+ nk_partial_store_b32x4_haswell_,
162
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
163
+ nk_define_cross_packed_(dots, e2m3, haswell, e2m3, e2m3, f32, nk_b256_vec_t, nk_dot_e2m3x32_state_haswell_t,
164
+ nk_b128_vec_t, nk_dot_e2m3x32_init_haswell, nk_load_b256_haswell_,
165
+ nk_partial_load_b8x32_serial_, nk_load_b256_haswell_, nk_partial_load_b8x32_serial_,
166
+ nk_dot_e2m3x32_update_haswell, nk_dot_e2m3x32_finalize_haswell, nk_store_b128_haswell_,
167
+ nk_partial_store_b32x4_haswell_,
168
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
169
+
170
+ /* E3M2 GEMM: integer LUT path, depth_simd_dimensions=32 (32 e3m2s = 32 bytes = AVX2 register width) */
171
+ nk_define_cross_pack_size_(dots, e3m2, haswell, e3m2, e3m2, /*norm_value_type=*/f32, /*depth_simd_dimensions=*/32,
172
+ /*dimensions_per_value=*/1)
173
+ nk_define_cross_pack_(dots, e3m2, haswell, e3m2, e3m2, nk_assign_from_to_, /*norm_value_type=*/f32,
174
+ nk_dots_reduce_sumsq_e3m2_,
175
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
176
+ nk_define_cross_symmetric_(dots, e3m2, haswell, e3m2, f32, nk_b256_vec_t, nk_dot_e3m2x32_state_haswell_t, nk_b128_vec_t,
177
+ nk_dot_e3m2x32_init_haswell, nk_load_b256_haswell_, nk_partial_load_b8x32_serial_,
178
+ nk_dot_e3m2x32_update_haswell, nk_dot_e3m2x32_finalize_haswell, nk_store_b128_haswell_,
179
+ nk_partial_store_b32x4_haswell_,
180
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
181
+ nk_define_cross_packed_(dots, e3m2, haswell, e3m2, e3m2, f32, nk_b256_vec_t, nk_dot_e3m2x32_state_haswell_t,
182
+ nk_b128_vec_t, nk_dot_e3m2x32_init_haswell, nk_load_b256_haswell_,
183
+ nk_partial_load_b8x32_serial_, nk_load_b256_haswell_, nk_partial_load_b8x32_serial_,
184
+ nk_dot_e3m2x32_update_haswell, nk_dot_e3m2x32_finalize_haswell, nk_store_b128_haswell_,
185
+ nk_partial_store_b32x4_haswell_,
186
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/1)
187
+
188
+ /* I8 GEMM: depth_simd_dimensions=16 (16 i8s = 16 bytes = 128-bit input) */
189
+ nk_define_cross_pack_size_(dots, i8, haswell, i8, i8, /*norm_value_type=*/u32, /*depth_simd_dimensions=*/16,
190
+ /*dimensions_per_value=*/1)
191
+ nk_define_cross_pack_(dots, i8, haswell, i8, i8, nk_assign_from_to_, /*norm_value_type=*/u32, nk_dots_reduce_sumsq_i8_,
192
+ /*depth_simd_dimensions=*/16, /*dimensions_per_value=*/1)
193
+ nk_define_cross_symmetric_(dots, i8, haswell, i8, i32, nk_b128_vec_t, nk_dot_i8x16_state_haswell_t, nk_b128_vec_t,
194
+ nk_dot_i8x16_init_haswell, nk_load_b128_haswell_, nk_partial_load_b8x16_serial_,
195
+ nk_dot_i8x16_update_haswell, nk_dot_i8x16_finalize_haswell, nk_store_b128_haswell_,
196
+ nk_partial_store_b32x4_haswell_,
197
+ /*depth_simd_dimensions=*/16, /*dimensions_per_value=*/1)
198
+ nk_define_cross_packed_(dots, i8, haswell, i8, i8, i32, nk_b128_vec_t, nk_dot_i8x16_state_haswell_t, nk_b128_vec_t,
199
+ nk_dot_i8x16_init_haswell, nk_load_b128_haswell_, nk_partial_load_b8x16_serial_,
200
+ nk_load_b128_haswell_, nk_partial_load_b8x16_serial_, nk_dot_i8x16_update_haswell,
201
+ nk_dot_i8x16_finalize_haswell, nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
202
+ /*depth_simd_dimensions=*/16, /*dimensions_per_value=*/1)
203
+
204
+ /* U8 GEMM: depth_simd_dimensions=16 (16 u8s = 16 bytes = 128-bit input) */
205
+ nk_define_cross_pack_size_(dots, u8, haswell, u8, u8, /*norm_value_type=*/u32, /*depth_simd_dimensions=*/16,
206
+ /*dimensions_per_value=*/1)
207
+ nk_define_cross_pack_(dots, u8, haswell, u8, u8, nk_assign_from_to_, /*norm_value_type=*/u32, nk_dots_reduce_sumsq_u8_,
208
+ /*depth_simd_dimensions=*/16, /*dimensions_per_value=*/1)
209
+ nk_define_cross_symmetric_(dots, u8, haswell, u8, u32, nk_b128_vec_t, nk_dot_u8x16_state_haswell_t, nk_b128_vec_t,
210
+ nk_dot_u8x16_init_haswell, nk_load_b128_haswell_, nk_partial_load_b8x16_serial_,
211
+ nk_dot_u8x16_update_haswell, nk_dot_u8x16_finalize_haswell, nk_store_b128_haswell_,
212
+ nk_partial_store_b32x4_haswell_,
213
+ /*depth_simd_dimensions=*/16, /*dimensions_per_value=*/1)
214
+ nk_define_cross_packed_(dots, u8, haswell, u8, u8, u32, nk_b128_vec_t, nk_dot_u8x16_state_haswell_t, nk_b128_vec_t,
215
+ nk_dot_u8x16_init_haswell, nk_load_b128_haswell_, nk_partial_load_b8x16_serial_,
216
+ nk_load_b128_haswell_, nk_partial_load_b8x16_serial_, nk_dot_u8x16_update_haswell,
217
+ nk_dot_u8x16_finalize_haswell, nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
218
+ /*depth_simd_dimensions=*/16, /*dimensions_per_value=*/1)
219
+
220
+ /* I4 GEMM: depth_simd_dimensions=32 — compensated (A+B sums precomputed)
221
+ * Note: dimensions_per_value=2 because 2 nibbles (i4 values) are packed per byte */
222
+ nk_define_cross_compensated_pack_size_(dots, i4, haswell, i4x2, i4x2,
223
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32,
224
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
225
+ nk_define_cross_compensated_pack_(dots, i4, haswell, i4x2, i4x2, nk_assign_from_to_,
226
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32, nk_dots_reduce_moments_i4_,
227
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
228
+ nk_define_cross_compensated_symmetric_(dots, i4, haswell, i4x2, i32,
229
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32, nk_b128_vec_t,
230
+ nk_dot_i4x32_state_haswell_t, nk_b128_vec_t, nk_dot_i4x32_init_haswell,
231
+ nk_load_b128_haswell_, nk_partial_load_b4x32_serial_,
232
+ nk_dot_i4x32_update_haswell, nk_dot_i4x32_finalize_haswell,
233
+ nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_, nk_load_b128_haswell_,
234
+ nk_partial_load_b32x4_haswell_, nk_sum_i4x32_state_serial_t,
235
+ nk_sum_i4x32_init_serial, nk_sum_i4x32_update_serial,
236
+ nk_sum_i4x32_finalize_serial,
237
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
238
+ nk_define_cross_compensated_packed_(dots, i4, haswell, i4x2, i4x2, i32,
239
+ /*sum_value_type=*/i32, /*norm_value_type=*/u32, nk_b128_vec_t,
240
+ nk_dot_i4x32_state_haswell_t, nk_b128_vec_t, nk_dot_i4x32_init_haswell,
241
+ nk_load_b128_haswell_, nk_partial_load_b4x32_serial_, nk_load_b128_haswell_,
242
+ nk_partial_load_b4x32_serial_, nk_dot_i4x32_update_haswell,
243
+ nk_dot_i4x32_finalize_haswell, nk_store_b128_haswell_,
244
+ nk_partial_store_b32x4_haswell_, nk_load_b128_haswell_,
245
+ nk_partial_load_b32x4_haswell_, nk_dots_reduce_sum_i4_,
246
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
247
+
248
+ /* U4 GEMM: depth_simd_dimensions=32 (32 nibbles = 16 bytes = 128-bit input)
249
+ * Note: dimensions_per_value=2 because 2 nibbles (u4 values) are packed per byte */
250
+ nk_define_cross_pack_size_(dots, u4, haswell, u4x2, u4x2, /*norm_value_type=*/u32, /*depth_simd_dimensions=*/32,
251
+ /*dimensions_per_value=*/2)
252
+ nk_define_cross_pack_(dots, u4, haswell, u4x2, u4x2, nk_assign_from_to_, /*norm_value_type=*/u32,
253
+ nk_dots_reduce_sumsq_u4_,
254
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
255
+ nk_define_cross_symmetric_(dots, u4, haswell, u4x2, u32, nk_b128_vec_t, nk_dot_u4x32_state_haswell_t, nk_b128_vec_t,
256
+ nk_dot_u4x32_init_haswell, nk_load_b128_haswell_, nk_partial_load_b4x32_serial_,
257
+ nk_dot_u4x32_update_haswell, nk_dot_u4x32_finalize_haswell, nk_store_b128_haswell_,
258
+ nk_partial_store_b32x4_haswell_,
259
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
260
+ nk_define_cross_packed_(dots, u4, haswell, u4x2, u4x2, u32, nk_b128_vec_t, nk_dot_u4x32_state_haswell_t, nk_b128_vec_t,
261
+ nk_dot_u4x32_init_haswell, nk_load_b128_haswell_, nk_partial_load_b4x32_serial_,
262
+ nk_load_b128_haswell_, nk_partial_load_b4x32_serial_, nk_dot_u4x32_update_haswell,
263
+ nk_dot_u4x32_finalize_haswell, nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
264
+ /*depth_simd_dimensions=*/32, /*dimensions_per_value=*/2)
265
+
266
+ /* U1 GEMM: depth_simd_dimensions=128 (128 bits = 16 bytes) */
267
+ nk_define_cross_pack_size_(dots, u1, haswell, u1x8, u1x8, /*norm_value_type=*/u32, /*depth_simd_dimensions=*/128,
268
+ /*dimensions_per_value=*/8)
269
+ nk_define_cross_pack_(dots, u1, haswell, u1x8, u1x8, nk_assign_from_to_, /*norm_value_type=*/u32,
270
+ nk_dots_reduce_sum_u1_,
271
+ /*depth_simd_dimensions=*/128, /*dimensions_per_value=*/8)
272
+ nk_define_cross_symmetric_(dots, u1, haswell, u1x8, u32, nk_b128_vec_t, nk_dot_u1x128_state_haswell_t, nk_b128_vec_t,
273
+ nk_dot_u1x128_init_haswell, nk_load_b128_haswell_, nk_partial_load_b1x128_serial_,
274
+ nk_dot_u1x128_update_haswell, nk_dot_u1x128_finalize_haswell, nk_store_b128_haswell_,
275
+ nk_partial_store_b32x4_haswell_,
276
+ /*depth_simd_dimensions=*/128, /*dimensions_per_value=*/8)
277
+ nk_define_cross_packed_(dots, u1, haswell, u1x8, u1x8, u32, nk_b128_vec_t, nk_dot_u1x128_state_haswell_t, nk_b128_vec_t,
278
+ nk_dot_u1x128_init_haswell, nk_load_b128_haswell_, nk_partial_load_b1x128_serial_,
279
+ nk_load_b128_haswell_, nk_partial_load_b1x128_serial_, nk_dot_u1x128_update_haswell,
280
+ nk_dot_u1x128_finalize_haswell, nk_store_b128_haswell_, nk_partial_store_b32x4_haswell_,
281
+ /*depth_simd_dimensions=*/128, /*dimensions_per_value=*/8)
282
+
283
+ #if defined(__clang__)
284
+ #pragma clang attribute pop
285
+ #elif defined(__GNUC__)
286
+ #pragma GCC pop_options
287
+ #endif
288
+
289
+ #if defined(__cplusplus)
290
+ } // extern "C"
291
+ #endif
292
+
293
+ #endif // NK_TARGET_HASWELL
294
+ #endif // NK_TARGET_X86_
295
+ #endif // NK_DOTS_HASWELL_H