whispercpp 1.3.1 → 1.3.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/.gitignore +7 -3
- data/README.md +161 -43
- data/Rakefile +45 -13
- data/ext/.gitignore +4 -8
- data/ext/dependencies.rb +73 -0
- data/ext/extconf.rb +21 -198
- data/ext/options.rb +85 -0
- data/ext/ruby_whisper.c +177 -0
- data/ext/ruby_whisper.h +17 -2
- data/ext/ruby_whisper_context.c +672 -0
- data/ext/ruby_whisper_error.c +52 -0
- data/ext/ruby_whisper_model.c +232 -0
- data/ext/ruby_whisper_params.c +1303 -0
- data/ext/ruby_whisper_segment.c +220 -0
- data/ext/ruby_whisper_transcribe.cpp +93 -0
- data/ext/ruby_whisper_vad_params.c +288 -0
- data/ext/sources/CMakeGraphVizOptions.cmake +8 -0
- data/ext/sources/CMakeLists.txt +255 -0
- data/ext/sources/bindings/javascript/CMakeLists.txt +41 -0
- data/ext/sources/bindings/javascript/emscripten.cpp +93 -0
- data/ext/sources/bindings/javascript/libwhisper.worker.js +1 -0
- data/ext/sources/bindings/javascript/package-tmpl.json +26 -0
- data/ext/sources/bindings/javascript/package.json +26 -0
- data/ext/sources/bindings/javascript/whisper.js +19 -0
- data/ext/sources/build-xcframework.sh +547 -0
- data/ext/sources/cmake/DefaultTargetOptions.cmake +16 -0
- data/ext/sources/cmake/FindFFmpeg.cmake +163 -0
- data/ext/sources/cmake/build-info.cmake +60 -0
- data/ext/sources/cmake/git-vars.cmake +22 -0
- data/ext/sources/cmake/whisper-config.cmake.in +65 -0
- data/ext/sources/cmake/whisper.pc.in +10 -0
- data/ext/sources/examples/CMakeLists.txt +124 -0
- data/ext/sources/examples/addon.node/CMakeLists.txt +31 -0
- data/ext/sources/examples/addon.node/__test__/whisper.spec.js +133 -0
- data/ext/sources/examples/addon.node/addon.cpp +557 -0
- data/ext/sources/examples/addon.node/index.js +57 -0
- data/ext/sources/examples/addon.node/package.json +16 -0
- data/ext/sources/examples/addon.node/vad-example.js +132 -0
- data/ext/sources/examples/bench/CMakeLists.txt +8 -0
- data/ext/sources/examples/bench/bench.cpp +176 -0
- data/ext/sources/examples/bench.wasm/CMakeLists.txt +49 -0
- data/ext/sources/examples/bench.wasm/emscripten.cpp +87 -0
- data/ext/sources/examples/bench.wasm/index-tmpl.html +284 -0
- data/ext/sources/examples/cli/CMakeLists.txt +8 -0
- data/ext/sources/examples/cli/cli.cpp +1295 -0
- data/ext/sources/examples/coi-serviceworker.js +146 -0
- data/ext/sources/examples/command/CMakeLists.txt +10 -0
- data/ext/sources/examples/command/command.cpp +800 -0
- data/ext/sources/examples/command/commands.txt +9 -0
- data/ext/sources/examples/command.wasm/CMakeLists.txt +50 -0
- data/ext/sources/examples/command.wasm/emscripten.cpp +327 -0
- data/ext/sources/examples/command.wasm/index-tmpl.html +414 -0
- data/ext/sources/examples/common-ggml.cpp +238 -0
- data/ext/sources/examples/common-ggml.h +18 -0
- data/ext/sources/examples/common-sdl.cpp +227 -0
- data/ext/sources/examples/common-sdl.h +49 -0
- data/ext/sources/examples/common-whisper.cpp +175 -0
- data/ext/sources/examples/common-whisper.h +24 -0
- data/ext/sources/examples/common.cpp +675 -0
- data/ext/sources/examples/common.h +322 -0
- data/ext/sources/examples/deprecation-warning/CMakeLists.txt +6 -0
- data/ext/sources/examples/deprecation-warning/deprecation-warning.cpp +38 -0
- data/ext/sources/examples/ffmpeg-transcode.cpp +368 -0
- data/ext/sources/examples/generate-karaoke.sh +57 -0
- data/ext/sources/examples/grammar-parser.cpp +423 -0
- data/ext/sources/examples/grammar-parser.h +29 -0
- data/ext/sources/examples/helpers.js +191 -0
- data/ext/sources/examples/json.hpp +24596 -0
- data/ext/sources/examples/livestream.sh +112 -0
- data/ext/sources/examples/lsp/CMakeLists.txt +9 -0
- data/ext/sources/examples/lsp/lsp.cpp +469 -0
- data/ext/sources/examples/lsp/whisper.vim +362 -0
- data/ext/sources/examples/miniaudio.h +93468 -0
- data/ext/sources/examples/python/test_whisper_processor.py +7 -0
- data/ext/sources/examples/python/whisper_processor.py +54 -0
- data/ext/sources/examples/quantize/CMakeLists.txt +6 -0
- data/ext/sources/examples/quantize/quantize.cpp +226 -0
- data/ext/sources/examples/server/CMakeLists.txt +15 -0
- data/ext/sources/examples/server/bench.js +29 -0
- data/ext/sources/examples/server/httplib.h +10497 -0
- data/ext/sources/examples/server/server.cpp +1238 -0
- data/ext/sources/examples/server.py +115 -0
- data/ext/sources/examples/stb_vorbis.c +5584 -0
- data/ext/sources/examples/stream/CMakeLists.txt +10 -0
- data/ext/sources/examples/stream/stream.cpp +435 -0
- data/ext/sources/examples/stream.wasm/CMakeLists.txt +49 -0
- data/ext/sources/examples/stream.wasm/emscripten.cpp +216 -0
- data/ext/sources/examples/stream.wasm/index-tmpl.html +414 -0
- data/ext/sources/examples/sycl/CMakeLists.txt +9 -0
- data/ext/sources/examples/sycl/build.sh +22 -0
- data/ext/sources/examples/sycl/ls-sycl-device.cpp +11 -0
- data/ext/sources/examples/sycl/run-whisper.sh +17 -0
- data/ext/sources/examples/talk-llama/CMakeLists.txt +43 -0
- data/ext/sources/examples/talk-llama/eleven-labs.py +80 -0
- data/ext/sources/examples/talk-llama/llama-adapter.cpp +388 -0
- data/ext/sources/examples/talk-llama/llama-adapter.h +76 -0
- data/ext/sources/examples/talk-llama/llama-arch.cpp +1914 -0
- data/ext/sources/examples/talk-llama/llama-arch.h +464 -0
- data/ext/sources/examples/talk-llama/llama-batch.cpp +843 -0
- data/ext/sources/examples/talk-llama/llama-batch.h +147 -0
- data/ext/sources/examples/talk-llama/llama-chat.cpp +685 -0
- data/ext/sources/examples/talk-llama/llama-chat.h +59 -0
- data/ext/sources/examples/talk-llama/llama-context.cpp +2845 -0
- data/ext/sources/examples/talk-llama/llama-context.h +297 -0
- data/ext/sources/examples/talk-llama/llama-cparams.cpp +5 -0
- data/ext/sources/examples/talk-llama/llama-cparams.h +41 -0
- data/ext/sources/examples/talk-llama/llama-grammar.cpp +1229 -0
- data/ext/sources/examples/talk-llama/llama-grammar.h +173 -0
- data/ext/sources/examples/talk-llama/llama-graph.cpp +1693 -0
- data/ext/sources/examples/talk-llama/llama-graph.h +710 -0
- data/ext/sources/examples/talk-llama/llama-hparams.cpp +103 -0
- data/ext/sources/examples/talk-llama/llama-hparams.h +207 -0
- data/ext/sources/examples/talk-llama/llama-impl.cpp +167 -0
- data/ext/sources/examples/talk-llama/llama-impl.h +61 -0
- data/ext/sources/examples/talk-llama/llama-io.cpp +15 -0
- data/ext/sources/examples/talk-llama/llama-io.h +35 -0
- data/ext/sources/examples/talk-llama/llama-kv-cache-unified-iswa.cpp +279 -0
- data/ext/sources/examples/talk-llama/llama-kv-cache-unified-iswa.h +128 -0
- data/ext/sources/examples/talk-llama/llama-kv-cache-unified.cpp +1841 -0
- data/ext/sources/examples/talk-llama/llama-kv-cache-unified.h +303 -0
- data/ext/sources/examples/talk-llama/llama-kv-cache.h +44 -0
- data/ext/sources/examples/talk-llama/llama-kv-cells.h +439 -0
- data/ext/sources/examples/talk-llama/llama-memory-hybrid.cpp +246 -0
- data/ext/sources/examples/talk-llama/llama-memory-hybrid.h +138 -0
- data/ext/sources/examples/talk-llama/llama-memory-recurrent.cpp +1125 -0
- data/ext/sources/examples/talk-llama/llama-memory-recurrent.h +183 -0
- data/ext/sources/examples/talk-llama/llama-memory.cpp +59 -0
- data/ext/sources/examples/talk-llama/llama-memory.h +116 -0
- data/ext/sources/examples/talk-llama/llama-mmap.cpp +600 -0
- data/ext/sources/examples/talk-llama/llama-mmap.h +68 -0
- data/ext/sources/examples/talk-llama/llama-model-loader.cpp +1163 -0
- data/ext/sources/examples/talk-llama/llama-model-loader.h +169 -0
- data/ext/sources/examples/talk-llama/llama-model-saver.cpp +282 -0
- data/ext/sources/examples/talk-llama/llama-model-saver.h +37 -0
- data/ext/sources/examples/talk-llama/llama-model.cpp +15114 -0
- data/ext/sources/examples/talk-llama/llama-model.h +452 -0
- data/ext/sources/examples/talk-llama/llama-quant.cpp +1049 -0
- data/ext/sources/examples/talk-llama/llama-quant.h +1 -0
- data/ext/sources/examples/talk-llama/llama-sampling.cpp +2575 -0
- data/ext/sources/examples/talk-llama/llama-sampling.h +32 -0
- data/ext/sources/examples/talk-llama/llama-vocab.cpp +3377 -0
- data/ext/sources/examples/talk-llama/llama-vocab.h +132 -0
- data/ext/sources/examples/talk-llama/llama.cpp +358 -0
- data/ext/sources/examples/talk-llama/llama.h +1484 -0
- data/ext/sources/examples/talk-llama/prompts/talk-alpaca.txt +23 -0
- data/ext/sources/examples/talk-llama/speak +40 -0
- data/ext/sources/examples/talk-llama/speak.bat +1 -0
- data/ext/sources/examples/talk-llama/speak.ps1 +14 -0
- data/ext/sources/examples/talk-llama/talk-llama.cpp +810 -0
- data/ext/sources/examples/talk-llama/unicode-data.cpp +7034 -0
- data/ext/sources/examples/talk-llama/unicode-data.h +20 -0
- data/ext/sources/examples/talk-llama/unicode.cpp +854 -0
- data/ext/sources/examples/talk-llama/unicode.h +66 -0
- data/ext/sources/examples/vad-speech-segments/CMakeLists.txt +8 -0
- data/ext/sources/examples/vad-speech-segments/speech.cpp +149 -0
- data/ext/sources/examples/wchess/CMakeLists.txt +10 -0
- data/ext/sources/examples/wchess/libwchess/CMakeLists.txt +19 -0
- data/ext/sources/examples/wchess/libwchess/Chessboard.cpp +803 -0
- data/ext/sources/examples/wchess/libwchess/Chessboard.h +33 -0
- data/ext/sources/examples/wchess/libwchess/WChess.cpp +193 -0
- data/ext/sources/examples/wchess/libwchess/WChess.h +63 -0
- data/ext/sources/examples/wchess/libwchess/test-chessboard.cpp +117 -0
- data/ext/sources/examples/wchess/wchess.cmd/CMakeLists.txt +8 -0
- data/ext/sources/examples/wchess/wchess.cmd/wchess.cmd.cpp +251 -0
- data/ext/sources/examples/whisper.wasm/CMakeLists.txt +50 -0
- data/ext/sources/examples/whisper.wasm/emscripten.cpp +118 -0
- data/ext/sources/examples/whisper.wasm/index-tmpl.html +658 -0
- data/ext/sources/ggml/CMakeLists.txt +435 -0
- data/ext/sources/ggml/cmake/BuildTypes.cmake +54 -0
- data/ext/sources/ggml/cmake/GitVars.cmake +22 -0
- data/ext/sources/ggml/cmake/common.cmake +50 -0
- data/ext/sources/ggml/cmake/ggml-config.cmake.in +152 -0
- data/ext/{ggml → sources/ggml}/include/ggml-alloc.h +1 -1
- data/ext/{ggml → sources/ggml}/include/ggml-backend.h +10 -8
- data/ext/{ggml → sources/ggml}/include/ggml-cpp.h +2 -1
- data/ext/{ggml → sources/ggml}/include/ggml-cpu.h +11 -1
- data/ext/{ggml → sources/ggml}/include/ggml-metal.h +1 -1
- data/ext/{ggml → sources/ggml}/include/ggml-opt.h +49 -28
- data/ext/{ggml → sources/ggml}/include/ggml-rpc.h +6 -1
- data/ext/{ggml → sources/ggml}/include/ggml-vulkan.h +0 -2
- data/ext/{ggml → sources/ggml}/include/ggml.h +325 -269
- data/ext/sources/ggml/include/gguf.h +202 -0
- data/ext/sources/ggml/src/CMakeLists.txt +404 -0
- data/ext/{ggml → sources/ggml}/src/ggml-alloc.c +34 -29
- data/ext/sources/ggml/src/ggml-amx/CMakeLists.txt +107 -0
- data/ext/{ggml → sources/ggml}/src/ggml-backend-impl.h +1 -2
- data/ext/{ggml → sources/ggml}/src/ggml-backend-reg.cpp +92 -53
- data/ext/{ggml → sources/ggml}/src/ggml-backend.cpp +69 -34
- data/ext/sources/ggml/src/ggml-blas/CMakeLists.txt +87 -0
- data/ext/sources/ggml/src/ggml-cann/CMakeLists.txt +75 -0
- data/ext/sources/ggml/src/ggml-cann/Doxyfile +2579 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cann/acl_tensor.cpp +10 -4
- data/ext/{ggml → sources/ggml}/src/ggml-cann/acl_tensor.h +5 -5
- data/ext/{ggml → sources/ggml}/src/ggml-cann/aclnn_ops.cpp +1272 -1506
- data/ext/sources/ggml/src/ggml-cann/aclnn_ops.h +1125 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cann/common.h +140 -1
- data/ext/{ggml → sources/ggml}/src/ggml-cann/ggml-cann.cpp +588 -146
- data/ext/sources/ggml/src/ggml-cann/kernels/CMakeLists.txt +30 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/dup.cpp +3 -5
- data/ext/{ggml → sources/ggml}/src/ggml-common.h +16 -8
- data/ext/sources/ggml/src/ggml-cpu/CMakeLists.txt +597 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/amx.cpp +3 -2
- data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/mmq.cpp +11 -10
- data/ext/sources/ggml/src/ggml-cpu/arch/arm/cpu-feats.cpp +94 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/arm/quants.c +4114 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/arm/repack.cpp +2163 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/loongarch/quants.c +2639 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/powerpc/cpu-feats.cpp +82 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/powerpc/quants.c +2732 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/riscv/quants.c +2069 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/riscv/repack.cpp +397 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/s390/quants.c +1300 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/wasm/quants.c +1481 -0
- data/ext/{ggml/src/ggml-cpu/cpu-feats-x86.cpp → sources/ggml/src/ggml-cpu/arch/x86/cpu-feats.cpp} +5 -1
- data/ext/sources/ggml/src/ggml-cpu/arch/x86/quants.c +4311 -0
- data/ext/sources/ggml/src/ggml-cpu/arch/x86/repack.cpp +3285 -0
- data/ext/sources/ggml/src/ggml-cpu/arch-fallback.h +184 -0
- data/ext/sources/ggml/src/ggml-cpu/binary-ops.cpp +158 -0
- data/ext/sources/ggml/src/ggml-cpu/binary-ops.h +16 -0
- data/ext/sources/ggml/src/ggml-cpu/cmake/FindSIMD.cmake +100 -0
- data/ext/sources/ggml/src/ggml-cpu/common.h +73 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu-impl.h +172 -41
- data/ext/sources/ggml/src/ggml-cpu/ggml-cpu.c +3551 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cpu/ggml-cpu.cpp +78 -25
- data/ext/{ggml/src/ggml-cpu/ggml-cpu-hbm.cpp → sources/ggml/src/ggml-cpu/hbm.cpp} +1 -1
- data/ext/sources/ggml/src/ggml-cpu/kleidiai/kernels.cpp +337 -0
- data/ext/sources/ggml/src/ggml-cpu/kleidiai/kernels.h +95 -0
- data/ext/sources/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp +482 -0
- data/ext/sources/ggml/src/ggml-cpu/kleidiai/kleidiai.h +17 -0
- data/ext/sources/ggml/src/ggml-cpu/llamafile/sgemm.cpp +3594 -0
- data/ext/sources/ggml/src/ggml-cpu/llamafile/sgemm.h +19 -0
- data/ext/sources/ggml/src/ggml-cpu/ops.cpp +9786 -0
- data/ext/sources/ggml/src/ggml-cpu/ops.h +118 -0
- data/ext/sources/ggml/src/ggml-cpu/quants.c +1158 -0
- data/ext/{ggml/src/ggml-cpu/ggml-cpu-quants.h → sources/ggml/src/ggml-cpu/quants.h} +26 -0
- data/ext/sources/ggml/src/ggml-cpu/repack.cpp +1571 -0
- data/ext/sources/ggml/src/ggml-cpu/repack.h +98 -0
- data/ext/sources/ggml/src/ggml-cpu/simd-mappings.h +1184 -0
- data/ext/{ggml/src/ggml-cpu/ggml-cpu-traits.cpp → sources/ggml/src/ggml-cpu/traits.cpp} +1 -1
- data/ext/sources/ggml/src/ggml-cpu/unary-ops.cpp +186 -0
- data/ext/sources/ggml/src/ggml-cpu/unary-ops.h +28 -0
- data/ext/sources/ggml/src/ggml-cpu/vec.cpp +345 -0
- data/ext/sources/ggml/src/ggml-cpu/vec.h +1027 -0
- data/ext/sources/ggml/src/ggml-cuda/CMakeLists.txt +184 -0
- data/ext/sources/ggml/src/ggml-cuda/acc.cu +61 -0
- data/ext/sources/ggml/src/ggml-cuda/acc.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/arange.cu +34 -0
- data/ext/sources/ggml/src/ggml-cuda/arange.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/argmax.cu +91 -0
- data/ext/sources/ggml/src/ggml-cuda/argmax.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/argsort.cu +104 -0
- data/ext/sources/ggml/src/ggml-cuda/argsort.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/binbcast.cu +363 -0
- data/ext/sources/ggml/src/ggml-cuda/binbcast.cuh +9 -0
- data/ext/sources/ggml/src/ggml-cuda/clamp.cu +45 -0
- data/ext/sources/ggml/src/ggml-cuda/clamp.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/common.cuh +851 -0
- data/ext/sources/ggml/src/ggml-cuda/concat.cu +221 -0
- data/ext/sources/ggml/src/ggml-cuda/concat.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/conv-transpose-1d.cu +89 -0
- data/ext/sources/ggml/src/ggml-cuda/conv-transpose-1d.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/conv2d-dw.cu +161 -0
- data/ext/sources/ggml/src/ggml-cuda/conv2d-dw.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/conv2d-transpose.cu +91 -0
- data/ext/sources/ggml/src/ggml-cuda/conv2d-transpose.cuh +4 -0
- data/ext/sources/ggml/src/ggml-cuda/convert.cu +752 -0
- data/ext/sources/ggml/src/ggml-cuda/convert.cuh +31 -0
- data/ext/sources/ggml/src/ggml-cuda/count-equal.cu +64 -0
- data/ext/sources/ggml/src/ggml-cuda/count-equal.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/cp-async.cuh +57 -0
- data/ext/sources/ggml/src/ggml-cuda/cpy.cu +705 -0
- data/ext/sources/ggml/src/ggml-cuda/cpy.cuh +11 -0
- data/ext/sources/ggml/src/ggml-cuda/cross-entropy-loss.cu +189 -0
- data/ext/sources/ggml/src/ggml-cuda/cross-entropy-loss.cuh +7 -0
- data/ext/sources/ggml/src/ggml-cuda/dequantize.cuh +103 -0
- data/ext/sources/ggml/src/ggml-cuda/diagmask.cu +40 -0
- data/ext/sources/ggml/src/ggml-cuda/diagmask.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-common.cuh +881 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-mma-f16.cuh +1474 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f16.cu +357 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f16.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f32.cu +365 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-tile-f32.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-vec-f16.cuh +482 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-vec-f32.cuh +472 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-wmma-f16.cu +638 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn-wmma-f16.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn.cu +346 -0
- data/ext/sources/ggml/src/ggml-cuda/fattn.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/getrows.cu +275 -0
- data/ext/sources/ggml/src/ggml-cuda/getrows.cuh +15 -0
- data/ext/sources/ggml/src/ggml-cuda/ggml-cuda.cu +3647 -0
- data/ext/sources/ggml/src/ggml-cuda/gla.cu +93 -0
- data/ext/sources/ggml/src/ggml-cuda/gla.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/im2col.cu +103 -0
- data/ext/sources/ggml/src/ggml-cuda/im2col.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/mean.cu +19 -0
- data/ext/sources/ggml/src/ggml-cuda/mean.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/mma.cuh +396 -0
- data/ext/sources/ggml/src/ggml-cuda/mmq.cu +324 -0
- data/ext/sources/ggml/src/ggml-cuda/mmq.cuh +3217 -0
- data/ext/sources/ggml/src/ggml-cuda/mmv.cu +506 -0
- data/ext/sources/ggml/src/ggml-cuda/mmv.cuh +11 -0
- data/ext/sources/ggml/src/ggml-cuda/mmvq.cu +595 -0
- data/ext/sources/ggml/src/ggml-cuda/mmvq.cuh +12 -0
- data/ext/sources/ggml/src/ggml-cuda/norm.cu +458 -0
- data/ext/sources/ggml/src/ggml-cuda/norm.cuh +11 -0
- data/ext/sources/ggml/src/ggml-cuda/opt-step-adamw.cu +78 -0
- data/ext/sources/ggml/src/ggml-cuda/opt-step-adamw.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/out-prod.cu +68 -0
- data/ext/sources/ggml/src/ggml-cuda/out-prod.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/pad.cu +49 -0
- data/ext/sources/ggml/src/ggml-cuda/pad.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/pool2d.cu +94 -0
- data/ext/sources/ggml/src/ggml-cuda/pool2d.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/quantize.cu +190 -0
- data/ext/sources/ggml/src/ggml-cuda/quantize.cuh +27 -0
- data/ext/sources/ggml/src/ggml-cuda/rope.cu +456 -0
- data/ext/sources/ggml/src/ggml-cuda/rope.cuh +7 -0
- data/ext/sources/ggml/src/ggml-cuda/scale.cu +31 -0
- data/ext/sources/ggml/src/ggml-cuda/scale.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/softmax.cu +283 -0
- data/ext/sources/ggml/src/ggml-cuda/softmax.cuh +7 -0
- data/ext/sources/ggml/src/ggml-cuda/ssm-conv.cu +148 -0
- data/ext/sources/ggml/src/ggml-cuda/ssm-conv.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/ssm-scan.cu +155 -0
- data/ext/sources/ggml/src/ggml-cuda/ssm-scan.cuh +3 -0
- data/ext/sources/ggml/src/ggml-cuda/sum.cu +45 -0
- data/ext/sources/ggml/src/ggml-cuda/sum.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/sumrows.cu +26 -0
- data/ext/sources/ggml/src/ggml-cuda/sumrows.cuh +4 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_1-ncols2_16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_1-ncols2_8.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_1.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_2.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_16-ncols2_4.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_4.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_2-ncols2_8.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_32-ncols2_1.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_32-ncols2_2.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_2.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_4.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_4-ncols2_8.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_64-ncols2_1.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_1.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_2.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_4.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_8-ncols2_8.cu +10 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-f16-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_0-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q4_1-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_0-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q5_1-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs128-q8_0-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs256-f16-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f16-instance-hs64-f16-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-f16-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_0-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q4_1-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_0-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q5_1-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs128-q8_0-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs256-f16-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-f16.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/fattn-vec-f32-instance-hs64-f16-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/generate_cu_files.py +78 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq1_s.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_s.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_xs.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq2_xxs.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq3_s.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq3_xxs.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq4_nl.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-iq4_xs.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q2_k.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q3_k.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q4_k.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_1.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q5_k.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q6_k.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/template-instances/mmq-instance-q8_0.cu +5 -0
- data/ext/sources/ggml/src/ggml-cuda/tsembd.cu +47 -0
- data/ext/sources/ggml/src/ggml-cuda/tsembd.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/unary.cu +378 -0
- data/ext/sources/ggml/src/ggml-cuda/unary.cuh +66 -0
- data/ext/sources/ggml/src/ggml-cuda/upscale.cu +51 -0
- data/ext/sources/ggml/src/ggml-cuda/upscale.cuh +5 -0
- data/ext/sources/ggml/src/ggml-cuda/vecdotq.cuh +1135 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/cuda.h +1 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/hip.h +57 -0
- data/ext/{ggml → sources/ggml}/src/ggml-cuda/vendors/musa.h +7 -1
- data/ext/sources/ggml/src/ggml-cuda/wkv.cu +199 -0
- data/ext/sources/ggml/src/ggml-cuda/wkv.cuh +7 -0
- data/ext/sources/ggml/src/ggml-hip/CMakeLists.txt +135 -0
- data/ext/{ggml → sources/ggml}/src/ggml-impl.h +147 -158
- data/ext/sources/ggml/src/ggml-kompute/CMakeLists.txt +166 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/common.comp +112 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_add.comp +58 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_addrow.comp +25 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f16_f16.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f16_f32.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f32_f16.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_cpy_f32_f32.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_diagmask.comp +30 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_gelu.comp +22 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows.comp +17 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_f16.comp +31 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_f32.comp +31 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q4_0.comp +38 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q4_1.comp +39 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_getrows_q6_k.comp +44 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_f16.comp +69 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_mat_f32.comp +51 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_0.comp +33 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_1.comp +35 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q4_k.comp +140 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q6_k.comp +106 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mat_q8_0.comp +73 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mv_q_n.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_mul_mv_q_n_pre.comp +28 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_norm.comp +84 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_relu.comp +21 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rmsnorm.comp +53 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_neox_f16.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_neox_f32.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_norm_f16.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_rope_norm_f32.comp +52 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_scale.comp +19 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_scale_8.comp +23 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_silu.comp +22 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/op_softmax.comp +72 -0
- data/ext/sources/ggml/src/ggml-kompute/kompute-shaders/rope_common.comp +71 -0
- data/ext/sources/ggml/src/ggml-metal/CMakeLists.txt +121 -0
- data/ext/sources/ggml/src/ggml-metal/ggml-metal-impl.h +649 -0
- data/ext/{ggml → sources/ggml}/src/ggml-metal/ggml-metal.m +2504 -1108
- data/ext/{ggml → sources/ggml}/src/ggml-metal/ggml-metal.metal +2102 -1463
- data/ext/sources/ggml/src/ggml-musa/CMakeLists.txt +113 -0
- data/ext/sources/ggml/src/ggml-musa/mudnn.cu +112 -0
- data/ext/sources/ggml/src/ggml-musa/mudnn.cuh +12 -0
- data/ext/sources/ggml/src/ggml-opencl/CMakeLists.txt +110 -0
- data/ext/sources/ggml/src/ggml-opencl/ggml-opencl.cpp +6494 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/add.cl +83 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/argsort.cl +86 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/clamp.cl +20 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/concat.cl +109 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/cpy.cl +184 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/cvt.cl +118 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/diag_mask_inf.cl +58 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/div.cl +72 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/embed_kernel.py +26 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/gelu.cl +62 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/gemv_noshuffle.cl +268 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/gemv_noshuffle_general.cl +274 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/get_rows.cl +163 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/glu.cl +201 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/group_norm.cl +72 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/im2col_f16.cl +57 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/im2col_f32.cl +57 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul.cl +79 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mat_Ab_Bi_8x4.cl +139 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f16.cl +118 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32.cl +118 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32_1row.cl +94 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f16_f32_l4.cl +84 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_f32_f32.cl +118 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_id_q4_0_f32_8x_flat.cl +283 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32.cl +192 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_1d_16x_flat.cl +307 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_1d_8x_flat.cl +265 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_8x_flat.cl +272 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q4_0_f32_v.cl +254 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/mul_mv_q6_k.cl +190 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/norm.cl +81 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/pad.cl +30 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/relu.cl +16 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/repeat.cl +39 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/rms_norm.cl +96 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/rope.cl +721 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/scale.cl +16 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/sigmoid.cl +29 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/silu.cl +30 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_4_f16.cl +87 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_4_f32.cl +87 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_f16.cl +86 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/softmax_f32.cl +86 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/sub.cl +72 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/sum_rows.cl +39 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/tanh.cl +63 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/transpose.cl +84 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/tsembd.cl +48 -0
- data/ext/sources/ggml/src/ggml-opencl/kernels/upscale.cl +121 -0
- data/ext/{ggml → sources/ggml}/src/ggml-opt.cpp +373 -190
- data/ext/{ggml → sources/ggml}/src/ggml-quants.c +120 -128
- data/ext/sources/ggml/src/ggml-rpc/CMakeLists.txt +9 -0
- data/ext/{ggml → sources/ggml}/src/ggml-rpc/ggml-rpc.cpp +494 -84
- data/ext/sources/ggml/src/ggml-sycl/CMakeLists.txt +189 -0
- data/ext/sources/ggml/src/ggml-sycl/backend.hpp +37 -0
- data/ext/sources/ggml/src/ggml-sycl/binbcast.cpp +344 -0
- data/ext/sources/ggml/src/ggml-sycl/binbcast.hpp +39 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/common.cpp +20 -32
- data/ext/sources/ggml/src/ggml-sycl/common.hpp +561 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/concat.cpp +56 -70
- data/ext/sources/ggml/src/ggml-sycl/concat.hpp +20 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/conv.cpp +8 -12
- data/ext/sources/ggml/src/ggml-sycl/conv.hpp +20 -0
- data/ext/sources/ggml/src/ggml-sycl/convert.cpp +575 -0
- data/ext/sources/ggml/src/ggml-sycl/convert.hpp +34 -0
- data/ext/sources/ggml/src/ggml-sycl/cpy.cpp +839 -0
- data/ext/sources/ggml/src/ggml-sycl/cpy.hpp +11 -0
- data/ext/sources/ggml/src/ggml-sycl/dequantize.hpp +823 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/dmmv.cpp +188 -67
- data/ext/sources/ggml/src/ggml-sycl/dmmv.hpp +27 -0
- data/ext/sources/ggml/src/ggml-sycl/dpct/helper.hpp +2987 -0
- data/ext/sources/ggml/src/ggml-sycl/element_wise.cpp +1120 -0
- data/ext/sources/ggml/src/ggml-sycl/element_wise.hpp +84 -0
- data/ext/sources/ggml/src/ggml-sycl/gemm.hpp +102 -0
- data/ext/sources/ggml/src/ggml-sycl/getrows.cpp +212 -0
- data/ext/sources/ggml/src/ggml-sycl/getrows.hpp +20 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/ggml-sycl.cpp +1197 -1295
- data/ext/sources/ggml/src/ggml-sycl/gla.cpp +106 -0
- data/ext/sources/ggml/src/ggml-sycl/gla.hpp +8 -0
- data/ext/sources/ggml/src/ggml-sycl/im2col.cpp +136 -0
- data/ext/sources/ggml/src/ggml-sycl/im2col.hpp +21 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/mmq.cpp +60 -81
- data/ext/sources/ggml/src/ggml-sycl/mmq.hpp +33 -0
- data/ext/sources/ggml/src/ggml-sycl/mmvq.cpp +1065 -0
- data/ext/sources/ggml/src/ggml-sycl/mmvq.hpp +27 -0
- data/ext/sources/ggml/src/ggml-sycl/norm.cpp +482 -0
- data/ext/sources/ggml/src/ggml-sycl/norm.hpp +26 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/outprod.cpp +8 -17
- data/ext/sources/ggml/src/ggml-sycl/outprod.hpp +10 -0
- data/ext/sources/ggml/src/ggml-sycl/presets.hpp +74 -0
- data/ext/sources/ggml/src/ggml-sycl/quants.hpp +111 -0
- data/ext/sources/ggml/src/ggml-sycl/rope.cpp +472 -0
- data/ext/sources/ggml/src/ggml-sycl/rope.hpp +20 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/softmax.cpp +38 -28
- data/ext/sources/ggml/src/ggml-sycl/softmax.hpp +20 -0
- data/ext/sources/ggml/src/ggml-sycl/sycl_hw.cpp +15 -0
- data/ext/sources/ggml/src/ggml-sycl/sycl_hw.hpp +26 -0
- data/ext/{ggml → sources/ggml}/src/ggml-sycl/tsembd.cpp +6 -11
- data/ext/sources/ggml/src/ggml-sycl/tsembd.hpp +20 -0
- data/ext/sources/ggml/src/ggml-sycl/vecdotq.hpp +1307 -0
- data/ext/sources/ggml/src/ggml-sycl/wkv.cpp +289 -0
- data/ext/sources/ggml/src/ggml-sycl/wkv.hpp +10 -0
- data/ext/sources/ggml/src/ggml-vulkan/CMakeLists.txt +200 -0
- data/ext/sources/ggml/src/ggml-vulkan/cmake/host-toolchain.cmake.in +15 -0
- data/ext/{ggml → sources/ggml}/src/ggml-vulkan/ggml-vulkan.cpp +3822 -1335
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/CMakeLists.txt +31 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/acc.comp +29 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/add.comp +29 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/argmax.comp +51 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/argsort.comp +69 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/clamp.comp +17 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/concat.comp +41 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/contig_copy.comp +49 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/conv2d_dw.comp +105 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/conv_transpose_1d.comp +98 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy.comp +23 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy_from_quant.comp +51 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/copy_to_quant.comp +242 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/cos.comp +17 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/count_equal.comp +31 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_f32.comp +20 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_funcs.comp +462 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_funcs_cm2.comp +699 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_head.comp +13 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq1_m.comp +42 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq1_s.comp +35 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_s.comp +44 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_xs.comp +43 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq2_xxs.comp +48 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq3_s.comp +39 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq3_xxs.comp +49 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq4_nl.comp +32 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_iq4_xs.comp +34 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q2_k.comp +34 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q3_k.comp +42 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_0.comp +30 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_1.comp +32 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q4_k.comp +68 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_0.comp +34 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_1.comp +35 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q5_k.comp +70 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q6_k.comp +33 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/dequant_q8_0.comp +31 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/diag_mask_inf.comp +34 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/div.comp +27 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn.comp +337 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_base.comp +162 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_cm1.comp +360 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_cm2.comp +267 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_split_k_reduce.comp +59 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/geglu.comp +13 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/gelu.comp +25 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/gelu_quick.comp +23 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_binary_head.comp +64 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_head.comp +9 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/generic_unary_head.comp +76 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/get_rows.comp +33 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/get_rows_quant.comp +41 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/glu_head.comp +15 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/glu_main.comp +29 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/group_norm.comp +66 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/im2col.comp +100 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/l2_norm.comp +41 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/leaky_relu.comp +22 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul.comp +27 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_split_k_reduce.comp +48 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec.comp +169 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_base.comp +118 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq1_m.comp +82 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq1_s.comp +79 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_s.comp +90 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_xs.comp +87 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq2_xxs.comp +87 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq3_s.comp +90 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_iq3_xxs.comp +88 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_nc.comp +118 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_p021.comp +154 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q2_k.comp +130 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q3_k.comp +132 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q4_k.comp +136 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q5_k.comp +167 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q6_k.comp +130 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mm.comp +868 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mm_cm2.comp +441 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq.comp +442 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq_funcs.comp +99 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/norm.comp +44 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/opt_step_adamw.comp +42 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/pad.comp +28 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/pool2d.comp +74 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/quantize_q8_1.comp +77 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/reglu.comp +9 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp +21 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/repeat.comp +26 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/repeat_back.comp +37 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rms_norm.comp +61 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rms_norm_back.comp +55 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_head.comp +58 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_multi.comp +60 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_neox.comp +43 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_norm.comp +43 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/rope_vision.comp +47 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/scale.comp +24 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sigmoid.comp +20 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/silu.comp +22 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/silu_back.comp +26 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sin.comp +17 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp +173 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/soft_max_back.comp +50 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/square.comp +17 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sub.comp +29 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/sum_rows.comp +37 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/swiglu.comp +9 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp +20 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_bfloat16_support.comp +7 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_coopmat2_support.comp +7 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_coopmat_support.comp +7 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/test_integer_dot_support.comp +7 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/timestep_embedding.comp +41 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/types.comp +1373 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/upscale.comp +36 -0
- data/ext/{ggml → sources/ggml}/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp +203 -36
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/wkv6.comp +87 -0
- data/ext/sources/ggml/src/ggml-vulkan/vulkan-shaders/wkv7.comp +91 -0
- data/ext/{ggml → sources/ggml}/src/ggml.c +918 -1782
- data/ext/sources/ggml/src/ggml.cpp +26 -0
- data/ext/sources/ggml/src/gguf.cpp +1351 -0
- data/ext/{include → sources/include}/whisper.h +70 -2
- data/ext/sources/src/CMakeLists.txt +145 -0
- data/ext/sources/src/coreml/whisper-compat.h +10 -0
- data/ext/sources/src/coreml/whisper-compat.m +35 -0
- data/ext/{src → sources/src}/coreml/whisper-decoder-impl.h +27 -15
- data/ext/{src → sources/src}/coreml/whisper-decoder-impl.m +36 -10
- data/ext/{src → sources/src}/coreml/whisper-encoder-impl.h +21 -9
- data/ext/{src → sources/src}/coreml/whisper-encoder-impl.m +29 -3
- data/ext/sources/src/coreml/whisper-encoder.mm +73 -0
- data/ext/sources/src/whisper-arch.h +197 -0
- data/ext/{src → sources/src}/whisper.cpp +1966 -386
- data/ext/sources/tests/CMakeLists.txt +105 -0
- data/ext/sources/tests/earnings21/eval.mk +58 -0
- data/ext/sources/tests/earnings21/eval.py +68 -0
- data/ext/sources/tests/earnings21/normalizers/__init__.py +2 -0
- data/ext/sources/tests/earnings21/normalizers/basic.py +80 -0
- data/ext/sources/tests/earnings21/normalizers/english.json +1741 -0
- data/ext/sources/tests/earnings21/normalizers/english.py +550 -0
- data/ext/sources/tests/earnings21/requirements.txt +6 -0
- data/ext/sources/tests/en-0-ref.txt +1 -0
- data/ext/sources/tests/en-1-ref.txt +1 -0
- data/ext/sources/tests/en-2-ref.txt +1 -0
- data/ext/sources/tests/es-0-ref.txt +1 -0
- data/ext/sources/tests/librispeech/eval.mk +39 -0
- data/ext/sources/tests/librispeech/eval.py +47 -0
- data/ext/sources/tests/librispeech/normalizers/__init__.py +2 -0
- data/ext/sources/tests/librispeech/normalizers/basic.py +80 -0
- data/ext/sources/tests/librispeech/normalizers/english.json +1741 -0
- data/ext/sources/tests/librispeech/normalizers/english.py +550 -0
- data/ext/sources/tests/librispeech/requirements.txt +6 -0
- data/ext/sources/tests/run-tests.sh +130 -0
- data/ext/sources/tests/test-c.c +3 -0
- data/ext/sources/tests/test-vad-full.cpp +54 -0
- data/ext/sources/tests/test-vad.cpp +83 -0
- data/ext/sources/tests/test-whisper.js +58 -0
- data/extsources.rb +39 -5
- data/lib/whisper/context.rb +15 -0
- data/lib/whisper/model/uri.rb +202 -126
- data/lib/whisper/segment.rb +58 -0
- data/sig/whisper.rbs +510 -0
- data/test/helper.rb +24 -0
- data/{tests → test}/test_callback.rb +45 -3
- data/{tests → test}/test_error.rb +2 -2
- data/{tests → test}/test_model.rb +47 -0
- data/test/test_package.rb +51 -0
- data/test/test_params.rb +297 -0
- data/test/test_segment.rb +146 -0
- data/test/test_vad.rb +19 -0
- data/test/test_vad_params.rb +103 -0
- data/{tests → test}/test_whisper.rb +106 -36
- data/whispercpp.gemspec +5 -5
- metadata +837 -134
- data/ext/cpu.mk +0 -9
- data/ext/examples/dr_wav.h +0 -8815
- data/ext/ggml/src/ggml-cann/aclnn_ops.h +0 -592
- data/ext/ggml/src/ggml-cpu/ggml-cpu-aarch64.cpp +0 -4262
- data/ext/ggml/src/ggml-cpu/ggml-cpu-aarch64.h +0 -8
- data/ext/ggml/src/ggml-cpu/ggml-cpu-quants.c +0 -10835
- data/ext/ggml/src/ggml-cpu/ggml-cpu.c +0 -14123
- data/ext/ggml/src/ggml-cpu/llamafile/sgemm.cpp +0 -1884
- data/ext/ggml/src/ggml-cpu/llamafile/sgemm.h +0 -14
- data/ext/ggml/src/ggml-metal/ggml-metal-impl.h +0 -288
- data/ext/ggml/src/ggml-sycl/convert.cpp +0 -547
- data/ext/ggml/src/ggml-sycl/element_wise.cpp +0 -1030
- data/ext/ggml/src/ggml-sycl/im2col.cpp +0 -126
- data/ext/ggml/src/ggml-sycl/mmvq.cpp +0 -1015
- data/ext/ggml/src/ggml-sycl/norm.cpp +0 -378
- data/ext/ggml/src/ggml-sycl/rope.cpp +0 -276
- data/ext/ggml/src/ggml-sycl/wkv6.cpp +0 -141
- data/ext/metal-embed.mk +0 -17
- data/ext/metal.mk +0 -6
- data/ext/ruby_whisper.cpp +0 -1909
- data/ext/scripts/get-flags.mk +0 -38
- data/lib/whisper.rb +0 -2
- data/tests/helper.rb +0 -7
- data/tests/test_package.rb +0 -31
- data/tests/test_params.rb +0 -160
- data/tests/test_segment.rb +0 -83
- /data/ext/{ggml → sources/ggml}/include/ggml-blas.h +0 -0
- /data/ext/{ggml → sources/ggml}/include/ggml-cann.h +0 -0
- /data/ext/{ggml → sources/ggml}/include/ggml-cuda.h +0 -0
- /data/ext/{ggml → sources/ggml}/include/ggml-kompute.h +0 -0
- /data/ext/{ggml → sources/ggml}/include/ggml-opencl.h +0 -0
- /data/ext/{ggml → sources/ggml}/include/ggml-sycl.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-amx/common.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-amx/ggml-amx.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-amx/mmq.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-amx/mmq.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-blas/ggml-blas.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/ascendc_kernels.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_f16.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_f32.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_q4_0.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/get_row_q8_0.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_f16_q8_0.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_f32_q8_0.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cann/kernels/quantize_float_to_q4_0.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/amx.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/common.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-cpu/amx/mmq.h +0 -0
- /data/ext/{ggml/src/ggml-cpu/ggml-cpu-hbm.h → sources/ggml/src/ggml-cpu/hbm.h} +0 -0
- /data/ext/{ggml/src/ggml-cpu/ggml-cpu-traits.h → sources/ggml/src/ggml-cpu/traits.h} +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-kompute/ggml-kompute.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-quants.h +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-threading.cpp +0 -0
- /data/ext/{ggml → sources/ggml}/src/ggml-threading.h +0 -0
- /data/ext/{src → sources/src}/coreml/whisper-encoder.h +0 -0
- /data/ext/{src → sources/src}/openvino/whisper-openvino-encoder.cpp +0 -0
- /data/ext/{src → sources/src}/openvino/whisper-openvino-encoder.h +0 -0
- /data/{tests → test}/jfk_reader/.gitignore +0 -0
- /data/{tests → test}/jfk_reader/extconf.rb +0 -0
- /data/{tests → test}/jfk_reader/jfk_reader.c +0 -0
@@ -0,0 +1,1120 @@
|
|
1
|
+
#include "common.hpp"
|
2
|
+
#include "ggml-sycl/presets.hpp"
|
3
|
+
#include "ggml.h"
|
4
|
+
#include "element_wise.hpp"
|
5
|
+
|
6
|
+
#define SYCL_GLOBAL_ID_LOOP(K, ITEM) \
|
7
|
+
for (auto i = ITEM.get_global_id(0); i < (size_t)K; i += ITEM.get_global_range(0))
|
8
|
+
|
9
|
+
#define SYCL_LOCAL_ID_CALC(ITEM, IDX) \
|
10
|
+
(ITEM.get_local_range(IDX) * ITEM.get_group(IDX) + ITEM.get_local_id(IDX))
|
11
|
+
|
12
|
+
|
13
|
+
static void acc_f32(const float * x, const float * y, float * dst, const int ne,
|
14
|
+
const int ne10, const int ne11, const int ne12,
|
15
|
+
const int nb1, const int nb2, int offset, const sycl::nd_item<1> &item_ct1) {
|
16
|
+
const int i = SYCL_LOCAL_ID_CALC(item_ct1, 0);
|
17
|
+
if (i >= ne) {
|
18
|
+
return;
|
19
|
+
}
|
20
|
+
int src1_idx = i - offset;
|
21
|
+
int oz = src1_idx / nb2;
|
22
|
+
int oy = (src1_idx - (oz * nb2)) / nb1;
|
23
|
+
int ox = src1_idx % nb1;
|
24
|
+
if (src1_idx >= 0 && ox < ne10 && oy < ne11 && oz < ne12) {
|
25
|
+
dst[i] = x[i] + y[ox + oy * ne10 + oz * ne10 * ne11];
|
26
|
+
} else {
|
27
|
+
dst[i] = x[i];
|
28
|
+
}
|
29
|
+
}
|
30
|
+
|
31
|
+
/* Unary OP funcs */
|
32
|
+
template<typename T>
|
33
|
+
static __dpct_inline__ T op_sgn(T x) {
|
34
|
+
return x > static_cast<T>(0.f) ? static_cast<T>(1.f) : ((x < static_cast<T>(0.f) ? static_cast<T>(-1.f) : static_cast<T>(0.f)));
|
35
|
+
}
|
36
|
+
|
37
|
+
template<typename T>
|
38
|
+
static __dpct_inline__ T op_abs(T x) {
|
39
|
+
return sycl::fabs(x);
|
40
|
+
}
|
41
|
+
|
42
|
+
template<typename T>
|
43
|
+
static __dpct_inline__ T op_elu(T x) {
|
44
|
+
return (x > static_cast<T>(0.f)) ? x : sycl::expm1(x);
|
45
|
+
}
|
46
|
+
|
47
|
+
template<typename T>
|
48
|
+
static __dpct_inline__ T op_gelu(T x) {
|
49
|
+
const T GELU_COEF_A = static_cast<T>(0.044715f);
|
50
|
+
const T SQRT_2_OVER_PI = static_cast<T>(0.79788456080286535587989211986876f);
|
51
|
+
return static_cast<T>(0.5f) * x *
|
52
|
+
(static_cast<T>(1.0f) +
|
53
|
+
sycl::tanh(SQRT_2_OVER_PI * x * (static_cast<T>(1.0f) + GELU_COEF_A * x * x)));
|
54
|
+
}
|
55
|
+
|
56
|
+
template<typename T>
|
57
|
+
static __dpct_inline__ T op_silu(T x) {
|
58
|
+
return x / (static_cast<T>(1.0f) + sycl::native::exp(-x));
|
59
|
+
}
|
60
|
+
|
61
|
+
template<typename T>
|
62
|
+
static __dpct_inline__ T op_gelu_quick(T x) {
|
63
|
+
const T GELU_QUICK_COEF_LOCAL = static_cast<T>(-1.702f);
|
64
|
+
return x * (static_cast<T>(1.0f) / (static_cast<T>(1.0f) + sycl::native::exp(GELU_QUICK_COEF_LOCAL * x)));
|
65
|
+
}
|
66
|
+
|
67
|
+
template<typename T>
|
68
|
+
static __dpct_inline__ T op_gelu_erf(T x) {
|
69
|
+
const T SQRT_2_INV = static_cast<T>(0.70710678118654752440084436210484f);
|
70
|
+
return static_cast<T>(0.5f) * x * (static_cast<T>(1.0f) + sycl::erf(x * SQRT_2_INV));
|
71
|
+
}
|
72
|
+
|
73
|
+
template<typename T>
|
74
|
+
static __dpct_inline__ T op_tanh(T x) {
|
75
|
+
return sycl::tanh(x);
|
76
|
+
}
|
77
|
+
|
78
|
+
template<typename T>
|
79
|
+
static __dpct_inline__ T op_relu(T x) {
|
80
|
+
return sycl::fmax(x, static_cast<T>(0));
|
81
|
+
}
|
82
|
+
|
83
|
+
template<typename T>
|
84
|
+
static __dpct_inline__ T op_sigmoid(T x) {
|
85
|
+
return static_cast<T>(1.0f) / (static_cast<T>(1.0f) + sycl::native::exp(-x));
|
86
|
+
}
|
87
|
+
|
88
|
+
template<typename T>
|
89
|
+
static __dpct_inline__ T op_sqrt(T x) {
|
90
|
+
return sycl::sqrt(x);
|
91
|
+
}
|
92
|
+
|
93
|
+
template<typename T>
|
94
|
+
static __dpct_inline__ T op_sin(T x) {
|
95
|
+
return sycl::sin(x);
|
96
|
+
}
|
97
|
+
|
98
|
+
template<typename T>
|
99
|
+
static __dpct_inline__ T op_cos(T x) {
|
100
|
+
return sycl::cos(x);
|
101
|
+
}
|
102
|
+
|
103
|
+
template<typename T>
|
104
|
+
static __dpct_inline__ T op_hardsigmoid(T x) {
|
105
|
+
return sycl::fmin(static_cast<T>(1.0f), sycl::fmax(static_cast<T>(0.0f), (x + static_cast<T>(3.0f)) / static_cast<T>(6.0f)));
|
106
|
+
}
|
107
|
+
|
108
|
+
template<typename T>
|
109
|
+
static __dpct_inline__ T op_hardswish(T x) {
|
110
|
+
return x * sycl::fmin(static_cast<T>(1.0f), sycl::fmax(static_cast<T>(0.0f), (x + static_cast<T>(3.0f)) / static_cast<T>(6.0f)));
|
111
|
+
}
|
112
|
+
|
113
|
+
template<typename T>
|
114
|
+
static __dpct_inline__ T op_exp(T x) {
|
115
|
+
return sycl::exp(x);
|
116
|
+
}
|
117
|
+
|
118
|
+
template<typename T>
|
119
|
+
static __dpct_inline__ T op_log(T x) {
|
120
|
+
if (x <= static_cast<T>(0)) {
|
121
|
+
return neg_infinity<T>();
|
122
|
+
}
|
123
|
+
return sycl::log(x);
|
124
|
+
}
|
125
|
+
|
126
|
+
template<typename T>
|
127
|
+
static __dpct_inline__ T op_neg(T x) {
|
128
|
+
return -x;
|
129
|
+
}
|
130
|
+
|
131
|
+
template<typename T>
|
132
|
+
static __dpct_inline__ T op_step(T x) {
|
133
|
+
return (x > static_cast<T>(0.0f)) ? static_cast<T>(1.0f) : static_cast<T>(0.0f);
|
134
|
+
}
|
135
|
+
|
136
|
+
template<typename T>
|
137
|
+
static __dpct_inline__ T op_leaky_relu(T x, float negative_slope) {
|
138
|
+
T neg_slope_T = static_cast<T>(negative_slope);
|
139
|
+
return sycl::fmax(x, static_cast<T>(0)) +
|
140
|
+
sycl::fmin(x, static_cast<T>(0.0f)) * neg_slope_T;
|
141
|
+
}
|
142
|
+
|
143
|
+
template<typename T>
|
144
|
+
static __dpct_inline__ T op_sqr(T x) {
|
145
|
+
return x * x;
|
146
|
+
}
|
147
|
+
|
148
|
+
template<typename T>
|
149
|
+
static __dpct_inline__ T op_clamp(T x, float min_val, float max_val) {
|
150
|
+
return x < static_cast<T>(min_val) ? static_cast<T>(min_val) : (x > static_cast<T>(max_val) ? static_cast<T>(max_val) : x);
|
151
|
+
}
|
152
|
+
|
153
|
+
template<typename T>
|
154
|
+
static void unary_op_sgn_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
155
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
156
|
+
dst[i] = op_sgn(x[i]);
|
157
|
+
}
|
158
|
+
}
|
159
|
+
|
160
|
+
template<typename T>
|
161
|
+
static void unary_op_abs_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
162
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
163
|
+
dst[i] = op_abs(x[i]);
|
164
|
+
}
|
165
|
+
}
|
166
|
+
|
167
|
+
template<typename T>
|
168
|
+
static void unary_op_elu_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
169
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
170
|
+
dst[i] = op_elu(x[i]);
|
171
|
+
}
|
172
|
+
}
|
173
|
+
|
174
|
+
template<typename T>
|
175
|
+
static void unary_op_gelu_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
176
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
177
|
+
dst[i] = op_gelu(x[i]);
|
178
|
+
}
|
179
|
+
}
|
180
|
+
|
181
|
+
template<typename T>
|
182
|
+
static void unary_op_silu_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
183
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
184
|
+
dst[i] = op_silu(x[i]);
|
185
|
+
}
|
186
|
+
}
|
187
|
+
|
188
|
+
template<typename T>
|
189
|
+
static void unary_op_gelu_quick_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
190
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
191
|
+
dst[i] = op_gelu_quick(x[i]);
|
192
|
+
}
|
193
|
+
}
|
194
|
+
|
195
|
+
template<typename T>
|
196
|
+
static void unary_op_gelu_erf_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
197
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
198
|
+
dst[i] = op_gelu_erf(x[i]);
|
199
|
+
}
|
200
|
+
}
|
201
|
+
|
202
|
+
template<typename T>
|
203
|
+
static void unary_op_tanh_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
204
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
205
|
+
dst[i] = op_tanh(x[i]);
|
206
|
+
}
|
207
|
+
}
|
208
|
+
|
209
|
+
template<typename T>
|
210
|
+
static void unary_op_relu_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
211
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
212
|
+
dst[i] = op_relu(x[i]);
|
213
|
+
}
|
214
|
+
}
|
215
|
+
|
216
|
+
template<typename T>
|
217
|
+
static void unary_op_sigmoid_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
218
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
219
|
+
dst[i] = op_sigmoid(x[i]);
|
220
|
+
}
|
221
|
+
}
|
222
|
+
|
223
|
+
template<typename T>
|
224
|
+
static void unary_op_sqrt_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
225
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
226
|
+
dst[i] = op_sqrt(x[i]);
|
227
|
+
}
|
228
|
+
}
|
229
|
+
|
230
|
+
template<typename T>
|
231
|
+
static void unary_op_sin_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
232
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
233
|
+
dst[i] = op_sin(x[i]);
|
234
|
+
}
|
235
|
+
}
|
236
|
+
|
237
|
+
template<typename T>
|
238
|
+
static void unary_op_cos_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
239
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
240
|
+
dst[i] = op_cos(x[i]);
|
241
|
+
}
|
242
|
+
}
|
243
|
+
|
244
|
+
template<typename T>
|
245
|
+
static void unary_op_hardsigmoid_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
246
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
247
|
+
dst[i] = op_hardsigmoid(x[i]);
|
248
|
+
}
|
249
|
+
}
|
250
|
+
|
251
|
+
template<typename T>
|
252
|
+
static void unary_op_hardswish_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
253
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
254
|
+
dst[i] = op_hardswish(x[i]);
|
255
|
+
}
|
256
|
+
}
|
257
|
+
|
258
|
+
template<typename T>
|
259
|
+
static void unary_op_exp_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
260
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
261
|
+
dst[i] = op_exp(x[i]);
|
262
|
+
}
|
263
|
+
}
|
264
|
+
|
265
|
+
template<typename T>
|
266
|
+
static void unary_op_log_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
267
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
268
|
+
dst[i] = op_log(x[i]);
|
269
|
+
}
|
270
|
+
}
|
271
|
+
|
272
|
+
template<typename T>
|
273
|
+
static void unary_op_neg_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
274
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
275
|
+
dst[i] = op_neg(x[i]);
|
276
|
+
}
|
277
|
+
}
|
278
|
+
|
279
|
+
template<typename T>
|
280
|
+
static void unary_op_step_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
281
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
282
|
+
dst[i] = op_step(x[i]);
|
283
|
+
}
|
284
|
+
}
|
285
|
+
|
286
|
+
template<typename T>
|
287
|
+
static void unary_op_leaky_relu_kernel(const T * x, T * dst, const int k, float negative_slope, const sycl::nd_item<1> &item_ct1) {
|
288
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
289
|
+
dst[i] = op_leaky_relu(x[i], negative_slope);
|
290
|
+
}
|
291
|
+
}
|
292
|
+
|
293
|
+
template<typename T>
|
294
|
+
static void unary_op_sqr_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1) {
|
295
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
296
|
+
dst[i] = op_sqr(x[i]);
|
297
|
+
}
|
298
|
+
}
|
299
|
+
|
300
|
+
template<typename T>
|
301
|
+
static void unary_op_clamp_kernel(const T * x, T * dst, const int k, const sycl::nd_item<1> &item_ct1, float min_val, float max_val) {
|
302
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
303
|
+
dst[i] = op_clamp(x[i], min_val, max_val);
|
304
|
+
}
|
305
|
+
}
|
306
|
+
|
307
|
+
template<typename T>
|
308
|
+
static void upscale(const T *x, T *dst, const int nb00, const int nb01,
|
309
|
+
const int nb02, const int nb03, const int ne10, const int ne11,
|
310
|
+
const int ne12, const int ne13, const float sf0, const float sf1,
|
311
|
+
const float sf2, const float sf3, const sycl::nd_item<1> &item_ct1) {
|
312
|
+
int index = item_ct1.get_local_id(0) +
|
313
|
+
item_ct1.get_group(0) * item_ct1.get_local_range(0);
|
314
|
+
if (index >= ne10 * ne11 * ne12 * ne13) {
|
315
|
+
return;
|
316
|
+
}
|
317
|
+
// operation
|
318
|
+
int i10 = index % ne10;
|
319
|
+
int i11 = (index / ne10) % ne11;
|
320
|
+
int i12 = (index / (ne10 * ne11)) % ne12;
|
321
|
+
int i13 = (index / (ne10 * ne11 * ne12)) % ne13;
|
322
|
+
|
323
|
+
int i00 = static_cast<int>(i10 / sf0);
|
324
|
+
int i01 = static_cast<int>(i11 / sf1);
|
325
|
+
int i02 = static_cast<int>(i12 / sf2);
|
326
|
+
int i03 = static_cast<int>(i13 / sf3);
|
327
|
+
|
328
|
+
dst[index] = *(const T *)((const char *)x + i03 * nb03 + i02 * nb02 + i01 * nb01 + i00 * nb00);
|
329
|
+
}
|
330
|
+
|
331
|
+
template <typename T>
|
332
|
+
static void pad(const T *x, T *dst, const int ne0, const int ne00, const int ne01, const int ne02,
|
333
|
+
const sycl::nd_item<3> &item_ct1) {
|
334
|
+
int nidx = SYCL_LOCAL_ID_CALC(item_ct1, 2);
|
335
|
+
if (nidx >= ne0) {
|
336
|
+
return;
|
337
|
+
}
|
338
|
+
|
339
|
+
// operation
|
340
|
+
int offset_dst = nidx + item_ct1.get_group(1) * ne0 +
|
341
|
+
item_ct1.get_group(0) * ne0 * item_ct1.get_group_range(1);
|
342
|
+
if (nidx < ne00 && item_ct1.get_group(1) < (size_t) ne01 && item_ct1.get_group(0) < (size_t) ne02) {
|
343
|
+
int offset_src = nidx + item_ct1.get_group(1) * ne00 +
|
344
|
+
item_ct1.get_group(0) * ne00 * ne01;
|
345
|
+
dst[offset_dst] = x[offset_src];
|
346
|
+
} else {
|
347
|
+
dst[offset_dst] = static_cast<T>(0.0f);
|
348
|
+
}
|
349
|
+
}
|
350
|
+
|
351
|
+
template<typename T>
|
352
|
+
static void clamp(const T * x, T * dst, const float min, const float max, const int k,
|
353
|
+
const sycl::nd_item<1> &item_ct1) {
|
354
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
355
|
+
dst[i] = x[i] < static_cast<T>(min) ? static_cast<T>(min) : (x[i] > static_cast<T>(max) ? static_cast<T>(max) : x[i]);
|
356
|
+
}
|
357
|
+
}
|
358
|
+
|
359
|
+
template<typename T>
|
360
|
+
static void gated_op_fused_geglu(const T * x, const T * g, T * dst, const uint64_t k, const uint64_t n, const uint64_t o0, const uint64_t o1, const sycl::nd_item<1> &item_ct1) {
|
361
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
362
|
+
const int64_t j0 = (i / n) * o0 + (i % n);
|
363
|
+
const int64_t j1 = o0 == o1 ? j0 : (i / n) * o1 + (i % n);
|
364
|
+
dst[i] = op_gelu(x[j0]) * g[j1];
|
365
|
+
}
|
366
|
+
}
|
367
|
+
|
368
|
+
template<typename T>
|
369
|
+
static void gated_op_fused_reglu(const T * x, const T * g, T * dst, const uint64_t k, const uint64_t n, const uint64_t o0, const uint64_t o1, const sycl::nd_item<1> &item_ct1) {
|
370
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
371
|
+
const int64_t j0 = (i / n) * o0 + (i % n);
|
372
|
+
const int64_t j1 = o0 == o1 ? j0 : (i / n) * o1 + (i % n);
|
373
|
+
dst[i] = op_relu(x[j0]) * g[j1];
|
374
|
+
}
|
375
|
+
}
|
376
|
+
|
377
|
+
template<typename T>
|
378
|
+
static void gated_op_fused_swiglu(const T * x, const T * g, T * dst, const uint64_t k, const uint64_t n, const uint64_t o0, const uint64_t o1, const sycl::nd_item<1> &item_ct1) {
|
379
|
+
SYCL_GLOBAL_ID_LOOP(k, item_ct1) {
|
380
|
+
const int64_t j0 = (i / n) * o0 + (i % n);
|
381
|
+
const int64_t j1 = o0 == o1 ? j0 : (i / n) * o1 + (i % n);
|
382
|
+
dst[i] = op_silu(x[j0]) * g[j1];
|
383
|
+
}
|
384
|
+
}
|
385
|
+
|
386
|
+
namespace ggml_sycl_detail {
|
387
|
+
static void acc_f32_sycl(const float *x, const float *y, float *dst,
|
388
|
+
const int n_elements, const int ne10, const int ne11,
|
389
|
+
const int ne12, const int nb1, const int nb2,
|
390
|
+
const int offset, queue_ptr stream) {
|
391
|
+
int num_blocks = ceil_div(n_elements, SYCL_ACC_BLOCK_SIZE);
|
392
|
+
sycl_parallel_for(stream,
|
393
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) *
|
394
|
+
sycl::range<1>(SYCL_ACC_BLOCK_SIZE),
|
395
|
+
sycl::range<1>(SYCL_ACC_BLOCK_SIZE)),
|
396
|
+
[=](sycl::nd_item<1> item_ct1) {
|
397
|
+
acc_f32(x, y, dst, n_elements, ne10, ne11, ne12, nb1, nb2, offset,
|
398
|
+
item_ct1);
|
399
|
+
});
|
400
|
+
}
|
401
|
+
|
402
|
+
template<typename T>
|
403
|
+
static void upscale_sycl(const T *x, T *dst, const int nb00, const int nb01,
|
404
|
+
const int nb02, const int nb03, const int ne10, const int ne11,
|
405
|
+
const int ne12, const int ne13, const float sf0, const float sf1,
|
406
|
+
const float sf2, const float sf3, queue_ptr stream) {
|
407
|
+
int dst_size = ne10 * ne11 * ne12 * ne13;
|
408
|
+
int num_blocks = ceil_div(dst_size, SYCL_UPSCALE_BLOCK_SIZE);
|
409
|
+
sycl::range<1> gridDim(num_blocks * SYCL_UPSCALE_BLOCK_SIZE);
|
410
|
+
sycl_parallel_for<1>(
|
411
|
+
stream, sycl::nd_range<1>(gridDim, sycl::range<1>(SYCL_UPSCALE_BLOCK_SIZE)), [=](sycl::nd_item<1> item_ct1) {
|
412
|
+
upscale(x, dst, nb00, nb01, nb02, nb03, ne10, ne11, ne12, ne13, sf0, sf1, sf2, sf3, item_ct1);
|
413
|
+
});
|
414
|
+
}
|
415
|
+
|
416
|
+
template<typename T>
|
417
|
+
static void pad_sycl(const T *x, T *dst, const int ne00,
|
418
|
+
const int ne01, const int ne02, const int ne0,
|
419
|
+
const int ne1, const int ne2, queue_ptr stream) {
|
420
|
+
int num_blocks = ceil_div(ne0, SYCL_PAD_BLOCK_SIZE);
|
421
|
+
sycl::range<3> gridDim(ne2, ne1, num_blocks);
|
422
|
+
sycl_parallel_for(stream,
|
423
|
+
sycl::nd_range<3>(gridDim * sycl::range<3>(1, 1, SYCL_PAD_BLOCK_SIZE),
|
424
|
+
sycl::range<3>(1, 1, SYCL_PAD_BLOCK_SIZE)),
|
425
|
+
[=](sycl::nd_item<3> item_ct1) { pad(x, dst, ne0, ne00, ne01, ne02, item_ct1); });
|
426
|
+
}
|
427
|
+
|
428
|
+
template<typename KernelInvoker, typename... Args>
|
429
|
+
static inline void dispatch_ggml_sycl_op_unary(ggml_backend_sycl_context & ctx, ggml_tensor * dst, KernelInvoker kernel_invoker, Args&&... args) {
|
430
|
+
#if defined (GGML_SYCL_F16)
|
431
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32 || dst->src[0]->type == GGML_TYPE_F16);
|
432
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32 || dst->type == GGML_TYPE_F16);
|
433
|
+
#else
|
434
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32);
|
435
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32);
|
436
|
+
#endif
|
437
|
+
GGML_ASSERT(dst->src[0]->type == dst->type);
|
438
|
+
dpct::queue_ptr main_stream = ctx.stream();
|
439
|
+
SYCL_CHECK(ggml_sycl_set_device(ctx.device));
|
440
|
+
switch (dst->type) {
|
441
|
+
#if defined (GGML_SYCL_F16)
|
442
|
+
case GGML_TYPE_F16:
|
443
|
+
{
|
444
|
+
auto data_pts = cast_data<sycl::half>(dst);
|
445
|
+
kernel_invoker(data_pts.src, data_pts.dst, (int)ggml_nelements(dst->src[0]), main_stream, std::forward<Args>(args)...);
|
446
|
+
break;
|
447
|
+
}
|
448
|
+
#endif
|
449
|
+
case GGML_TYPE_F32:
|
450
|
+
{
|
451
|
+
auto data_pts = cast_data<float>(dst);
|
452
|
+
kernel_invoker(data_pts.src, data_pts.dst, (int)ggml_nelements(dst->src[0]), main_stream, std::forward<Args>(args)...);
|
453
|
+
break;
|
454
|
+
}
|
455
|
+
default:
|
456
|
+
GGML_ABORT("GGML tensor type not supported!\n");
|
457
|
+
}
|
458
|
+
}
|
459
|
+
|
460
|
+
template<typename KernelInvoker, typename... Args>
|
461
|
+
static inline void dispatch_ggml_sycl_op_fused_glu(ggml_backend_sycl_context & ctx, ggml_tensor * dst, KernelInvoker kernel_invoker, Args&&... args) {
|
462
|
+
#if defined (GGML_SYCL_F16)
|
463
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32 || dst->src[0]->type == GGML_TYPE_F16);
|
464
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32 || dst->type == GGML_TYPE_F16);
|
465
|
+
#else
|
466
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32);
|
467
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32);
|
468
|
+
#endif
|
469
|
+
GGML_ASSERT(dst->src[0]->type == dst->type);
|
470
|
+
dpct::queue_ptr main_stream = ctx.stream();
|
471
|
+
SYCL_CHECK(ggml_sycl_set_device(ctx.device));
|
472
|
+
const ggml_tensor * src0 = dst->src[0];
|
473
|
+
const ggml_tensor * src1 = dst->src[1];
|
474
|
+
const int64_t nc = src1 ? src0->ne[0] : src0->ne[0] / 2;;
|
475
|
+
GGML_ASSERT(dst->ne[0] == nc);
|
476
|
+
GGML_ASSERT(ggml_is_contiguous_1(dst->src[0]));
|
477
|
+
GGML_ASSERT(ggml_is_contiguous(dst));
|
478
|
+
const int32_t swapped = ((const int32_t *) dst->op_params)[1];
|
479
|
+
void * src0_d = src0->data;
|
480
|
+
void * src1_d = src1 ? src1->data : src0->data;
|
481
|
+
const int64_t src0_o = src0->nb[1];
|
482
|
+
const int64_t src1_o = src1 ? src1->nb[1] : src0->nb[1];
|
483
|
+
void * dst_d = dst->data;
|
484
|
+
if (src1) {
|
485
|
+
GGML_ASSERT(ggml_is_contiguous_1(src1));
|
486
|
+
GGML_ASSERT(src1->nb[0] == ggml_element_size(src1));
|
487
|
+
GGML_ASSERT(src1->ne[0] == nc);
|
488
|
+
GGML_ASSERT(src0->type == src1->type);
|
489
|
+
}
|
490
|
+
switch (dst->type) {
|
491
|
+
#if defined (GGML_SYCL_F16)
|
492
|
+
case GGML_TYPE_F16:
|
493
|
+
{
|
494
|
+
sycl::half * src0_p = (sycl::half *) src0_d;
|
495
|
+
sycl::half * src1_p = (sycl::half *) src1_d;
|
496
|
+
|
497
|
+
if (!src1) {
|
498
|
+
src0_p += swapped ? nc : 0;
|
499
|
+
src1_p += swapped ? 0 : nc;
|
500
|
+
}
|
501
|
+
kernel_invoker(src0_p,
|
502
|
+
src1_p,
|
503
|
+
(sycl::half *) dst_d,
|
504
|
+
ggml_nelements(dst),
|
505
|
+
nc,
|
506
|
+
src0_o / sizeof(sycl::half),
|
507
|
+
src1_o / sizeof(sycl::half),
|
508
|
+
main_stream,
|
509
|
+
std::forward<Args>(args)...);
|
510
|
+
break;
|
511
|
+
}
|
512
|
+
#endif
|
513
|
+
case GGML_TYPE_F32:
|
514
|
+
{
|
515
|
+
float * src0_p = (float *) src0_d;
|
516
|
+
float * src1_p = (float *) src1_d;
|
517
|
+
|
518
|
+
if (!src1) {
|
519
|
+
src0_p += swapped ? nc : 0;
|
520
|
+
src1_p += swapped ? 0 : nc;
|
521
|
+
}
|
522
|
+
|
523
|
+
kernel_invoker(src0_p,
|
524
|
+
src1_p,
|
525
|
+
(float *) dst_d,
|
526
|
+
ggml_nelements(dst),
|
527
|
+
nc,
|
528
|
+
src0_o / sizeof(float),
|
529
|
+
src1_o / sizeof(float),
|
530
|
+
main_stream,
|
531
|
+
std::forward<Args>(args)...);
|
532
|
+
break;
|
533
|
+
}
|
534
|
+
default:
|
535
|
+
GGML_ABORT("GGML tensor type not supported!\n");
|
536
|
+
}
|
537
|
+
}
|
538
|
+
|
539
|
+
template<typename KernelInvoker, typename... Args>
|
540
|
+
static inline void dispatch_ggml_sycl_op_upscale(ggml_backend_sycl_context & ctx, ggml_tensor * dst, KernelInvoker kernel_invoker, Args&&... args) {
|
541
|
+
#if defined (GGML_SYCL_F16)
|
542
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32 || dst->src[0]->type == GGML_TYPE_F16);
|
543
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32 || dst->type == GGML_TYPE_F16);
|
544
|
+
#else
|
545
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32);
|
546
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32);
|
547
|
+
#endif
|
548
|
+
GGML_ASSERT(dst->src[0]->type == dst->type);
|
549
|
+
|
550
|
+
dpct::queue_ptr main_stream = ctx.stream();
|
551
|
+
SYCL_CHECK(ggml_sycl_set_device(ctx.device));
|
552
|
+
|
553
|
+
const float sf0 = (float) dst->ne[0] / dst->src[0]->ne[0];
|
554
|
+
const float sf1 = (float) dst->ne[1] / dst->src[0]->ne[1];
|
555
|
+
const float sf2 = (float) dst->ne[2] / dst->src[0]->ne[2];
|
556
|
+
const float sf3 = (float) dst->ne[3] / dst->src[0]->ne[3];
|
557
|
+
switch (dst->type) {
|
558
|
+
#if defined (GGML_SYCL_F16)
|
559
|
+
case GGML_TYPE_F16:
|
560
|
+
{
|
561
|
+
auto data_pts = cast_data<sycl::half>(dst);
|
562
|
+
kernel_invoker(data_pts.src, data_pts.dst, (int)dst->src[0]->nb[0], (int)dst->src[0]->nb[1], (int)dst->src[0]->nb[2],
|
563
|
+
(int)dst->src[0]->nb[3], (int)dst->ne[0], (int)dst->ne[1], (int)dst->ne[2], (int)dst->ne[3], sf0, sf1, sf2, sf3,
|
564
|
+
main_stream, std::forward<Args>(args)...);
|
565
|
+
break;
|
566
|
+
}
|
567
|
+
#endif
|
568
|
+
case GGML_TYPE_F32:
|
569
|
+
{
|
570
|
+
auto data_pts = cast_data<float>(dst);
|
571
|
+
kernel_invoker(data_pts.src, data_pts.dst, (int)dst->src[0]->nb[0], (int)dst->src[0]->nb[1], (int)dst->src[0]->nb[2],
|
572
|
+
(int)dst->src[0]->nb[3], (int)dst->ne[0], (int)dst->ne[1], (int)dst->ne[2], (int)dst->ne[3], sf0, sf1, sf2, sf3,
|
573
|
+
main_stream, std::forward<Args>(args)...);
|
574
|
+
break;
|
575
|
+
}
|
576
|
+
default:
|
577
|
+
GGML_ABORT("GGML tensor type not supported!\n");
|
578
|
+
}
|
579
|
+
}
|
580
|
+
|
581
|
+
template<typename KernelInvoker, typename... Args>
|
582
|
+
static inline void dispatch_ggml_sycl_op_pad(ggml_backend_sycl_context & ctx, ggml_tensor * dst, KernelInvoker kernel_invoker, Args&&... args) {
|
583
|
+
#if defined (GGML_SYCL_F16)
|
584
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32 || dst->src[0]->type == GGML_TYPE_F16);
|
585
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32 || dst->type == GGML_TYPE_F16);
|
586
|
+
#else
|
587
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32);
|
588
|
+
GGML_ASSERT(dst->type == GGML_TYPE_F32);
|
589
|
+
#endif
|
590
|
+
GGML_ASSERT(dst->src[0]->type == dst->type);
|
591
|
+
GGML_ASSERT(dst->src[0]->ne[3] == 1 && dst->ne[3] == 1); // just 3D tensors
|
592
|
+
dpct::queue_ptr main_stream = ctx.stream();
|
593
|
+
SYCL_CHECK(ggml_sycl_set_device(ctx.device));
|
594
|
+
switch (dst->type) {
|
595
|
+
#if defined (GGML_SYCL_F16)
|
596
|
+
case GGML_TYPE_F16:
|
597
|
+
{
|
598
|
+
auto data_pts = cast_data<sycl::half>(dst);
|
599
|
+
kernel_invoker(data_pts.src, data_pts.dst, (int)dst->src[0]->ne[0], (int)dst->src[0]->ne[1], (int)dst->src[0]->ne[2], (int)dst->ne[0],
|
600
|
+
(int)dst->ne[1], (int)dst->ne[2], main_stream, std::forward<Args>(args)...);
|
601
|
+
break;
|
602
|
+
}
|
603
|
+
#endif
|
604
|
+
case GGML_TYPE_F32:
|
605
|
+
{
|
606
|
+
auto data_pts = cast_data<float>(dst);
|
607
|
+
kernel_invoker(data_pts.src, data_pts.dst, (int)dst->src[0]->ne[0], (int)dst->src[0]->ne[1], (int)dst->src[0]->ne[2], (int)dst->ne[0],
|
608
|
+
(int)dst->ne[1], (int)dst->ne[2], main_stream, std::forward<Args>(args)...);
|
609
|
+
break;
|
610
|
+
}
|
611
|
+
default:
|
612
|
+
GGML_ABORT("GGML tensor type not supported!\n");
|
613
|
+
}
|
614
|
+
}
|
615
|
+
|
616
|
+
} // namespace ggml_sycl_detail
|
617
|
+
|
618
|
+
|
619
|
+
|
620
|
+
static inline void ggml_sycl_op_sgn(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
621
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
622
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
623
|
+
const int num_blocks = ceil_div(k_elements, 256);
|
624
|
+
sycl_parallel_for(stream,
|
625
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(256),
|
626
|
+
sycl::range<1>(256)),
|
627
|
+
[=](sycl::nd_item<1> item_ct1) {
|
628
|
+
unary_op_sgn_kernel(src, dst_ptr, k_elements, item_ct1);
|
629
|
+
});
|
630
|
+
});
|
631
|
+
}
|
632
|
+
|
633
|
+
static inline void ggml_sycl_op_abs(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
634
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
635
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
636
|
+
const int num_blocks = ceil_div(k_elements, 256);
|
637
|
+
sycl_parallel_for(stream,
|
638
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(256),
|
639
|
+
sycl::range<1>(256)),
|
640
|
+
[=](sycl::nd_item<1> item_ct1) {
|
641
|
+
unary_op_abs_kernel(src, dst_ptr, k_elements, item_ct1);
|
642
|
+
});
|
643
|
+
});
|
644
|
+
}
|
645
|
+
|
646
|
+
static inline void ggml_sycl_op_elu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
647
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
648
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
649
|
+
const int num_blocks = ceil_div(k_elements, 256);
|
650
|
+
sycl_parallel_for(stream,
|
651
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(256),
|
652
|
+
sycl::range<1>(256)),
|
653
|
+
[=](sycl::nd_item<1> item_ct1) {
|
654
|
+
unary_op_elu_kernel(src, dst_ptr, k_elements, item_ct1);
|
655
|
+
});
|
656
|
+
});
|
657
|
+
}
|
658
|
+
|
659
|
+
static inline void ggml_sycl_op_silu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
660
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
661
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
662
|
+
const int num_blocks = ceil_div(k_elements, SYCL_SILU_BLOCK_SIZE);
|
663
|
+
sycl_parallel_for(stream,
|
664
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_SILU_BLOCK_SIZE),
|
665
|
+
sycl::range<1>(SYCL_SILU_BLOCK_SIZE)),
|
666
|
+
[=](sycl::nd_item<1> item_ct1) {
|
667
|
+
unary_op_silu_kernel(src, dst_ptr, k_elements, item_ct1);
|
668
|
+
});
|
669
|
+
});
|
670
|
+
}
|
671
|
+
|
672
|
+
static inline void ggml_sycl_op_gelu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
673
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
674
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
675
|
+
const int num_blocks = ceil_div(k_elements, SYCL_GELU_BLOCK_SIZE);
|
676
|
+
sycl_parallel_for(stream,
|
677
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_GELU_BLOCK_SIZE),
|
678
|
+
sycl::range<1>(SYCL_GELU_BLOCK_SIZE)),
|
679
|
+
[=](sycl::nd_item<1> item_ct1) {
|
680
|
+
unary_op_gelu_kernel(src, dst_ptr, k_elements, item_ct1);
|
681
|
+
});
|
682
|
+
});
|
683
|
+
}
|
684
|
+
|
685
|
+
static inline void ggml_sycl_op_gelu_quick(ggml_backend_sycl_context & ctx, ggml_tensor *dst) {
|
686
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
687
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
688
|
+
const int num_blocks = ceil_div(k_elements, SYCL_GELU_BLOCK_SIZE);
|
689
|
+
sycl_parallel_for(stream,
|
690
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_GELU_BLOCK_SIZE),
|
691
|
+
sycl::range<1>(SYCL_GELU_BLOCK_SIZE)),
|
692
|
+
[=](sycl::nd_item<1> item_ct1) {
|
693
|
+
unary_op_gelu_quick_kernel(src, dst_ptr, k_elements, item_ct1);
|
694
|
+
});
|
695
|
+
});
|
696
|
+
}
|
697
|
+
|
698
|
+
static inline void ggml_sycl_op_gelu_erf(ggml_backend_sycl_context & ctx, ggml_tensor *dst) {
|
699
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
700
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
701
|
+
const int num_blocks = ceil_div(k_elements, SYCL_GELU_BLOCK_SIZE);
|
702
|
+
sycl_parallel_for(stream,
|
703
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_GELU_BLOCK_SIZE),
|
704
|
+
sycl::range<1>(SYCL_GELU_BLOCK_SIZE)),
|
705
|
+
[=](sycl::nd_item<1> item_ct1) {
|
706
|
+
unary_op_gelu_erf_kernel(src, dst_ptr, k_elements, item_ct1);
|
707
|
+
});
|
708
|
+
});
|
709
|
+
}
|
710
|
+
|
711
|
+
static inline void ggml_sycl_op_tanh(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
712
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
713
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
714
|
+
const int num_blocks = ceil_div(k_elements, SYCL_TANH_BLOCK_SIZE);
|
715
|
+
sycl_parallel_for(stream,
|
716
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_TANH_BLOCK_SIZE),
|
717
|
+
sycl::range<1>(SYCL_TANH_BLOCK_SIZE)),
|
718
|
+
[=](sycl::nd_item<1> item_ct1) {
|
719
|
+
unary_op_tanh_kernel(src, dst_ptr, k_elements, item_ct1);
|
720
|
+
});
|
721
|
+
});
|
722
|
+
}
|
723
|
+
|
724
|
+
static inline void ggml_sycl_op_relu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
725
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
726
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
727
|
+
const int num_blocks = ceil_div(k_elements, SYCL_RELU_BLOCK_SIZE);
|
728
|
+
sycl_parallel_for(stream,
|
729
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_RELU_BLOCK_SIZE),
|
730
|
+
sycl::range<1>(SYCL_RELU_BLOCK_SIZE)),
|
731
|
+
[=](sycl::nd_item<1> item_ct1) {
|
732
|
+
unary_op_relu_kernel(src, dst_ptr, k_elements, item_ct1);
|
733
|
+
});
|
734
|
+
});
|
735
|
+
}
|
736
|
+
|
737
|
+
static inline void ggml_sycl_op_hardsigmoid(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
738
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
739
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
740
|
+
const int num_blocks = ceil_div(k_elements, SYCL_HARDSIGMOID_BLOCK_SIZE);
|
741
|
+
sycl_parallel_for(stream,
|
742
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_HARDSIGMOID_BLOCK_SIZE),
|
743
|
+
sycl::range<1>(SYCL_HARDSIGMOID_BLOCK_SIZE)),
|
744
|
+
[=](sycl::nd_item<1> item_ct1) {
|
745
|
+
unary_op_hardsigmoid_kernel(src, dst_ptr, k_elements, item_ct1);
|
746
|
+
});
|
747
|
+
});
|
748
|
+
}
|
749
|
+
|
750
|
+
static inline void ggml_sycl_op_hardswish(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
751
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
752
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
753
|
+
const int num_blocks = ceil_div(k_elements, SYCL_HARDSWISH_BLOCK_SIZE);
|
754
|
+
sycl_parallel_for(stream,
|
755
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_HARDSWISH_BLOCK_SIZE),
|
756
|
+
sycl::range<1>(SYCL_HARDSWISH_BLOCK_SIZE)),
|
757
|
+
[=](sycl::nd_item<1> item_ct1) {
|
758
|
+
unary_op_hardswish_kernel(src, dst_ptr, k_elements, item_ct1);
|
759
|
+
});
|
760
|
+
});
|
761
|
+
}
|
762
|
+
|
763
|
+
static inline void ggml_sycl_op_exp(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
764
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
765
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
766
|
+
const int num_blocks = ceil_div(k_elements, SYCL_EXP_BLOCK_SIZE);
|
767
|
+
sycl_parallel_for(stream,
|
768
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_EXP_BLOCK_SIZE),
|
769
|
+
sycl::range<1>(SYCL_EXP_BLOCK_SIZE)),
|
770
|
+
[=](sycl::nd_item<1> item_ct1) {
|
771
|
+
unary_op_exp_kernel(src, dst_ptr, k_elements, item_ct1);
|
772
|
+
});
|
773
|
+
});
|
774
|
+
}
|
775
|
+
|
776
|
+
static inline void ggml_sycl_op_log(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
777
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
778
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
779
|
+
const int num_blocks = ceil_div(k_elements, SYCL_EXP_BLOCK_SIZE); // Using EXP block size
|
780
|
+
sycl_parallel_for(stream,
|
781
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_EXP_BLOCK_SIZE),
|
782
|
+
sycl::range<1>(SYCL_EXP_BLOCK_SIZE)),
|
783
|
+
[=](sycl::nd_item<1> item_ct1) {
|
784
|
+
unary_op_log_kernel(src, dst_ptr, k_elements, item_ct1);
|
785
|
+
});
|
786
|
+
});
|
787
|
+
}
|
788
|
+
|
789
|
+
static inline void ggml_sycl_op_neg(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
790
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
791
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
792
|
+
const int num_blocks = ceil_div(k_elements, SYCL_NEG_BLOCK_SIZE);
|
793
|
+
sycl_parallel_for(stream,
|
794
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_NEG_BLOCK_SIZE),
|
795
|
+
sycl::range<1>(SYCL_NEG_BLOCK_SIZE)),
|
796
|
+
[=](sycl::nd_item<1> item_ct1) {
|
797
|
+
unary_op_neg_kernel(src, dst_ptr, k_elements, item_ct1);
|
798
|
+
});
|
799
|
+
});
|
800
|
+
}
|
801
|
+
|
802
|
+
static inline void ggml_sycl_op_step(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
803
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
804
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
805
|
+
const int num_blocks = ceil_div(k_elements, SYCL_NEG_BLOCK_SIZE); // Using NEG block size
|
806
|
+
sycl_parallel_for(stream,
|
807
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_NEG_BLOCK_SIZE),
|
808
|
+
sycl::range<1>(SYCL_NEG_BLOCK_SIZE)),
|
809
|
+
[=](sycl::nd_item<1> item_ct1) {
|
810
|
+
unary_op_step_kernel(src, dst_ptr, k_elements, item_ct1);
|
811
|
+
});
|
812
|
+
});
|
813
|
+
}
|
814
|
+
|
815
|
+
static inline void ggml_sycl_op_sigmoid(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
816
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
817
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
818
|
+
const int num_blocks = ceil_div(k_elements, SYCL_SIGMOID_BLOCK_SIZE);
|
819
|
+
sycl_parallel_for(stream,
|
820
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_SIGMOID_BLOCK_SIZE),
|
821
|
+
sycl::range<1>(SYCL_SIGMOID_BLOCK_SIZE)),
|
822
|
+
[=](sycl::nd_item<1> item_ct1) {
|
823
|
+
unary_op_sigmoid_kernel(src, dst_ptr, k_elements, item_ct1);
|
824
|
+
});
|
825
|
+
});
|
826
|
+
}
|
827
|
+
|
828
|
+
static inline void ggml_sycl_op_sqrt(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
829
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
830
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
831
|
+
const int num_blocks = ceil_div(k_elements, SYCL_SQRT_BLOCK_SIZE);
|
832
|
+
sycl_parallel_for(stream,
|
833
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_SQRT_BLOCK_SIZE),
|
834
|
+
sycl::range<1>(SYCL_SQRT_BLOCK_SIZE)),
|
835
|
+
[=](sycl::nd_item<1> item_ct1) {
|
836
|
+
unary_op_sqrt_kernel(src, dst_ptr, k_elements, item_ct1);
|
837
|
+
});
|
838
|
+
});
|
839
|
+
}
|
840
|
+
|
841
|
+
static inline void ggml_sycl_op_sin(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
842
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
843
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
844
|
+
const int num_blocks = ceil_div(k_elements, SYCL_SIN_BLOCK_SIZE);
|
845
|
+
sycl_parallel_for(stream,
|
846
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_SIN_BLOCK_SIZE),
|
847
|
+
sycl::range<1>(SYCL_SIN_BLOCK_SIZE)),
|
848
|
+
[=](sycl::nd_item<1> item_ct1) {
|
849
|
+
unary_op_sin_kernel(src, dst_ptr, k_elements, item_ct1);
|
850
|
+
});
|
851
|
+
});
|
852
|
+
}
|
853
|
+
|
854
|
+
static inline void ggml_sycl_op_cos(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
855
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
856
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
857
|
+
const int num_blocks = ceil_div(k_elements, SYCL_SIN_BLOCK_SIZE); // Using SIN block size
|
858
|
+
sycl_parallel_for(stream,
|
859
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_SIN_BLOCK_SIZE),
|
860
|
+
sycl::range<1>(SYCL_SIN_BLOCK_SIZE)),
|
861
|
+
[=](sycl::nd_item<1> item_ct1) {
|
862
|
+
unary_op_cos_kernel(src, dst_ptr, k_elements, item_ct1);
|
863
|
+
});
|
864
|
+
});
|
865
|
+
}
|
866
|
+
|
867
|
+
static inline void ggml_sycl_op_leaky_relu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
868
|
+
float negative_slope;
|
869
|
+
memcpy(&negative_slope, dst->op_params, sizeof(float));
|
870
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
871
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream, float slope) {
|
872
|
+
const int num_blocks = ceil_div(k_elements, SYCL_RELU_BLOCK_SIZE);
|
873
|
+
sycl_parallel_for(stream,
|
874
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_RELU_BLOCK_SIZE),
|
875
|
+
sycl::range<1>(SYCL_RELU_BLOCK_SIZE)),
|
876
|
+
[=](sycl::nd_item<1> item_ct1) {
|
877
|
+
unary_op_leaky_relu_kernel(src, dst_ptr, k_elements, slope, item_ct1);
|
878
|
+
});
|
879
|
+
}, negative_slope);
|
880
|
+
}
|
881
|
+
|
882
|
+
static inline void ggml_sycl_op_sqr(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
883
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
884
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream) {
|
885
|
+
const int num_blocks = ceil_div(k_elements, SYCL_SQR_BLOCK_SIZE);
|
886
|
+
sycl_parallel_for(stream,
|
887
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_SQR_BLOCK_SIZE),
|
888
|
+
sycl::range<1>(SYCL_SQR_BLOCK_SIZE)),
|
889
|
+
[=](sycl::nd_item<1> item_ct1) {
|
890
|
+
unary_op_sqr_kernel(src, dst_ptr, k_elements, item_ct1);
|
891
|
+
});
|
892
|
+
});
|
893
|
+
}
|
894
|
+
|
895
|
+
static inline void ggml_sycl_op_upscale(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
896
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_upscale(ctx, dst,
|
897
|
+
[](const auto* src, auto* dst_ptr, int nb00, int nb01, int nb02, int nb03,
|
898
|
+
int ne10, int ne11, int ne12, int ne13, float sf0, float sf1, float sf2, float sf3,
|
899
|
+
queue_ptr stream) {
|
900
|
+
ggml_sycl_detail::upscale_sycl(src, dst_ptr, nb00, nb01, nb02, nb03, ne10, ne11, ne12, ne13, sf0, sf1, sf2, sf3, stream);
|
901
|
+
});
|
902
|
+
}
|
903
|
+
|
904
|
+
static inline void ggml_sycl_op_pad(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
905
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_pad(ctx, dst,
|
906
|
+
[](const auto* src, auto* dst_ptr, int ne00, int ne01, int ne02, int ne0, int ne1, int ne2,
|
907
|
+
queue_ptr stream) {
|
908
|
+
ggml_sycl_detail::pad_sycl(src, dst_ptr, ne00, ne01, ne02, ne0, ne1, ne2, stream);
|
909
|
+
});
|
910
|
+
}
|
911
|
+
|
912
|
+
static inline void ggml_sycl_op_clamp(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
913
|
+
float min_val;
|
914
|
+
float max_val;
|
915
|
+
memcpy(&min_val, dst->op_params, sizeof(float));
|
916
|
+
memcpy(&max_val, (float *) dst->op_params + 1, sizeof(float));
|
917
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_unary(ctx, dst,
|
918
|
+
[](const auto* src, auto* dst_ptr, int k_elements, queue_ptr stream, float min_arg, float max_arg) {
|
919
|
+
const int num_blocks = ceil_div(k_elements, SYCL_CLAMP_BLOCK_SIZE);
|
920
|
+
sycl_parallel_for(stream,
|
921
|
+
sycl::nd_range<1>(sycl::range<1>(num_blocks) * sycl::range<1>(SYCL_CLAMP_BLOCK_SIZE),
|
922
|
+
sycl::range<1>(SYCL_CLAMP_BLOCK_SIZE)),
|
923
|
+
[=](sycl::nd_item<1> item_ct1) {
|
924
|
+
clamp(src, dst_ptr, min_arg, max_arg, k_elements, item_ct1);
|
925
|
+
});
|
926
|
+
}, min_val, max_val);
|
927
|
+
}
|
928
|
+
|
929
|
+
static inline void ggml_sycl_op_acc(ggml_backend_sycl_context & ctx, ggml_tensor *dst) {
|
930
|
+
GGML_ASSERT(dst->src[0]->type == GGML_TYPE_F32);
|
931
|
+
GGML_ASSERT(dst->src[1]->type == GGML_TYPE_F32);
|
932
|
+
GGML_ASSERT( dst->type == GGML_TYPE_F32);
|
933
|
+
GGML_ASSERT(dst->ne[3] == 1); // just 3D tensors supported
|
934
|
+
dpct::queue_ptr main_stream = ctx.stream();
|
935
|
+
SYCL_CHECK(ggml_sycl_set_device(ctx.device));
|
936
|
+
const float * src0_dd = static_cast<const float *>(dst->src[0]->data);
|
937
|
+
const float * src1_dd = static_cast<const float*>(dst->src[1]->data);
|
938
|
+
float * dst_dd = static_cast<float *>(dst->data);
|
939
|
+
|
940
|
+
int nb1 = dst->op_params[0] / 4; // 4 bytes of float32
|
941
|
+
int nb2 = dst->op_params[1] / 4; // 4 bytes of float32
|
942
|
+
// int nb3 = dst->op_params[2] / 4; // 4 bytes of float32 - unused
|
943
|
+
int offset = dst->op_params[3] / 4; // offset in bytes
|
944
|
+
|
945
|
+
ggml_sycl_detail::acc_f32_sycl(src0_dd, src1_dd, dst_dd, (int)ggml_nelements(dst), (int)dst->src[1]->ne[0], (int)dst->src[1]->ne[1], (int)dst->src[1]->ne[2], nb1, nb2, offset, main_stream);
|
946
|
+
}
|
947
|
+
|
948
|
+
static inline void ggml_sycl_op_geglu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
949
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_fused_glu(ctx, dst,
|
950
|
+
[](const auto* x_ptr, const auto* g_ptr, auto* dst_ptr, uint64_t k, uint64_t n, uint64_t o0, uint64_t o1, queue_ptr main_stream) {
|
951
|
+
const uint32_t num_blocks = ceil_div(k, SYCL_GELU_BLOCK_SIZE);
|
952
|
+
sycl_parallel_for(main_stream,
|
953
|
+
sycl::nd_range<1>((num_blocks * sycl::range<1>(SYCL_GELU_BLOCK_SIZE)), sycl::range<1>(SYCL_GELU_BLOCK_SIZE)), [=](sycl::nd_item<1> item_ct1) {
|
954
|
+
gated_op_fused_geglu(x_ptr, g_ptr, dst_ptr, k, n, o0, o1, item_ct1);
|
955
|
+
});
|
956
|
+
});
|
957
|
+
}
|
958
|
+
|
959
|
+
static inline void ggml_sycl_op_reglu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
960
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_fused_glu(ctx, dst,
|
961
|
+
[](const auto* x_ptr, const auto* g_ptr, auto* dst_ptr, uint64_t k, uint64_t n, uint64_t o0, uint64_t o1, queue_ptr main_stream) {
|
962
|
+
const uint32_t num_blocks = ceil_div((uint32_t)k, SYCL_RELU_BLOCK_SIZE); // Using RELU block size for reglu
|
963
|
+
sycl_parallel_for(main_stream,
|
964
|
+
sycl::nd_range<1>((num_blocks * sycl::range<1>(SYCL_RELU_BLOCK_SIZE)), sycl::range<1>(SYCL_RELU_BLOCK_SIZE)), [=](sycl::nd_item<1> item_ct1) {
|
965
|
+
gated_op_fused_reglu(x_ptr, g_ptr, dst_ptr, k, n, o0, o1, item_ct1);
|
966
|
+
});
|
967
|
+
});
|
968
|
+
}
|
969
|
+
|
970
|
+
static inline void ggml_sycl_op_swiglu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
971
|
+
ggml_sycl_detail::dispatch_ggml_sycl_op_fused_glu(ctx, dst,
|
972
|
+
[](const auto* x_ptr, const auto* g_ptr, auto* dst_ptr, uint64_t k, uint64_t n, uint64_t o0, uint64_t o1, queue_ptr main_stream) {
|
973
|
+
const uint32_t num_blocks = ceil_div((uint32_t)k, SYCL_SILU_BLOCK_SIZE); // Using SILU block size for swiglu
|
974
|
+
sycl_parallel_for(main_stream,
|
975
|
+
sycl::nd_range<1>((num_blocks * sycl::range<1>(SYCL_SILU_BLOCK_SIZE)), sycl::range<1>(SYCL_SILU_BLOCK_SIZE)), [=](sycl::nd_item<1> item_ct1) {
|
976
|
+
gated_op_fused_swiglu(x_ptr, g_ptr, dst_ptr, k, n, o0, o1, item_ct1);
|
977
|
+
});
|
978
|
+
});
|
979
|
+
}
|
980
|
+
|
981
|
+
|
982
|
+
void ggml_sycl_sqrt(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
983
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
984
|
+
ggml_sycl_op_sqrt(ctx, dst);
|
985
|
+
}
|
986
|
+
|
987
|
+
void ggml_sycl_sin(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
988
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
989
|
+
ggml_sycl_op_sin(ctx, dst);
|
990
|
+
}
|
991
|
+
|
992
|
+
void ggml_sycl_cos(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
993
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
994
|
+
ggml_sycl_op_cos(ctx, dst);
|
995
|
+
}
|
996
|
+
|
997
|
+
void ggml_sycl_acc(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
998
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/2);
|
999
|
+
ggml_sycl_op_acc(ctx, dst);
|
1000
|
+
}
|
1001
|
+
|
1002
|
+
void ggml_sycl_gelu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1003
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1004
|
+
ggml_sycl_op_gelu(ctx, dst);
|
1005
|
+
}
|
1006
|
+
|
1007
|
+
void ggml_sycl_silu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1008
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1009
|
+
ggml_sycl_op_silu(ctx, dst);
|
1010
|
+
}
|
1011
|
+
|
1012
|
+
void ggml_sycl_gelu_quick(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1013
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1014
|
+
ggml_sycl_op_gelu_quick(ctx, dst);
|
1015
|
+
}
|
1016
|
+
|
1017
|
+
void ggml_sycl_gelu_erf(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1018
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1019
|
+
ggml_sycl_op_gelu_erf(ctx, dst);
|
1020
|
+
}
|
1021
|
+
|
1022
|
+
void ggml_sycl_tanh(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1023
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1024
|
+
ggml_sycl_op_tanh(ctx, dst);
|
1025
|
+
}
|
1026
|
+
|
1027
|
+
void ggml_sycl_relu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1028
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1029
|
+
ggml_sycl_op_relu(ctx, dst);
|
1030
|
+
}
|
1031
|
+
|
1032
|
+
void ggml_sycl_sigmoid(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1033
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1034
|
+
ggml_sycl_op_sigmoid(ctx, dst);
|
1035
|
+
}
|
1036
|
+
|
1037
|
+
void ggml_sycl_hardsigmoid(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1038
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1039
|
+
ggml_sycl_op_hardsigmoid(ctx, dst);
|
1040
|
+
}
|
1041
|
+
|
1042
|
+
void ggml_sycl_hardswish(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1043
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1044
|
+
ggml_sycl_op_hardswish(ctx, dst);
|
1045
|
+
}
|
1046
|
+
|
1047
|
+
void ggml_sycl_exp(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1048
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1049
|
+
ggml_sycl_op_exp(ctx, dst);
|
1050
|
+
}
|
1051
|
+
|
1052
|
+
void ggml_sycl_log(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1053
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1054
|
+
ggml_sycl_op_log(ctx, dst);
|
1055
|
+
}
|
1056
|
+
|
1057
|
+
void ggml_sycl_neg(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1058
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1059
|
+
ggml_sycl_op_neg(ctx, dst);
|
1060
|
+
}
|
1061
|
+
|
1062
|
+
void ggml_sycl_step(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1063
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1064
|
+
ggml_sycl_op_step(ctx, dst);
|
1065
|
+
}
|
1066
|
+
|
1067
|
+
void ggml_sycl_leaky_relu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1068
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1069
|
+
ggml_sycl_op_leaky_relu(ctx, dst);
|
1070
|
+
}
|
1071
|
+
|
1072
|
+
void ggml_sycl_sqr(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1073
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1074
|
+
ggml_sycl_op_sqr(ctx, dst);
|
1075
|
+
}
|
1076
|
+
|
1077
|
+
void ggml_sycl_upscale(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1078
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1079
|
+
ggml_sycl_op_upscale(ctx, dst);
|
1080
|
+
}
|
1081
|
+
|
1082
|
+
void ggml_sycl_pad(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1083
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1084
|
+
ggml_sycl_op_pad(ctx, dst);
|
1085
|
+
}
|
1086
|
+
|
1087
|
+
void ggml_sycl_clamp(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1088
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1089
|
+
ggml_sycl_op_clamp(ctx, dst);
|
1090
|
+
}
|
1091
|
+
|
1092
|
+
void ggml_sycl_sgn(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1093
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1094
|
+
ggml_sycl_op_sgn(ctx, dst);
|
1095
|
+
}
|
1096
|
+
|
1097
|
+
void ggml_sycl_abs(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1098
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1099
|
+
ggml_sycl_op_abs(ctx, dst);
|
1100
|
+
}
|
1101
|
+
|
1102
|
+
void ggml_sycl_elu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1103
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1104
|
+
ggml_sycl_op_elu(ctx, dst);
|
1105
|
+
}
|
1106
|
+
|
1107
|
+
void ggml_sycl_geglu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1108
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1109
|
+
ggml_sycl_op_geglu(ctx, dst);
|
1110
|
+
}
|
1111
|
+
|
1112
|
+
void ggml_sycl_reglu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1113
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1114
|
+
ggml_sycl_op_reglu(ctx, dst);
|
1115
|
+
}
|
1116
|
+
|
1117
|
+
void ggml_sycl_swiglu(ggml_backend_sycl_context & ctx, ggml_tensor * dst) {
|
1118
|
+
scope_op_debug_print scope_dbg_print(__func__, dst, /*num_src=*/1);
|
1119
|
+
ggml_sycl_op_swiglu(ctx, dst);
|
1120
|
+
}
|