]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
vulkan: implement several ops relevant for ggml_opt (#11769)
authorRémy O <redacted>
Mon, 17 Feb 2025 06:55:57 +0000 (07:55 +0100)
committerGitHub <redacted>
Mon, 17 Feb 2025 06:55:57 +0000 (07:55 +0100)
commit2eea03d86a2d132c8245468c26290ce07a27a8e8
tree3b58eba2ee05b8e595354eff7d36d729e9126e13
parent0f2bbe656473177538956d22b6842bcaa0449fab
vulkan: implement several ops relevant for ggml_opt (#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
ggml/src/ggml-vulkan/ggml-vulkan.cpp
ggml/src/ggml-vulkan/vulkan-shaders/argmax.comp [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/count_equal.comp [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/opt_step_adamw.comp [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/repeat_back.comp [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/sub.comp [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp
tests/test-backend-ops.cpp