]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
rwkv6: add wkv6 support for Vulkan backend (#10829)
authorZhiyuan Li <redacted>
Mon, 16 Dec 2024 21:00:46 +0000 (05:00 +0800)
committerGitHub <redacted>
Mon, 16 Dec 2024 21:00:46 +0000 (22:00 +0100)
commit160bc039c862c10b9938269a90ddb09d794a7b0d
tree00fee9495f1d27c8f908c6115d9a7bade7373bf5
parent08ea539df211e46bb4d0dd275e541cb591d5ebc8
rwkv6: add wkv6 support for Vulkan backend (#10829)

* rwkv_wkv6 vulkan shader

* RWKV_WKV6 Vulkan op tests passed

Signed-off-by: Molly Sophia <redacted>
* Apply code format changes

Signed-off-by: Molly Sophia <redacted>
* add [[unroll]] and remove unnecessary conditions

* add uma support

* fix erros in EditorConfig Checker

---------

Signed-off-by: Molly Sophia <redacted>
Co-authored-by: Molly Sophia <redacted>
ggml/src/ggml-vulkan/ggml-vulkan.cpp
ggml/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp
ggml/src/ggml-vulkan/vulkan-shaders/wkv6.comp [new file with mode: 0644]