]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
rwkv6: add wkv6 support for Vulkan backend (llama/10829)
authorZhiyuan Li <redacted>
Mon, 16 Dec 2024 21:00:46 +0000 (05:00 +0800)
committerGeorgi Gerganov <redacted>
Tue, 17 Dec 2024 17:23:40 +0000 (19:23 +0200)
commitf46b8e1126052547e25d17206e7072419087d1d6
tree1d819ca693382afa5647192b0b0be01d3e4c3b0c
parent2db5dd291e82748c6eadf60d72f485fe006268ba
rwkv6: add wkv6 support for Vulkan backend (llama/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>
src/ggml-vulkan/ggml-vulkan.cpp
src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp
src/ggml-vulkan/vulkan-shaders/wkv6.comp [new file with mode: 0644]