whispercpp 1.3.1 → 1.3.3

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 (857) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +7 -3
  3. data/README.md +161 -43
  4. data/Rakefile +45 -13
  5. data/ext/.gitignore +4 -8
  6. data/ext/dependencies.rb +73 -0
  7. data/ext/extconf.rb +21 -198
  8. data/ext/options.rb +85 -0
  9. data/ext/ruby_whisper.c +177 -0
  10. data/ext/ruby_whisper.h +17 -2
  11. data/ext/ruby_whisper_context.c +672 -0
  12. data/ext/ruby_whisper_error.c +52 -0
  13. data/ext/ruby_whisper_model.c +232 -0
  14. data/ext/ruby_whisper_params.c +1303 -0
  15. data/ext/ruby_whisper_segment.c +220 -0
  16. data/ext/ruby_whisper_transcribe.cpp +93 -0
  17. data/ext/ruby_whisper_vad_params.c +288 -0
  18. data/ext/sources/CMakeGraphVizOptions.cmake +8 -0
  19. data/ext/sources/CMakeLists.txt +255 -0
  20. data/ext/sources/bindings/javascript/CMakeLists.txt +41 -0
  21. data/ext/sources/bindings/javascript/emscripten.cpp +93 -0
  22. data/ext/sources/bindings/javascript/libwhisper.worker.js +1 -0
  23. data/ext/sources/bindings/javascript/package-tmpl.json +26 -0
  24. data/ext/sources/bindings/javascript/package.json +26 -0
  25. data/ext/sources/bindings/javascript/whisper.js +19 -0
  26. data/ext/sources/build-xcframework.sh +547 -0
  27. data/ext/sources/cmake/DefaultTargetOptions.cmake +16 -0
  28. data/ext/sources/cmake/FindFFmpeg.cmake +163 -0
  29. data/ext/sources/cmake/build-info.cmake +60 -0
  30. data/ext/sources/cmake/git-vars.cmake +22 -0
  31. data/ext/sources/cmake/whisper-config.cmake.in +65 -0
  32. data/ext/sources/cmake/whisper.pc.in +10 -0
  33. data/ext/sources/examples/CMakeLists.txt +124 -0
  34. data/ext/sources/examples/addon.node/CMakeLists.txt +31 -0
  35. data/ext/sources/examples/addon.node/__test__/whisper.spec.js +133 -0
  36. data/ext/sources/examples/addon.node/addon.cpp +557 -0
  37. data/ext/sources/examples/addon.node/index.js +57 -0
  38. data/ext/sources/examples/addon.node/package.json +16 -0
  39. data/ext/sources/examples/addon.node/vad-example.js +132 -0
  40. data/ext/sources/examples/bench/CMakeLists.txt +8 -0
  41. data/ext/sources/examples/bench/bench.cpp +176 -0
  42. data/ext/sources/examples/bench.wasm/CMakeLists.txt +49 -0
  43. data/ext/sources/examples/bench.wasm/emscripten.cpp +87 -0
  44. data/ext/sources/examples/bench.wasm/index-tmpl.html +284 -0
  45. data/ext/sources/examples/cli/CMakeLists.txt +8 -0
  46. data/ext/sources/examples/cli/cli.cpp +1295 -0
  47. data/ext/sources/examples/coi-serviceworker.js +146 -0
  48. data/ext/sources/examples/command/CMakeLists.txt +10 -0
  49. data/ext/sources/examples/command/command.cpp +800 -0
  50. data/ext/sources/examples/command/commands.txt +9 -0
  51. data/ext/sources/examples/command.wasm/CMakeLists.txt +50 -0
  52. data/ext/sources/examples/command.wasm/emscripten.cpp +327 -0
  53. data/ext/sources/examples/command.wasm/index-tmpl.html +414 -0
  54. data/ext/sources/examples/common-ggml.cpp +238 -0
  55. data/ext/sources/examples/common-ggml.h +18 -0
  56. data/ext/sources/examples/common-sdl.cpp +227 -0
  57. data/ext/sources/examples/common-sdl.h +49 -0
  58. data/ext/sources/examples/common-whisper.cpp +175 -0
  59. data/ext/sources/examples/common-whisper.h +24 -0
  60. data/ext/sources/examples/common.cpp +675 -0
  61. data/ext/sources/examples/common.h +322 -0
  62. data/ext/sources/examples/deprecation-warning/CMakeLists.txt +6 -0
  63. data/ext/sources/examples/deprecation-warning/deprecation-warning.cpp +38 -0
  64. data/ext/sources/examples/ffmpeg-transcode.cpp +368 -0
  65. data/ext/sources/examples/generate-karaoke.sh +57 -0
  66. data/ext/sources/examples/grammar-parser.cpp +423 -0
  67. data/ext/sources/examples/grammar-parser.h +29 -0
  68. data/ext/sources/examples/helpers.js +191 -0
  69. data/ext/sources/examples/json.hpp +24596 -0
  70. data/ext/sources/examples/livestream.sh +112 -0
  71. data/ext/sources/examples/lsp/CMakeLists.txt +9 -0
  72. data/ext/sources/examples/lsp/lsp.cpp +469 -0
  73. data/ext/sources/examples/lsp/whisper.vim +362 -0
  74. data/ext/sources/examples/miniaudio.h +93468 -0
  75. data/ext/sources/examples/python/test_whisper_processor.py +7 -0
  76. data/ext/sources/examples/python/whisper_processor.py +54 -0
  77. data/ext/sources/examples/quantize/CMakeLists.txt +6 -0
  78. data/ext/sources/examples/quantize/quantize.cpp +226 -0
  79. data/ext/sources/examples/server/CMakeLists.txt +15 -0
  80. data/ext/sources/examples/server/bench.js +29 -0
  81. data/ext/sources/examples/server/httplib.h +10497 -0
  82. data/ext/sources/examples/server/server.cpp +1238 -0
  83. data/ext/sources/examples/server.py +115 -0
  84. data/ext/sources/examples/stb_vorbis.c +5584 -0
  85. data/ext/sources/examples/stream/CMakeLists.txt +10 -0
  86. data/ext/sources/examples/stream/stream.cpp +435 -0
  87. data/ext/sources/examples/stream.wasm/CMakeLists.txt +49 -0
  88. data/ext/sources/examples/stream.wasm/emscripten.cpp +216 -0
  89. data/ext/sources/examples/stream.wasm/index-tmpl.html +414 -0
  90. data/ext/sources/examples/sycl/CMakeLists.txt +9 -0
  91. data/ext/sources/examples/sycl/build.sh +22 -0
  92. data/ext/sources/examples/sycl/ls-sycl-device.cpp +11 -0
  93. data/ext/sources/examples/sycl/run-whisper.sh +17 -0
  94. data/ext/sources/examples/talk-llama/CMakeLists.txt +43 -0
  95. data/ext/sources/examples/talk-llama/eleven-labs.py +80 -0
  96. data/ext/sources/examples/talk-llama/llama-adapter.cpp +388 -0
  97. data/ext/sources/examples/talk-llama/llama-adapter.h +76 -0
  98. data/ext/sources/examples/talk-llama/llama-arch.cpp +1914 -0
  99. data/ext/sources/examples/talk-llama/llama-arch.h +464 -0
  100. data/ext/sources/examples/talk-llama/llama-batch.cpp +843 -0
  101. data/ext/sources/examples/talk-llama/llama-batch.h +147 -0
  102. data/ext/sources/examples/talk-llama/llama-chat.cpp +685 -0
  103. data/ext/sources/examples/talk-llama/llama-chat.h +59 -0
  104. data/ext/sources/examples/talk-llama/llama-context.cpp +2845 -0
  105. data/ext/sources/examples/talk-llama/llama-context.h +297 -0
  106. data/ext/sources/examples/talk-llama/llama-cparams.cpp +5 -0
  107. data/ext/sources/examples/talk-llama/llama-cparams.h +41 -0
  108. data/ext/sources/examples/talk-llama/llama-grammar.cpp +1229 -0
  109. data/ext/sources/examples/talk-llama/llama-grammar.h +173 -0
  110. data/ext/sources/examples/talk-llama/llama-graph.cpp +1693 -0
  111. data/ext/sources/examples/talk-llama/llama-graph.h +710 -0
  112. data/ext/sources/examples/talk-llama/llama-hparams.cpp +103 -0
  113. data/ext/sources/examples/talk-llama/llama-hparams.h +207 -0
  114. data/ext/sources/examples/talk-llama/llama-impl.cpp +167 -0
  115. data/ext/sources/examples/talk-llama/llama-impl.h +61 -0
  116. data/ext/sources/examples/talk-llama/llama-io.cpp +15 -0
  117. data/ext/sources/examples/talk-llama/llama-io.h +35 -0
  118. data/ext/sources/examples/talk-llama/llama-kv-cache-unified-iswa.cpp +279 -0
  119. data/ext/sources/examples/talk-llama/llama-kv-cache-unified-iswa.h +128 -0
  120. data/ext/sources/examples/talk-llama/llama-kv-cache-unified.cpp +1841 -0
  121. data/ext/sources/examples/talk-llama/llama-kv-cache-unified.h +303 -0
  122. data/ext/sources/examples/talk-llama/llama-kv-cache.h +44 -0
  123. data/ext/sources/examples/talk-llama/llama-kv-cells.h +439 -0
  124. data/ext/sources/examples/talk-llama/llama-memory-hybrid.cpp +246 -0
  125. data/ext/sources/examples/talk-llama/llama-memory-hybrid.h +138 -0
  126. data/ext/sources/examples/talk-llama/llama-memory-recurrent.cpp +1125 -0
  127. data/ext/sources/examples/talk-llama/llama-memory-recurrent.h +183 -0
  128. data/ext/sources/examples/talk-llama/llama-memory.cpp +59 -0
  129. data/ext/sources/examples/talk-llama/llama-memory.h +116 -0
  130. data/ext/sources/examples/talk-llama/llama-mmap.cpp +600 -0
  131. data/ext/sources/examples/talk-llama/llama-mmap.h +68 -0
  132. data/ext/sources/examples/talk-llama/llama-model-loader.cpp +1163 -0
  133. data/ext/sources/examples/talk-llama/llama-model-loader.h +169 -0
  134. data/ext/sources/examples/talk-llama/llama-model-saver.cpp +282 -0
  135. data/ext/sources/examples/talk-llama/llama-model-saver.h +37 -0
  136. data/ext/sources/examples/talk-llama/llama-model.cpp +15114 -0
  137. data/ext/sources/examples/talk-llama/llama-model.h +452 -0
  138. data/ext/sources/examples/talk-llama/llama-quant.cpp +1049 -0
  139. data/ext/sources/examples/talk-llama/llama-quant.h +1 -0
  140. data/ext/sources/examples/talk-llama/llama-sampling.cpp +2575 -0
  141. data/ext/sources/examples/talk-llama/llama-sampling.h +32 -0
  142. data/ext/sources/examples/talk-llama/llama-vocab.cpp +3377 -0
  143. data/ext/sources/examples/talk-llama/llama-vocab.h +132 -0
  144. data/ext/sources/examples/talk-llama/llama.cpp +358 -0
  145. data/ext/sources/examples/talk-llama/llama.h +1484 -0
  146. data/ext/sources/examples/talk-llama/prompts/talk-alpaca.txt +23 -0
  147. data/ext/sources/examples/talk-llama/speak +40 -0
  148. data/ext/sources/examples/talk-llama/speak.bat +1 -0
  149. data/ext/sources/examples/talk-llama/speak.ps1 +14 -0
  150. data/ext/sources/examples/talk-llama/talk-llama.cpp +810 -0
  151. data/ext/sources/examples/talk-llama/unicode-data.cpp +7034 -0
  152. data/ext/sources/examples/talk-llama/unicode-data.h +20 -0
  153. data/ext/sources/examples/talk-llama/unicode.cpp +854 -0
  154. data/ext/sources/examples/talk-llama/unicode.h +66 -0
  155. data/ext/sources/examples/vad-speech-segments/CMakeLists.txt +8 -0
  156. data/ext/sources/examples/vad-speech-segments/speech.cpp +149 -0
  157. data/ext/sources/examples/wchess/CMakeLists.txt +10 -0
  158. data/ext/sources/examples/wchess/libwchess/CMakeLists.txt +19 -0
  159. data/ext/sources/examples/wchess/libwchess/Chessboard.cpp +803 -0
  160. data/ext/sources/examples/wchess/libwchess/Chessboard.h +33 -0
  161. data/ext/sources/examples/wchess/libwchess/WChess.cpp +193 -0
  162. data/ext/sources/examples/wchess/libwchess/WChess.h +63 -0
  163. data/ext/sources/examples/wchess/libwchess/test-chessboard.cpp +117 -0
  164. data/ext/sources/examples/wchess/wchess.cmd/CMakeLists.txt +8 -0
  165. data/ext/sources/examples/wchess/wchess.cmd/wchess.cmd.cpp +251 -0
  166. data/ext/sources/examples/whisper.wasm/CMakeLists.txt +50 -0
  167. data/ext/sources/examples/whisper.wasm/emscripten.cpp +118 -0
  168. data/ext/sources/examples/whisper.wasm/index-tmpl.html +658 -0
  169. data/ext/sources/ggml/CMakeLists.txt +435 -0
  170. data/ext/sources/ggml/cmake/BuildTypes.cmake +54 -0
  171. data/ext/sources/ggml/cmake/GitVars.cmake +22 -0
  172. data/ext/sources/ggml/cmake/common.cmake +50 -0
  173. data/ext/sources/ggml/cmake/ggml-config.cmake.in +152 -0
  174. data/ext/{ggml → sources/ggml}/include/ggml-alloc.h +1 -1
  175. data/ext/{ggml → sources/ggml}/include/ggml-backend.h +10 -8
  176. data/ext/{ggml → sources/ggml}/include/ggml-cpp.h +2 -1
  177. data/ext/{ggml → sources/ggml}/include/ggml-cpu.h +11 -1
  178. data/ext/{ggml → sources/ggml}/include/ggml-metal.h +1 -1
  179. data/ext/{ggml → sources/ggml}/include/ggml-opt.h +49 -28
  180. data/ext/{ggml → sources/ggml}/include/ggml-rpc.h +6 -1
  181. data/ext/{ggml → sources/ggml}/include/ggml-vulkan.h +0 -2
  182. data/ext/{ggml → sources/ggml}/include/ggml.h +325 -269
  183. data/ext/sources/ggml/include/gguf.h +202 -0
  184. data/ext/sources/ggml/src/CMakeLists.txt +404 -0
  185. data/ext/{ggml → sources/ggml}/src/ggml-alloc.c +34 -29
  186. data/ext/sources/ggml/src/ggml-amx/CMakeLists.txt +107 -0
  187. data/ext/{ggml → sources/ggml}/src/ggml-backend-impl.h +1 -2
  188. data/ext/{ggml → sources/ggml}/src/ggml-backend-reg.cpp +92 -53
  189. data/ext/{ggml → sources/ggml}/src/ggml-backend.cpp +69 -34
  190. data/ext/sources/ggml/src/ggml-blas/CMakeLists.txt +87 -0
  191. data/ext/sources/ggml/src/ggml-cann/CMakeLists.txt +75 -0
  192. data/ext/sources/ggml/src/ggml-cann/Doxyfile +2579 -0
  193. data/ext/{ggml → sources/ggml}/src/ggml-cann/acl_tensor.cpp +10 -4
  194. data/ext/{ggml → sources/ggml}/src/ggml-cann/acl_tensor.h +5 -5
  195. data/ext/{ggml → sources/ggml}/src/ggml-cann/aclnn_ops.cpp +1272 -1506
  196. data/ext/sources/ggml/src/ggml-cann/aclnn_ops.h +1125 -0
  197. data/ext/{ggml → sources/ggml}/src/ggml-cann/common.h +140 -1
  198. data/ext/{ggml → sources/ggml}/src/ggml-cann/ggml-cann.cpp +588 -146
  199. data/ext/sources/ggml/src/ggml-cann/kernels/CMakeLists.txt +30 -0
  200. data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/dup.cpp +3 -5
  201. data/ext/{ggml → sources/ggml}/src/ggml-common.h +16 -8
  202. data/ext/sources/ggml/src/ggml-cpu/CMakeLists.txt +597 -0
  203. data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/amx.cpp +3 -2
  204. data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/mmq.cpp +11 -10
  205. data/ext/sources/ggml/src/ggml-cpu/arch/arm/cpu-feats.cpp +94 -0
  206. data/ext/sources/ggml/src/ggml-cpu/arch/arm/quants.c +4114 -0
  207. data/ext/sources/ggml/src/ggml-cpu/arch/arm/repack.cpp +2163 -0
  208. data/ext/sources/ggml/src/ggml-cpu/arch/loongarch/quants.c +2639 -0
  209. data/ext/sources/ggml/src/ggml-cpu/arch/powerpc/cpu-feats.cpp +82 -0
  210. data/ext/sources/ggml/src/ggml-cpu/arch/powerpc/quants.c +2732 -0
  211. data/ext/sources/ggml/src/ggml-cpu/arch/riscv/quants.c +2069 -0
  212. data/ext/sources/ggml/src/ggml-cpu/arch/riscv/repack.cpp +397 -0
  213. data/ext/sources/ggml/src/ggml-cpu/arch/s390/quants.c +1300 -0
  214. data/ext/sources/ggml/src/ggml-cpu/arch/wasm/quants.c +1481 -0
  215. data/ext/{ggml/src/ggml-cpu/cpu-feats-x86.cpp → sources/ggml/src/ggml-cpu/arch/x86/cpu-feats.cpp} +5 -1
  216. data/ext/sources/ggml/src/ggml-cpu/arch/x86/quants.c +4311 -0
  217. data/ext/sources/ggml/src/ggml-cpu/arch/x86/repack.cpp +3285 -0
  218. data/ext/sources/ggml/src/ggml-cpu/arch-fallback.h +184 -0
  219. data/ext/sources/ggml/src/ggml-cpu/binary-ops.cpp +158 -0
  220. data/ext/sources/ggml/src/ggml-cpu/binary-ops.h +16 -0
  221. data/ext/sources/ggml/src/ggml-cpu/cmake/FindSIMD.cmake +100 -0
  222. data/ext/sources/ggml/src/ggml-cpu/common.h +73 -0
  223. data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-impl.h +172 -41
  224. data/ext/sources/ggml/src/ggml-cpu/ggml-cpu.c +3551 -0
  225. data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu.cpp +78 -25
  226. data/ext/{ggml/src/ggml-cpu/ggml-cpu-hbm.cpp → sources/ggml/src/ggml-cpu/hbm.cpp} +1 -1
  227. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kernels.cpp +337 -0
  228. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kernels.h +95 -0
  229. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp +482 -0
  230. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kleidiai.h +17 -0
  231. data/ext/sources/ggml/src/ggml-cpu/llamafile/sgemm.cpp +3594 -0
  232. data/ext/sources/ggml/src/ggml-cpu/llamafile/sgemm.h +19 -0
  233. data/ext/sources/ggml/src/ggml-cpu/ops.cpp +9786 -0
  234. data/ext/sources/ggml/src/ggml-cpu/ops.h +118 -0
  235. data/ext/sources/ggml/src/ggml-cpu/quants.c +1158 -0
  236. data/ext/{ggml/src/ggml-cpu/ggml-cpu-quants.h → sources/ggml/src/ggml-cpu/quants.h} +26 -0
  237. data/ext/sources/ggml/src/ggml-cpu/repack.cpp +1571 -0
  238. data/ext/sources/ggml/src/ggml-cpu/repack.h +98 -0
  239. data/ext/sources/ggml/src/ggml-cpu/simd-mappings.h +1184 -0
  240. data/ext/{ggml/src/ggml-cpu/ggml-cpu-traits.cpp → sources/ggml/src/ggml-cpu/traits.cpp} +1 -1
  241. data/ext/sources/ggml/src/ggml-cpu/unary-ops.cpp +186 -0
  242. data/ext/sources/ggml/src/ggml-cpu/unary-ops.h +28 -0
  243. data/ext/sources/ggml/src/ggml-cpu/vec.cpp +345 -0
  244. data/ext/sources/ggml/src/ggml-cpu/vec.h +1027 -0
  245. data/ext/sources/ggml/src/ggml-cuda/CMakeLists.txt +184 -0
  246. data/ext/sources/ggml/src/ggml-cuda/acc.cu +61 -0
  247. data/ext/sources/ggml/src/ggml-cuda/acc.cuh +5 -0
  248. data/ext/sources/ggml/src/ggml-cuda/arange.cu +34 -0
  249. data/ext/sources/ggml/src/ggml-cuda/arange.cuh +5 -0
  250. data/ext/sources/ggml/src/ggml-cuda/argmax.cu +91 -0
  251. data/ext/sources/ggml/src/ggml-cuda/argmax.cuh +3 -0
  252. data/ext/sources/ggml/src/ggml-cuda/argsort.cu +104 -0
  253. data/ext/sources/ggml/src/ggml-cuda/argsort.cuh +3 -0
  254. data/ext/sources/ggml/src/ggml-cuda/binbcast.cu +363 -0
  255. data/ext/sources/ggml/src/ggml-cuda/binbcast.cuh +9 -0
  256. data/ext/sources/ggml/src/ggml-cuda/clamp.cu +45 -0
  257. data/ext/sources/ggml/src/ggml-cuda/clamp.cuh +5 -0
  258. data/ext/sources/ggml/src/ggml-cuda/common.cuh +851 -0
  259. data/ext/sources/ggml/src/ggml-cuda/concat.cu +221 -0
  260. data/ext/sources/ggml/src/ggml-cuda/concat.cuh +5 -0
  261. data/ext/sources/ggml/src/ggml-cuda/conv-transpose-1d.cu +89 -0
  262. data/ext/sources/ggml/src/ggml-cuda/conv-transpose-1d.cuh +5 -0
  263. data/ext/sources/ggml/src/ggml-cuda/conv2d-dw.cu +161 -0
  264. data/ext/sources/ggml/src/ggml-cuda/conv2d-dw.cuh +5 -0
  265. data/ext/sources/ggml/src/ggml-cuda/conv2d-transpose.cu +91 -0
  266. data/ext/sources/ggml/src/ggml-cuda/conv2d-transpose.cuh +4 -0
  267. data/ext/sources/ggml/src/ggml-cuda/convert.cu +752 -0
  268. data/ext/sources/ggml/src/ggml-cuda/convert.cuh +31 -0
  269. data/ext/sources/ggml/src/ggml-cuda/count-equal.cu +64 -0
  270. data/ext/sources/ggml/src/ggml-cuda/count-equal.cuh +5 -0
  271. data/ext/sources/ggml/src/ggml-cuda/cp-async.cuh +57 -0
  272. data/ext/sources/ggml/src/ggml-cuda/cpy.cu +705 -0
  273. data/ext/sources/ggml/src/ggml-cuda/cpy.cuh +11 -0
  274. data/ext/sources/ggml/src/ggml-cuda/cross-entropy-loss.cu +189 -0
  275. data/ext/sources/ggml/src/ggml-cuda/cross-entropy-loss.cuh +7 -0
  276. data/ext/sources/ggml/src/ggml-cuda/dequantize.cuh +103 -0
  277. data/ext/sources/ggml/src/ggml-cuda/diagmask.cu +40 -0
  278. data/ext/sources/ggml/src/ggml-cuda/diagmask.cuh +5 -0
  279. data/ext/sources/ggml/src/ggml-cuda/fattn-common.cuh +881 -0
  280. data/ext/sources/ggml/src/ggml-cuda/fattn-mma-f16.cuh +1474 -0
  281. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f16.cu +357 -0
  282. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f16.cuh +3 -0
  283. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f32.cu +365 -0
  284. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f32.cuh +3 -0
  285. data/ext/sources/ggml/src/ggml-cuda/fattn-vec-f16.cuh +482 -0
  286. data/ext/sources/ggml/src/ggml-cuda/fattn-vec-f32.cuh +472 -0
  287. data/ext/sources/ggml/src/ggml-cuda/fattn-wmma-f16.cu +638 -0
  288. data/ext/sources/ggml/src/ggml-cuda/fattn-wmma-f16.cuh +3 -0
  289. data/ext/sources/ggml/src/ggml-cuda/fattn.cu +346 -0
  290. data/ext/sources/ggml/src/ggml-cuda/fattn.cuh +3 -0
  291. data/ext/sources/ggml/src/ggml-cuda/getrows.cu +275 -0
  292. data/ext/sources/ggml/src/ggml-cuda/getrows.cuh +15 -0
  293. data/ext/sources/ggml/src/ggml-cuda/ggml-cuda.cu +3647 -0
  294. data/ext/sources/ggml/src/ggml-cuda/gla.cu +93 -0
  295. data/ext/sources/ggml/src/ggml-cuda/gla.cuh +3 -0
  296. data/ext/sources/ggml/src/ggml-cuda/im2col.cu +103 -0
  297. data/ext/sources/ggml/src/ggml-cuda/im2col.cuh +5 -0
  298. data/ext/sources/ggml/src/ggml-cuda/mean.cu +19 -0
  299. data/ext/sources/ggml/src/ggml-cuda/mean.cuh +3 -0
  300. data/ext/sources/ggml/src/ggml-cuda/mma.cuh +396 -0
  301. data/ext/sources/ggml/src/ggml-cuda/mmq.cu +324 -0
  302. data/ext/sources/ggml/src/ggml-cuda/mmq.cuh +3217 -0
  303. data/ext/sources/ggml/src/ggml-cuda/mmv.cu +506 -0
  304. data/ext/sources/ggml/src/ggml-cuda/mmv.cuh +11 -0
  305. data/ext/sources/ggml/src/ggml-cuda/mmvq.cu +595 -0
  306. data/ext/sources/ggml/src/ggml-cuda/mmvq.cuh +12 -0
  307. data/ext/sources/ggml/src/ggml-cuda/norm.cu +458 -0
  308. data/ext/sources/ggml/src/ggml-cuda/norm.cuh +11 -0
  309. data/ext/sources/ggml/src/ggml-cuda/opt-step-adamw.cu +78 -0
  310. data/ext/sources/ggml/src/ggml-cuda/opt-step-adamw.cuh +5 -0
  311. data/ext/sources/ggml/src/ggml-cuda/out-prod.cu +68 -0
  312. data/ext/sources/ggml/src/ggml-cuda/out-prod.cuh +3 -0
  313. data/ext/sources/ggml/src/ggml-cuda/pad.cu +49 -0
  314. data/ext/sources/ggml/src/ggml-cuda/pad.cuh +5 -0
  315. data/ext/sources/ggml/src/ggml-cuda/pool2d.cu +94 -0
  316. data/ext/sources/ggml/src/ggml-cuda/pool2d.cuh +5 -0
  317. data/ext/sources/ggml/src/ggml-cuda/quantize.cu +190 -0
  318. data/ext/sources/ggml/src/ggml-cuda/quantize.cuh +27 -0
  319. data/ext/sources/ggml/src/ggml-cuda/rope.cu +456 -0
  320. data/ext/sources/ggml/src/ggml-cuda/rope.cuh +7 -0
  321. data/ext/sources/ggml/src/ggml-cuda/scale.cu +31 -0
  322. data/ext/sources/ggml/src/ggml-cuda/scale.cuh +5 -0
  323. data/ext/sources/ggml/src/ggml-cuda/softmax.cu +283 -0
  324. data/ext/sources/ggml/src/ggml-cuda/softmax.cuh +7 -0
  325. data/ext/sources/ggml/src/ggml-cuda/ssm-conv.cu +148 -0
  326. data/ext/sources/ggml/src/ggml-cuda/ssm-conv.cuh +3 -0
  327. data/ext/sources/ggml/src/ggml-cuda/ssm-scan.cu +155 -0
  328. data/ext/sources/ggml/src/ggml-cuda/ssm-scan.cuh +3 -0
  329. data/ext/sources/ggml/src/ggml-cuda/sum.cu +45 -0
  330. data/ext/sources/ggml/src/ggml-cuda/sum.cuh +5 -0
  331. data/ext/sources/ggml/src/ggml-cuda/sumrows.cu +26 -0
  332. data/ext/sources/ggml/src/ggml-cuda/sumrows.cuh +4 -0
  333. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_1-ncols2_16.cu +5 -0
  334. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_1-ncols2_8.cu +10 -0
  335. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_1.cu +10 -0
  336. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_2.cu +10 -0
  337. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_4.cu +10 -0
  338. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_16.cu +5 -0
  339. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_4.cu +10 -0
  340. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_8.cu +10 -0
  341. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_32-ncols2_1.cu +10 -0
  342. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_32-ncols2_2.cu +10 -0
  343. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_16.cu +5 -0
  344. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_2.cu +10 -0
  345. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_4.cu +10 -0
  346. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_8.cu +10 -0
  347. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_64-ncols2_1.cu +10 -0
  348. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_1.cu +10 -0
  349. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_2.cu +10 -0
  350. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_4.cu +10 -0
  351. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_8.cu +10 -0
  352. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-f16.cu +5 -0
  353. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q4_0.cu +5 -0
  354. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q4_1.cu +5 -0
  355. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q5_0.cu +5 -0
  356. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q5_1.cu +5 -0
  357. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q8_0.cu +5 -0
  358. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-f16.cu +5 -0
  359. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q4_0.cu +5 -0
  360. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q4_1.cu +5 -0
  361. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q5_0.cu +5 -0
  362. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q5_1.cu +5 -0
  363. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q8_0.cu +5 -0
  364. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-f16.cu +5 -0
  365. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q4_0.cu +5 -0
  366. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q4_1.cu +5 -0
  367. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q5_0.cu +5 -0
  368. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q5_1.cu +5 -0
  369. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q8_0.cu +5 -0
  370. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-f16.cu +5 -0
  371. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q4_0.cu +5 -0
  372. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q4_1.cu +5 -0
  373. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q5_0.cu +5 -0
  374. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q5_1.cu +5 -0
  375. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q8_0.cu +5 -0
  376. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-f16.cu +5 -0
  377. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q4_0.cu +5 -0
  378. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q4_1.cu +5 -0
  379. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q5_0.cu +5 -0
  380. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q5_1.cu +5 -0
  381. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q8_0.cu +5 -0
  382. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-f16.cu +5 -0
  383. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q4_0.cu +5 -0
  384. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q4_1.cu +5 -0
  385. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q5_0.cu +5 -0
  386. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q5_1.cu +5 -0
  387. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q8_0.cu +5 -0
  388. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs256-f16-f16.cu +5 -0
  389. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-f16.cu +5 -0
  390. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q4_0.cu +5 -0
  391. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q4_1.cu +5 -0
  392. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q5_0.cu +5 -0
  393. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q5_1.cu +5 -0
  394. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q8_0.cu +5 -0
  395. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-f16.cu +5 -0
  396. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q4_0.cu +5 -0
  397. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q4_1.cu +5 -0
  398. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q5_0.cu +5 -0
  399. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q5_1.cu +5 -0
  400. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q8_0.cu +5 -0
  401. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-f16.cu +5 -0
  402. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q4_0.cu +5 -0
  403. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q4_1.cu +5 -0
  404. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q5_0.cu +5 -0
  405. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q5_1.cu +5 -0
  406. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q8_0.cu +5 -0
  407. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-f16.cu +5 -0
  408. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q4_0.cu +5 -0
  409. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q4_1.cu +5 -0
  410. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q5_0.cu +5 -0
  411. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q5_1.cu +5 -0
  412. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q8_0.cu +5 -0
  413. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-f16.cu +5 -0
  414. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q4_0.cu +5 -0
  415. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q4_1.cu +5 -0
  416. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q5_0.cu +5 -0
  417. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q5_1.cu +5 -0
  418. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q8_0.cu +5 -0
  419. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-f16.cu +5 -0
  420. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q4_0.cu +5 -0
  421. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q4_1.cu +5 -0
  422. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q5_0.cu +5 -0
  423. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q5_1.cu +5 -0
  424. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q8_0.cu +5 -0
  425. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-f16.cu +5 -0
  426. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q4_0.cu +5 -0
  427. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q4_1.cu +5 -0
  428. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q5_0.cu +5 -0
  429. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q5_1.cu +5 -0
  430. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q8_0.cu +5 -0
  431. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs256-f16-f16.cu +5 -0
  432. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-f16.cu +5 -0
  433. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q4_0.cu +5 -0
  434. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q4_1.cu +5 -0
  435. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q5_0.cu +5 -0
  436. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q5_1.cu +5 -0
  437. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q8_0.cu +5 -0
  438. data/ext/sources/ggml/src/ggml-cuda/template-instances/generate_cu_files.py +78 -0
  439. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq1_s.cu +5 -0
  440. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_s.cu +5 -0
  441. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_xs.cu +5 -0
  442. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_xxs.cu +5 -0
  443. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq3_s.cu +5 -0
  444. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq3_xxs.cu +5 -0
  445. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq4_nl.cu +5 -0
  446. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq4_xs.cu +5 -0
  447. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q2_k.cu +5 -0
  448. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q3_k.cu +5 -0
  449. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_0.cu +5 -0
  450. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_1.cu +5 -0
  451. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_k.cu +5 -0
  452. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_0.cu +5 -0
  453. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_1.cu +5 -0
  454. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_k.cu +5 -0
  455. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q6_k.cu +5 -0
  456. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q8_0.cu +5 -0
  457. data/ext/sources/ggml/src/ggml-cuda/tsembd.cu +47 -0
  458. data/ext/sources/ggml/src/ggml-cuda/tsembd.cuh +5 -0
  459. data/ext/sources/ggml/src/ggml-cuda/unary.cu +378 -0
  460. data/ext/sources/ggml/src/ggml-cuda/unary.cuh +66 -0
  461. data/ext/sources/ggml/src/ggml-cuda/upscale.cu +51 -0
  462. data/ext/sources/ggml/src/ggml-cuda/upscale.cuh +5 -0
  463. data/ext/sources/ggml/src/ggml-cuda/vecdotq.cuh +1135 -0
  464. data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/cuda.h +1 -0
  465. data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/hip.h +57 -0
  466. data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/musa.h +7 -1
  467. data/ext/sources/ggml/src/ggml-cuda/wkv.cu +199 -0
  468. data/ext/sources/ggml/src/ggml-cuda/wkv.cuh +7 -0
  469. data/ext/sources/ggml/src/ggml-hip/CMakeLists.txt +135 -0
  470. data/ext/{ggml → sources/ggml}/src/ggml-impl.h +147 -158
  471. data/ext/sources/ggml/src/ggml-kompute/CMakeLists.txt +166 -0
  472. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/common.comp +112 -0
  473. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_add.comp +58 -0
  474. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_addrow.comp +25 -0
  475. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f16_f16.comp +52 -0
  476. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f16_f32.comp +52 -0
  477. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f32_f16.comp +52 -0
  478. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f32_f32.comp +52 -0
  479. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_diagmask.comp +30 -0
  480. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_gelu.comp +22 -0
  481. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows.comp +17 -0
  482. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_f16.comp +31 -0
  483. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_f32.comp +31 -0
  484. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q4_0.comp +38 -0
  485. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q4_1.comp +39 -0
  486. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q6_k.comp +44 -0
  487. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul.comp +52 -0
  488. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_f16.comp +69 -0
  489. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_mat_f32.comp +51 -0
  490. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_0.comp +33 -0
  491. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_1.comp +35 -0
  492. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_k.comp +140 -0
  493. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q6_k.comp +106 -0
  494. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q8_0.comp +73 -0
  495. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mv_q_n.comp +52 -0
  496. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mv_q_n_pre.comp +28 -0
  497. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_norm.comp +84 -0
  498. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_relu.comp +21 -0
  499. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rmsnorm.comp +53 -0
  500. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_neox_f16.comp +52 -0
  501. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_neox_f32.comp +52 -0
  502. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_norm_f16.comp +52 -0
  503. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_norm_f32.comp +52 -0
  504. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_scale.comp +19 -0
  505. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_scale_8.comp +23 -0
  506. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_silu.comp +22 -0
  507. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_softmax.comp +72 -0
  508. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/rope_common.comp +71 -0
  509. data/ext/sources/ggml/src/ggml-metal/CMakeLists.txt +121 -0
  510. data/ext/sources/ggml/src/ggml-metal/ggml-metal-impl.h +649 -0
  511. data/ext/{ggml → sources/ggml}/src/ggml-metal/ggml-metal.m +2504 -1108
  512. data/ext/{ggml → sources/ggml}/src/ggml-metal/ggml-metal.metal +2102 -1463
  513. data/ext/sources/ggml/src/ggml-musa/CMakeLists.txt +113 -0
  514. data/ext/sources/ggml/src/ggml-musa/mudnn.cu +112 -0
  515. data/ext/sources/ggml/src/ggml-musa/mudnn.cuh +12 -0
  516. data/ext/sources/ggml/src/ggml-opencl/CMakeLists.txt +110 -0
  517. data/ext/sources/ggml/src/ggml-opencl/ggml-opencl.cpp +6494 -0
  518. data/ext/sources/ggml/src/ggml-opencl/kernels/add.cl +83 -0
  519. data/ext/sources/ggml/src/ggml-opencl/kernels/argsort.cl +86 -0
  520. data/ext/sources/ggml/src/ggml-opencl/kernels/clamp.cl +20 -0
  521. data/ext/sources/ggml/src/ggml-opencl/kernels/concat.cl +109 -0
  522. data/ext/sources/ggml/src/ggml-opencl/kernels/cpy.cl +184 -0
  523. data/ext/sources/ggml/src/ggml-opencl/kernels/cvt.cl +118 -0
  524. data/ext/sources/ggml/src/ggml-opencl/kernels/diag_mask_inf.cl +58 -0
  525. data/ext/sources/ggml/src/ggml-opencl/kernels/div.cl +72 -0
  526. data/ext/sources/ggml/src/ggml-opencl/kernels/embed_kernel.py +26 -0
  527. data/ext/sources/ggml/src/ggml-opencl/kernels/gelu.cl +62 -0
  528. data/ext/sources/ggml/src/ggml-opencl/kernels/gemv_noshuffle.cl +268 -0
  529. data/ext/sources/ggml/src/ggml-opencl/kernels/gemv_noshuffle_general.cl +274 -0
  530. data/ext/sources/ggml/src/ggml-opencl/kernels/get_rows.cl +163 -0
  531. data/ext/sources/ggml/src/ggml-opencl/kernels/glu.cl +201 -0
  532. data/ext/sources/ggml/src/ggml-opencl/kernels/group_norm.cl +72 -0
  533. data/ext/sources/ggml/src/ggml-opencl/kernels/im2col_f16.cl +57 -0
  534. data/ext/sources/ggml/src/ggml-opencl/kernels/im2col_f32.cl +57 -0
  535. data/ext/sources/ggml/src/ggml-opencl/kernels/mul.cl +79 -0
  536. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mat_Ab_Bi_8x4.cl +139 -0
  537. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f16.cl +118 -0
  538. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32.cl +118 -0
  539. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32_1row.cl +94 -0
  540. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32_l4.cl +84 -0
  541. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f32_f32.cl +118 -0
  542. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_id_q4_0_f32_8x_flat.cl +283 -0
  543. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32.cl +192 -0
  544. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_1d_16x_flat.cl +307 -0
  545. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_1d_8x_flat.cl +265 -0
  546. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_8x_flat.cl +272 -0
  547. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_v.cl +254 -0
  548. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q6_k.cl +190 -0
  549. data/ext/sources/ggml/src/ggml-opencl/kernels/norm.cl +81 -0
  550. data/ext/sources/ggml/src/ggml-opencl/kernels/pad.cl +30 -0
  551. data/ext/sources/ggml/src/ggml-opencl/kernels/relu.cl +16 -0
  552. data/ext/sources/ggml/src/ggml-opencl/kernels/repeat.cl +39 -0
  553. data/ext/sources/ggml/src/ggml-opencl/kernels/rms_norm.cl +96 -0
  554. data/ext/sources/ggml/src/ggml-opencl/kernels/rope.cl +721 -0
  555. data/ext/sources/ggml/src/ggml-opencl/kernels/scale.cl +16 -0
  556. data/ext/sources/ggml/src/ggml-opencl/kernels/sigmoid.cl +29 -0
  557. data/ext/sources/ggml/src/ggml-opencl/kernels/silu.cl +30 -0
  558. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_4_f16.cl +87 -0
  559. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_4_f32.cl +87 -0
  560. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_f16.cl +86 -0
  561. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_f32.cl +86 -0
  562. data/ext/sources/ggml/src/ggml-opencl/kernels/sub.cl +72 -0
  563. data/ext/sources/ggml/src/ggml-opencl/kernels/sum_rows.cl +39 -0
  564. data/ext/sources/ggml/src/ggml-opencl/kernels/tanh.cl +63 -0
  565. data/ext/sources/ggml/src/ggml-opencl/kernels/transpose.cl +84 -0
  566. data/ext/sources/ggml/src/ggml-opencl/kernels/tsembd.cl +48 -0
  567. data/ext/sources/ggml/src/ggml-opencl/kernels/upscale.cl +121 -0
  568. data/ext/{ggml → sources/ggml}/src/ggml-opt.cpp +373 -190
  569. data/ext/{ggml → sources/ggml}/src/ggml-quants.c +120 -128
  570. data/ext/sources/ggml/src/ggml-rpc/CMakeLists.txt +9 -0
  571. data/ext/{ggml → sources/ggml}/src/ggml-rpc/ggml-rpc.cpp +494 -84
  572. data/ext/sources/ggml/src/ggml-sycl/CMakeLists.txt +189 -0
  573. data/ext/sources/ggml/src/ggml-sycl/backend.hpp +37 -0
  574. data/ext/sources/ggml/src/ggml-sycl/binbcast.cpp +344 -0
  575. data/ext/sources/ggml/src/ggml-sycl/binbcast.hpp +39 -0
  576. data/ext/{ggml → sources/ggml}/src/ggml-sycl/common.cpp +20 -32
  577. data/ext/sources/ggml/src/ggml-sycl/common.hpp +561 -0
  578. data/ext/{ggml → sources/ggml}/src/ggml-sycl/concat.cpp +56 -70
  579. data/ext/sources/ggml/src/ggml-sycl/concat.hpp +20 -0
  580. data/ext/{ggml → sources/ggml}/src/ggml-sycl/conv.cpp +8 -12
  581. data/ext/sources/ggml/src/ggml-sycl/conv.hpp +20 -0
  582. data/ext/sources/ggml/src/ggml-sycl/convert.cpp +575 -0
  583. data/ext/sources/ggml/src/ggml-sycl/convert.hpp +34 -0
  584. data/ext/sources/ggml/src/ggml-sycl/cpy.cpp +839 -0
  585. data/ext/sources/ggml/src/ggml-sycl/cpy.hpp +11 -0
  586. data/ext/sources/ggml/src/ggml-sycl/dequantize.hpp +823 -0
  587. data/ext/{ggml → sources/ggml}/src/ggml-sycl/dmmv.cpp +188 -67
  588. data/ext/sources/ggml/src/ggml-sycl/dmmv.hpp +27 -0
  589. data/ext/sources/ggml/src/ggml-sycl/dpct/helper.hpp +2987 -0
  590. data/ext/sources/ggml/src/ggml-sycl/element_wise.cpp +1120 -0
  591. data/ext/sources/ggml/src/ggml-sycl/element_wise.hpp +84 -0
  592. data/ext/sources/ggml/src/ggml-sycl/gemm.hpp +102 -0
  593. data/ext/sources/ggml/src/ggml-sycl/getrows.cpp +212 -0
  594. data/ext/sources/ggml/src/ggml-sycl/getrows.hpp +20 -0
  595. data/ext/{ggml → sources/ggml}/src/ggml-sycl/ggml-sycl.cpp +1197 -1295
  596. data/ext/sources/ggml/src/ggml-sycl/gla.cpp +106 -0
  597. data/ext/sources/ggml/src/ggml-sycl/gla.hpp +8 -0
  598. data/ext/sources/ggml/src/ggml-sycl/im2col.cpp +136 -0
  599. data/ext/sources/ggml/src/ggml-sycl/im2col.hpp +21 -0
  600. data/ext/{ggml → sources/ggml}/src/ggml-sycl/mmq.cpp +60 -81
  601. data/ext/sources/ggml/src/ggml-sycl/mmq.hpp +33 -0
  602. data/ext/sources/ggml/src/ggml-sycl/mmvq.cpp +1065 -0
  603. data/ext/sources/ggml/src/ggml-sycl/mmvq.hpp +27 -0
  604. data/ext/sources/ggml/src/ggml-sycl/norm.cpp +482 -0
  605. data/ext/sources/ggml/src/ggml-sycl/norm.hpp +26 -0
  606. data/ext/{ggml → sources/ggml}/src/ggml-sycl/outprod.cpp +8 -17
  607. data/ext/sources/ggml/src/ggml-sycl/outprod.hpp +10 -0
  608. data/ext/sources/ggml/src/ggml-sycl/presets.hpp +74 -0
  609. data/ext/sources/ggml/src/ggml-sycl/quants.hpp +111 -0
  610. data/ext/sources/ggml/src/ggml-sycl/rope.cpp +472 -0
  611. data/ext/sources/ggml/src/ggml-sycl/rope.hpp +20 -0
  612. data/ext/{ggml → sources/ggml}/src/ggml-sycl/softmax.cpp +38 -28
  613. data/ext/sources/ggml/src/ggml-sycl/softmax.hpp +20 -0
  614. data/ext/sources/ggml/src/ggml-sycl/sycl_hw.cpp +15 -0
  615. data/ext/sources/ggml/src/ggml-sycl/sycl_hw.hpp +26 -0
  616. data/ext/{ggml → sources/ggml}/src/ggml-sycl/tsembd.cpp +6 -11
  617. data/ext/sources/ggml/src/ggml-sycl/tsembd.hpp +20 -0
  618. data/ext/sources/ggml/src/ggml-sycl/vecdotq.hpp +1307 -0
  619. data/ext/sources/ggml/src/ggml-sycl/wkv.cpp +289 -0
  620. data/ext/sources/ggml/src/ggml-sycl/wkv.hpp +10 -0
  621. data/ext/sources/ggml/src/ggml-vulkan/CMakeLists.txt +200 -0
  622. data/ext/sources/ggml/src/ggml-vulkan/cmake/host-toolchain.cmake.in +15 -0
  623. data/ext/{ggml → sources/ggml}/src/ggml-vulkan/ggml-vulkan.cpp +3822 -1335
  624. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/CMakeLists.txt +31 -0
  625. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/acc.comp +29 -0
  626. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/add.comp +29 -0
  627. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/argmax.comp +51 -0
  628. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/argsort.comp +69 -0
  629. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/clamp.comp +17 -0
  630. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/concat.comp +41 -0
  631. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/contig_copy.comp +49 -0
  632. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/conv2d_dw.comp +105 -0
  633. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/conv_transpose_1d.comp +98 -0
  634. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy.comp +23 -0
  635. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy_from_quant.comp +51 -0
  636. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy_to_quant.comp +242 -0
  637. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/cos.comp +17 -0
  638. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/count_equal.comp +31 -0
  639. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_f32.comp +20 -0
  640. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_funcs.comp +462 -0
  641. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_funcs_cm2.comp +699 -0
  642. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_head.comp +13 -0
  643. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq1_m.comp +42 -0
  644. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq1_s.comp +35 -0
  645. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_s.comp +44 -0
  646. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_xs.comp +43 -0
  647. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_xxs.comp +48 -0
  648. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq3_s.comp +39 -0
  649. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq3_xxs.comp +49 -0
  650. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq4_nl.comp +32 -0
  651. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq4_xs.comp +34 -0
  652. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q2_k.comp +34 -0
  653. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q3_k.comp +42 -0
  654. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_0.comp +30 -0
  655. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_1.comp +32 -0
  656. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_k.comp +68 -0
  657. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_0.comp +34 -0
  658. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_1.comp +35 -0
  659. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_k.comp +70 -0
  660. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q6_k.comp +33 -0
  661. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q8_0.comp +31 -0
  662. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/diag_mask_inf.comp +34 -0
  663. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/div.comp +27 -0
  664. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn.comp +337 -0
  665. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_base.comp +162 -0
  666. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_cm1.comp +360 -0
  667. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_cm2.comp +267 -0
  668. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_split_k_reduce.comp +59 -0
  669. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/geglu.comp +13 -0
  670. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/gelu.comp +25 -0
  671. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/gelu_quick.comp +23 -0
  672. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_binary_head.comp +64 -0
  673. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_head.comp +9 -0
  674. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_unary_head.comp +76 -0
  675. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/get_rows.comp +33 -0
  676. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/get_rows_quant.comp +41 -0
  677. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/glu_head.comp +15 -0
  678. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/glu_main.comp +29 -0
  679. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/group_norm.comp +66 -0
  680. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/im2col.comp +100 -0
  681. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/l2_norm.comp +41 -0
  682. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/leaky_relu.comp +22 -0
  683. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul.comp +27 -0
  684. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_split_k_reduce.comp +48 -0
  685. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec.comp +169 -0
  686. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_base.comp +118 -0
  687. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq1_m.comp +82 -0
  688. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq1_s.comp +79 -0
  689. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_s.comp +90 -0
  690. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_xs.comp +87 -0
  691. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_xxs.comp +87 -0
  692. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq3_s.comp +90 -0
  693. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq3_xxs.comp +88 -0
  694. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_nc.comp +118 -0
  695. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_p021.comp +154 -0
  696. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q2_k.comp +130 -0
  697. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q3_k.comp +132 -0
  698. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q4_k.comp +136 -0
  699. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q5_k.comp +167 -0
  700. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q6_k.comp +130 -0
  701. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mm.comp +868 -0
  702. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mm_cm2.comp +441 -0
  703. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq.comp +442 -0
  704. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq_funcs.comp +99 -0
  705. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/norm.comp +44 -0
  706. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/opt_step_adamw.comp +42 -0
  707. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/pad.comp +28 -0
  708. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/pool2d.comp +74 -0
  709. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/quantize_q8_1.comp +77 -0
  710. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/reglu.comp +9 -0
  711. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp +21 -0
  712. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/repeat.comp +26 -0
  713. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/repeat_back.comp +37 -0
  714. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rms_norm.comp +61 -0
  715. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rms_norm_back.comp +55 -0
  716. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_head.comp +58 -0
  717. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_multi.comp +60 -0
  718. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_neox.comp +43 -0
  719. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_norm.comp +43 -0
  720. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_vision.comp +47 -0
  721. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/scale.comp +24 -0
  722. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sigmoid.comp +20 -0
  723. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/silu.comp +22 -0
  724. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/silu_back.comp +26 -0
  725. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sin.comp +17 -0
  726. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp +173 -0
  727. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/soft_max_back.comp +50 -0
  728. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/square.comp +17 -0
  729. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sub.comp +29 -0
  730. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sum_rows.comp +37 -0
  731. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/swiglu.comp +9 -0
  732. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp +20 -0
  733. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_bfloat16_support.comp +7 -0
  734. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_coopmat2_support.comp +7 -0
  735. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_coopmat_support.comp +7 -0
  736. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_integer_dot_support.comp +7 -0
  737. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/timestep_embedding.comp +41 -0
  738. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/types.comp +1373 -0
  739. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/upscale.comp +36 -0
  740. data/ext/{ggml → sources/ggml}/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp +203 -36
  741. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/wkv6.comp +87 -0
  742. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/wkv7.comp +91 -0
  743. data/ext/{ggml → sources/ggml}/src/ggml.c +918 -1782
  744. data/ext/sources/ggml/src/ggml.cpp +26 -0
  745. data/ext/sources/ggml/src/gguf.cpp +1351 -0
  746. data/ext/{include → sources/include}/whisper.h +70 -2
  747. data/ext/sources/src/CMakeLists.txt +145 -0
  748. data/ext/sources/src/coreml/whisper-compat.h +10 -0
  749. data/ext/sources/src/coreml/whisper-compat.m +35 -0
  750. data/ext/{src → sources/src}/coreml/whisper-decoder-impl.h +27 -15
  751. data/ext/{src → sources/src}/coreml/whisper-decoder-impl.m +36 -10
  752. data/ext/{src → sources/src}/coreml/whisper-encoder-impl.h +21 -9
  753. data/ext/{src → sources/src}/coreml/whisper-encoder-impl.m +29 -3
  754. data/ext/sources/src/coreml/whisper-encoder.mm +73 -0
  755. data/ext/sources/src/whisper-arch.h +197 -0
  756. data/ext/{src → sources/src}/whisper.cpp +1966 -386
  757. data/ext/sources/tests/CMakeLists.txt +105 -0
  758. data/ext/sources/tests/earnings21/eval.mk +58 -0
  759. data/ext/sources/tests/earnings21/eval.py +68 -0
  760. data/ext/sources/tests/earnings21/normalizers/__init__.py +2 -0
  761. data/ext/sources/tests/earnings21/normalizers/basic.py +80 -0
  762. data/ext/sources/tests/earnings21/normalizers/english.json +1741 -0
  763. data/ext/sources/tests/earnings21/normalizers/english.py +550 -0
  764. data/ext/sources/tests/earnings21/requirements.txt +6 -0
  765. data/ext/sources/tests/en-0-ref.txt +1 -0
  766. data/ext/sources/tests/en-1-ref.txt +1 -0
  767. data/ext/sources/tests/en-2-ref.txt +1 -0
  768. data/ext/sources/tests/es-0-ref.txt +1 -0
  769. data/ext/sources/tests/librispeech/eval.mk +39 -0
  770. data/ext/sources/tests/librispeech/eval.py +47 -0
  771. data/ext/sources/tests/librispeech/normalizers/__init__.py +2 -0
  772. data/ext/sources/tests/librispeech/normalizers/basic.py +80 -0
  773. data/ext/sources/tests/librispeech/normalizers/english.json +1741 -0
  774. data/ext/sources/tests/librispeech/normalizers/english.py +550 -0
  775. data/ext/sources/tests/librispeech/requirements.txt +6 -0
  776. data/ext/sources/tests/run-tests.sh +130 -0
  777. data/ext/sources/tests/test-c.c +3 -0
  778. data/ext/sources/tests/test-vad-full.cpp +54 -0
  779. data/ext/sources/tests/test-vad.cpp +83 -0
  780. data/ext/sources/tests/test-whisper.js +58 -0
  781. data/extsources.rb +39 -5
  782. data/lib/whisper/context.rb +15 -0
  783. data/lib/whisper/model/uri.rb +202 -126
  784. data/lib/whisper/segment.rb +58 -0
  785. data/sig/whisper.rbs +510 -0
  786. data/test/helper.rb +24 -0
  787. data/{tests → test}/test_callback.rb +45 -3
  788. data/{tests → test}/test_error.rb +2 -2
  789. data/{tests → test}/test_model.rb +47 -0
  790. data/test/test_package.rb +51 -0
  791. data/test/test_params.rb +297 -0
  792. data/test/test_segment.rb +146 -0
  793. data/test/test_vad.rb +19 -0
  794. data/test/test_vad_params.rb +103 -0
  795. data/{tests → test}/test_whisper.rb +106 -36
  796. data/whispercpp.gemspec +5 -5
  797. metadata +837 -134
  798. data/ext/cpu.mk +0 -9
  799. data/ext/examples/dr_wav.h +0 -8815
  800. data/ext/ggml/src/ggml-cann/aclnn_ops.h +0 -592
  801. data/ext/ggml/src/ggml-cpu/ggml-cpu-aarch64.cpp +0 -4262
  802. data/ext/ggml/src/ggml-cpu/ggml-cpu-aarch64.h +0 -8
  803. data/ext/ggml/src/ggml-cpu/ggml-cpu-quants.c +0 -10835
  804. data/ext/ggml/src/ggml-cpu/ggml-cpu.c +0 -14123
  805. data/ext/ggml/src/ggml-cpu/llamafile/sgemm.cpp +0 -1884
  806. data/ext/ggml/src/ggml-cpu/llamafile/sgemm.h +0 -14
  807. data/ext/ggml/src/ggml-metal/ggml-metal-impl.h +0 -288
  808. data/ext/ggml/src/ggml-sycl/convert.cpp +0 -547
  809. data/ext/ggml/src/ggml-sycl/element_wise.cpp +0 -1030
  810. data/ext/ggml/src/ggml-sycl/im2col.cpp +0 -126
  811. data/ext/ggml/src/ggml-sycl/mmvq.cpp +0 -1015
  812. data/ext/ggml/src/ggml-sycl/norm.cpp +0 -378
  813. data/ext/ggml/src/ggml-sycl/rope.cpp +0 -276
  814. data/ext/ggml/src/ggml-sycl/wkv6.cpp +0 -141
  815. data/ext/metal-embed.mk +0 -17
  816. data/ext/metal.mk +0 -6
  817. data/ext/ruby_whisper.cpp +0 -1909
  818. data/ext/scripts/get-flags.mk +0 -38
  819. data/lib/whisper.rb +0 -2
  820. data/tests/helper.rb +0 -7
  821. data/tests/test_package.rb +0 -31
  822. data/tests/test_params.rb +0 -160
  823. data/tests/test_segment.rb +0 -83
  824. /data/ext/{ggml → sources/ggml}/include/ggml-blas.h +0 -0
  825. /data/ext/{ggml → sources/ggml}/include/ggml-cann.h +0 -0
  826. /data/ext/{ggml → sources/ggml}/include/ggml-cuda.h +0 -0
  827. /data/ext/{ggml → sources/ggml}/include/ggml-kompute.h +0 -0
  828. /data/ext/{ggml → sources/ggml}/include/ggml-opencl.h +0 -0
  829. /data/ext/{ggml → sources/ggml}/include/ggml-sycl.h +0 -0
  830. /data/ext/{ggml → sources/ggml}/src/ggml-amx/common.h +0 -0
  831. /data/ext/{ggml → sources/ggml}/src/ggml-amx/ggml-amx.cpp +0 -0
  832. /data/ext/{ggml → sources/ggml}/src/ggml-amx/mmq.cpp +0 -0
  833. /data/ext/{ggml → sources/ggml}/src/ggml-amx/mmq.h +0 -0
  834. /data/ext/{ggml → sources/ggml}/src/ggml-blas/ggml-blas.cpp +0 -0
  835. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/ascendc_kernels.h +0 -0
  836. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_f16.cpp +0 -0
  837. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_f32.cpp +0 -0
  838. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_q4_0.cpp +0 -0
  839. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_q8_0.cpp +0 -0
  840. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_f16_q8_0.cpp +0 -0
  841. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_f32_q8_0.cpp +0 -0
  842. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_float_to_q4_0.cpp +0 -0
  843. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/amx.h +0 -0
  844. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/common.h +0 -0
  845. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/mmq.h +0 -0
  846. /data/ext/{ggml/src/ggml-cpu/ggml-cpu-hbm.h → sources/ggml/src/ggml-cpu/hbm.h} +0 -0
  847. /data/ext/{ggml/src/ggml-cpu/ggml-cpu-traits.h → sources/ggml/src/ggml-cpu/traits.h} +0 -0
  848. /data/ext/{ggml → sources/ggml}/src/ggml-kompute/ggml-kompute.cpp +0 -0
  849. /data/ext/{ggml → sources/ggml}/src/ggml-quants.h +0 -0
  850. /data/ext/{ggml → sources/ggml}/src/ggml-threading.cpp +0 -0
  851. /data/ext/{ggml → sources/ggml}/src/ggml-threading.h +0 -0
  852. /data/ext/{src → sources/src}/coreml/whisper-encoder.h +0 -0
  853. /data/ext/{src → sources/src}/openvino/whisper-openvino-encoder.cpp +0 -0
  854. /data/ext/{src → sources/src}/openvino/whisper-openvino-encoder.h +0 -0
  855. /data/{tests → test}/jfk_reader/.gitignore +0 -0
  856. /data/{tests → test}/jfk_reader/extconf.rb +0 -0
  857. /data/{tests → test}/jfk_reader/jfk_reader.c +0 -0
@@ -0,0 +1,1307 @@
1
+ //
2
+ // MIT license
3
+ // Copyright (C) 2025 Intel Corporation
4
+ // SPDX-License-Identifier: MIT
5
+ //
6
+
7
+ //
8
+ // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
9
+ // See https://llvm.org/LICENSE.txt for license information.
10
+ // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
11
+ //
12
+
13
+ #ifndef GGML_SYCL_VECDOTQ_HPP
14
+ #define GGML_SYCL_VECDOTQ_HPP
15
+
16
+ #include "dpct/helper.hpp"
17
+ #include "ggml.h"
18
+ #include "quants.hpp"
19
+
20
+ typedef float (*vec_dot_q_sycl_t)(const void * __restrict__ vbq, const block_q8_1 * __restrict__ bq8_1,
21
+ const int & iqs);
22
+
23
+ static __dpct_inline__ int get_int_from_int8(const int8_t* x8, const int& i32) {
24
+ const uint16_t* x16 =
25
+ (const uint16_t*)(x8 + sizeof(int) * i32); // assume at least 2 byte
26
+ // alignment
27
+
28
+ int x32 = 0;
29
+ x32 |= x16[0] << 0;
30
+ x32 |= x16[1] << 16;
31
+
32
+ return x32;
33
+ }
34
+
35
+ static __dpct_inline__ int get_int_from_uint8(
36
+ const uint8_t* x8,
37
+ const int& i32) {
38
+ const uint16_t* x16 =
39
+ (const uint16_t*)(x8 + sizeof(int) * i32); // assume at least 2 byte
40
+ // alignment
41
+
42
+ int x32 = 0;
43
+ x32 |= x16[0] << 0;
44
+ x32 |= x16[1] << 16;
45
+
46
+ return x32;
47
+ }
48
+
49
+ static __dpct_inline__ int get_int_from_int8_aligned(
50
+ const int8_t* x8,
51
+ const int& i32) {
52
+ return *(
53
+ (const int*)(x8 + sizeof(int) * i32)); // assume at least 4 byte alignment
54
+ }
55
+
56
+ static __dpct_inline__ int get_int_from_uint8_aligned(
57
+ const uint8_t* x8,
58
+ const int& i32) {
59
+ return *(
60
+ (const int*)(x8 + sizeof(int) * i32)); // assume at least 4 byte alignment
61
+ }
62
+
63
+ static __dpct_inline__ void get_int_from_table_16(const uint32_t &q4,
64
+ const uint8_t *values,
65
+ int &val1, int &val2) {
66
+
67
+ uint32_t aux32; const uint8_t * q8 = (const uint8_t *)&aux32;
68
+ aux32 = q4 & 0x0f0f0f0f;
69
+ uint16_t v1 = values[q8[0]] | (values[q8[1]] << 8);
70
+ uint16_t v2 = values[q8[2]] | (values[q8[3]] << 8);
71
+ val1 = v1 | (v2 << 16);
72
+ aux32 = (q4 >> 4) & 0x0f0f0f0f;
73
+ v1 = values[q8[0]] | (values[q8[1]] << 8);
74
+ v2 = values[q8[2]] | (values[q8[3]] << 8);
75
+ val2 = v1 | (v2 << 16);
76
+ }
77
+
78
+ #define VDR_Q2_K_Q8_1_MMVQ 1
79
+
80
+ // contiguous v/x values
81
+ static __dpct_inline__ float vec_dot_q2_K_q8_1_impl_mmvq(
82
+ const int &v, const int *__restrict__ u, const uint8_t *__restrict__ scales,
83
+ const sycl::half2 &dm2, const float *__restrict__ d8) {
84
+
85
+ float sumf_d = 0.0f;
86
+ float sumf_m = 0.0f;
87
+
88
+ #pragma unroll
89
+ for (int i = 0; i < QR2_K; ++i) {
90
+ const int sc = scales[2*i];
91
+
92
+ const int vi = (v >> (2*i)) & 0x03030303;
93
+
94
+ sumf_d +=
95
+ d8[i] * (dpct::dp4a(vi, u[i], 0) * (sc & 0xF)); // SIMD dot product
96
+
97
+ // fill int with 4x m
98
+ int m = sc >> 4;
99
+ m |= m << 8;
100
+ m |= m << 16;
101
+ sumf_m += d8[i] *
102
+ dpct::dp4a(
103
+ m, u[i],
104
+ 0); // multiply constant q2_K part with sum of q8_1 values
105
+ }
106
+
107
+ const sycl::float2 dm2f =
108
+ dm2.convert<float, sycl::rounding_mode::automatic>();
109
+
110
+ return dm2f.x() * sumf_d - dm2f.y() * sumf_m;
111
+ }
112
+
113
+
114
+ #define VDR_Q3_K_Q8_1_MMVQ 1
115
+
116
+ // contiguous v/x values
117
+ static __dpct_inline__ float vec_dot_q3_K_q8_1_impl_mmvq(
118
+ const int &vl, const int &vh, const int *__restrict__ u,
119
+ const uint8_t *__restrict__ scales, const int &scale_offset,
120
+ const float &d3, const float *__restrict__ d8) {
121
+
122
+ float sumf = 0.0f;
123
+
124
+ #pragma unroll
125
+ for (int i = 0; i < QR3_K; ++i) {
126
+ const int isc = scale_offset + 2*i;
127
+
128
+ const int isc_low = isc % (QK_K/32);
129
+ const int sc_shift_low = 4 * (isc / (QK_K/32));
130
+ const int sc_low = (scales[isc_low] >> sc_shift_low) & 0xF;
131
+
132
+ const int isc_high = isc % (QK_K/64);
133
+ const int sc_shift_high = 2 * (isc / (QK_K/64));
134
+ const int sc_high = ((scales[(QK_K/32) + isc_high] >> sc_shift_high) & 3) << 4;
135
+
136
+ const int sc = (sc_low | sc_high) - 32;
137
+
138
+ const int vil = (vl >> (2*i)) & 0x03030303;
139
+
140
+ const int vih = ((vh >> i) << 2) & 0x04040404;
141
+
142
+ const int vi =
143
+ dpct::vectorized_binary<sycl::char4>(vil, vih, dpct::sub_sat());
144
+
145
+ sumf += d8[i] * (dpct::dp4a(vi, u[i], 0) * sc); // SIMD dot product
146
+ }
147
+
148
+ return d3 * sumf;
149
+ }
150
+
151
+ #define VDR_Q4_K_Q8_1_MMVQ 2
152
+
153
+ // contiguous v/x values
154
+ static __dpct_inline__ float vec_dot_q4_K_q8_1_impl_vmmq(
155
+ const int *__restrict__ v, const int *__restrict__ u,
156
+ const uint8_t *__restrict__ sc, const uint8_t *__restrict__ m,
157
+ const sycl::half2 &dm4, const float *__restrict__ d8) {
158
+
159
+ float sumf_d = 0.0f;
160
+ float sumf_m = 0.0f;
161
+
162
+ #pragma unroll
163
+ for (int i = 0; i < QR4_K; ++i) {
164
+ const int v0i = (v[0] >> (4*i)) & 0x0F0F0F0F;
165
+ const int v1i = (v[1] >> (4*i)) & 0x0F0F0F0F;
166
+
167
+ const int dot1 =
168
+ dpct::dp4a(v1i, u[2 * i + 1],
169
+ dpct::dp4a(v0i, u[2 * i + 0], 0)); // SIMD dot product
170
+ const int dot2 =
171
+ dpct::dp4a(0x01010101, u[2 * i + 1],
172
+ dpct::dp4a(0x01010101, u[2 * i + 0], 0)); // sum of u
173
+
174
+ sumf_d += d8[i] * (dot1 * sc[i]);
175
+ sumf_m += d8[i] * (dot2 * m[i]); // multiply constant part of q4_K with sum of q8_1 values
176
+ }
177
+
178
+ const sycl::float2 dm4f =
179
+ dm4.convert<float, sycl::rounding_mode::automatic>();
180
+
181
+ return dm4f.x() * sumf_d - dm4f.y() * sumf_m;
182
+ }
183
+
184
+
185
+ #define VDR_Q5_K_Q8_1_MMVQ 2
186
+
187
+ // contiguous v/x values
188
+ static __dpct_inline__ float vec_dot_q5_K_q8_1_impl_vmmq(
189
+ const int *__restrict__ vl, const int *__restrict__ vh,
190
+ const int *__restrict__ u, const uint8_t *__restrict__ sc,
191
+ const uint8_t *__restrict__ m, const sycl::half2 &dm5,
192
+ const float *__restrict__ d8) {
193
+
194
+ float sumf_d = 0.0f;
195
+ float sumf_m = 0.0f;
196
+
197
+ #pragma unroll
198
+ for (int i = 0; i < QR5_K; ++i) {
199
+ const int vl0i = (vl[0] >> (4*i)) & 0x0F0F0F0F;
200
+ const int vl1i = (vl[1] >> (4*i)) & 0x0F0F0F0F;
201
+
202
+ const int vh0i = ((vh[0] >> i) << 4) & 0x10101010;
203
+ const int vh1i = ((vh[1] >> i) << 4) & 0x10101010;
204
+
205
+ const int v0i = vl0i | vh0i;
206
+ const int v1i = vl1i | vh1i;
207
+
208
+ const int dot1 =
209
+ dpct::dp4a(v0i, u[2 * i + 0],
210
+ dpct::dp4a(v1i, u[2 * i + 1], 0)); // SIMD dot product
211
+ const int dot2 =
212
+ dpct::dp4a(0x01010101, u[2 * i + 0],
213
+ dpct::dp4a(0x01010101, u[2 * i + 1], 0)); // sum of u
214
+
215
+ sumf_d += d8[i] * (dot1 * sc[i]);
216
+ sumf_m += d8[i] * (dot2 * m[i]);
217
+
218
+ }
219
+
220
+ const sycl::float2 dm5f =
221
+ dm5.convert<float, sycl::rounding_mode::automatic>();
222
+
223
+ return dm5f.x() * sumf_d - dm5f.y() * sumf_m;
224
+ }
225
+
226
+
227
+ #define VDR_Q6_K_Q8_1_MMVQ 1
228
+
229
+ // contiguous v/x values
230
+ static __dpct_inline__ float
231
+ vec_dot_q6_K_q8_1_impl_mmvq(const int &vl, const int &vh,
232
+ const int *__restrict__ u,
233
+ const int8_t *__restrict__ scales, const float &d,
234
+ const float *__restrict__ d8) {
235
+
236
+ float sumf = 0.0f;
237
+
238
+ #pragma unroll
239
+ for (int i = 0; i < QR6_K; ++i) {
240
+ const int sc = scales[4*i];
241
+
242
+ const int vil = (vl >> (4*i)) & 0x0F0F0F0F;
243
+
244
+ const int vih = ((vh >> (4*i)) << 4) & 0x30303030;
245
+
246
+ const int vi = dpct::vectorized_binary<sycl::char4>(
247
+ (vil | vih), 0x20202020, dpct::sub_sat()); // vi = (vil | vih) - 32
248
+
249
+ sumf += d8[i] * (dpct::dp4a(vi, u[i], 0) * sc); // SIMD dot product
250
+ }
251
+
252
+ return d*sumf;
253
+ }
254
+
255
+ // VDR = vec dot ratio, how many contiguous integers each thread processes when the vec dot kernel is called
256
+ // MMVQ = mul_mat_vec_q, MMQ = mul_mat_q
257
+
258
+ template <ggml_type T> struct reorder_vec_dot_q_sycl {
259
+ static_assert(T != T, "ggml_type for reorder vecdot not implemented");
260
+ };
261
+
262
+ template <> struct reorder_vec_dot_q_sycl<GGML_TYPE_Q4_0> {
263
+ static constexpr ggml_type gtype = GGML_TYPE_Q4_0;
264
+
265
+ using q4_0_block = ggml_sycl_reordered::block_q_t<GGML_TYPE_Q4_0>;
266
+ using q4_0_traits = typename q4_0_block::traits;
267
+
268
+ __dpct_inline__ float vec_dot_q4_0_q8_1_impl(const int * v, const int * u, const float & d4, const sycl::half2 & ds8) {
269
+ int sumi = 0;
270
+
271
+ #pragma unroll
272
+ for (size_t i = 0; i < q4_0_traits::vdr_mmvq; ++i) {
273
+ const int vi0 = (v[i] >> 0) & 0x0F0F0F0F;
274
+ const int vi1 = (v[i] >> 4) & 0x0F0F0F0F;
275
+
276
+ // SIMD dot product of quantized values
277
+ sumi = dpct::dp4a(vi0, u[2 * i + 0], sumi);
278
+ sumi = dpct::dp4a(vi1, u[2 * i + 1], sumi);
279
+ }
280
+
281
+ const sycl::float2 ds8f = ds8.convert<float, sycl::rounding_mode::automatic>();
282
+
283
+ // second part effectively subtracts 8 from each quant value
284
+ return d4 * (sumi * ds8f.x() - (8 * q4_0_traits::vdr_mmvq / q4_0_traits::qi) * ds8f.y());
285
+ }
286
+
287
+ __dpct_inline__ float operator()(const void * __restrict__ vbq, const std::pair<int, int> ibx_offset,
288
+ const std::pair<int, int> d_offset, const int8_t * q8_1_quant_ptr,
289
+ const sycl::half2 * q8_1_ds, const int & iqs) {
290
+ const uint8_t * bq4_0 = static_cast<const uint8_t *>(vbq) + ibx_offset.first;
291
+ const ggml_half d = *(reinterpret_cast<const ggml_half *>(static_cast<const uint8_t *>(vbq) + d_offset.first));
292
+ int v[q4_0_traits::vdr_mmvq];
293
+ int u[2 * q4_0_traits::vdr_mmvq];
294
+
295
+
296
+ #pragma unroll
297
+ for (size_t i = 0; i < q4_0_traits::vdr_mmvq; ++i) {
298
+ v[i] = get_int_from_uint8(bq4_0, iqs + i);
299
+ u[2 * i + 0] = get_int_from_int8_aligned(q8_1_quant_ptr, iqs + i);
300
+ u[2 * i + 1] = get_int_from_int8_aligned(q8_1_quant_ptr, iqs + i + q4_0_traits::qi);
301
+ }
302
+
303
+ return vec_dot_q4_0_q8_1_impl(v, u, d, *q8_1_ds);
304
+ };
305
+ };
306
+
307
+ static inline float vec_dot_q4_K_q8_1_common(const int * __restrict__ q4, const uint16_t * __restrict__ scales,
308
+ const ggml_half2 & dm, const block_q8_1 * __restrict__ bq8_1,
309
+ const int & iqs) {
310
+ int v[2];
311
+ int u[2 * QR4_K];
312
+ float d8[QR4_K];
313
+
314
+ v[0] = q4[0];
315
+ v[1] = q4[4];
316
+
317
+ uint16_t aux[2];
318
+ const int j = (QR4_K * ((iqs / 2) / (QI8_1 / 2))) / 2;
319
+ if (j < 2) {
320
+ aux[0] = scales[j + 0] & 0x3f3f;
321
+ aux[1] = scales[j + 2] & 0x3f3f;
322
+ } else {
323
+ aux[0] = ((scales[j + 2] >> 0) & 0x0f0f) | ((scales[j - 2] & 0xc0c0) >> 2);
324
+ aux[1] = ((scales[j + 2] >> 4) & 0x0f0f) | ((scales[j - 0] & 0xc0c0) >> 2);
325
+ }
326
+
327
+ const uint8_t * sc = (const uint8_t *) aux;
328
+ const uint8_t * m = sc + 2;
329
+
330
+ const int bq8_offset = QR4_K * ((iqs / 2) / (QI8_1 / 2));
331
+
332
+ for (int i = 0; i < QR4_K; ++i) {
333
+ const block_q8_1 * bq8i = bq8_1 + bq8_offset + i;
334
+ d8[i] = bq8i->ds[0];
335
+
336
+ const int * q8 = (const int *) bq8i->qs + ((iqs / 2) % 4);
337
+ u[2 * i + 0] = q8[0];
338
+ u[2 * i + 1] = q8[4];
339
+ }
340
+
341
+ return vec_dot_q4_K_q8_1_impl_vmmq(v, u, sc, m, dm, d8);
342
+ }
343
+
344
+ template <> struct reorder_vec_dot_q_sycl<GGML_TYPE_Q4_K> {
345
+ static constexpr ggml_type gtype = GGML_TYPE_Q4_K;
346
+
347
+ using q4_k_block = ggml_sycl_reordered::block_q_t<GGML_TYPE_Q4_K>;
348
+ using q4_k_traits = typename q4_k_block::traits;
349
+
350
+ __dpct_inline__ float operator()(const void * __restrict__ vbq, const std::pair<int, int> ibx_offset,
351
+ const std::pair<int, int> d_offset, const int8_t * q8_1_quant_ptr,
352
+ const sycl::half2 * q8_1_ds, const int & iqs) {
353
+ const int ib = ibx_offset.first / (QK_K / 2);
354
+
355
+ const uint8_t * base = static_cast<const uint8_t *>(vbq);
356
+ const uint8_t * qs = base + ibx_offset.first;
357
+ const uint8_t * scs = base + d_offset.first + ib * K_SCALE_SIZE;
358
+ const ggml_half2 * dms = reinterpret_cast<const ggml_half2 *>(base + d_offset.second);
359
+
360
+ const int bq8_offset = QR4_K * ((iqs / 2) / (QI8_1 / 2));
361
+ const int * q4 = (const int *) (qs + 16 * bq8_offset + 4 * ((iqs / 2) % 4));
362
+ const uint16_t * scales = (const uint16_t *) scs;
363
+
364
+ int v[2];
365
+ int u[2 * QR4_K];
366
+ float d8[QR4_K];
367
+
368
+ v[0] = q4[0];
369
+ v[1] = q4[4];
370
+
371
+ uint16_t aux[2];
372
+ const int j = (QR4_K * ((iqs / 2) / (QI8_1 / 2))) / 2;
373
+ if (j < 2) {
374
+ aux[0] = scales[j + 0] & 0x3f3f;
375
+ aux[1] = scales[j + 2] & 0x3f3f;
376
+ } else {
377
+ aux[0] = ((scales[j + 2] >> 0) & 0x0f0f) | ((scales[j - 2] & 0xc0c0) >> 2);
378
+ aux[1] = ((scales[j + 2] >> 4) & 0x0f0f) | ((scales[j - 0] & 0xc0c0) >> 2);
379
+ }
380
+
381
+ const uint8_t * sc = (const uint8_t *) aux;
382
+ const uint8_t * m = sc + 2;
383
+
384
+ for (int i = 0; i < QR4_K; ++i) {
385
+ const int8_t* quant_base_ptr = q8_1_quant_ptr + (bq8_offset + i) * QK8_1;
386
+ sycl::half2 ds_values = *(q8_1_ds + bq8_offset + i);
387
+
388
+ d8[i] = ds_values[0];
389
+
390
+ const int * q8 = (const int *) quant_base_ptr + ((iqs / 2) % 4);
391
+ u[2 * i + 0] = q8[0];
392
+ u[2 * i + 1] = q8[4];
393
+ }
394
+
395
+ return vec_dot_q4_K_q8_1_impl_vmmq(v, u, sc, m, *dms, d8);
396
+ }
397
+ };
398
+
399
+ template <> struct reorder_vec_dot_q_sycl<GGML_TYPE_Q6_K> {
400
+ static constexpr ggml_type gtype = GGML_TYPE_Q6_K;
401
+
402
+ using q6_k_block = ggml_sycl_reordered::block_q_t<GGML_TYPE_Q6_K>;
403
+ using q6_k_traits = typename q6_k_block::traits;
404
+
405
+ __dpct_inline__ float vec_dot_q6_K_q8_1_impl_mmvq(const int vl, const int vh, const int * __restrict__ u,
406
+ const int8_t * __restrict__ scales, const float d,
407
+ const float * __restrict__ d8) {
408
+ float sumf = 0.0f;
409
+
410
+ #pragma unroll
411
+ for (int i = 0; i < QR6_K; ++i) {
412
+ const int sc = scales[4 * i];
413
+
414
+ const int vil = (vl >> (4 * i)) & 0x0F0F0F0F;
415
+
416
+ const int vih = ((vh >> (4 * i)) << 4) & 0x30303030;
417
+
418
+ const int vi = dpct::vectorized_binary<sycl::char4>((vil | vih), 0x20202020,
419
+ dpct::sub_sat()); // vi = (vil | vih) - 32
420
+
421
+ sumf += d8[i] * (dpct::dp4a(vi, u[i], 0) * sc); // SIMD dot product
422
+ }
423
+
424
+ return d * sumf;
425
+ }
426
+
427
+ __dpct_inline__ float operator()(const void * __restrict__ vbq, const std::pair<int, int> ibx_offset,
428
+ const std::pair<int, int> d_offset, const int8_t * q8_1_quant_ptr, const sycl::half2 * q8_1_ds,
429
+ const int iqs) {
430
+ const int ib = ibx_offset.first / (QK_K / 2);
431
+
432
+ const uint8_t * base = static_cast<const uint8_t *>(vbq);
433
+ const uint8_t * ql = base + ibx_offset.first;
434
+ const uint8_t * qh = base + ibx_offset.second;
435
+ const int8_t * scales = reinterpret_cast<const int8_t *>(base + d_offset.first);
436
+ const ggml_half * d = (const ggml_half *) (base + d_offset.second) + ib;
437
+
438
+ const int bq8_offset = 2 * QR6_K * (iqs / (QI6_K / 2)) + (iqs % (QI6_K / 2)) / (QI6_K / 4);
439
+ const int scale_offset = (QI6_K / 4) * (iqs / (QI6_K / 2)) + (iqs % (QI6_K / 2)) / (QI6_K / 8);
440
+ const int vh_shift = 2 * ((iqs % (QI6_K / 2)) / (QI6_K / 4));
441
+
442
+ const int vl = get_int_from_uint8(ql, iqs);
443
+ const int vh = get_int_from_uint8(qh, (QI6_K / 4) * (iqs / (QI6_K / 2)) + iqs % (QI6_K / 4)) >> vh_shift;
444
+
445
+ const int8_t * scs = scales + scale_offset;
446
+
447
+ int u[QR6_K];
448
+ float d8[QR6_K];
449
+
450
+ #pragma unroll
451
+ for (int i = 0; i < QR6_K; ++i) {
452
+ u[i] = get_int_from_int8_aligned(q8_1_quant_ptr + (bq8_offset + 2 * i) * QK8_1, iqs % QI8_1);
453
+ const sycl::half2 ds_values = *(q8_1_ds + bq8_offset + 2 * i);
454
+ d8[i] = ds_values[0];
455
+ }
456
+ return vec_dot_q6_K_q8_1_impl_mmvq(vl, vh, u, scs, *d, d8);
457
+ }
458
+ };
459
+ #define VDR_Q4_0_Q8_1_MMVQ 2
460
+ #define VDR_Q4_0_Q8_1_MMQ 4
461
+
462
+ template <int vdr>
463
+ static __dpct_inline__ float vec_dot_q4_0_q8_1_impl(const int * v, const int * u, const float & d4,
464
+ const sycl::half2 & ds8) {
465
+ int sumi = 0;
466
+ #pragma unroll
467
+ for (int i = 0; i < vdr; ++i) {
468
+ const int vi0 = (v[i] >> 0) & 0x0F0F0F0F;
469
+ const int vi1 = (v[i] >> 4) & 0x0F0F0F0F;
470
+
471
+ // SIMD dot product of quantized values
472
+ sumi = dpct::dp4a(vi0, u[2 * i + 0], sumi);
473
+ sumi = dpct::dp4a(vi1, u[2 * i + 1], sumi);
474
+ }
475
+
476
+ const sycl::float2 ds8f = ds8.convert<float, sycl::rounding_mode::automatic>();
477
+
478
+ // second part effectively subtracts 8 from each quant value
479
+ return d4 * (sumi * ds8f.x() - (8 * vdr / QI4_0) * ds8f.y());
480
+ }
481
+
482
+ #define VDR_Q4_1_Q8_1_MMVQ 2
483
+ #define VDR_Q4_1_Q8_1_MMQ 4
484
+
485
+ template <int vdr>
486
+ static __dpct_inline__ float vec_dot_q4_1_q8_1_impl(const int *v, const int *u,
487
+ const sycl::half2 &dm4,
488
+ const sycl::half2 &ds8) {
489
+
490
+ int sumi = 0;
491
+
492
+ #pragma unroll
493
+ for (int i = 0; i < vdr; ++i) {
494
+ const int vi0 = (v[i] >> 0) & 0x0F0F0F0F;
495
+ const int vi1 = (v[i] >> 4) & 0x0F0F0F0F;
496
+
497
+ // SIMD dot product of quantized values
498
+ sumi = dpct::dp4a(vi0, u[2 * i + 0], sumi);
499
+ sumi = dpct::dp4a(vi1, u[2 * i + 1], sumi);
500
+ }
501
+
502
+ #ifdef GGML_SYCL_F16
503
+ const sycl::float2 tmp =
504
+ (dm4 * ds8).convert<float, sycl::rounding_mode::automatic>();
505
+ const float d4d8 = tmp.x();
506
+ const float m4s8 = tmp.y();
507
+ #else
508
+ const sycl::float2 dm4f =
509
+ dm4.convert<float, sycl::rounding_mode::automatic>();
510
+ const sycl::float2 ds8f =
511
+ ds8.convert<float, sycl::rounding_mode::automatic>();
512
+ const float d4d8 = dm4f.x() * ds8f.x();
513
+ const float m4s8 = dm4f.y() * ds8f.y();
514
+ #endif // GGML_SYCL_F16
515
+
516
+ // scale second part of sum by QI8_1/(vdr * QR4_1) to compensate for multiple threads adding it
517
+ return sumi * d4d8 + m4s8 / (QI8_1 / (vdr * QR4_1));
518
+ }
519
+
520
+ #define VDR_Q5_0_Q8_1_MMVQ 2
521
+ #define VDR_Q5_0_Q8_1_MMQ 4
522
+
523
+ template <int vdr>
524
+ static __dpct_inline__ float
525
+ vec_dot_q5_0_q8_1_impl(const int *vl, const int *vh, const int *u,
526
+ const float &d5, const sycl::half2 &ds8) {
527
+ int sumi = 0;
528
+
529
+ #pragma unroll
530
+ for (int i = 0; i < vdr; ++i) {
531
+ int vi0 = (vl[i] >> 0) & 0x0F0F0F0F; // lower 4 qs bits, still need qh as 5th bits
532
+ vi0 |= (vh[i] << 4) & 0x00000010; // 0 -> 4
533
+ vi0 |= (vh[i] << 11) & 0x00001000; // 1 -> 12
534
+ vi0 |= (vh[i] << 18) & 0x00100000; // 2 -> 20
535
+ vi0 |= (vh[i] << 25) & 0x10000000; // 3 -> 28
536
+ sumi = dpct::dp4a(vi0, u[2 * i + 0],
537
+ sumi); // SIMD dot product of quantized values
538
+
539
+ int vi1 = (vl[i] >> 4) & 0x0F0F0F0F; // upper 4 qs bits, still need qh as 5th bits
540
+ vi1 |= (vh[i] >> 12) & 0x00000010; // 16 -> 4
541
+ vi1 |= (vh[i] >> 5) & 0x00001000; // 17 -> 12
542
+ vi1 |= (vh[i] << 2) & 0x00100000; // 18 -> 20
543
+ vi1 |= (vh[i] << 9) & 0x10000000; // 19 -> 28
544
+ sumi = dpct::dp4a(vi1, u[2 * i + 1],
545
+ sumi); // SIMD dot product of quantized values
546
+ }
547
+
548
+ const sycl::float2 ds8f =
549
+ ds8.convert<float, sycl::rounding_mode::automatic>();
550
+
551
+ // second part effectively subtracts 16 from each quant value
552
+ return d5 * (sumi * ds8f.x() - (16 * vdr / QI5_0) * ds8f.y());
553
+ }
554
+
555
+ #define VDR_Q5_1_Q8_1_MMVQ 2
556
+ #define VDR_Q5_1_Q8_1_MMQ 4
557
+
558
+ template <int vdr>
559
+ static __dpct_inline__ float
560
+ vec_dot_q5_1_q8_1_impl(const int *vl, const int *vh, const int *u,
561
+ const sycl::half2 &dm5, const sycl::half2 &ds8) {
562
+
563
+ int sumi = 0;
564
+
565
+ #pragma unroll
566
+ for (int i = 0; i < vdr; ++i) {
567
+ int vi0 = (vl[i] >> 0) & 0x0F0F0F0F; // lower 4 qs bits, still need qh as 5th bits
568
+ vi0 |= (vh[i] << 4) & 0x00000010; // 0 -> 4
569
+ vi0 |= (vh[i] << 11) & 0x00001000; // 1 -> 12
570
+ vi0 |= (vh[i] << 18) & 0x00100000; // 2 -> 20
571
+ vi0 |= (vh[i] << 25) & 0x10000000; // 3 -> 28
572
+ sumi = dpct::dp4a(vi0, u[2 * i + 0],
573
+ sumi); // SIMD dot product of quantized values
574
+
575
+ int vi1 = (vl[i] >> 4) & 0x0F0F0F0F; // upper 4 qs bits, still need qh as 5th bits
576
+ vi1 |= (vh[i] >> 12) & 0x00000010; // 16 -> 4
577
+ vi1 |= (vh[i] >> 5) & 0x00001000; // 17 -> 12
578
+ vi1 |= (vh[i] << 2) & 0x00100000; // 18 -> 20
579
+ vi1 |= (vh[i] << 9) & 0x10000000; // 19 -> 28
580
+ sumi = dpct::dp4a(vi1, u[2 * i + 1],
581
+ sumi); // SIMD dot product of quantized values
582
+ }
583
+
584
+ #ifdef GGML_SYCL_F16
585
+ const sycl::float2 tmp =
586
+ (dm5 * ds8).convert<float, sycl::rounding_mode::automatic>();
587
+ const float d5d8 = tmp.x();
588
+ const float m5s8 = tmp.y();
589
+
590
+
591
+ #else
592
+ const sycl::float2 dm5f =
593
+ dm5.convert<float, sycl::rounding_mode::automatic>();
594
+ const sycl::float2 ds8f =
595
+ ds8.convert<float, sycl::rounding_mode::automatic>();
596
+ const float d5d8 = dm5f.x() * ds8f.x();
597
+ const float m5s8 = dm5f.y() * ds8f.y();
598
+ #endif // GGML_SYCL_F16
599
+
600
+ // scale second part of sum by QI5_1 / vdr to compensate for multiple threads adding it
601
+ return sumi*d5d8 + m5s8 / (QI5_1 / vdr);
602
+ }
603
+
604
+ #define VDR_Q8_0_Q8_1_MMVQ 2
605
+ #define VDR_Q8_0_Q8_1_MMQ 8
606
+
607
+ template <int vdr>
608
+ static __dpct_inline__ float vec_dot_q8_0_q8_1_impl(const int *v, const int *u,
609
+ const float &d8_0,
610
+ const float &d8_1) {
611
+
612
+ int sumi = 0;
613
+
614
+ #pragma unroll
615
+ for (int i = 0; i < vdr; ++i) {
616
+ // SIMD dot product of quantized values
617
+ sumi = dpct::dp4a(v[i], u[i], sumi);
618
+ }
619
+
620
+ return d8_0*d8_1 * sumi;
621
+ }
622
+
623
+ template <int vdr>
624
+ static __dpct_inline__ float vec_dot_q8_1_q8_1_impl(const int *v, const int *u,
625
+ const sycl::half2 &dm8,
626
+ const sycl::half2 &ds8) {
627
+
628
+ int sumi = 0;
629
+
630
+ #pragma unroll
631
+ for (int i = 0; i < vdr; ++i) {
632
+ // SIMD dot product of quantized values
633
+ sumi = dpct::dp4a(v[i], u[i], sumi);
634
+ }
635
+
636
+ #ifdef GGML_SYCL_F16
637
+ const sycl::float2 tmp =
638
+ (dm8 * ds8).convert<float, sycl::rounding_mode::automatic>();
639
+ const float d8d8 = tmp.x();
640
+ const float m8s8 = tmp.y();
641
+ #else
642
+ const sycl::float2 dm8f =
643
+ dm8.convert<float, sycl::rounding_mode::automatic>();
644
+ const sycl::float2 ds8f =
645
+ ds8.convert<float, sycl::rounding_mode::automatic>();
646
+ const float d8d8 = dm8f.x() * ds8f.x();
647
+ const float m8s8 = dm8f.y() * ds8f.y();
648
+ #endif // GGML_SYCL_F16
649
+
650
+ // scale second part of sum by QI8_1/ vdr to compensate for multiple threads adding it
651
+ return sumi*d8d8 + m8s8 / (QI8_1 / vdr);
652
+ }
653
+
654
+ static __dpct_inline__ float
655
+ vec_dot_q4_0_q8_1(const void *__restrict__ vbq,
656
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
657
+
658
+ const block_q4_0 * bq4_0 = (const block_q4_0 *) vbq;
659
+
660
+ int v[VDR_Q4_0_Q8_1_MMVQ];
661
+ int u[2 * VDR_Q4_0_Q8_1_MMVQ];
662
+
663
+ #pragma unroll
664
+ for (int i = 0; i < VDR_Q4_0_Q8_1_MMVQ; ++i) {
665
+ v[i] = get_int_from_uint8(bq4_0->qs, iqs + i);
666
+ u[2 * i + 0] = get_int_from_int8_aligned(bq8_1->qs, iqs + i);
667
+ u[2 * i + 1] = get_int_from_int8_aligned(bq8_1->qs, iqs + i + QI4_0);
668
+ }
669
+
670
+ return vec_dot_q4_0_q8_1_impl<VDR_Q4_0_Q8_1_MMVQ>(v, u, bq4_0->d, bq8_1->ds);
671
+ }
672
+
673
+ static __dpct_inline__ float
674
+ vec_dot_q4_1_q8_1(const void *__restrict__ vbq,
675
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
676
+
677
+ const block_q4_1 * bq4_1 = (const block_q4_1 *) vbq;
678
+
679
+ int v[VDR_Q4_1_Q8_1_MMVQ];
680
+ int u[2*VDR_Q4_1_Q8_1_MMVQ];
681
+
682
+ #pragma unroll
683
+ for (int i = 0; i < VDR_Q4_1_Q8_1_MMVQ; ++i) {
684
+ v[i] = get_int_from_uint8_aligned(bq4_1->qs, iqs + i);
685
+ u[2*i+0] = get_int_from_int8_aligned(bq8_1->qs, iqs + i);
686
+ u[2*i+1] = get_int_from_int8_aligned(bq8_1->qs, iqs + i + QI4_1);
687
+ }
688
+
689
+ return vec_dot_q4_1_q8_1_impl<VDR_Q4_1_Q8_1_MMVQ>(v, u, bq4_1->dm, bq8_1->ds);
690
+ }
691
+
692
+ static __dpct_inline__ float
693
+ vec_dot_q5_0_q8_1(const void *__restrict__ vbq,
694
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
695
+
696
+ const block_q5_0 * bq5_0 = (const block_q5_0 *) vbq;
697
+
698
+ int vl[VDR_Q5_0_Q8_1_MMVQ];
699
+ int vh[VDR_Q5_0_Q8_1_MMVQ];
700
+ int u[2*VDR_Q5_0_Q8_1_MMVQ];
701
+
702
+ #pragma unroll
703
+ for (int i = 0; i < VDR_Q5_0_Q8_1_MMVQ; ++i) {
704
+ vl[i] = get_int_from_uint8(bq5_0->qs, iqs + i);
705
+ vh[i] = get_int_from_uint8(bq5_0->qh, 0) >> (4 * (iqs + i));
706
+ u[2*i+0] = get_int_from_int8_aligned(bq8_1->qs, iqs + i);
707
+ u[2*i+1] = get_int_from_int8_aligned(bq8_1->qs, iqs + i + QI5_0);
708
+ }
709
+
710
+ return vec_dot_q5_0_q8_1_impl<VDR_Q5_0_Q8_1_MMVQ>(vl, vh, u, bq5_0->d, bq8_1->ds);
711
+ }
712
+
713
+ static __dpct_inline__ float
714
+ vec_dot_q5_1_q8_1(const void *__restrict__ vbq,
715
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
716
+
717
+ const block_q5_1 * bq5_1 = (const block_q5_1 *) vbq;
718
+
719
+ int vl[VDR_Q5_1_Q8_1_MMVQ];
720
+ int vh[VDR_Q5_1_Q8_1_MMVQ];
721
+ int u[2*VDR_Q5_1_Q8_1_MMVQ];
722
+
723
+ #pragma unroll
724
+ for (int i = 0; i < VDR_Q5_1_Q8_1_MMVQ; ++i) {
725
+ vl[i] = get_int_from_uint8_aligned(bq5_1->qs, iqs + i);
726
+ vh[i] = get_int_from_uint8_aligned(bq5_1->qh, 0) >> (4 * (iqs + i));
727
+ u[2*i+0] = get_int_from_int8_aligned(bq8_1->qs, iqs + i);
728
+ u[2*i+1] = get_int_from_int8_aligned(bq8_1->qs, iqs + i + QI5_1);
729
+ }
730
+
731
+ return vec_dot_q5_1_q8_1_impl<VDR_Q5_1_Q8_1_MMVQ>(vl, vh, u, bq5_1->dm, bq8_1->ds);
732
+ }
733
+
734
+ static __dpct_inline__ float
735
+ vec_dot_q8_0_q8_1(const void *__restrict__ vbq,
736
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
737
+
738
+ const block_q8_0 * bq8_0 = (const block_q8_0 *) vbq;
739
+
740
+ int v[VDR_Q8_0_Q8_1_MMVQ];
741
+ int u[VDR_Q8_0_Q8_1_MMVQ];
742
+
743
+ #pragma unroll
744
+ for (int i = 0; i < VDR_Q8_0_Q8_1_MMVQ; ++i) {
745
+ v[i] = get_int_from_int8(bq8_0->qs, iqs + i);
746
+ u[i] = get_int_from_int8_aligned(bq8_1->qs, iqs + i);
747
+ }
748
+
749
+ return vec_dot_q8_0_q8_1_impl<VDR_Q8_0_Q8_1_MMVQ>(v, u, bq8_0->d,
750
+ bq8_1->ds[0]);
751
+ }
752
+
753
+ static __dpct_inline__ float
754
+ vec_dot_q2_K_q8_1(const void *__restrict__ vbq,
755
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
756
+
757
+ const block_q2_K * bq2_K = (const block_q2_K *) vbq;
758
+
759
+ const int bq8_offset = QR2_K * (iqs / QI8_1);
760
+ const int scale_offset = iqs - iqs % QI8_1 + (iqs % QI8_1) / (QI8_1/2);
761
+
762
+ const uint8_t * scales = bq2_K->scales + scale_offset;
763
+
764
+ const int v = get_int_from_uint8_aligned(bq2_K->qs, iqs);
765
+ int u[QR2_K];
766
+ float d8[QR2_K];
767
+
768
+ #pragma unroll
769
+ for (int i = 0; i < QR2_K; ++ i) {
770
+ u[i] = get_int_from_int8_aligned(bq8_1[bq8_offset + i].qs, iqs % QI8_1);
771
+ d8[i] = bq8_1[bq8_offset + i].ds[0];
772
+ }
773
+
774
+ return vec_dot_q2_K_q8_1_impl_mmvq(v, u, scales, bq2_K->dm, d8);
775
+ }
776
+
777
+ static __dpct_inline__ float
778
+ vec_dot_q3_K_q8_1(const void *__restrict__ vbq,
779
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
780
+
781
+ const block_q3_K * bq3_K = (const block_q3_K *) vbq;
782
+
783
+ const int bq8_offset = QR3_K * (iqs / (QI3_K/2));
784
+ const int scale_offset = iqs - iqs % QI8_1 + (iqs % QI8_1) / (QI8_1/2);
785
+
786
+ const float d = bq3_K->d;
787
+
788
+ const int vl = get_int_from_uint8(bq3_K->qs, iqs);
789
+
790
+ // invert the mask with ~ so that a 0/1 results in 4/0 being subtracted
791
+ const int vh = ~get_int_from_uint8(bq3_K->hmask, iqs % (QI3_K/2)) >> bq8_offset;
792
+
793
+ int u[QR3_K];
794
+ float d8[QR3_K];
795
+
796
+ #pragma unroll
797
+ for (int i = 0; i < QR3_K; ++i) {
798
+ u[i] = get_int_from_int8_aligned(bq8_1[bq8_offset + i].qs, iqs % QI8_1);
799
+ d8[i] = bq8_1[bq8_offset + i].ds[0];
800
+ }
801
+
802
+ return vec_dot_q3_K_q8_1_impl_mmvq(vl, vh, u, bq3_K->scales, scale_offset, d, d8);
803
+ }
804
+
805
+ static __dpct_inline__ float vec_dot_q4_K_q8_1(const void * __restrict__ vbq, const block_q8_1 * __restrict__ bq8_1,
806
+ const int & iqs) {
807
+ #ifndef GGML_QKK_64
808
+
809
+ const block_q4_K * bq4_K = (const block_q4_K *) vbq;
810
+
811
+ const int bq8_offset = QR4_K * ((iqs / 2) / (QI8_1 / 2));
812
+ const int * q4 = (const int *) (bq4_K->qs + 16 * bq8_offset + 4 * ((iqs / 2) % 4));
813
+ const uint16_t * scales = (const uint16_t *) bq4_K->scales;
814
+
815
+ return vec_dot_q4_K_q8_1_common(q4, scales, bq4_K->dm, bq8_1, iqs);
816
+
817
+ #else
818
+
819
+ #if __SYCL_ARCH__ >= VER_4VEC // lowest compute capability for integer intrinsics
820
+ const block_q4_K * bq4_K = (const block_q4_K *) vbq;
821
+
822
+ float sumf_d = 0.0f;
823
+ float sumf_m = 0.0f;
824
+
825
+ uint16_t aux16[2];
826
+ const uint8_t * s = (const uint8_t *)aux16;
827
+
828
+ const uint16_t * a = (const uint16_t *)bq4_K->scales;
829
+ aux16[0] = a[0] & 0x0f0f;
830
+ aux16[1] = (a[0] >> 4) & 0x0f0f;
831
+
832
+ const float dall = bq4_K->dm[0];
833
+ const float dmin = bq4_K->dm[1];
834
+
835
+ const float d8_1 = bq8_1[0].ds[0];
836
+ const float d8_2 = bq8_1[1].ds[1];
837
+
838
+ const int ui1 = *((const int *)bq8_1[0].qs + (iqs/2));
839
+ const int ui2 = *((const int *)bq8_1[0].qs + (iqs/2) + 4);
840
+ const int ui3 = *((const int *)bq8_1[1].qs + (iqs/2));
841
+ const int ui4 = *((const int *)bq8_1[1].qs + (iqs/2) + 4);
842
+
843
+ const int * q4 = (const int *)bq4_K->qs + (iqs/2);
844
+ const int v1 = q4[0];
845
+ const int v2 = q4[4];
846
+
847
+ const int dot1 = dpct::dp4a(ui2, v2 & 0x0f0f0f0f, dpct::dp4a(ui1, v1 & 0x0f0f0f0f, 0));
848
+ const int dot2 = dpct::dp4a(ui4, (v2 >> 4) & 0x0f0f0f0f, dpct::dp4a(ui3, (v1 >> 4) & 0x0f0f0f0f, 0));
849
+ const int dot3 = dpct::dp4a(0x01010101, ui2, dpct::dp4a(0x01010101, ui1, 0));
850
+ const int dot4 = dpct::dp4a(0x01010101, ui4, dpct::dp4a(0x01010101, ui3, 0));
851
+
852
+ sumf_d += d8_1 * (dot1 * s[0]) + d8_2 * (dot2 * s[1]);
853
+ sumf_m += d8_1 * (dot3 * s[2]) + d8_2 * (dot4 * s[3]);
854
+
855
+ return dall * sumf_d - dmin * sumf_m;
856
+
857
+ #else
858
+ bad_arch();
859
+ #endif // __SYCL_ARCH__ >= VER_4VEC
860
+
861
+ #endif
862
+ }
863
+
864
+ static __dpct_inline__ float
865
+ vec_dot_q5_K_q8_1(const void *__restrict__ vbq,
866
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
867
+
868
+ #ifndef GGML_QKK_64
869
+ const block_q5_K * bq5_K = (const block_q5_K *) vbq;
870
+
871
+ int vl[2];
872
+ int vh[2];
873
+ int u[2*QR5_K];
874
+ float d8[QR5_K];
875
+
876
+ const int bq8_offset = QR5_K * ((iqs/2) / (QI8_1/2));
877
+ const int * ql = (const int *)(bq5_K->qs + 16 * bq8_offset + 4 * ((iqs/2)%4));
878
+ const int * qh = (const int *)(bq5_K->qh + 4 * ((iqs/2)%4));
879
+
880
+ vl[0] = ql[0];
881
+ vl[1] = ql[4];
882
+
883
+ vh[0] = qh[0] >> bq8_offset;
884
+ vh[1] = qh[4] >> bq8_offset;
885
+
886
+ const uint16_t * scales = (const uint16_t *)bq5_K->scales;
887
+ uint16_t aux[2];
888
+ const int j = bq8_offset/2;
889
+ if (j < 2) {
890
+ aux[0] = scales[j+0] & 0x3f3f;
891
+ aux[1] = scales[j+2] & 0x3f3f;
892
+ } else {
893
+ aux[0] = ((scales[j+2] >> 0) & 0x0f0f) | ((scales[j-2] & 0xc0c0) >> 2);
894
+ aux[1] = ((scales[j+2] >> 4) & 0x0f0f) | ((scales[j-0] & 0xc0c0) >> 2);
895
+ }
896
+ const uint8_t * sc = (const uint8_t *)aux;
897
+ const uint8_t * m = sc + 2;
898
+
899
+ #pragma unroll
900
+ for (int i = 0; i < QR5_K; ++i) {
901
+ const block_q8_1 * bq8i = bq8_1 + bq8_offset + i;
902
+ d8[i] = bq8i->ds[0];
903
+
904
+ const int * q8 = (const int *)bq8i->qs + ((iqs/2)%4);
905
+ u[2*i+0] = q8[0];
906
+ u[2*i+1] = q8[4];
907
+ }
908
+
909
+ return vec_dot_q5_K_q8_1_impl_vmmq(vl, vh, u, sc, m, bq5_K->dm, d8);
910
+
911
+ #else
912
+
913
+ #if __SYCL_ARCH__ >= VER_4VEC // lowest compute capability for integer intrinsics
914
+ const block_q5_K * bq5_K = (const block_q5_K *) vbq;
915
+
916
+ const int8_t * s = bq5_K->scales;
917
+
918
+ const float d = bq5_K->d;
919
+
920
+ const float d8_1 = bq8_1[0].ds[0];
921
+ const float d8_2 = bq8_1[1].ds[1];
922
+
923
+ const int ui1 = *((const int *)bq8_1[0].qs + (iqs/2));
924
+ const int ui2 = *((const int *)bq8_1[0].qs + (iqs/2) + 4);
925
+ const int ui3 = *((const int *)bq8_1[1].qs + (iqs/2));
926
+ const int ui4 = *((const int *)bq8_1[1].qs + (iqs/2) + 4);
927
+
928
+ const int * ql = (const int *)bq5_K->qs + (iqs/2);
929
+ const int vl1 = ql[0];
930
+ const int vl2 = ql[4];
931
+
932
+ const int step = 4 * (iqs/2); // 0, 4, 8, 12
933
+ const int im = step/8; // = 0 for iqs = 0, 2, = 1 for iqs = 4, 6
934
+ const int in = step%8; // 0, 4, 0, 4
935
+ const int vh = (*((const int *)(bq5_K->qh + in))) >> im;
936
+
937
+ const int v1 = (((vh << 4) & 0x10101010) ^ 0x10101010) | ((vl1 >> 0) & 0x0f0f0f0f);
938
+ const int v2 = (((vh << 2) & 0x10101010) ^ 0x10101010) | ((vl2 >> 0) & 0x0f0f0f0f);
939
+ const int v3 = (((vh >> 0) & 0x10101010) ^ 0x10101010) | ((vl1 >> 4) & 0x0f0f0f0f);
940
+ const int v4 = (((vh >> 2) & 0x10101010) ^ 0x10101010) | ((vl2 >> 4) & 0x0f0f0f0f);
941
+
942
+ const float sumf_d = d8_1 * (dpct::dp4a(ui1, v1, 0) * s[0] + dpct::dp4a(ui2, v2, 0) * s[1])
943
+ + d8_2 * (dpct::dp4a(ui3, v3, 0) * s[2] + dpct::dp4a(ui4, v4, 0) * s[3]);
944
+
945
+ return d * sumf_d;
946
+
947
+ #else
948
+ bad_arch();
949
+ #endif // __SYCL_ARCH__ >= VER_4VEC
950
+
951
+ #endif
952
+ }
953
+
954
+ static __dpct_inline__ float
955
+ vec_dot_q6_K_q8_1(const void *__restrict__ vbq,
956
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
957
+
958
+ const block_q6_K * bq6_K = (const block_q6_K *) vbq;
959
+
960
+ const int bq8_offset = 2 * QR6_K * (iqs / (QI6_K/2)) + (iqs % (QI6_K/2)) / (QI6_K/4);
961
+ const int scale_offset = (QI6_K/4) * (iqs / (QI6_K/2)) + (iqs % (QI6_K/2)) / (QI6_K/8);
962
+ const int vh_shift = 2 * ((iqs % (QI6_K/2)) / (QI6_K/4));
963
+
964
+ const int vl = get_int_from_uint8(bq6_K->ql, iqs);
965
+ const int vh = get_int_from_uint8(bq6_K->qh, (QI6_K/4) * (iqs / (QI6_K/2)) + iqs % (QI6_K/4)) >> vh_shift;
966
+
967
+ const int8_t * scales = bq6_K->scales + scale_offset;
968
+
969
+ int u[QR6_K];
970
+ float d8[QR6_K];
971
+
972
+ #pragma unroll
973
+ for (int i = 0; i < QR6_K; ++i) {
974
+ u[i] = get_int_from_int8_aligned(bq8_1[bq8_offset + 2*i].qs, iqs % QI8_1);
975
+ d8[i] = bq8_1[bq8_offset + 2 * i].ds[0];
976
+ }
977
+
978
+ return vec_dot_q6_K_q8_1_impl_mmvq(vl, vh, u, scales, bq6_K->d, d8);
979
+ }
980
+
981
+
982
+ static __dpct_inline__ float
983
+ vec_dot_iq2_xxs_q8_1(const void *__restrict__ vbq,
984
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs,
985
+ const uint64_t *iq2xxs_grid, const uint8_t *ksigns_iq2xs,
986
+ const uint8_t *kmask_iq2xs) {
987
+ #if QK_K == 256
988
+ const block_iq2_xxs * bq2 = (const block_iq2_xxs *) vbq;
989
+
990
+ const int ib32 = iqs;
991
+ const uint16_t * q2 = bq2->qs + 4*ib32;
992
+ const uint8_t * aux8 = (const uint8_t *)q2;
993
+ const int8_t * q8 = bq8_1[ib32].qs;
994
+ uint32_t aux32 = q2[2] | (q2[3] << 16);
995
+ int sumi = 0;
996
+ for (int l = 0; l < 4; ++l) {
997
+ const uint8_t * grid = (const uint8_t *)(iq2xxs_grid + aux8[l]);
998
+ const uint8_t signs = ksigns_iq2xs[aux32 & 127];
999
+ for (int j = 0; j < 8; ++j) {
1000
+ sumi += q8[j] * grid[j] * (signs & kmask_iq2xs[j] ? -1 : 1);
1001
+ }
1002
+ q8 += 8;
1003
+ aux32 >>= 7;
1004
+ }
1005
+ const float d = (float)bq2->d * (0.5f + aux32) * bq8_1[ib32].ds[0] * 0.25f;
1006
+ return d * sumi;
1007
+ #else
1008
+ assert(false);
1009
+ return 0.f;
1010
+ #endif
1011
+ }
1012
+
1013
+ static __dpct_inline__ float
1014
+ vec_dot_iq2_xs_q8_1(const void *__restrict__ vbq,
1015
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs,
1016
+ const uint64_t *iq2xs_grid, const uint64_t *ksigns64) {
1017
+ #if DPCT_COMPATIBILITY_TEMP >= \
1018
+ MIN_CC_DP4A // lowest compute capability for integer intrinsics
1019
+ #if QK_K == 256
1020
+ const block_iq2_xs * bq2 = (const block_iq2_xs *) vbq;
1021
+
1022
+ const int ib32 = iqs;
1023
+ const uint16_t * q2 = bq2->qs + 4*ib32;
1024
+ const int8_t * q8 = bq8_1[ib32].qs;
1025
+ const uint8_t ls1 = bq2->scales[ib32] & 0xf;
1026
+ const uint8_t ls2 = bq2->scales[ib32] >> 4;
1027
+ int sumi1 = 0;
1028
+ for (int l = 0; l < 2; ++l) {
1029
+ const uint32_t * grid = (const uint32_t *)(iq2xs_grid + (q2[l] & 511));
1030
+ const uint32_t * signs = (const uint32_t *)(ksigns64 + (q2[l] >> 9));
1031
+ const int grid_l = dpct::vectorized_binary<sycl::uchar4>(
1032
+ grid[0] ^ signs[0], signs[0], std::minus<>());
1033
+ const int grid_h = dpct::vectorized_binary<sycl::uchar4>(
1034
+ grid[1] ^ signs[1], signs[1], std::minus<>());
1035
+ sumi1 = dpct::dp4a(grid_l, *((const int *)q8 + 0), sumi1);
1036
+ sumi1 = dpct::dp4a(grid_h, *((const int *)q8 + 1), sumi1);
1037
+ q8 += 8;
1038
+ }
1039
+ int sumi2 = 0;
1040
+ for (int l = 2; l < 4; ++l) {
1041
+ const uint32_t * grid = (const uint32_t *)(iq2xs_grid + (q2[l] & 511));
1042
+ const uint32_t * signs = (const uint32_t *)(ksigns64 + (q2[l] >> 9));
1043
+ const int grid_l = dpct::vectorized_binary<sycl::uchar4>(
1044
+ grid[0] ^ signs[0], signs[0], std::minus<>());
1045
+ const int grid_h = dpct::vectorized_binary<sycl::uchar4>(
1046
+ grid[1] ^ signs[1], signs[1], std::minus<>());
1047
+ sumi2 = dpct::dp4a(grid_l, *((const int *)q8 + 0), sumi2);
1048
+ sumi2 = dpct::dp4a(grid_h, *((const int *)q8 + 1), sumi2);
1049
+ q8 += 8;
1050
+ }
1051
+ const float d = (float)bq2->d * bq8_1[ib32].ds[0] * 0.25f;
1052
+ return d * ((0.5f + ls1) * sumi1 + (0.5f + ls2) * sumi2);
1053
+ #else
1054
+ assert(false);
1055
+ return 0.f;
1056
+ #endif
1057
+ #else
1058
+ assert(false);
1059
+ return 0.f;
1060
+ #endif
1061
+ }
1062
+
1063
+ static __dpct_inline__ float
1064
+ vec_dot_iq2_s_q8_1(const void *__restrict__ vbq,
1065
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
1066
+ #if QK_K == 256
1067
+ const block_iq2_s * bq2 = (const block_iq2_s *) vbq;
1068
+
1069
+ const int ib32 = iqs;
1070
+ const int8_t * q8 = bq8_1[ib32].qs;
1071
+ const uint8_t * signs = bq2->qs + QK_K/8 + 4*ib32;
1072
+ const uint8_t ls1 = bq2->scales[ib32] & 0xf;
1073
+ const uint8_t ls2 = bq2->scales[ib32] >> 4;
1074
+ int sumi1 = 0;
1075
+ for (int l = 0; l < 2; ++l) {
1076
+ const uint32_t * grid = (const uint32_t *)(iq2s_grid + (bq2->qs[4*ib32+l] | ((bq2->qh[ib32] << (8-2*l)) & 0x300)));
1077
+ const uint32_t signs0 = dpct::vectorized_binary<sycl::uchar4>(
1078
+ ((signs[l] & 0xf) * 0x01010101) & 0x08040201, 0x08040201,
1079
+ std::equal_to<>());
1080
+ const uint32_t signs1 = dpct::vectorized_binary<sycl::uchar4>(
1081
+ ((signs[l] >> 4) * 0x01010101) & 0x08040201, 0x08040201,
1082
+ std::equal_to<>());
1083
+ const int grid_l = dpct::vectorized_binary<sycl::uchar4>(
1084
+ grid[0] ^ signs0, signs0, std::minus<>());
1085
+ const int grid_h = dpct::vectorized_binary<sycl::uchar4>(
1086
+ grid[1] ^ signs1, signs1, std::minus<>());
1087
+ sumi1 = dpct::dp4a(grid_l, *((const int *)q8 + 0), sumi1);
1088
+ sumi1 = dpct::dp4a(grid_h, *((const int *)q8 + 1), sumi1);
1089
+ q8 += 8;
1090
+ }
1091
+ int sumi2 = 0;
1092
+ for (int l = 2; l < 4; ++l) {
1093
+ const uint32_t * grid = (const uint32_t *)(iq2s_grid + (bq2->qs[4*ib32+l] | ((bq2->qh[ib32] << (8-2*l)) & 0x300)));
1094
+ const uint32_t signs0 = dpct::vectorized_binary<sycl::uchar4>(
1095
+ ((signs[l] & 0xf) * 0x01010101) & 0x08040201, 0x08040201,
1096
+ std::equal_to<>());
1097
+ const uint32_t signs1 = dpct::vectorized_binary<sycl::uchar4>(
1098
+ ((signs[l] >> 4) * 0x01010101) & 0x08040201, 0x08040201,
1099
+ std::equal_to<>());
1100
+ const int grid_l = dpct::vectorized_binary<sycl::uchar4>(
1101
+ grid[0] ^ signs0, signs0, std::minus<>());
1102
+ const int grid_h = dpct::vectorized_binary<sycl::uchar4>(
1103
+ grid[1] ^ signs1, signs1, std::minus<>());
1104
+ sumi2 = dpct::dp4a(grid_l, *((const int *)q8 + 0), sumi2);
1105
+ sumi2 = dpct::dp4a(grid_h, *((const int *)q8 + 1), sumi2);
1106
+ q8 += 8;
1107
+ }
1108
+ const float d = (float)bq2->d * bq8_1[ib32].ds[0] * 0.25f;
1109
+ return d * ((0.5f + ls1) * sumi1 + (0.5f + ls2) * sumi2);
1110
+ #else
1111
+ assert(false);
1112
+ #endif
1113
+ }
1114
+
1115
+ static __dpct_inline__ float
1116
+ vec_dot_iq3_xxs_q8_1(const void *__restrict__ vbq,
1117
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs,
1118
+ const uint32_t *iq3xxs_grid, const uint64_t *ksigns64) {
1119
+ #if DPCT_COMPATIBILITY_TEMP >= \
1120
+ MIN_CC_DP4A // lowest compute capability for integer intrinsics
1121
+ #if QK_K == 256
1122
+ const block_iq3_xxs * bq2 = (const block_iq3_xxs *) vbq;
1123
+
1124
+ const int ib32 = iqs;
1125
+ const uint8_t * q3 = bq2->qs + 8*ib32;
1126
+ const uint16_t * gas = (const uint16_t *)(bq2->qs + QK_K/4) + 2*ib32;
1127
+ const int8_t * q8 = bq8_1[ib32].qs;
1128
+ uint32_t aux32 = gas[0] | (gas[1] << 16);
1129
+ int sumi = 0;
1130
+ for (int l = 0; l < 4; ++l) {
1131
+ const uint32_t * grid1 = iq3xxs_grid + q3[2*l+0];
1132
+ const uint32_t * grid2 = iq3xxs_grid + q3[2*l+1];
1133
+ const uint32_t * signs = (const uint32_t *)(ksigns64 + (aux32 & 127));
1134
+ const int grid_l = dpct::vectorized_binary<sycl::uchar4>(
1135
+ grid1[0] ^ signs[0], signs[0], std::minus<>());
1136
+ const int grid_h = dpct::vectorized_binary<sycl::uchar4>(
1137
+ grid2[0] ^ signs[1], signs[1], std::minus<>());
1138
+ sumi = dpct::dp4a(grid_l, *((const int *)q8 + 0), sumi);
1139
+ sumi = dpct::dp4a(grid_h, *((const int *)q8 + 1), sumi);
1140
+ q8 += 8;
1141
+ aux32 >>= 7;
1142
+ }
1143
+ const float d = (float)bq2->d * (0.5f + aux32) * bq8_1[ib32].ds[0] * 0.5f;
1144
+ return d * sumi;
1145
+ #else
1146
+ assert(false);
1147
+ return 0.f;
1148
+ #endif
1149
+ #else
1150
+ assert(false);
1151
+ return 0.f;
1152
+ #endif
1153
+ }
1154
+
1155
+ static __dpct_inline__ float
1156
+ vec_dot_iq3_s_q8_1(const void *__restrict__ vbq,
1157
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs,
1158
+ const uint32_t *iq3s_grid) {
1159
+ #if QK_K == 256
1160
+ const block_iq3_s * bq2 = (const block_iq3_s *) vbq;
1161
+
1162
+ const int ib32 = iqs;
1163
+ const uint8_t * qs = bq2->qs + 8*ib32;
1164
+ const int8_t * q8 = bq8_1[ib32].qs;
1165
+ int sumi = 0;
1166
+ for (int l = 0; l < 4; ++l) {
1167
+ const uint32_t * grid1 = iq3s_grid + (qs[2*l+0] | ((bq2->qh[ib32] << (8 - 2*l)) & 256));
1168
+ const uint32_t * grid2 = iq3s_grid + (qs[2*l+1] | ((bq2->qh[ib32] << (7 - 2*l)) & 256));
1169
+ uint32_t signs0 = dpct::vectorized_binary<sycl::uchar4>(
1170
+ ((bq2->signs[4 * ib32 + l] & 0xf) * 0x01010101) & 0x08040201,
1171
+ 0x08040201, std::equal_to<>());
1172
+ uint32_t signs1 = dpct::vectorized_binary<sycl::uchar4>(
1173
+ ((bq2->signs[4 * ib32 + l] >> 4) * 0x01010101) & 0x08040201,
1174
+ 0x08040201, std::equal_to<>());
1175
+ const int grid_l = dpct::vectorized_binary<sycl::uchar4>(
1176
+ grid1[0] ^ signs0, signs0, std::minus<>());
1177
+ const int grid_h = dpct::vectorized_binary<sycl::uchar4>(
1178
+ grid2[0] ^ signs1, signs1, std::minus<>());
1179
+ sumi = dpct::dp4a(grid_l, *((const int *)q8 + 0), sumi);
1180
+ sumi = dpct::dp4a(grid_h, *((const int *)q8 + 1), sumi);
1181
+ q8 += 8;
1182
+ }
1183
+ const float d =
1184
+ (float)bq2->d *
1185
+ (1 + 2 * ((bq2->scales[ib32 / 2] >> 4 * (ib32 % 2)) & 0xf)) *
1186
+ bq8_1[ib32].ds[0];
1187
+ return d * sumi;
1188
+ #else
1189
+ assert(false);
1190
+ #endif
1191
+ }
1192
+
1193
+ static __dpct_inline__ float
1194
+ vec_dot_iq1_s_q8_1(const void *__restrict__ vbq,
1195
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs,
1196
+ const uint32_t *iq1s_grid_gpu) {
1197
+ #if QK_K == 256
1198
+ const block_iq1_s * bq1 = (const block_iq1_s *) vbq;
1199
+
1200
+ const int ib32 = iqs;
1201
+ int sumi = 0;
1202
+ const int * q8 = (const int *)bq8_1[ib32].qs;
1203
+ for (int l = 0; l < 4; ++l) {
1204
+ const int * grid = (const int *)(iq1s_grid_gpu + (bq1->qs[4*ib32+l] | (((bq1->qh[ib32] >> 3*l) & 7) << 8)));
1205
+ int grid0 = grid[0] & 0x0f0f0f0f;
1206
+ int grid1 = (grid[0] >> 4) & 0x0f0f0f0f;
1207
+ sumi = dpct::dp4a(q8[2 * l + 1], grid1,
1208
+ dpct::dp4a(q8[2 * l + 0], grid0, sumi));
1209
+ }
1210
+
1211
+ const float delta = bq1->qh[ib32] & 0x8000 ? -1-IQ1S_DELTA : -1+IQ1S_DELTA;
1212
+ const float d1q = (float)bq1->d * (2*((bq1->qh[ib32] >> 12) & 7) + 1);
1213
+ const float d = d1q * bq8_1[ib32].ds[0];
1214
+ const float m = d1q * bq8_1[ib32].ds[1];
1215
+ return d * sumi + m * delta;
1216
+ #else
1217
+ assert(false);
1218
+ #endif
1219
+ }
1220
+
1221
+ static __dpct_inline__ float
1222
+ vec_dot_iq1_m_q8_1(const void *__restrict__ vbq,
1223
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
1224
+ #if QK_K == 256
1225
+ const block_iq1_m * bq1 = (const block_iq1_m *) vbq;
1226
+
1227
+ const int ib32 = iqs;
1228
+ int sumi[2] = {0, 0};
1229
+ float sumf[2] = {0.f, 0.f};
1230
+
1231
+ const int * q8 = (const int *)bq8_1[ib32].qs;
1232
+ for (int l = 0; l < 4; ++l) {
1233
+ const int * grid = (const int *)(iq1s_grid_gpu + (bq1->qs[4*ib32+l] | (((bq1->qh[2*ib32+l/2] >> 4*(l%2)) & 7) << 8)));
1234
+ int grid0 = grid[0] & 0x0f0f0f0f;
1235
+ int grid1 = (grid[0] >> 4) & 0x0f0f0f0f;
1236
+ sumi[l / 2] = dpct::dp4a(q8[2 * l + 1], grid1,
1237
+ dpct::dp4a(q8[2 * l + 0], grid0, sumi[l / 2]));
1238
+ const float delta = (bq1->qh[2*ib32+l/2] >> 4*(l%2)) & 0x08 ? -1-IQ1M_DELTA : -1+IQ1M_DELTA;
1239
+ const int sumy = dpct::dp4a(q8[2 * l + 1], 0x01010101,
1240
+ dpct::dp4a(q8[2 * l + 0], 0x01010101, 0));
1241
+ sumf[l/2] += delta*sumy;
1242
+ }
1243
+
1244
+ iq1m_scale_t scale;
1245
+ const uint16_t * sc = (const uint16_t *)bq1->scales;
1246
+ scale.u16 = (sc[0] >> 12) | ((sc[1] >> 8) & 0x00f0) | ((sc[2] >> 4) & 0x0f00) | (sc[3] & 0xf000);
1247
+ const float d = (float)scale.f16 * bq8_1[ib32].ds[0];
1248
+ return d * ((sumi[0] + sumf[0]) * (2*((sc[ib32/2] >> 6*(ib32%2)) & 0x7) + 1) + (sumi[1] + sumf[1]) * (2*((sc[ib32/2] >> (6*(ib32%2)+3)) & 0x7) + 1));
1249
+ #else
1250
+ assert(false);
1251
+ #endif
1252
+ }
1253
+
1254
+
1255
+ static __dpct_inline__ float
1256
+ vec_dot_iq4_nl_q8_1(const void *__restrict__ vbq,
1257
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
1258
+
1259
+ const block_iq4_nl * bq = (const block_iq4_nl *) vbq;
1260
+
1261
+ const uint16_t * q4 = (const uint16_t *)bq->qs + 2*iqs;
1262
+ const int32_t * q8 = (const int32_t *)bq8_1->qs + iqs;
1263
+
1264
+ const uint8_t * values = (const uint8_t *)kvalues_iq4nl;
1265
+
1266
+ int v1, v2;
1267
+ int sumi1 = 0, sumi2 = 0;
1268
+ for (int l = 0; l < VDR_Q4_0_Q8_1_MMVQ; ++l) {
1269
+ const uint32_t aux = q4[2*l] | (q4[2*l+1] << 16);
1270
+ get_int_from_table_16(aux, values, v1, v2);
1271
+ sumi1 = dpct::dp4a(v1, q8[l + 0], sumi1);
1272
+ sumi2 = dpct::dp4a(v2, q8[l + 4], sumi2);
1273
+ }
1274
+
1275
+ const float d = (float)bq->d * bq8_1->ds[0];
1276
+ return d * (sumi1 + sumi2);
1277
+ }
1278
+
1279
+
1280
+ static __dpct_inline__ float
1281
+ vec_dot_iq4_xs_q8_1(const void *__restrict__ vbq,
1282
+ const block_q8_1 *__restrict__ bq8_1, const int &iqs) {
1283
+
1284
+ #if QK_K == 256
1285
+ const block_iq4_xs * bq4 = (const block_iq4_xs *) vbq;
1286
+ const uint8_t * values = (const uint8_t *)kvalues_iq4nl;
1287
+
1288
+ // iqs is 0...7
1289
+ const int ib32 = iqs;
1290
+ const int32_t * q8 = (const int *)bq8_1[ib32].qs;
1291
+ const uint32_t * q4 = (const uint32_t *)bq4->qs + 4*ib32;
1292
+ const int8_t ls = ((bq4->scales_l[ib32/2] >> 4*(ib32%2)) & 0xf) | (((bq4->scales_h >> 2*ib32) & 3) << 4);
1293
+ const float d = (float)bq4->d * (ls - 32) * bq8_1[ib32].ds[0];
1294
+ int v1, v2;
1295
+ int sumi1 = 0, sumi2 = 0;
1296
+ for (int j = 0; j < 4; ++j) {
1297
+ get_int_from_table_16(q4[j], values, v1, v2);
1298
+ sumi1 = dpct::dp4a(v1, q8[j + 0], sumi1);
1299
+ sumi2 = dpct::dp4a(v2, q8[j + 4], sumi2);
1300
+ }
1301
+ return d * (sumi1 + sumi2);
1302
+ #else
1303
+ assert(false);
1304
+ #endif
1305
+ }
1306
+
1307
+ #endif // GGML_SYCL_VECDOTQ_HPP