]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
RWKV v6: RWKV_WKV op CUDA implementation (#9454)
authorMolly Sophia <redacted>
Sun, 22 Sep 2024 02:29:12 +0000 (10:29 +0800)
committerGitHub <redacted>
Sun, 22 Sep 2024 02:29:12 +0000 (04:29 +0200)
commit2a63caaa69fad6c2afddc47bae052cf2afb01529
tree7376473763f17b5af3e49e9076a8cf6895532235
parentd09770cae71b416c032ec143dda530f7413c4038
RWKV v6: RWKV_WKV op CUDA implementation (#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
tests/test-backend-ops.cpp