]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commit
RWKV v6: RWKV_WKV op CUDA implementation (llama/9454)
authorMolly Sophia <redacted>
Sun, 22 Sep 2024 02:29:12 +0000 (10:29 +0800)
committerGeorgi Gerganov <redacted>
Tue, 24 Sep 2024 16:45:08 +0000 (19:45 +0300)
commit008816a25771be239b29152690c71f8abb91f8a3
treecb63ca096b4e41539f4bc1399fca92b7f8a3dfc6
parent33e5a6612e0cb0c6ee441ef5033f25a20e11c87b
RWKV v6: RWKV_WKV op CUDA implementation (llama/9454)

* ggml: CUDA unary op EXP

Signed-off-by: Molly Sophia <redacted>
* ggml: rwkv_wkv op CUDA impl

Signed-off-by: Molly Sophia <redacted>
---------

Signed-off-by: Molly Sophia <redacted>
ggml/src/ggml-cuda.cu
ggml/src/ggml-cuda/rwkv-wkv.cu [new file with mode: 0644]
ggml/src/ggml-cuda/rwkv-wkv.cuh [new file with mode: 0644]
ggml/src/ggml-cuda/unary.cu
ggml/src/ggml-cuda/unary.cuh