whispercpp 1.3.1 → 1.3.2

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 (797) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +4 -3
  3. data/README.md +92 -31
  4. data/Rakefile +26 -7
  5. data/ext/.gitignore +5 -7
  6. data/ext/dependencies.rb +61 -0
  7. data/ext/extconf.rb +21 -198
  8. data/ext/options.rb +221 -0
  9. data/ext/ruby_whisper.c +159 -0
  10. data/ext/ruby_whisper.h +17 -2
  11. data/ext/ruby_whisper_context.c +641 -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 +1301 -0
  15. data/ext/ruby_whisper_segment.c +143 -0
  16. data/ext/ruby_whisper_transcribe.cpp +87 -0
  17. data/ext/ruby_whisper_vad_params.c +288 -0
  18. data/ext/sources/.dockerignore +3 -0
  19. data/ext/sources/.github/workflows/bindings-ruby.yml +21 -0
  20. data/ext/sources/CMakeGraphVizOptions.cmake +8 -0
  21. data/ext/sources/CMakeLists.txt +251 -0
  22. data/ext/sources/bindings/javascript/CMakeLists.txt +41 -0
  23. data/ext/sources/bindings/javascript/emscripten.cpp +93 -0
  24. data/ext/sources/bindings/javascript/libwhisper.worker.js +1 -0
  25. data/ext/sources/bindings/javascript/package-tmpl.json +26 -0
  26. data/ext/sources/bindings/javascript/package.json +26 -0
  27. data/ext/sources/bindings/javascript/whisper.js +19 -0
  28. data/ext/sources/build-xcframework.sh +547 -0
  29. data/ext/sources/ci/run.sh +336 -0
  30. data/ext/sources/close-issue.yml +28 -0
  31. data/ext/sources/cmake/DefaultTargetOptions.cmake +16 -0
  32. data/ext/sources/cmake/FindFFmpeg.cmake +163 -0
  33. data/ext/sources/cmake/build-info.cmake +60 -0
  34. data/ext/sources/cmake/git-vars.cmake +22 -0
  35. data/ext/sources/cmake/whisper-config.cmake.in +65 -0
  36. data/ext/sources/cmake/whisper.pc.in +10 -0
  37. data/ext/sources/examples/CMakeLists.txt +124 -0
  38. data/ext/sources/examples/addon.node/CMakeLists.txt +31 -0
  39. data/ext/sources/examples/addon.node/__test__/whisper.spec.js +37 -0
  40. data/ext/sources/examples/addon.node/addon.cpp +438 -0
  41. data/ext/sources/examples/addon.node/index.js +54 -0
  42. data/ext/sources/examples/addon.node/package.json +16 -0
  43. data/ext/sources/examples/bench/CMakeLists.txt +8 -0
  44. data/ext/sources/examples/bench/bench.cpp +175 -0
  45. data/ext/sources/examples/bench.wasm/CMakeLists.txt +49 -0
  46. data/ext/sources/examples/bench.wasm/emscripten.cpp +87 -0
  47. data/ext/sources/examples/bench.wasm/index-tmpl.html +284 -0
  48. data/ext/sources/examples/cli/CMakeLists.txt +8 -0
  49. data/ext/sources/examples/cli/cli.cpp +1294 -0
  50. data/ext/sources/examples/coi-serviceworker.js +146 -0
  51. data/ext/sources/examples/command/CMakeLists.txt +10 -0
  52. data/ext/sources/examples/command/command.cpp +776 -0
  53. data/ext/sources/examples/command/commands.txt +9 -0
  54. data/ext/sources/examples/command.wasm/CMakeLists.txt +50 -0
  55. data/ext/sources/examples/command.wasm/emscripten.cpp +327 -0
  56. data/ext/sources/examples/command.wasm/index-tmpl.html +414 -0
  57. data/ext/sources/examples/common-ggml.cpp +238 -0
  58. data/ext/sources/examples/common-ggml.h +18 -0
  59. data/ext/sources/examples/common-sdl.cpp +227 -0
  60. data/ext/sources/examples/common-sdl.h +49 -0
  61. data/ext/sources/examples/common-whisper.cpp +168 -0
  62. data/ext/sources/examples/common-whisper.h +24 -0
  63. data/ext/sources/examples/common.cpp +675 -0
  64. data/ext/sources/examples/common.h +322 -0
  65. data/ext/sources/examples/deprecation-warning/CMakeLists.txt +6 -0
  66. data/ext/sources/examples/deprecation-warning/deprecation-warning.cpp +38 -0
  67. data/ext/sources/examples/ffmpeg-transcode.cpp +368 -0
  68. data/ext/sources/examples/generate-karaoke.sh +57 -0
  69. data/ext/sources/examples/grammar-parser.cpp +423 -0
  70. data/ext/sources/examples/grammar-parser.h +29 -0
  71. data/ext/sources/examples/helpers.js +191 -0
  72. data/ext/sources/examples/json.hpp +24596 -0
  73. data/ext/sources/examples/livestream.sh +112 -0
  74. data/ext/sources/examples/lsp/CMakeLists.txt +9 -0
  75. data/ext/sources/examples/lsp/lsp.cpp +467 -0
  76. data/ext/sources/examples/lsp/whisper.vim +362 -0
  77. data/ext/sources/examples/miniaudio.h +93468 -0
  78. data/ext/sources/examples/python/test_whisper_processor.py +7 -0
  79. data/ext/sources/examples/python/whisper_processor.py +54 -0
  80. data/ext/sources/examples/quantize/CMakeLists.txt +6 -0
  81. data/ext/sources/examples/quantize/quantize.cpp +223 -0
  82. data/ext/sources/examples/server/CMakeLists.txt +12 -0
  83. data/ext/sources/examples/server/bench.js +29 -0
  84. data/ext/sources/examples/server/httplib.h +10497 -0
  85. data/ext/sources/examples/server/server.cpp +1091 -0
  86. data/ext/sources/examples/server.py +115 -0
  87. data/ext/sources/examples/stb_vorbis.c +5584 -0
  88. data/ext/sources/examples/stream/CMakeLists.txt +10 -0
  89. data/ext/sources/examples/stream/stream.cpp +429 -0
  90. data/ext/sources/examples/stream.wasm/CMakeLists.txt +49 -0
  91. data/ext/sources/examples/stream.wasm/emscripten.cpp +216 -0
  92. data/ext/sources/examples/stream.wasm/index-tmpl.html +414 -0
  93. data/ext/sources/examples/sycl/CMakeLists.txt +9 -0
  94. data/ext/sources/examples/sycl/build.sh +22 -0
  95. data/ext/sources/examples/sycl/ls-sycl-device.cpp +11 -0
  96. data/ext/sources/examples/sycl/run-whisper.sh +17 -0
  97. data/ext/sources/examples/talk-llama/CMakeLists.txt +40 -0
  98. data/ext/sources/examples/talk-llama/eleven-labs.py +80 -0
  99. data/ext/sources/examples/talk-llama/llama-adapter.cpp +388 -0
  100. data/ext/sources/examples/talk-llama/llama-adapter.h +76 -0
  101. data/ext/sources/examples/talk-llama/llama-arch.cpp +1746 -0
  102. data/ext/sources/examples/talk-llama/llama-arch.h +437 -0
  103. data/ext/sources/examples/talk-llama/llama-batch.cpp +374 -0
  104. data/ext/sources/examples/talk-llama/llama-batch.h +89 -0
  105. data/ext/sources/examples/talk-llama/llama-chat.cpp +663 -0
  106. data/ext/sources/examples/talk-llama/llama-chat.h +58 -0
  107. data/ext/sources/examples/talk-llama/llama-context.cpp +2676 -0
  108. data/ext/sources/examples/talk-llama/llama-context.h +276 -0
  109. data/ext/sources/examples/talk-llama/llama-cparams.cpp +5 -0
  110. data/ext/sources/examples/talk-llama/llama-cparams.h +41 -0
  111. data/ext/sources/examples/talk-llama/llama-grammar.cpp +1229 -0
  112. data/ext/sources/examples/talk-llama/llama-grammar.h +173 -0
  113. data/ext/sources/examples/talk-llama/llama-graph.cpp +1618 -0
  114. data/ext/sources/examples/talk-llama/llama-graph.h +640 -0
  115. data/ext/sources/examples/talk-llama/llama-hparams.cpp +95 -0
  116. data/ext/sources/examples/talk-llama/llama-hparams.h +190 -0
  117. data/ext/sources/examples/talk-llama/llama-impl.cpp +167 -0
  118. data/ext/sources/examples/talk-llama/llama-impl.h +61 -0
  119. data/ext/sources/examples/talk-llama/llama-io.cpp +15 -0
  120. data/ext/sources/examples/talk-llama/llama-io.h +35 -0
  121. data/ext/sources/examples/talk-llama/llama-kv-cache.cpp +2739 -0
  122. data/ext/sources/examples/talk-llama/llama-kv-cache.h +502 -0
  123. data/ext/sources/examples/talk-llama/llama-kv-cells.h +379 -0
  124. data/ext/sources/examples/talk-llama/llama-memory.cpp +1 -0
  125. data/ext/sources/examples/talk-llama/llama-memory.h +32 -0
  126. data/ext/sources/examples/talk-llama/llama-mmap.cpp +600 -0
  127. data/ext/sources/examples/talk-llama/llama-mmap.h +68 -0
  128. data/ext/sources/examples/talk-llama/llama-model-loader.cpp +1138 -0
  129. data/ext/sources/examples/talk-llama/llama-model-loader.h +169 -0
  130. data/ext/sources/examples/talk-llama/llama-model-saver.cpp +281 -0
  131. data/ext/sources/examples/talk-llama/llama-model-saver.h +37 -0
  132. data/ext/sources/examples/talk-llama/llama-model.cpp +13814 -0
  133. data/ext/sources/examples/talk-llama/llama-model.h +425 -0
  134. data/ext/sources/examples/talk-llama/llama-quant.cpp +966 -0
  135. data/ext/sources/examples/talk-llama/llama-quant.h +1 -0
  136. data/ext/sources/examples/talk-llama/llama-sampling.cpp +2575 -0
  137. data/ext/sources/examples/talk-llama/llama-sampling.h +32 -0
  138. data/ext/sources/examples/talk-llama/llama-vocab.cpp +3340 -0
  139. data/ext/sources/examples/talk-llama/llama-vocab.h +131 -0
  140. data/ext/sources/examples/talk-llama/llama.cpp +354 -0
  141. data/ext/sources/examples/talk-llama/llama.h +1377 -0
  142. data/ext/sources/examples/talk-llama/prompts/talk-alpaca.txt +23 -0
  143. data/ext/sources/examples/talk-llama/speak +40 -0
  144. data/ext/sources/examples/talk-llama/speak.bat +1 -0
  145. data/ext/sources/examples/talk-llama/speak.ps1 +14 -0
  146. data/ext/sources/examples/talk-llama/talk-llama.cpp +808 -0
  147. data/ext/sources/examples/talk-llama/unicode-data.cpp +7034 -0
  148. data/ext/sources/examples/talk-llama/unicode-data.h +20 -0
  149. data/ext/sources/examples/talk-llama/unicode.cpp +849 -0
  150. data/ext/sources/examples/talk-llama/unicode.h +66 -0
  151. data/ext/sources/examples/vad-speech-segments/CMakeLists.txt +8 -0
  152. data/ext/sources/examples/vad-speech-segments/speech.cpp +143 -0
  153. data/ext/sources/examples/wchess/CMakeLists.txt +10 -0
  154. data/ext/sources/examples/wchess/libwchess/CMakeLists.txt +19 -0
  155. data/ext/sources/examples/wchess/libwchess/Chessboard.cpp +803 -0
  156. data/ext/sources/examples/wchess/libwchess/Chessboard.h +33 -0
  157. data/ext/sources/examples/wchess/libwchess/WChess.cpp +193 -0
  158. data/ext/sources/examples/wchess/libwchess/WChess.h +63 -0
  159. data/ext/sources/examples/wchess/libwchess/test-chessboard.cpp +117 -0
  160. data/ext/sources/examples/wchess/wchess.cmd/CMakeLists.txt +8 -0
  161. data/ext/sources/examples/wchess/wchess.cmd/wchess.cmd.cpp +249 -0
  162. data/ext/sources/examples/whisper.wasm/CMakeLists.txt +50 -0
  163. data/ext/sources/examples/whisper.wasm/emscripten.cpp +118 -0
  164. data/ext/sources/examples/whisper.wasm/index-tmpl.html +658 -0
  165. data/ext/sources/ggml/CMakeLists.txt +390 -0
  166. data/ext/sources/ggml/cmake/BuildTypes.cmake +54 -0
  167. data/ext/sources/ggml/cmake/GitVars.cmake +22 -0
  168. data/ext/sources/ggml/cmake/common.cmake +26 -0
  169. data/ext/sources/ggml/cmake/ggml-config.cmake.in +152 -0
  170. data/ext/{ggml → sources/ggml}/include/ggml-alloc.h +1 -1
  171. data/ext/{ggml → sources/ggml}/include/ggml-backend.h +9 -7
  172. data/ext/{ggml → sources/ggml}/include/ggml-cpp.h +2 -1
  173. data/ext/{ggml → sources/ggml}/include/ggml-cpu.h +9 -1
  174. data/ext/{ggml → sources/ggml}/include/ggml-metal.h +1 -1
  175. data/ext/{ggml → sources/ggml}/include/ggml-opt.h +49 -28
  176. data/ext/{ggml → sources/ggml}/include/ggml-rpc.h +6 -1
  177. data/ext/{ggml → sources/ggml}/include/ggml-vulkan.h +0 -2
  178. data/ext/{ggml → sources/ggml}/include/ggml.h +182 -265
  179. data/ext/sources/ggml/include/gguf.h +202 -0
  180. data/ext/sources/ggml/src/CMakeLists.txt +346 -0
  181. data/ext/{ggml → sources/ggml}/src/ggml-alloc.c +34 -29
  182. data/ext/sources/ggml/src/ggml-amx/CMakeLists.txt +107 -0
  183. data/ext/{ggml → sources/ggml}/src/ggml-backend-impl.h +1 -2
  184. data/ext/{ggml → sources/ggml}/src/ggml-backend-reg.cpp +87 -53
  185. data/ext/{ggml → sources/ggml}/src/ggml-backend.cpp +26 -14
  186. data/ext/sources/ggml/src/ggml-blas/CMakeLists.txt +87 -0
  187. data/ext/sources/ggml/src/ggml-cann/CMakeLists.txt +74 -0
  188. data/ext/sources/ggml/src/ggml-cann/Doxyfile +2579 -0
  189. data/ext/{ggml → sources/ggml}/src/ggml-cann/acl_tensor.cpp +10 -4
  190. data/ext/{ggml → sources/ggml}/src/ggml-cann/acl_tensor.h +5 -5
  191. data/ext/{ggml → sources/ggml}/src/ggml-cann/aclnn_ops.cpp +1272 -1506
  192. data/ext/sources/ggml/src/ggml-cann/aclnn_ops.h +1125 -0
  193. data/ext/{ggml → sources/ggml}/src/ggml-cann/common.h +135 -1
  194. data/ext/{ggml → sources/ggml}/src/ggml-cann/ggml-cann.cpp +564 -146
  195. data/ext/sources/ggml/src/ggml-cann/kernels/CMakeLists.txt +30 -0
  196. data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/dup.cpp +3 -5
  197. data/ext/{ggml → sources/ggml}/src/ggml-common.h +12 -8
  198. data/ext/sources/ggml/src/ggml-cpu/CMakeLists.txt +504 -0
  199. data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/amx.cpp +2 -1
  200. data/ext/sources/ggml/src/ggml-cpu/binary-ops.cpp +158 -0
  201. data/ext/sources/ggml/src/ggml-cpu/binary-ops.h +16 -0
  202. data/ext/sources/ggml/src/ggml-cpu/cmake/FindSIMD.cmake +100 -0
  203. data/ext/sources/ggml/src/ggml-cpu/common.h +72 -0
  204. data/ext/{ggml → sources/ggml}/src/ggml-cpu/cpu-feats-x86.cpp +5 -1
  205. data/ext/sources/ggml/src/ggml-cpu/ggml-cpu-aarch64.cpp +6431 -0
  206. data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-impl.h +163 -41
  207. data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-quants.c +4029 -1117
  208. data/ext/sources/ggml/src/ggml-cpu/ggml-cpu.c +3510 -0
  209. data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu.cpp +67 -18
  210. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kernels.cpp +337 -0
  211. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kernels.h +95 -0
  212. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp +482 -0
  213. data/ext/sources/ggml/src/ggml-cpu/kleidiai/kleidiai.h +17 -0
  214. data/ext/sources/ggml/src/ggml-cpu/llamafile/sgemm.cpp +3544 -0
  215. data/ext/sources/ggml/src/ggml-cpu/llamafile/sgemm.h +14 -0
  216. data/ext/sources/ggml/src/ggml-cpu/ops.cpp +8903 -0
  217. data/ext/sources/ggml/src/ggml-cpu/ops.h +110 -0
  218. data/ext/sources/ggml/src/ggml-cpu/simd-mappings.h +892 -0
  219. data/ext/sources/ggml/src/ggml-cpu/unary-ops.cpp +186 -0
  220. data/ext/sources/ggml/src/ggml-cpu/unary-ops.h +28 -0
  221. data/ext/sources/ggml/src/ggml-cpu/vec.cpp +252 -0
  222. data/ext/sources/ggml/src/ggml-cpu/vec.h +818 -0
  223. data/ext/sources/ggml/src/ggml-cuda/CMakeLists.txt +184 -0
  224. data/ext/sources/ggml/src/ggml-cuda/acc.cu +61 -0
  225. data/ext/sources/ggml/src/ggml-cuda/acc.cuh +5 -0
  226. data/ext/sources/ggml/src/ggml-cuda/arange.cu +34 -0
  227. data/ext/sources/ggml/src/ggml-cuda/arange.cuh +5 -0
  228. data/ext/sources/ggml/src/ggml-cuda/argmax.cu +91 -0
  229. data/ext/sources/ggml/src/ggml-cuda/argmax.cuh +3 -0
  230. data/ext/sources/ggml/src/ggml-cuda/argsort.cu +104 -0
  231. data/ext/sources/ggml/src/ggml-cuda/argsort.cuh +3 -0
  232. data/ext/sources/ggml/src/ggml-cuda/binbcast.cu +363 -0
  233. data/ext/sources/ggml/src/ggml-cuda/binbcast.cuh +9 -0
  234. data/ext/sources/ggml/src/ggml-cuda/clamp.cu +45 -0
  235. data/ext/sources/ggml/src/ggml-cuda/clamp.cuh +5 -0
  236. data/ext/sources/ggml/src/ggml-cuda/common.cuh +828 -0
  237. data/ext/sources/ggml/src/ggml-cuda/concat.cu +221 -0
  238. data/ext/sources/ggml/src/ggml-cuda/concat.cuh +5 -0
  239. data/ext/sources/ggml/src/ggml-cuda/conv-transpose-1d.cu +89 -0
  240. data/ext/sources/ggml/src/ggml-cuda/conv-transpose-1d.cuh +5 -0
  241. data/ext/sources/ggml/src/ggml-cuda/convert.cu +730 -0
  242. data/ext/sources/ggml/src/ggml-cuda/convert.cuh +26 -0
  243. data/ext/sources/ggml/src/ggml-cuda/count-equal.cu +64 -0
  244. data/ext/sources/ggml/src/ggml-cuda/count-equal.cuh +5 -0
  245. data/ext/sources/ggml/src/ggml-cuda/cp-async.cuh +57 -0
  246. data/ext/sources/ggml/src/ggml-cuda/cpy.cu +705 -0
  247. data/ext/sources/ggml/src/ggml-cuda/cpy.cuh +11 -0
  248. data/ext/sources/ggml/src/ggml-cuda/cross-entropy-loss.cu +189 -0
  249. data/ext/sources/ggml/src/ggml-cuda/cross-entropy-loss.cuh +7 -0
  250. data/ext/sources/ggml/src/ggml-cuda/dequantize.cuh +103 -0
  251. data/ext/sources/ggml/src/ggml-cuda/diagmask.cu +40 -0
  252. data/ext/sources/ggml/src/ggml-cuda/diagmask.cuh +5 -0
  253. data/ext/sources/ggml/src/ggml-cuda/fattn-common.cuh +881 -0
  254. data/ext/sources/ggml/src/ggml-cuda/fattn-mma-f16.cuh +1471 -0
  255. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f16.cu +357 -0
  256. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f16.cuh +3 -0
  257. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f32.cu +365 -0
  258. data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f32.cuh +3 -0
  259. data/ext/sources/ggml/src/ggml-cuda/fattn-vec-f16.cuh +482 -0
  260. data/ext/sources/ggml/src/ggml-cuda/fattn-vec-f32.cuh +472 -0
  261. data/ext/sources/ggml/src/ggml-cuda/fattn-wmma-f16.cu +634 -0
  262. data/ext/sources/ggml/src/ggml-cuda/fattn-wmma-f16.cuh +3 -0
  263. data/ext/sources/ggml/src/ggml-cuda/fattn.cu +346 -0
  264. data/ext/sources/ggml/src/ggml-cuda/fattn.cuh +3 -0
  265. data/ext/sources/ggml/src/ggml-cuda/getrows.cu +275 -0
  266. data/ext/sources/ggml/src/ggml-cuda/getrows.cuh +15 -0
  267. data/ext/sources/ggml/src/ggml-cuda/ggml-cuda.cu +3505 -0
  268. data/ext/sources/ggml/src/ggml-cuda/gla.cu +93 -0
  269. data/ext/sources/ggml/src/ggml-cuda/gla.cuh +3 -0
  270. data/ext/sources/ggml/src/ggml-cuda/im2col.cu +103 -0
  271. data/ext/sources/ggml/src/ggml-cuda/im2col.cuh +5 -0
  272. data/ext/sources/ggml/src/ggml-cuda/mma.cuh +396 -0
  273. data/ext/sources/ggml/src/ggml-cuda/mmq.cu +324 -0
  274. data/ext/sources/ggml/src/ggml-cuda/mmq.cuh +3217 -0
  275. data/ext/sources/ggml/src/ggml-cuda/mmv.cu +336 -0
  276. data/ext/sources/ggml/src/ggml-cuda/mmv.cuh +12 -0
  277. data/ext/sources/ggml/src/ggml-cuda/mmvq.cu +595 -0
  278. data/ext/sources/ggml/src/ggml-cuda/mmvq.cuh +12 -0
  279. data/ext/sources/ggml/src/ggml-cuda/norm.cu +458 -0
  280. data/ext/sources/ggml/src/ggml-cuda/norm.cuh +11 -0
  281. data/ext/sources/ggml/src/ggml-cuda/opt-step-adamw.cu +78 -0
  282. data/ext/sources/ggml/src/ggml-cuda/opt-step-adamw.cuh +5 -0
  283. data/ext/sources/ggml/src/ggml-cuda/out-prod.cu +68 -0
  284. data/ext/sources/ggml/src/ggml-cuda/out-prod.cuh +3 -0
  285. data/ext/sources/ggml/src/ggml-cuda/pad.cu +49 -0
  286. data/ext/sources/ggml/src/ggml-cuda/pad.cuh +5 -0
  287. data/ext/sources/ggml/src/ggml-cuda/pool2d.cu +94 -0
  288. data/ext/sources/ggml/src/ggml-cuda/pool2d.cuh +5 -0
  289. data/ext/sources/ggml/src/ggml-cuda/quantize.cu +190 -0
  290. data/ext/sources/ggml/src/ggml-cuda/quantize.cuh +27 -0
  291. data/ext/sources/ggml/src/ggml-cuda/rope.cu +456 -0
  292. data/ext/sources/ggml/src/ggml-cuda/rope.cuh +7 -0
  293. data/ext/sources/ggml/src/ggml-cuda/scale.cu +31 -0
  294. data/ext/sources/ggml/src/ggml-cuda/scale.cuh +5 -0
  295. data/ext/sources/ggml/src/ggml-cuda/softmax.cu +283 -0
  296. data/ext/sources/ggml/src/ggml-cuda/softmax.cuh +7 -0
  297. data/ext/sources/ggml/src/ggml-cuda/ssm-conv.cu +148 -0
  298. data/ext/sources/ggml/src/ggml-cuda/ssm-conv.cuh +3 -0
  299. data/ext/sources/ggml/src/ggml-cuda/ssm-scan.cu +153 -0
  300. data/ext/sources/ggml/src/ggml-cuda/ssm-scan.cuh +3 -0
  301. data/ext/sources/ggml/src/ggml-cuda/sum.cu +45 -0
  302. data/ext/sources/ggml/src/ggml-cuda/sum.cuh +5 -0
  303. data/ext/sources/ggml/src/ggml-cuda/sumrows.cu +39 -0
  304. data/ext/sources/ggml/src/ggml-cuda/sumrows.cuh +5 -0
  305. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_1-ncols2_16.cu +5 -0
  306. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_1-ncols2_8.cu +10 -0
  307. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_1.cu +10 -0
  308. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_2.cu +10 -0
  309. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_4.cu +10 -0
  310. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_16.cu +5 -0
  311. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_4.cu +10 -0
  312. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_8.cu +10 -0
  313. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_32-ncols2_1.cu +10 -0
  314. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_32-ncols2_2.cu +10 -0
  315. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_16.cu +5 -0
  316. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_2.cu +10 -0
  317. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_4.cu +10 -0
  318. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_8.cu +10 -0
  319. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_64-ncols2_1.cu +10 -0
  320. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_1.cu +10 -0
  321. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_2.cu +10 -0
  322. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_4.cu +10 -0
  323. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_8.cu +10 -0
  324. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-f16.cu +5 -0
  325. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q4_0.cu +5 -0
  326. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q4_1.cu +5 -0
  327. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q5_0.cu +5 -0
  328. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q5_1.cu +5 -0
  329. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q8_0.cu +5 -0
  330. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-f16.cu +5 -0
  331. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q4_0.cu +5 -0
  332. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q4_1.cu +5 -0
  333. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q5_0.cu +5 -0
  334. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q5_1.cu +5 -0
  335. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q8_0.cu +5 -0
  336. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-f16.cu +5 -0
  337. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q4_0.cu +5 -0
  338. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q4_1.cu +5 -0
  339. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q5_0.cu +5 -0
  340. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q5_1.cu +5 -0
  341. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q8_0.cu +5 -0
  342. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-f16.cu +5 -0
  343. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q4_0.cu +5 -0
  344. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q4_1.cu +5 -0
  345. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q5_0.cu +5 -0
  346. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q5_1.cu +5 -0
  347. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q8_0.cu +5 -0
  348. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-f16.cu +5 -0
  349. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q4_0.cu +5 -0
  350. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q4_1.cu +5 -0
  351. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q5_0.cu +5 -0
  352. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q5_1.cu +5 -0
  353. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q8_0.cu +5 -0
  354. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-f16.cu +5 -0
  355. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q4_0.cu +5 -0
  356. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q4_1.cu +5 -0
  357. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q5_0.cu +5 -0
  358. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q5_1.cu +5 -0
  359. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q8_0.cu +5 -0
  360. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs256-f16-f16.cu +5 -0
  361. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-f16.cu +5 -0
  362. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q4_0.cu +5 -0
  363. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q4_1.cu +5 -0
  364. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q5_0.cu +5 -0
  365. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q5_1.cu +5 -0
  366. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q8_0.cu +5 -0
  367. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-f16.cu +5 -0
  368. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q4_0.cu +5 -0
  369. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q4_1.cu +5 -0
  370. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q5_0.cu +5 -0
  371. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q5_1.cu +5 -0
  372. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q8_0.cu +5 -0
  373. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-f16.cu +5 -0
  374. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q4_0.cu +5 -0
  375. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q4_1.cu +5 -0
  376. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q5_0.cu +5 -0
  377. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q5_1.cu +5 -0
  378. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q8_0.cu +5 -0
  379. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-f16.cu +5 -0
  380. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q4_0.cu +5 -0
  381. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q4_1.cu +5 -0
  382. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q5_0.cu +5 -0
  383. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q5_1.cu +5 -0
  384. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q8_0.cu +5 -0
  385. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-f16.cu +5 -0
  386. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q4_0.cu +5 -0
  387. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q4_1.cu +5 -0
  388. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q5_0.cu +5 -0
  389. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q5_1.cu +5 -0
  390. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q8_0.cu +5 -0
  391. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-f16.cu +5 -0
  392. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q4_0.cu +5 -0
  393. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q4_1.cu +5 -0
  394. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q5_0.cu +5 -0
  395. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q5_1.cu +5 -0
  396. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q8_0.cu +5 -0
  397. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-f16.cu +5 -0
  398. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q4_0.cu +5 -0
  399. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q4_1.cu +5 -0
  400. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q5_0.cu +5 -0
  401. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q5_1.cu +5 -0
  402. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q8_0.cu +5 -0
  403. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs256-f16-f16.cu +5 -0
  404. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-f16.cu +5 -0
  405. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q4_0.cu +5 -0
  406. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q4_1.cu +5 -0
  407. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q5_0.cu +5 -0
  408. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q5_1.cu +5 -0
  409. data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q8_0.cu +5 -0
  410. data/ext/sources/ggml/src/ggml-cuda/template-instances/generate_cu_files.py +78 -0
  411. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq1_s.cu +5 -0
  412. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_s.cu +5 -0
  413. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_xs.cu +5 -0
  414. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_xxs.cu +5 -0
  415. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq3_s.cu +5 -0
  416. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq3_xxs.cu +5 -0
  417. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq4_nl.cu +5 -0
  418. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq4_xs.cu +5 -0
  419. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q2_k.cu +5 -0
  420. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q3_k.cu +5 -0
  421. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_0.cu +5 -0
  422. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_1.cu +5 -0
  423. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_k.cu +5 -0
  424. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_0.cu +5 -0
  425. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_1.cu +5 -0
  426. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_k.cu +5 -0
  427. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q6_k.cu +5 -0
  428. data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q8_0.cu +5 -0
  429. data/ext/sources/ggml/src/ggml-cuda/tsembd.cu +47 -0
  430. data/ext/sources/ggml/src/ggml-cuda/tsembd.cuh +5 -0
  431. data/ext/sources/ggml/src/ggml-cuda/unary.cu +289 -0
  432. data/ext/sources/ggml/src/ggml-cuda/unary.cuh +59 -0
  433. data/ext/sources/ggml/src/ggml-cuda/upscale.cu +51 -0
  434. data/ext/sources/ggml/src/ggml-cuda/upscale.cuh +5 -0
  435. data/ext/sources/ggml/src/ggml-cuda/vecdotq.cuh +1135 -0
  436. data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/cuda.h +1 -0
  437. data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/hip.h +57 -0
  438. data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/musa.h +7 -1
  439. data/ext/sources/ggml/src/ggml-cuda/wkv.cu +199 -0
  440. data/ext/sources/ggml/src/ggml-cuda/wkv.cuh +7 -0
  441. data/ext/sources/ggml/src/ggml-hip/CMakeLists.txt +131 -0
  442. data/ext/{ggml → sources/ggml}/src/ggml-impl.h +64 -19
  443. data/ext/sources/ggml/src/ggml-kompute/CMakeLists.txt +166 -0
  444. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/common.comp +112 -0
  445. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_add.comp +58 -0
  446. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_addrow.comp +25 -0
  447. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f16_f16.comp +52 -0
  448. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f16_f32.comp +52 -0
  449. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f32_f16.comp +52 -0
  450. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f32_f32.comp +52 -0
  451. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_diagmask.comp +30 -0
  452. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_gelu.comp +22 -0
  453. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows.comp +17 -0
  454. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_f16.comp +31 -0
  455. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_f32.comp +31 -0
  456. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q4_0.comp +38 -0
  457. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q4_1.comp +39 -0
  458. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q6_k.comp +44 -0
  459. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul.comp +52 -0
  460. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_f16.comp +69 -0
  461. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_mat_f32.comp +51 -0
  462. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_0.comp +33 -0
  463. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_1.comp +35 -0
  464. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_k.comp +140 -0
  465. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q6_k.comp +106 -0
  466. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q8_0.comp +73 -0
  467. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mv_q_n.comp +52 -0
  468. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mv_q_n_pre.comp +28 -0
  469. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_norm.comp +84 -0
  470. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_relu.comp +21 -0
  471. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rmsnorm.comp +53 -0
  472. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_neox_f16.comp +52 -0
  473. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_neox_f32.comp +52 -0
  474. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_norm_f16.comp +52 -0
  475. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_norm_f32.comp +52 -0
  476. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_scale.comp +19 -0
  477. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_scale_8.comp +23 -0
  478. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_silu.comp +22 -0
  479. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_softmax.comp +72 -0
  480. data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/rope_common.comp +71 -0
  481. data/ext/sources/ggml/src/ggml-metal/CMakeLists.txt +120 -0
  482. data/ext/sources/ggml/src/ggml-metal/ggml-metal-impl.h +622 -0
  483. data/ext/{ggml → sources/ggml}/src/ggml-metal/ggml-metal.m +2178 -1064
  484. data/ext/{ggml → sources/ggml}/src/ggml-metal/ggml-metal.metal +1575 -1218
  485. data/ext/sources/ggml/src/ggml-musa/CMakeLists.txt +113 -0
  486. data/ext/sources/ggml/src/ggml-musa/mudnn.cu +112 -0
  487. data/ext/sources/ggml/src/ggml-musa/mudnn.cuh +12 -0
  488. data/ext/sources/ggml/src/ggml-opencl/CMakeLists.txt +96 -0
  489. data/ext/sources/ggml/src/ggml-opencl/ggml-opencl.cpp +5124 -0
  490. data/ext/sources/ggml/src/ggml-opencl/kernels/add.cl +83 -0
  491. data/ext/sources/ggml/src/ggml-opencl/kernels/clamp.cl +20 -0
  492. data/ext/sources/ggml/src/ggml-opencl/kernels/cpy.cl +184 -0
  493. data/ext/sources/ggml/src/ggml-opencl/kernels/cvt.cl +118 -0
  494. data/ext/sources/ggml/src/ggml-opencl/kernels/diag_mask_inf.cl +58 -0
  495. data/ext/sources/ggml/src/ggml-opencl/kernels/embed_kernel.py +26 -0
  496. data/ext/sources/ggml/src/ggml-opencl/kernels/gelu.cl +62 -0
  497. data/ext/sources/ggml/src/ggml-opencl/kernels/gemv_noshuffle.cl +268 -0
  498. data/ext/sources/ggml/src/ggml-opencl/kernels/gemv_noshuffle_general.cl +274 -0
  499. data/ext/sources/ggml/src/ggml-opencl/kernels/get_rows.cl +163 -0
  500. data/ext/sources/ggml/src/ggml-opencl/kernels/im2col_f16.cl +57 -0
  501. data/ext/sources/ggml/src/ggml-opencl/kernels/im2col_f32.cl +57 -0
  502. data/ext/sources/ggml/src/ggml-opencl/kernels/mul.cl +79 -0
  503. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mat_Ab_Bi_8x4.cl +139 -0
  504. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f16.cl +118 -0
  505. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32.cl +118 -0
  506. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32_1row.cl +94 -0
  507. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32_l4.cl +84 -0
  508. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f32_f32.cl +118 -0
  509. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32.cl +192 -0
  510. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_1d_16x_flat.cl +307 -0
  511. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_1d_8x_flat.cl +265 -0
  512. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_8x_flat.cl +272 -0
  513. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_v.cl +254 -0
  514. data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q6_k.cl +190 -0
  515. data/ext/sources/ggml/src/ggml-opencl/kernels/norm.cl +81 -0
  516. data/ext/sources/ggml/src/ggml-opencl/kernels/relu.cl +16 -0
  517. data/ext/sources/ggml/src/ggml-opencl/kernels/rms_norm.cl +96 -0
  518. data/ext/sources/ggml/src/ggml-opencl/kernels/rope.cl +721 -0
  519. data/ext/sources/ggml/src/ggml-opencl/kernels/scale.cl +16 -0
  520. data/ext/sources/ggml/src/ggml-opencl/kernels/silu.cl +30 -0
  521. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_4_f16.cl +87 -0
  522. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_4_f32.cl +87 -0
  523. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_f16.cl +86 -0
  524. data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_f32.cl +86 -0
  525. data/ext/sources/ggml/src/ggml-opencl/kernels/transpose.cl +84 -0
  526. data/ext/{ggml → sources/ggml}/src/ggml-opt.cpp +373 -190
  527. data/ext/{ggml → sources/ggml}/src/ggml-quants.c +114 -120
  528. data/ext/sources/ggml/src/ggml-rpc/CMakeLists.txt +9 -0
  529. data/ext/{ggml → sources/ggml}/src/ggml-rpc/ggml-rpc.cpp +480 -73
  530. data/ext/sources/ggml/src/ggml-sycl/CMakeLists.txt +189 -0
  531. data/ext/sources/ggml/src/ggml-sycl/backend.hpp +37 -0
  532. data/ext/sources/ggml/src/ggml-sycl/binbcast.cpp +345 -0
  533. data/ext/sources/ggml/src/ggml-sycl/binbcast.hpp +39 -0
  534. data/ext/{ggml → sources/ggml}/src/ggml-sycl/common.cpp +20 -32
  535. data/ext/sources/ggml/src/ggml-sycl/common.hpp +589 -0
  536. data/ext/{ggml → sources/ggml}/src/ggml-sycl/concat.cpp +32 -33
  537. data/ext/sources/ggml/src/ggml-sycl/concat.hpp +20 -0
  538. data/ext/{ggml → sources/ggml}/src/ggml-sycl/conv.cpp +4 -2
  539. data/ext/sources/ggml/src/ggml-sycl/conv.hpp +20 -0
  540. data/ext/{ggml → sources/ggml}/src/ggml-sycl/convert.cpp +104 -28
  541. data/ext/sources/ggml/src/ggml-sycl/convert.hpp +34 -0
  542. data/ext/sources/ggml/src/ggml-sycl/cpy.cpp +700 -0
  543. data/ext/sources/ggml/src/ggml-sycl/cpy.hpp +11 -0
  544. data/ext/sources/ggml/src/ggml-sycl/dequantize.hpp +791 -0
  545. data/ext/{ggml → sources/ggml}/src/ggml-sycl/dmmv.cpp +156 -17
  546. data/ext/sources/ggml/src/ggml-sycl/dmmv.hpp +27 -0
  547. data/ext/sources/ggml/src/ggml-sycl/dpct/helper.hpp +2957 -0
  548. data/ext/sources/ggml/src/ggml-sycl/element_wise.cpp +1511 -0
  549. data/ext/sources/ggml/src/ggml-sycl/element_wise.hpp +75 -0
  550. data/ext/sources/ggml/src/ggml-sycl/gemm.hpp +99 -0
  551. data/ext/sources/ggml/src/ggml-sycl/getrows.cpp +309 -0
  552. data/ext/sources/ggml/src/ggml-sycl/getrows.hpp +20 -0
  553. data/ext/{ggml → sources/ggml}/src/ggml-sycl/ggml-sycl.cpp +1004 -1240
  554. data/ext/sources/ggml/src/ggml-sycl/gla.cpp +106 -0
  555. data/ext/sources/ggml/src/ggml-sycl/gla.hpp +8 -0
  556. data/ext/sources/ggml/src/ggml-sycl/im2col.cpp +136 -0
  557. data/ext/sources/ggml/src/ggml-sycl/im2col.hpp +21 -0
  558. data/ext/{ggml → sources/ggml}/src/ggml-sycl/mmq.cpp +0 -1
  559. data/ext/sources/ggml/src/ggml-sycl/mmq.hpp +33 -0
  560. data/ext/{ggml → sources/ggml}/src/ggml-sycl/mmvq.cpp +261 -166
  561. data/ext/sources/ggml/src/ggml-sycl/mmvq.hpp +27 -0
  562. data/ext/{ggml → sources/ggml}/src/ggml-sycl/norm.cpp +204 -81
  563. data/ext/sources/ggml/src/ggml-sycl/norm.hpp +26 -0
  564. data/ext/{ggml → sources/ggml}/src/ggml-sycl/outprod.cpp +8 -17
  565. data/ext/sources/ggml/src/ggml-sycl/outprod.hpp +10 -0
  566. data/ext/sources/ggml/src/ggml-sycl/presets.hpp +74 -0
  567. data/ext/sources/ggml/src/ggml-sycl/quants.hpp +83 -0
  568. data/ext/sources/ggml/src/ggml-sycl/rope.cpp +361 -0
  569. data/ext/sources/ggml/src/ggml-sycl/rope.hpp +20 -0
  570. data/ext/{ggml → sources/ggml}/src/ggml-sycl/softmax.cpp +35 -25
  571. data/ext/sources/ggml/src/ggml-sycl/softmax.hpp +20 -0
  572. data/ext/sources/ggml/src/ggml-sycl/sycl_hw.cpp +13 -0
  573. data/ext/sources/ggml/src/ggml-sycl/sycl_hw.hpp +23 -0
  574. data/ext/{ggml → sources/ggml}/src/ggml-sycl/tsembd.cpp +3 -3
  575. data/ext/sources/ggml/src/ggml-sycl/tsembd.hpp +20 -0
  576. data/ext/sources/ggml/src/ggml-sycl/vecdotq.hpp +1215 -0
  577. data/ext/sources/ggml/src/ggml-sycl/wkv.cpp +293 -0
  578. data/ext/sources/ggml/src/ggml-sycl/wkv.hpp +10 -0
  579. data/ext/sources/ggml/src/ggml-vulkan/CMakeLists.txt +196 -0
  580. data/ext/sources/ggml/src/ggml-vulkan/cmake/host-toolchain.cmake.in +15 -0
  581. data/ext/{ggml → sources/ggml}/src/ggml-vulkan/ggml-vulkan.cpp +3130 -1087
  582. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/CMakeLists.txt +39 -0
  583. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/acc.comp +29 -0
  584. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/add.comp +29 -0
  585. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/argmax.comp +51 -0
  586. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/argsort.comp +69 -0
  587. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/clamp.comp +17 -0
  588. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/concat.comp +41 -0
  589. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/contig_copy.comp +49 -0
  590. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/conv2d_dw.comp +105 -0
  591. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy.comp +23 -0
  592. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy_from_quant.comp +51 -0
  593. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy_to_quant.comp +242 -0
  594. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/cos.comp +17 -0
  595. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/count_equal.comp +31 -0
  596. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_f32.comp +20 -0
  597. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_funcs.comp +462 -0
  598. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_funcs_cm2.comp +699 -0
  599. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_head.comp +13 -0
  600. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq1_m.comp +42 -0
  601. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq1_s.comp +35 -0
  602. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_s.comp +44 -0
  603. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_xs.comp +43 -0
  604. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_xxs.comp +48 -0
  605. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq3_s.comp +39 -0
  606. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq3_xxs.comp +49 -0
  607. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq4_nl.comp +32 -0
  608. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq4_xs.comp +34 -0
  609. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q2_k.comp +34 -0
  610. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q3_k.comp +42 -0
  611. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_0.comp +30 -0
  612. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_1.comp +32 -0
  613. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_k.comp +68 -0
  614. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_0.comp +34 -0
  615. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_1.comp +35 -0
  616. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_k.comp +70 -0
  617. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q6_k.comp +33 -0
  618. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q8_0.comp +31 -0
  619. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/diag_mask_inf.comp +34 -0
  620. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/div.comp +27 -0
  621. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn.comp +337 -0
  622. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_base.comp +162 -0
  623. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_cm1.comp +360 -0
  624. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_cm2.comp +267 -0
  625. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_split_k_reduce.comp +59 -0
  626. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/gelu.comp +25 -0
  627. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/gelu_quick.comp +23 -0
  628. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_binary_head.comp +64 -0
  629. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_head.comp +9 -0
  630. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_unary_head.comp +76 -0
  631. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/get_rows.comp +33 -0
  632. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/get_rows_quant.comp +41 -0
  633. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/group_norm.comp +66 -0
  634. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/im2col.comp +100 -0
  635. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/l2_norm.comp +41 -0
  636. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/leaky_relu.comp +22 -0
  637. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul.comp +27 -0
  638. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_split_k_reduce.comp +48 -0
  639. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec.comp +169 -0
  640. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_base.comp +118 -0
  641. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq1_m.comp +82 -0
  642. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq1_s.comp +79 -0
  643. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_s.comp +90 -0
  644. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_xs.comp +87 -0
  645. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_xxs.comp +87 -0
  646. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq3_s.comp +90 -0
  647. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq3_xxs.comp +88 -0
  648. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_nc.comp +118 -0
  649. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_p021.comp +154 -0
  650. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q2_k.comp +130 -0
  651. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q3_k.comp +132 -0
  652. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q4_k.comp +136 -0
  653. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q5_k.comp +167 -0
  654. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q6_k.comp +130 -0
  655. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mm.comp +868 -0
  656. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mm_cm2.comp +441 -0
  657. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq.comp +442 -0
  658. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq_funcs.comp +99 -0
  659. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/norm.comp +44 -0
  660. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/opt_step_adamw.comp +42 -0
  661. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/pad.comp +28 -0
  662. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/pool2d.comp +74 -0
  663. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/quantize_q8_1.comp +77 -0
  664. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp +21 -0
  665. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/repeat.comp +26 -0
  666. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/repeat_back.comp +37 -0
  667. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rms_norm.comp +52 -0
  668. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rms_norm_back.comp +55 -0
  669. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_head.comp +58 -0
  670. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_multi.comp +60 -0
  671. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_neox.comp +43 -0
  672. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_norm.comp +43 -0
  673. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_vision.comp +47 -0
  674. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/scale.comp +24 -0
  675. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sigmoid.comp +20 -0
  676. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/silu.comp +22 -0
  677. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/silu_back.comp +26 -0
  678. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sin.comp +17 -0
  679. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp +173 -0
  680. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/soft_max_back.comp +50 -0
  681. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/square.comp +17 -0
  682. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sub.comp +29 -0
  683. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sum_rows.comp +37 -0
  684. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp +20 -0
  685. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_bfloat16_support.comp +7 -0
  686. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_coopmat2_support.comp +7 -0
  687. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_coopmat_support.comp +7 -0
  688. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_integer_dot_support.comp +7 -0
  689. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/timestep_embedding.comp +41 -0
  690. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/types.comp +1373 -0
  691. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/upscale.comp +36 -0
  692. data/ext/{ggml → sources/ggml}/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp +193 -35
  693. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/wkv6.comp +87 -0
  694. data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/wkv7.comp +91 -0
  695. data/ext/{ggml → sources/ggml}/src/ggml.c +676 -1820
  696. data/ext/sources/ggml/src/gguf.cpp +1330 -0
  697. data/ext/{include → sources/include}/whisper.h +68 -2
  698. data/ext/sources/src/CMakeLists.txt +143 -0
  699. data/ext/{src → sources/src}/coreml/whisper-decoder-impl.h +27 -15
  700. data/ext/{src → sources/src}/coreml/whisper-decoder-impl.m +35 -10
  701. data/ext/{src → sources/src}/coreml/whisper-encoder-impl.h +21 -9
  702. data/ext/{src → sources/src}/coreml/whisper-encoder-impl.m +28 -3
  703. data/ext/sources/src/coreml/whisper-encoder.mm +73 -0
  704. data/ext/sources/src/whisper-arch.h +197 -0
  705. data/ext/{src → sources/src}/whisper.cpp +1905 -374
  706. data/ext/sources/tests/CMakeLists.txt +105 -0
  707. data/ext/sources/tests/earnings21/eval.mk +58 -0
  708. data/ext/sources/tests/earnings21/eval.py +68 -0
  709. data/ext/sources/tests/earnings21/normalizers/__init__.py +2 -0
  710. data/ext/sources/tests/earnings21/normalizers/basic.py +80 -0
  711. data/ext/sources/tests/earnings21/normalizers/english.json +1741 -0
  712. data/ext/sources/tests/earnings21/normalizers/english.py +550 -0
  713. data/ext/sources/tests/earnings21/requirements.txt +6 -0
  714. data/ext/sources/tests/en-0-ref.txt +1 -0
  715. data/ext/sources/tests/en-1-ref.txt +1 -0
  716. data/ext/sources/tests/en-2-ref.txt +1 -0
  717. data/ext/sources/tests/es-0-ref.txt +1 -0
  718. data/ext/sources/tests/librispeech/eval.mk +39 -0
  719. data/ext/sources/tests/librispeech/eval.py +47 -0
  720. data/ext/sources/tests/librispeech/normalizers/__init__.py +2 -0
  721. data/ext/sources/tests/librispeech/normalizers/basic.py +80 -0
  722. data/ext/sources/tests/librispeech/normalizers/english.json +1741 -0
  723. data/ext/sources/tests/librispeech/normalizers/english.py +550 -0
  724. data/ext/sources/tests/librispeech/requirements.txt +6 -0
  725. data/ext/sources/tests/run-tests.sh +130 -0
  726. data/ext/sources/tests/test-c.c +3 -0
  727. data/ext/sources/tests/test-vad-full.cpp +54 -0
  728. data/ext/sources/tests/test-vad.cpp +83 -0
  729. data/ext/sources/tests/test-whisper.js +58 -0
  730. data/extsources.rb +33 -5
  731. data/lib/whisper/model/uri.rb +149 -128
  732. data/sig/whisper.rbs +480 -0
  733. data/tests/helper.rb +28 -0
  734. data/tests/test_callback.rb +45 -3
  735. data/tests/test_error.rb +2 -2
  736. data/tests/test_model.rb +38 -0
  737. data/tests/test_package.rb +18 -3
  738. data/tests/test_params.rb +145 -8
  739. data/tests/test_segment.rb +10 -19
  740. data/tests/test_vad.rb +19 -0
  741. data/tests/test_vad_params.rb +103 -0
  742. data/tests/test_whisper.rb +37 -37
  743. data/whispercpp.gemspec +5 -4
  744. metadata +766 -111
  745. data/ext/cpu.mk +0 -9
  746. data/ext/examples/dr_wav.h +0 -8815
  747. data/ext/ggml/src/ggml-cann/aclnn_ops.h +0 -592
  748. data/ext/ggml/src/ggml-cpu/ggml-cpu-aarch64.cpp +0 -4262
  749. data/ext/ggml/src/ggml-cpu/ggml-cpu.c +0 -14123
  750. data/ext/ggml/src/ggml-cpu/llamafile/sgemm.cpp +0 -1884
  751. data/ext/ggml/src/ggml-cpu/llamafile/sgemm.h +0 -14
  752. data/ext/ggml/src/ggml-metal/ggml-metal-impl.h +0 -288
  753. data/ext/ggml/src/ggml-sycl/element_wise.cpp +0 -1030
  754. data/ext/ggml/src/ggml-sycl/im2col.cpp +0 -126
  755. data/ext/ggml/src/ggml-sycl/rope.cpp +0 -276
  756. data/ext/ggml/src/ggml-sycl/wkv6.cpp +0 -141
  757. data/ext/metal-embed.mk +0 -17
  758. data/ext/metal.mk +0 -6
  759. data/ext/ruby_whisper.cpp +0 -1909
  760. data/ext/scripts/get-flags.mk +0 -38
  761. data/lib/whisper.rb +0 -2
  762. /data/ext/{ggml → sources/ggml}/include/ggml-blas.h +0 -0
  763. /data/ext/{ggml → sources/ggml}/include/ggml-cann.h +0 -0
  764. /data/ext/{ggml → sources/ggml}/include/ggml-cuda.h +0 -0
  765. /data/ext/{ggml → sources/ggml}/include/ggml-kompute.h +0 -0
  766. /data/ext/{ggml → sources/ggml}/include/ggml-opencl.h +0 -0
  767. /data/ext/{ggml → sources/ggml}/include/ggml-sycl.h +0 -0
  768. /data/ext/{ggml → sources/ggml}/src/ggml-amx/common.h +0 -0
  769. /data/ext/{ggml → sources/ggml}/src/ggml-amx/ggml-amx.cpp +0 -0
  770. /data/ext/{ggml → sources/ggml}/src/ggml-amx/mmq.cpp +0 -0
  771. /data/ext/{ggml → sources/ggml}/src/ggml-amx/mmq.h +0 -0
  772. /data/ext/{ggml → sources/ggml}/src/ggml-blas/ggml-blas.cpp +0 -0
  773. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/ascendc_kernels.h +0 -0
  774. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_f16.cpp +0 -0
  775. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_f32.cpp +0 -0
  776. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_q4_0.cpp +0 -0
  777. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_q8_0.cpp +0 -0
  778. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_f16_q8_0.cpp +0 -0
  779. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_f32_q8_0.cpp +0 -0
  780. /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_float_to_q4_0.cpp +0 -0
  781. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/amx.h +0 -0
  782. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/common.h +0 -0
  783. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/mmq.cpp +0 -0
  784. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/mmq.h +0 -0
  785. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-aarch64.h +0 -0
  786. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-hbm.cpp +0 -0
  787. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-hbm.h +0 -0
  788. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-quants.h +0 -0
  789. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-traits.cpp +0 -0
  790. /data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-traits.h +0 -0
  791. /data/ext/{ggml → sources/ggml}/src/ggml-kompute/ggml-kompute.cpp +0 -0
  792. /data/ext/{ggml → sources/ggml}/src/ggml-quants.h +0 -0
  793. /data/ext/{ggml → sources/ggml}/src/ggml-threading.cpp +0 -0
  794. /data/ext/{ggml → sources/ggml}/src/ggml-threading.h +0 -0
  795. /data/ext/{src → sources/src}/coreml/whisper-encoder.h +0 -0
  796. /data/ext/{src → sources/src}/openvino/whisper-openvino-encoder.cpp +0 -0
  797. /data/ext/{src → sources/src}/openvino/whisper-openvino-encoder.h +0 -0
@@ -0,0 +1,1301 @@
1
+ #include <ruby.h>
2
+ #include "ruby_whisper.h"
3
+
4
+ #define BOOL_PARAMS_SETTER(self, prop, value) \
5
+ ruby_whisper_params *rwp; \
6
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp); \
7
+ if (value == Qfalse || value == Qnil) { \
8
+ rwp->params.prop = false; \
9
+ } else { \
10
+ rwp->params.prop = true; \
11
+ } \
12
+ return value; \
13
+
14
+ #define BOOL_PARAMS_GETTER(self, prop) \
15
+ ruby_whisper_params *rwp; \
16
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp); \
17
+ if (rwp->params.prop) { \
18
+ return Qtrue; \
19
+ } else { \
20
+ return Qfalse; \
21
+ }
22
+
23
+ #define DEFINE_PARAM(param_name, nth) \
24
+ id_ ## param_name = rb_intern(#param_name); \
25
+ param_names[nth] = id_ ## param_name; \
26
+ rb_define_method(cParams, #param_name, ruby_whisper_params_get_ ## param_name, 0); \
27
+ rb_define_method(cParams, #param_name "=", ruby_whisper_params_set_ ## param_name, 1);
28
+
29
+ #define RUBY_WHISPER_PARAMS_PARAM_NAMES_COUNT 35
30
+
31
+ extern VALUE cParams;
32
+ extern VALUE cVADParams;
33
+
34
+ extern ID id_call;
35
+
36
+ extern VALUE ruby_whisper_normalize_model_path(VALUE model_path);
37
+ extern VALUE rb_whisper_segment_initialize(VALUE context, int index);
38
+ extern const rb_data_type_t ruby_whisper_vad_params_type;
39
+
40
+ static ID param_names[RUBY_WHISPER_PARAMS_PARAM_NAMES_COUNT];
41
+ static ID id_language;
42
+ static ID id_translate;
43
+ static ID id_no_context;
44
+ static ID id_single_segment;
45
+ static ID id_print_special;
46
+ static ID id_print_progress;
47
+ static ID id_print_realtime;
48
+ static ID id_print_timestamps;
49
+ static ID id_suppress_blank;
50
+ static ID id_suppress_nst;
51
+ static ID id_token_timestamps;
52
+ static ID id_split_on_word;
53
+ static ID id_initial_prompt;
54
+ static ID id_diarize;
55
+ static ID id_offset;
56
+ static ID id_duration;
57
+ static ID id_max_text_tokens;
58
+ static ID id_temperature;
59
+ static ID id_max_initial_ts;
60
+ static ID id_length_penalty;
61
+ static ID id_temperature_inc;
62
+ static ID id_entropy_thold;
63
+ static ID id_logprob_thold;
64
+ static ID id_no_speech_thold;
65
+ static ID id_new_segment_callback;
66
+ static ID id_new_segment_callback_user_data;
67
+ static ID id_progress_callback;
68
+ static ID id_progress_callback_user_data;
69
+ static ID id_encoder_begin_callback;
70
+ static ID id_encoder_begin_callback_user_data;
71
+ static ID id_abort_callback;
72
+ static ID id_abort_callback_user_data;
73
+ static ID id_vad;
74
+ static ID id_vad_model_path;
75
+ static ID id_vad_params;
76
+
77
+ static void
78
+ rb_whisper_callbcack_container_mark(ruby_whisper_callback_container *rwc)
79
+ {
80
+ rb_gc_mark(rwc->user_data);
81
+ rb_gc_mark(rwc->callback);
82
+ rb_gc_mark(rwc->callbacks);
83
+ }
84
+
85
+ static ruby_whisper_callback_container*
86
+ rb_whisper_callback_container_allocate() {
87
+ ruby_whisper_callback_container *container;
88
+ container = ALLOC(ruby_whisper_callback_container);
89
+ container->context = NULL;
90
+ container->user_data = Qnil;
91
+ container->callback = Qnil;
92
+ container->callbacks = rb_ary_new();
93
+ return container;
94
+ }
95
+
96
+ static void new_segment_callback(struct whisper_context *ctx, struct whisper_state *state, int n_new, void *user_data) {
97
+ const ruby_whisper_callback_container *container = (ruby_whisper_callback_container *)user_data;
98
+
99
+ // Currently, doesn't support state because
100
+ // those require to resolve GC-related problems.
101
+ if (!NIL_P(container->callback)) {
102
+ rb_funcall(container->callback, id_call, 4, *container->context, Qnil, INT2NUM(n_new), container->user_data);
103
+ }
104
+ const long callbacks_len = RARRAY_LEN(container->callbacks);
105
+ if (0 == callbacks_len) {
106
+ return;
107
+ }
108
+ const int n_segments = whisper_full_n_segments_from_state(state);
109
+ for (int i = n_new; i > 0; i--) {
110
+ int i_segment = n_segments - i;
111
+ VALUE segment = rb_whisper_segment_initialize(*container->context, i_segment);
112
+ for (int j = 0; j < callbacks_len; j++) {
113
+ VALUE cb = rb_ary_entry(container->callbacks, j);
114
+ rb_funcall(cb, id_call, 1, segment);
115
+ }
116
+ }
117
+ }
118
+
119
+ static void progress_callback(struct whisper_context *ctx, struct whisper_state *state, int progress_cur, void *user_data) {
120
+ const ruby_whisper_callback_container *container = (ruby_whisper_callback_container *)user_data;
121
+ const VALUE progress = INT2NUM(progress_cur);
122
+ // Currently, doesn't support state because
123
+ // those require to resolve GC-related problems.
124
+ if (!NIL_P(container->callback)) {
125
+ rb_funcall(container->callback, id_call, 4, *container->context, Qnil, progress, container->user_data);
126
+ }
127
+ const long callbacks_len = RARRAY_LEN(container->callbacks);
128
+ if (0 == callbacks_len) {
129
+ return;
130
+ }
131
+ for (int j = 0; j < callbacks_len; j++) {
132
+ VALUE cb = rb_ary_entry(container->callbacks, j);
133
+ rb_funcall(cb, id_call, 1, progress);
134
+ }
135
+ }
136
+
137
+ static bool encoder_begin_callback(struct whisper_context *ctx, struct whisper_state *state, void *user_data) {
138
+ const ruby_whisper_callback_container *container = (ruby_whisper_callback_container *)user_data;
139
+ bool is_aborted = false;
140
+ VALUE result;
141
+
142
+ // Currently, doesn't support state because
143
+ // those require to resolve GC-related problems.
144
+ if (!NIL_P(container->callback)) {
145
+ result = rb_funcall(container->callback, id_call, 3, *container->context, Qnil, container->user_data);
146
+ if (result == Qfalse) {
147
+ is_aborted = true;
148
+ }
149
+ }
150
+ const long callbacks_len = RARRAY_LEN(container->callbacks);
151
+ if (0 == callbacks_len) {
152
+ return !is_aborted;
153
+ }
154
+ for (int j = 0; j < callbacks_len; j++) {
155
+ VALUE cb = rb_ary_entry(container->callbacks, j);
156
+ result = rb_funcall(cb, id_call, 0);
157
+ if (result == Qfalse) {
158
+ is_aborted = true;
159
+ }
160
+ }
161
+ return !is_aborted;
162
+ }
163
+
164
+ static bool abort_callback(void * user_data) {
165
+ const ruby_whisper_callback_container *container = (ruby_whisper_callback_container *)user_data;
166
+ if (!NIL_P(container->callback)) {
167
+ VALUE result = rb_funcall(container->callback, id_call, 1, container->user_data);
168
+ if (!NIL_P(result) && Qfalse != result) {
169
+ return true;
170
+ }
171
+ }
172
+ const long callbacks_len = RARRAY_LEN(container->callbacks);
173
+ if (0 == callbacks_len) {
174
+ return false;
175
+ }
176
+ for (int j = 0; j < callbacks_len; j++) {
177
+ VALUE cb = rb_ary_entry(container->callbacks, j);
178
+ VALUE result = rb_funcall(cb, id_call, 1, container->user_data);
179
+ if (!NIL_P(result) && Qfalse != result) {
180
+ return true;
181
+ }
182
+ }
183
+ return false;
184
+ }
185
+
186
+ static void register_callbacks(ruby_whisper_params * rwp, VALUE * context) {
187
+ if (!NIL_P(rwp->new_segment_callback_container->callback) || 0 != RARRAY_LEN(rwp->new_segment_callback_container->callbacks)) {
188
+ rwp->new_segment_callback_container->context = context;
189
+ rwp->params.new_segment_callback = new_segment_callback;
190
+ rwp->params.new_segment_callback_user_data = rwp->new_segment_callback_container;
191
+ }
192
+
193
+ if (!NIL_P(rwp->progress_callback_container->callback) || 0 != RARRAY_LEN(rwp->progress_callback_container->callbacks)) {
194
+ rwp->progress_callback_container->context = context;
195
+ rwp->params.progress_callback = progress_callback;
196
+ rwp->params.progress_callback_user_data = rwp->progress_callback_container;
197
+ }
198
+
199
+ if (!NIL_P(rwp->encoder_begin_callback_container->callback) || 0 != RARRAY_LEN(rwp->encoder_begin_callback_container->callbacks)) {
200
+ rwp->encoder_begin_callback_container->context = context;
201
+ rwp->params.encoder_begin_callback = encoder_begin_callback;
202
+ rwp->params.encoder_begin_callback_user_data = rwp->encoder_begin_callback_container;
203
+ }
204
+
205
+ if (!NIL_P(rwp->abort_callback_container->callback) || 0 != RARRAY_LEN(rwp->abort_callback_container->callbacks)) {
206
+ rwp->abort_callback_container->context = context;
207
+ rwp->params.abort_callback = abort_callback;
208
+ rwp->params.abort_callback_user_data = rwp->abort_callback_container;
209
+ }
210
+ }
211
+
212
+ static void set_vad_params(ruby_whisper_params *rwp)
213
+ {
214
+ ruby_whisper_vad_params * rwvp;
215
+ TypedData_Get_Struct(rwp->vad_params, ruby_whisper_vad_params, &ruby_whisper_vad_params_type, rwvp);
216
+ rwp->params.vad_params = rwvp->params;
217
+ }
218
+
219
+ void
220
+ prepare_transcription(ruby_whisper_params *rwp, VALUE *context)
221
+ {
222
+ register_callbacks(rwp, context);
223
+ set_vad_params(rwp);
224
+ }
225
+
226
+ void
227
+ rb_whisper_params_mark(void *p)
228
+ {
229
+ ruby_whisper_params *rwp = (ruby_whisper_params *)p;
230
+ rb_whisper_callbcack_container_mark(rwp->new_segment_callback_container);
231
+ rb_whisper_callbcack_container_mark(rwp->progress_callback_container);
232
+ rb_whisper_callbcack_container_mark(rwp->encoder_begin_callback_container);
233
+ rb_whisper_callbcack_container_mark(rwp->abort_callback_container);
234
+ rb_gc_mark(rwp->vad_params);
235
+ }
236
+
237
+ void
238
+ ruby_whisper_params_free(ruby_whisper_params *rwp)
239
+ {
240
+ }
241
+
242
+ void
243
+ rb_whisper_params_free(void *p)
244
+ {
245
+ ruby_whisper_params *rwp = (ruby_whisper_params *)p;
246
+ // How to free user_data and callback only when not referred to by others?
247
+ ruby_whisper_params_free(rwp);
248
+ free(rwp);
249
+ }
250
+
251
+ static size_t
252
+ ruby_whisper_params_memsize(const void *p)
253
+ {
254
+ const ruby_whisper_params *rwp = (const ruby_whisper_params *)p;
255
+
256
+ return sizeof(ruby_whisper_params) + sizeof(rwp->params) + sizeof(rwp->vad_params);
257
+ }
258
+
259
+ const rb_data_type_t ruby_whisper_params_type = {
260
+ "ruby_whisper_params",
261
+ {
262
+ rb_whisper_params_mark,
263
+ rb_whisper_params_free,
264
+ ruby_whisper_params_memsize,
265
+ },
266
+ 0, 0,
267
+ 0
268
+ };
269
+
270
+ static VALUE
271
+ ruby_whisper_params_allocate(VALUE klass)
272
+ {
273
+ ruby_whisper_params *rwp;
274
+ VALUE obj = TypedData_Make_Struct(klass, ruby_whisper_params, &ruby_whisper_params_type, rwp);
275
+ rwp->params = whisper_full_default_params(WHISPER_SAMPLING_GREEDY);
276
+ rwp->diarize = false;
277
+ rwp->vad_params = TypedData_Wrap_Struct(cVADParams, &ruby_whisper_vad_params_type, (void *)&rwp->params.vad_params);
278
+ rwp->new_segment_callback_container = rb_whisper_callback_container_allocate();
279
+ rwp->progress_callback_container = rb_whisper_callback_container_allocate();
280
+ rwp->encoder_begin_callback_container = rb_whisper_callback_container_allocate();
281
+ rwp->abort_callback_container = rb_whisper_callback_container_allocate();
282
+ return obj;
283
+ }
284
+
285
+ /*
286
+ * params.language = "auto" | "en", etc...
287
+ *
288
+ * call-seq:
289
+ * language = lang_name -> lang_name
290
+ */
291
+ static VALUE
292
+ ruby_whisper_params_set_language(VALUE self, VALUE value)
293
+ {
294
+ ruby_whisper_params *rwp;
295
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
296
+ if (value == Qfalse || value == Qnil) {
297
+ rwp->params.language = "auto";
298
+ } else {
299
+ rwp->params.language = StringValueCStr(value);
300
+ }
301
+ return value;
302
+ }
303
+ /*
304
+ * call-seq:
305
+ * language -> String
306
+ */
307
+ static VALUE
308
+ ruby_whisper_params_get_language(VALUE self)
309
+ {
310
+ ruby_whisper_params *rwp;
311
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
312
+ if (rwp->params.language) {
313
+ return rb_str_new2(rwp->params.language);
314
+ } else {
315
+ return rb_str_new2("auto");
316
+ }
317
+ }
318
+ /*
319
+ * call-seq:
320
+ * translate = do_translate -> do_translate
321
+ */
322
+ static VALUE
323
+ ruby_whisper_params_set_translate(VALUE self, VALUE value)
324
+ {
325
+ BOOL_PARAMS_SETTER(self, translate, value)
326
+ }
327
+ /*
328
+ * call-seq:
329
+ * translate -> bool
330
+ */
331
+ static VALUE
332
+ ruby_whisper_params_get_translate(VALUE self)
333
+ {
334
+ BOOL_PARAMS_GETTER(self, translate)
335
+ }
336
+ /*
337
+ * call-seq:
338
+ * no_context = dont_use_context -> dont_use_context
339
+ */
340
+ static VALUE
341
+ ruby_whisper_params_set_no_context(VALUE self, VALUE value)
342
+ {
343
+ BOOL_PARAMS_SETTER(self, no_context, value)
344
+ }
345
+ /*
346
+ * If true, does not use past transcription (if any) as initial prompt for the decoder.
347
+ *
348
+ * call-seq:
349
+ * no_context -> bool
350
+ */
351
+ static VALUE
352
+ ruby_whisper_params_get_no_context(VALUE self)
353
+ {
354
+ BOOL_PARAMS_GETTER(self, no_context)
355
+ }
356
+ /*
357
+ * call-seq:
358
+ * single_segment = force_single -> force_single
359
+ */
360
+ static VALUE
361
+ ruby_whisper_params_set_single_segment(VALUE self, VALUE value)
362
+ {
363
+ BOOL_PARAMS_SETTER(self, single_segment, value)
364
+ }
365
+ /*
366
+ * If true, forces single segment output (useful for streaming).
367
+ *
368
+ * call-seq:
369
+ * single_segment -> bool
370
+ */
371
+ static VALUE
372
+ ruby_whisper_params_get_single_segment(VALUE self)
373
+ {
374
+ BOOL_PARAMS_GETTER(self, single_segment)
375
+ }
376
+ /*
377
+ * call-seq:
378
+ * print_special = force_print -> force_print
379
+ */
380
+ static VALUE
381
+ ruby_whisper_params_set_print_special(VALUE self, VALUE value)
382
+ {
383
+ BOOL_PARAMS_SETTER(self, print_special, value)
384
+ }
385
+ /*
386
+ * If true, prints special tokens (e.g. <SOT>, <EOT>, <BEG>, etc.).
387
+ *
388
+ * call-seq:
389
+ * print_special -> bool
390
+ */
391
+ static VALUE
392
+ ruby_whisper_params_get_print_special(VALUE self)
393
+ {
394
+ BOOL_PARAMS_GETTER(self, print_special)
395
+ }
396
+ /*
397
+ * call-seq:
398
+ * print_progress = force_print -> force_print
399
+ */
400
+ static VALUE
401
+ ruby_whisper_params_set_print_progress(VALUE self, VALUE value)
402
+ {
403
+ BOOL_PARAMS_SETTER(self, print_progress, value)
404
+ }
405
+ /*
406
+ * If true, prints progress information.
407
+ *
408
+ * call-seq:
409
+ * print_progress -> bool
410
+ */
411
+ static VALUE
412
+ ruby_whisper_params_get_print_progress(VALUE self)
413
+ {
414
+ BOOL_PARAMS_GETTER(self, print_progress)
415
+ }
416
+ /*
417
+ * call-seq:
418
+ * print_realtime = force_print -> force_print
419
+ */
420
+ static VALUE
421
+ ruby_whisper_params_set_print_realtime(VALUE self, VALUE value)
422
+ {
423
+ BOOL_PARAMS_SETTER(self, print_realtime, value)
424
+ }
425
+ /*
426
+ * If true, prints results from within whisper.cpp. (avoid it, use callback instead)
427
+ * call-seq:
428
+ * print_realtime -> bool
429
+ */
430
+ static VALUE
431
+ ruby_whisper_params_get_print_realtime(VALUE self)
432
+ {
433
+ BOOL_PARAMS_GETTER(self, print_realtime)
434
+ }
435
+ /*
436
+ * call-seq:
437
+ * print_timestamps = force_print -> force_print
438
+ */
439
+ static VALUE
440
+ ruby_whisper_params_set_print_timestamps(VALUE self, VALUE value)
441
+ {
442
+ BOOL_PARAMS_SETTER(self, print_timestamps, value)
443
+ }
444
+ /*
445
+ * If true, prints timestamps for each text segment when printing realtime.
446
+ *
447
+ * call-seq:
448
+ * print_timestamps -> bool
449
+ */
450
+ static VALUE
451
+ ruby_whisper_params_get_print_timestamps(VALUE self)
452
+ {
453
+ BOOL_PARAMS_GETTER(self, print_timestamps)
454
+ }
455
+ /*
456
+ * call-seq:
457
+ * suppress_blank = force_suppress -> force_suppress
458
+ */
459
+ static VALUE
460
+ ruby_whisper_params_set_suppress_blank(VALUE self, VALUE value)
461
+ {
462
+ BOOL_PARAMS_SETTER(self, suppress_blank, value)
463
+ }
464
+ /*
465
+ * If true, suppresses blank outputs.
466
+ *
467
+ * call-seq:
468
+ * suppress_blank -> bool
469
+ */
470
+ static VALUE
471
+ ruby_whisper_params_get_suppress_blank(VALUE self)
472
+ {
473
+ BOOL_PARAMS_GETTER(self, suppress_blank)
474
+ }
475
+ /*
476
+ * call-seq:
477
+ * suppress_nst = force_suppress -> force_suppress
478
+ */
479
+ static VALUE
480
+ ruby_whisper_params_set_suppress_nst(VALUE self, VALUE value)
481
+ {
482
+ BOOL_PARAMS_SETTER(self, suppress_nst, value)
483
+ }
484
+ /*
485
+ * If true, suppresses non-speech-tokens.
486
+ *
487
+ * call-seq:
488
+ * suppress_nst -> bool
489
+ */
490
+ static VALUE
491
+ ruby_whisper_params_get_suppress_nst(VALUE self)
492
+ {
493
+ BOOL_PARAMS_GETTER(self, suppress_nst)
494
+ }
495
+ /*
496
+ * If true, enables token-level timestamps.
497
+ *
498
+ * call-seq:
499
+ * token_timestamps -> bool
500
+ */
501
+ static VALUE
502
+ ruby_whisper_params_get_token_timestamps(VALUE self)
503
+ {
504
+ BOOL_PARAMS_GETTER(self, token_timestamps)
505
+ }
506
+ /*
507
+ * call-seq:
508
+ * token_timestamps = force_timestamps -> force_timestamps
509
+ */
510
+ static VALUE
511
+ ruby_whisper_params_set_token_timestamps(VALUE self, VALUE value)
512
+ {
513
+ BOOL_PARAMS_SETTER(self, token_timestamps, value)
514
+ }
515
+ /*
516
+ * If true, split on word rather than on token (when used with max_len).
517
+ *
518
+ * call-seq:
519
+ * translate -> bool
520
+ */
521
+ static VALUE
522
+ ruby_whisper_params_get_split_on_word(VALUE self)
523
+ {
524
+ BOOL_PARAMS_GETTER(self, split_on_word)
525
+ }
526
+ /*
527
+ * call-seq:
528
+ * split_on_word = force_split -> force_split
529
+ */
530
+ static VALUE
531
+ ruby_whisper_params_set_split_on_word(VALUE self, VALUE value)
532
+ {
533
+ BOOL_PARAMS_SETTER(self, split_on_word, value)
534
+ }
535
+ /*
536
+ * Tokens to provide to the whisper decoder as initial prompt
537
+ * these are prepended to any existing text context from a previous call
538
+ * use whisper_tokenize() to convert text to tokens.
539
+ * Maximum of whisper_n_text_ctx()/2 tokens are used (typically 224).
540
+ *
541
+ * call-seq:
542
+ * initial_prompt -> String
543
+ */
544
+ static VALUE
545
+ ruby_whisper_params_get_initial_prompt(VALUE self)
546
+ {
547
+ ruby_whisper_params *rwp;
548
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
549
+ return rwp->params.initial_prompt == NULL ? Qnil : rb_str_new2(rwp->params.initial_prompt);
550
+ }
551
+ /*
552
+ * call-seq:
553
+ * initial_prompt = prompt -> prompt
554
+ */
555
+ static VALUE
556
+ ruby_whisper_params_set_initial_prompt(VALUE self, VALUE value)
557
+ {
558
+ ruby_whisper_params *rwp;
559
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
560
+ rwp->params.initial_prompt = StringValueCStr(value);
561
+ return value;
562
+ }
563
+ /*
564
+ * If true, enables diarization.
565
+ *
566
+ * call-seq:
567
+ * diarize -> bool
568
+ */
569
+ static VALUE
570
+ ruby_whisper_params_get_diarize(VALUE self)
571
+ {
572
+ ruby_whisper_params *rwp;
573
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
574
+ if (rwp->diarize) {
575
+ return Qtrue;
576
+ } else {
577
+ return Qfalse;
578
+ }
579
+ }
580
+ /*
581
+ * call-seq:
582
+ * diarize = force_diarize -> force_diarize
583
+ */
584
+ static VALUE
585
+ ruby_whisper_params_set_diarize(VALUE self, VALUE value)
586
+ {
587
+ ruby_whisper_params *rwp;
588
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
589
+ if (value == Qfalse || value == Qnil) {
590
+ rwp->diarize = false;
591
+ } else {
592
+ rwp->diarize = true;
593
+ } \
594
+ return value;
595
+ }
596
+
597
+ /*
598
+ * Start offset in ms.
599
+ *
600
+ * call-seq:
601
+ * offset -> Integer
602
+ */
603
+ static VALUE
604
+ ruby_whisper_params_get_offset(VALUE self)
605
+ {
606
+ ruby_whisper_params *rwp;
607
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
608
+ return INT2NUM(rwp->params.offset_ms);
609
+ }
610
+ /*
611
+ * call-seq:
612
+ * offset = offset_ms -> offset_ms
613
+ */
614
+ static VALUE
615
+ ruby_whisper_params_set_offset(VALUE self, VALUE value)
616
+ {
617
+ ruby_whisper_params *rwp;
618
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
619
+ rwp->params.offset_ms = NUM2INT(value);
620
+ return value;
621
+ }
622
+ /*
623
+ * Audio duration to process in ms.
624
+ *
625
+ * call-seq:
626
+ * duration -> Integer
627
+ */
628
+ static VALUE
629
+ ruby_whisper_params_get_duration(VALUE self)
630
+ {
631
+ ruby_whisper_params *rwp;
632
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
633
+ return INT2NUM(rwp->params.duration_ms);
634
+ }
635
+ /*
636
+ * call-seq:
637
+ * duration = duration_ms -> duration_ms
638
+ */
639
+ static VALUE
640
+ ruby_whisper_params_set_duration(VALUE self, VALUE value)
641
+ {
642
+ ruby_whisper_params *rwp;
643
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
644
+ rwp->params.duration_ms = NUM2INT(value);
645
+ return value;
646
+ }
647
+
648
+ /*
649
+ * Max tokens to use from past text as prompt for the decoder.
650
+ *
651
+ * call-seq:
652
+ * max_text_tokens -> Integer
653
+ */
654
+ static VALUE
655
+ ruby_whisper_params_get_max_text_tokens(VALUE self)
656
+ {
657
+ ruby_whisper_params *rwp;
658
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
659
+ return INT2NUM(rwp->params.n_max_text_ctx);
660
+ }
661
+ /*
662
+ * call-seq:
663
+ * max_text_tokens = n_tokens -> n_tokens
664
+ */
665
+ static VALUE
666
+ ruby_whisper_params_set_max_text_tokens(VALUE self, VALUE value)
667
+ {
668
+ ruby_whisper_params *rwp;
669
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
670
+ rwp->params.n_max_text_ctx = NUM2INT(value);
671
+ return value;
672
+ }
673
+ /*
674
+ * call-seq:
675
+ * temperature -> Float
676
+ */
677
+ static VALUE
678
+ ruby_whisper_params_get_temperature(VALUE self)
679
+ {
680
+ ruby_whisper_params *rwp;
681
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
682
+ return DBL2NUM(rwp->params.temperature);
683
+ }
684
+ /*
685
+ * call-seq:
686
+ * temperature = temp -> temp
687
+ */
688
+ static VALUE
689
+ ruby_whisper_params_set_temperature(VALUE self, VALUE value)
690
+ {
691
+ ruby_whisper_params *rwp;
692
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
693
+ rwp->params.temperature = RFLOAT_VALUE(value);
694
+ return value;
695
+ }
696
+ /*
697
+ * See https://github.com/openai/whisper/blob/f82bc59f5ea234d4b97fb2860842ed38519f7e65/whisper/decoding.py#L97
698
+ *
699
+ * call-seq:
700
+ * max_initial_ts -> Flaot
701
+ */
702
+ static VALUE
703
+ ruby_whisper_params_get_max_initial_ts(VALUE self)
704
+ {
705
+ ruby_whisper_params *rwp;
706
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
707
+ return DBL2NUM(rwp->params.max_initial_ts);
708
+ }
709
+ /*
710
+ * call-seq:
711
+ * max_initial_ts = timestamp -> timestamp
712
+ */
713
+ static VALUE
714
+ ruby_whisper_params_set_max_initial_ts(VALUE self, VALUE value)
715
+ {
716
+ ruby_whisper_params *rwp;
717
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
718
+ rwp->params.max_initial_ts = RFLOAT_VALUE(value);
719
+ return value;
720
+ }
721
+ /*
722
+ * call-seq:
723
+ * length_penalty -> Float
724
+ */
725
+ static VALUE
726
+ ruby_whisper_params_get_length_penalty(VALUE self)
727
+ {
728
+ ruby_whisper_params *rwp;
729
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
730
+ return DBL2NUM(rwp->params.length_penalty);
731
+ }
732
+ /*
733
+ * call-seq:
734
+ * length_penalty = penalty -> penalty
735
+ */
736
+ static VALUE
737
+ ruby_whisper_params_set_length_penalty(VALUE self, VALUE value)
738
+ {
739
+ ruby_whisper_params *rwp;
740
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
741
+ rwp->params.length_penalty = RFLOAT_VALUE(value);
742
+ return value;
743
+ }
744
+ /*
745
+ * call-seq:
746
+ * temperature_inc -> Float
747
+ */
748
+ static VALUE
749
+ ruby_whisper_params_get_temperature_inc(VALUE self)
750
+ {
751
+ ruby_whisper_params *rwp;
752
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
753
+ return DBL2NUM(rwp->params.temperature_inc);
754
+ }
755
+ /*
756
+ * call-seq:
757
+ * temperature_inc = inc -> inc
758
+ */
759
+ static VALUE
760
+ ruby_whisper_params_set_temperature_inc(VALUE self, VALUE value)
761
+ {
762
+ ruby_whisper_params *rwp;
763
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
764
+ rwp->params.temperature_inc = RFLOAT_VALUE(value);
765
+ return value;
766
+ }
767
+ /*
768
+ * Similar to OpenAI's "compression_ratio_threshold"
769
+ *
770
+ * call-seq:
771
+ * entropy_thold -> Float
772
+ */
773
+ static VALUE
774
+ ruby_whisper_params_get_entropy_thold(VALUE self)
775
+ {
776
+ ruby_whisper_params *rwp;
777
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
778
+ return DBL2NUM(rwp->params.entropy_thold);
779
+ }
780
+ /*
781
+ * call-seq:
782
+ * entropy_thold = threshold -> threshold
783
+ */
784
+ static VALUE
785
+ ruby_whisper_params_set_entropy_thold(VALUE self, VALUE value)
786
+ {
787
+ ruby_whisper_params *rwp;
788
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
789
+ rwp->params.entropy_thold = RFLOAT_VALUE(value);
790
+ return value;
791
+ }
792
+ /*
793
+ * call-seq:
794
+ * logprob_thold -> Float
795
+ */
796
+ static VALUE
797
+ ruby_whisper_params_get_logprob_thold(VALUE self)
798
+ {
799
+ ruby_whisper_params *rwp;
800
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
801
+ return DBL2NUM(rwp->params.logprob_thold);
802
+ }
803
+ /*
804
+ * call-seq:
805
+ * logprob_thold = threshold -> threshold
806
+ */
807
+ static VALUE
808
+ ruby_whisper_params_set_logprob_thold(VALUE self, VALUE value)
809
+ {
810
+ ruby_whisper_params *rwp;
811
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
812
+ rwp->params.logprob_thold = RFLOAT_VALUE(value);
813
+ return value;
814
+ }
815
+ /*
816
+ * call-seq:
817
+ * no_speech_thold -> Float
818
+ */
819
+ static VALUE
820
+ ruby_whisper_params_get_no_speech_thold(VALUE self)
821
+ {
822
+ ruby_whisper_params *rwp;
823
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
824
+ return DBL2NUM(rwp->params.no_speech_thold);
825
+ }
826
+ /*
827
+ * call-seq:
828
+ * no_speech_thold = threshold -> threshold
829
+ */
830
+ static VALUE
831
+ ruby_whisper_params_set_no_speech_thold(VALUE self, VALUE value)
832
+ {
833
+ ruby_whisper_params *rwp;
834
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
835
+ rwp->params.no_speech_thold = RFLOAT_VALUE(value);
836
+ return value;
837
+ }
838
+ static VALUE
839
+ ruby_whisper_params_get_new_segment_callback(VALUE self)
840
+ {
841
+ ruby_whisper_params *rwp;
842
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
843
+ return rwp->new_segment_callback_container->callback;
844
+ }
845
+ /*
846
+ * Sets new segment callback, called for every newly generated text segment.
847
+ *
848
+ * params.new_segment_callback = ->(context, _, n_new, user_data) {
849
+ * # ...
850
+ * }
851
+ *
852
+ * call-seq:
853
+ * new_segment_callback = callback -> callback
854
+ */
855
+ static VALUE
856
+ ruby_whisper_params_set_new_segment_callback(VALUE self, VALUE value)
857
+ {
858
+ ruby_whisper_params *rwp;
859
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
860
+ rwp->new_segment_callback_container->callback = value;
861
+ return value;
862
+ }
863
+ static VALUE
864
+ ruby_whisper_params_get_new_segment_callback_user_data(VALUE self)
865
+ {
866
+ ruby_whisper_params *rwp;
867
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
868
+ return rwp->new_segment_callback_container->user_data;
869
+ }
870
+ /*
871
+ * Sets user data passed to the last argument of new segment callback.
872
+ *
873
+ * call-seq:
874
+ * new_segment_callback_user_data = user_data -> use_data
875
+ */
876
+ static VALUE
877
+ ruby_whisper_params_set_new_segment_callback_user_data(VALUE self, VALUE value)
878
+ {
879
+ ruby_whisper_params *rwp;
880
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
881
+ rwp->new_segment_callback_container->user_data = value;
882
+ return value;
883
+ }
884
+ static VALUE
885
+ ruby_whisper_params_get_progress_callback(VALUE self)
886
+ {
887
+ ruby_whisper_params *rwp;
888
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
889
+ return rwp->progress_callback_container->callback;
890
+ }
891
+ /*
892
+ * Sets progress callback, called on each progress update.
893
+ *
894
+ * params.new_segment_callback = ->(context, _, progress, user_data) {
895
+ * # ...
896
+ * }
897
+ *
898
+ * +progress+ is an Integer between 0 and 100.
899
+ *
900
+ * call-seq:
901
+ * progress_callback = callback -> callback
902
+ */
903
+ static VALUE
904
+ ruby_whisper_params_set_progress_callback(VALUE self, VALUE value)
905
+ {
906
+ ruby_whisper_params *rwp;
907
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
908
+ rwp->progress_callback_container->callback = value;
909
+ return value;
910
+ }
911
+ static VALUE
912
+ ruby_whisper_params_get_progress_callback_user_data(VALUE self)
913
+ {
914
+ ruby_whisper_params *rwp;
915
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
916
+ return rwp->progress_callback_container->user_data;
917
+ }
918
+ /*
919
+ * Sets user data passed to the last argument of progress callback.
920
+ *
921
+ * call-seq:
922
+ * progress_callback_user_data = user_data -> use_data
923
+ */
924
+ static VALUE
925
+ ruby_whisper_params_set_progress_callback_user_data(VALUE self, VALUE value)
926
+ {
927
+ ruby_whisper_params *rwp;
928
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
929
+ rwp->progress_callback_container->user_data = value;
930
+ return value;
931
+ }
932
+
933
+ static VALUE
934
+ ruby_whisper_params_get_encoder_begin_callback(VALUE self)
935
+ {
936
+ ruby_whisper_params *rwp;
937
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
938
+ return rwp->encoder_begin_callback_container->callback;
939
+ }
940
+
941
+ /*
942
+ * Sets encoder begin callback, called when the encoder starts.
943
+ *
944
+ * params.encoder_begin_callback = ->(context, _, user_data) {
945
+ * # ...
946
+ * }
947
+ *
948
+ * call-seq:
949
+ * encoder_begin_callback = callback -> callback
950
+ */
951
+ static VALUE
952
+ ruby_whisper_params_set_encoder_begin_callback(VALUE self, VALUE value)
953
+ {
954
+ ruby_whisper_params *rwp;
955
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
956
+ rwp->encoder_begin_callback_container->callback = value;
957
+ return value;
958
+ }
959
+
960
+ static VALUE
961
+ ruby_whisper_params_get_encoder_begin_callback_user_data(VALUE self)
962
+ {
963
+ ruby_whisper_params *rwp;
964
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
965
+ return rwp->encoder_begin_callback_container->user_data;
966
+ }
967
+
968
+ /*
969
+ * Sets user data passed to the last argument of encoder begin callback.
970
+ *
971
+ * call-seq:
972
+ * encoder_begin_callback_user_data = user_data -> use_data
973
+ */
974
+ static VALUE
975
+ ruby_whisper_params_set_encoder_begin_callback_user_data(VALUE self, VALUE value)
976
+ {
977
+ ruby_whisper_params *rwp;
978
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
979
+ rwp->encoder_begin_callback_container->user_data = value;
980
+ return value;
981
+ }
982
+
983
+ static VALUE
984
+ ruby_whisper_params_get_abort_callback(VALUE self)
985
+ {
986
+ ruby_whisper_params *rwp;
987
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
988
+ return rwp->abort_callback_container->callback;
989
+ }
990
+ /*
991
+ * Sets abort callback, called to check if the process should be aborted.
992
+ *
993
+ * params.abort_callback = ->(user_data) {
994
+ * # ...
995
+ * }
996
+ *
997
+ * call-seq:
998
+ * abort_callback = callback -> callback
999
+ */
1000
+ static VALUE
1001
+ ruby_whisper_params_set_abort_callback(VALUE self, VALUE value)
1002
+ {
1003
+ ruby_whisper_params *rwp;
1004
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1005
+ rwp->abort_callback_container->callback = value;
1006
+ return value;
1007
+ }
1008
+ static VALUE
1009
+ ruby_whisper_params_get_abort_callback_user_data(VALUE self)
1010
+ {
1011
+ ruby_whisper_params *rwp;
1012
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1013
+ return rwp->abort_callback_container->user_data;
1014
+ }
1015
+ /*
1016
+ * Sets user data passed to the last argument of abort callback.
1017
+ *
1018
+ * call-seq:
1019
+ * abort_callback_user_data = user_data -> use_data
1020
+ */
1021
+ static VALUE
1022
+ ruby_whisper_params_set_abort_callback_user_data(VALUE self, VALUE value)
1023
+ {
1024
+ ruby_whisper_params *rwp;
1025
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1026
+ rwp->abort_callback_container->user_data = value;
1027
+ return value;
1028
+ }
1029
+
1030
+ /*
1031
+ * call-seq:
1032
+ * vad = use_vad -> use_vad
1033
+ */
1034
+ static VALUE
1035
+ ruby_whisper_params_get_vad(VALUE self)
1036
+ {
1037
+ BOOL_PARAMS_GETTER(self, vad)
1038
+ }
1039
+
1040
+ static VALUE
1041
+ ruby_whisper_params_set_vad(VALUE self, VALUE value)
1042
+ {
1043
+ BOOL_PARAMS_SETTER(self, vad, value)
1044
+ }
1045
+
1046
+ /*
1047
+ * call-seq:
1048
+ * vad_model_path = model_path -> model_path
1049
+ */
1050
+ static VALUE
1051
+ ruby_whisper_params_set_vad_model_path(VALUE self, VALUE value)
1052
+ {
1053
+ ruby_whisper_params *rwp;
1054
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1055
+ if (NIL_P(value)) {
1056
+ rwp->params.vad_model_path = NULL;
1057
+ return value;
1058
+ }
1059
+ VALUE path = ruby_whisper_normalize_model_path(value);
1060
+ rwp->params.vad_model_path = StringValueCStr(path);
1061
+ return value;
1062
+ }
1063
+
1064
+ static VALUE
1065
+ ruby_whisper_params_get_vad_model_path(VALUE self)
1066
+ {
1067
+ ruby_whisper_params *rwp;
1068
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1069
+ return rwp->params.vad_model_path == NULL ? Qnil : rb_str_new2(rwp->params.vad_model_path);
1070
+ }
1071
+
1072
+ /*
1073
+ * call-seq:
1074
+ * vad_params = params -> params
1075
+ */
1076
+ static VALUE
1077
+ ruby_whisper_params_set_vad_params(VALUE self, VALUE value)
1078
+ {
1079
+ ruby_whisper_params *rwp;
1080
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1081
+ rwp->vad_params = value;
1082
+ return value;
1083
+ }
1084
+
1085
+ static VALUE
1086
+ ruby_whisper_params_get_vad_params(VALUE self)
1087
+ {
1088
+ ruby_whisper_params *rwp;
1089
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1090
+ return rwp->vad_params;
1091
+ }
1092
+
1093
+ #define SET_PARAM_IF_SAME(param_name) \
1094
+ if (id == id_ ## param_name) { \
1095
+ ruby_whisper_params_set_ ## param_name(self, value); \
1096
+ continue; \
1097
+ }
1098
+
1099
+ static VALUE
1100
+ ruby_whisper_params_initialize(int argc, VALUE *argv, VALUE self)
1101
+ {
1102
+ VALUE kw_hash;
1103
+ VALUE values[RUBY_WHISPER_PARAMS_PARAM_NAMES_COUNT] = {Qundef};
1104
+ VALUE value;
1105
+ ruby_whisper_params *rwp;
1106
+ ID id;
1107
+ int i;
1108
+
1109
+ rb_scan_args_kw(RB_SCAN_ARGS_KEYWORDS, argc, argv, ":", &kw_hash);
1110
+ if (NIL_P(kw_hash)) {
1111
+ return self;
1112
+ }
1113
+
1114
+ rb_get_kwargs(kw_hash, param_names, 0, RUBY_WHISPER_PARAMS_PARAM_NAMES_COUNT, values);
1115
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1116
+
1117
+ for (i = 0; i < RUBY_WHISPER_PARAMS_PARAM_NAMES_COUNT; i++) {
1118
+ id = param_names[i];
1119
+ value = values[i];
1120
+ if (value == Qundef) {
1121
+ continue;
1122
+ }
1123
+ if (id == id_diarize) {
1124
+ rwp->diarize = value;
1125
+ continue;
1126
+ } else {
1127
+ SET_PARAM_IF_SAME(language)
1128
+ SET_PARAM_IF_SAME(translate)
1129
+ SET_PARAM_IF_SAME(no_context)
1130
+ SET_PARAM_IF_SAME(single_segment)
1131
+ SET_PARAM_IF_SAME(print_special)
1132
+ SET_PARAM_IF_SAME(print_progress)
1133
+ SET_PARAM_IF_SAME(print_realtime)
1134
+ SET_PARAM_IF_SAME(print_timestamps)
1135
+ SET_PARAM_IF_SAME(suppress_blank)
1136
+ SET_PARAM_IF_SAME(suppress_nst)
1137
+ SET_PARAM_IF_SAME(token_timestamps)
1138
+ SET_PARAM_IF_SAME(split_on_word)
1139
+ SET_PARAM_IF_SAME(initial_prompt)
1140
+ SET_PARAM_IF_SAME(offset)
1141
+ SET_PARAM_IF_SAME(duration)
1142
+ SET_PARAM_IF_SAME(max_text_tokens)
1143
+ SET_PARAM_IF_SAME(temperature)
1144
+ SET_PARAM_IF_SAME(max_initial_ts)
1145
+ SET_PARAM_IF_SAME(length_penalty)
1146
+ SET_PARAM_IF_SAME(temperature_inc)
1147
+ SET_PARAM_IF_SAME(entropy_thold)
1148
+ SET_PARAM_IF_SAME(logprob_thold)
1149
+ SET_PARAM_IF_SAME(no_speech_thold)
1150
+ SET_PARAM_IF_SAME(new_segment_callback)
1151
+ SET_PARAM_IF_SAME(new_segment_callback_user_data)
1152
+ SET_PARAM_IF_SAME(progress_callback)
1153
+ SET_PARAM_IF_SAME(progress_callback_user_data)
1154
+ SET_PARAM_IF_SAME(encoder_begin_callback)
1155
+ SET_PARAM_IF_SAME(encoder_begin_callback_user_data)
1156
+ SET_PARAM_IF_SAME(abort_callback)
1157
+ SET_PARAM_IF_SAME(abort_callback_user_data)
1158
+ SET_PARAM_IF_SAME(vad)
1159
+ SET_PARAM_IF_SAME(vad_model_path)
1160
+ SET_PARAM_IF_SAME(vad_params)
1161
+ }
1162
+ }
1163
+
1164
+ return self;
1165
+ }
1166
+
1167
+ #undef SET_PARAM_IF_SAME
1168
+
1169
+ /*
1170
+ * Hook called on new segment. Yields each Whisper::Segment.
1171
+ *
1172
+ * whisper.on_new_segment do |segment|
1173
+ * # ...
1174
+ * end
1175
+ *
1176
+ * call-seq:
1177
+ * on_new_segment {|segment| ... }
1178
+ */
1179
+ static VALUE
1180
+ ruby_whisper_params_on_new_segment(VALUE self)
1181
+ {
1182
+ ruby_whisper_params *rwp;
1183
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1184
+ const VALUE blk = rb_block_proc();
1185
+ rb_ary_push(rwp->new_segment_callback_container->callbacks, blk);
1186
+ return Qnil;
1187
+ }
1188
+
1189
+ /*
1190
+ * Hook called on progress update. Yields each progress Integer between 0 and 100.
1191
+ *
1192
+ * whisper.on_progress do |progress|
1193
+ * # ...
1194
+ * end
1195
+ *
1196
+ * call-seq:
1197
+ * on_progress {|progress| ... }
1198
+ */
1199
+ static VALUE
1200
+ ruby_whisper_params_on_progress(VALUE self)
1201
+ {
1202
+ ruby_whisper_params *rwp;
1203
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1204
+ const VALUE blk = rb_block_proc();
1205
+ rb_ary_push(rwp->progress_callback_container->callbacks, blk);
1206
+ return Qnil;
1207
+ }
1208
+
1209
+ /*
1210
+ * Hook called when the encoder starts.
1211
+ *
1212
+ * whisper.on_encoder_begin do
1213
+ * # ...
1214
+ * end
1215
+ *
1216
+ * call-seq:
1217
+ * on_encoder_begin { ... }
1218
+ */
1219
+ static VALUE
1220
+ ruby_whisper_params_on_encoder_begin(VALUE self)
1221
+ {
1222
+ ruby_whisper_params *rwp;
1223
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1224
+ const VALUE blk = rb_block_proc();
1225
+ rb_ary_push(rwp->encoder_begin_callback_container->callbacks, blk);
1226
+ return Qnil;
1227
+ }
1228
+
1229
+ /*
1230
+ * Call block to determine whether abort or not. Return +true+ when you want to abort.
1231
+ *
1232
+ * params.abort_on do
1233
+ * if some_condition
1234
+ * true # abort
1235
+ * else
1236
+ * false # continue
1237
+ * end
1238
+ * end
1239
+ *
1240
+ * call-seq:
1241
+ * abort_on { ... }
1242
+ */
1243
+ static VALUE
1244
+ ruby_whisper_params_abort_on(VALUE self)
1245
+ {
1246
+ ruby_whisper_params *rwp;
1247
+ TypedData_Get_Struct(self, ruby_whisper_params, &ruby_whisper_params_type, rwp);
1248
+ const VALUE blk = rb_block_proc();
1249
+ rb_ary_push(rwp->abort_callback_container->callbacks, blk);
1250
+ return Qnil;
1251
+ }
1252
+
1253
+ void
1254
+ init_ruby_whisper_params(VALUE *mWhisper)
1255
+ {
1256
+ cParams = rb_define_class_under(*mWhisper, "Params", rb_cObject);
1257
+
1258
+ rb_define_alloc_func(cParams, ruby_whisper_params_allocate);
1259
+ rb_define_method(cParams, "initialize", ruby_whisper_params_initialize, -1);
1260
+
1261
+ DEFINE_PARAM(language, 0)
1262
+ DEFINE_PARAM(translate, 1)
1263
+ DEFINE_PARAM(no_context, 2)
1264
+ DEFINE_PARAM(single_segment, 3)
1265
+ DEFINE_PARAM(print_special, 4)
1266
+ DEFINE_PARAM(print_progress, 5)
1267
+ DEFINE_PARAM(print_realtime, 6)
1268
+ DEFINE_PARAM(print_timestamps, 7)
1269
+ DEFINE_PARAM(suppress_blank, 8)
1270
+ DEFINE_PARAM(suppress_nst, 9)
1271
+ DEFINE_PARAM(token_timestamps, 10)
1272
+ DEFINE_PARAM(split_on_word, 11)
1273
+ DEFINE_PARAM(initial_prompt, 12)
1274
+ DEFINE_PARAM(diarize, 13)
1275
+ DEFINE_PARAM(offset, 14)
1276
+ DEFINE_PARAM(duration, 15)
1277
+ DEFINE_PARAM(max_text_tokens, 16)
1278
+ DEFINE_PARAM(temperature, 17)
1279
+ DEFINE_PARAM(max_initial_ts, 18)
1280
+ DEFINE_PARAM(length_penalty, 19)
1281
+ DEFINE_PARAM(temperature_inc, 20)
1282
+ DEFINE_PARAM(entropy_thold, 21)
1283
+ DEFINE_PARAM(logprob_thold, 22)
1284
+ DEFINE_PARAM(no_speech_thold, 23)
1285
+ DEFINE_PARAM(new_segment_callback, 24)
1286
+ DEFINE_PARAM(new_segment_callback_user_data, 25)
1287
+ DEFINE_PARAM(progress_callback, 26)
1288
+ DEFINE_PARAM(progress_callback_user_data, 27)
1289
+ DEFINE_PARAM(encoder_begin_callback, 28)
1290
+ DEFINE_PARAM(encoder_begin_callback_user_data, 29)
1291
+ DEFINE_PARAM(abort_callback, 30)
1292
+ DEFINE_PARAM(abort_callback_user_data, 31)
1293
+ DEFINE_PARAM(vad, 32)
1294
+ DEFINE_PARAM(vad_model_path, 33)
1295
+ DEFINE_PARAM(vad_params, 34)
1296
+
1297
+ rb_define_method(cParams, "on_new_segment", ruby_whisper_params_on_new_segment, 0);
1298
+ rb_define_method(cParams, "on_progress", ruby_whisper_params_on_progress, 0);
1299
+ rb_define_method(cParams, "on_encoder_begin", ruby_whisper_params_on_encoder_begin, 0);
1300
+ rb_define_method(cParams, "abort_on", ruby_whisper_params_abort_on, 0);
1301
+ }