]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
vulkan: implement several ops relevant for ggml_opt (llama/11769)
authorRémy O <redacted>
Mon, 17 Feb 2025 06:55:57 +0000 (07:55 +0100)
committerGeorgi Gerganov <redacted>
Tue, 25 Feb 2025 11:33:09 +0000 (13:33 +0200)
commita80831e6e96ea9bbb281795e2cbcd677da9e0146
tree298a0701af059cd3806ffbe41f3eeb47970b2144
parent96dfb5d9cc7cc822402b7e63eb94c36e1adb536e
vulkan: implement several ops relevant for ggml_opt (llama/11769)

* vulkan: support memset_tensor

* vulkan: support GGML_OP_SUM

* vulkan: implement GGML_OP_ARGMAX

* vulkan: implement GGML_OP_SUB

* vulkan: implement GGML_OP_COUNT_EQUAL

* vulkan: implement GGML_OP_OPT_STEP_ADAMW

* vulkan: fix check_results RWKV_WKV6 crash and memory leaks

* vulkan: implement GGML_OP_REPEAT_BACK

* tests: remove invalid test-backend-ops REPEAT_BACK tests

* vulkan: fix COUNT_EQUAL memset using a fillBuffer command
src/ggml-vulkan/ggml-vulkan.cpp
src/ggml-vulkan/vulkan-shaders/argmax.comp [new file with mode: 0644]
src/ggml-vulkan/vulkan-shaders/count_equal.comp [new file with mode: 0644]
src/ggml-vulkan/vulkan-shaders/opt_step_adamw.comp [new file with mode: 0644]
src/ggml-vulkan/vulkan-shaders/repeat_back.comp [new file with mode: 0644]
src/ggml-vulkan/vulkan-shaders/sub.comp [new file with mode: 0644]
src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp
tests/test-backend-ops.cpp