]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
CUDA: use async data loading for FlashAttention (llama/11894)
authorJohannes Gäßler <redacted>
Mon, 17 Feb 2025 13:03:24 +0000 (14:03 +0100)
committerGeorgi Gerganov <redacted>
Tue, 25 Feb 2025 11:33:09 +0000 (13:33 +0200)
commit929463763595d8101f8d1132b17fb850849eec4d
treed1ae13bbd527d2ba8bd2f2dfc620d1fd5f87527c
parenta80831e6e96ea9bbb281795e2cbcd677da9e0146
CUDA: use async data loading for FlashAttention (llama/11894)

* CUDA: use async data loading for FlashAttention

---------

Co-authored-by: Diego Devesa <redacted>
src/ggml-cuda/common.cuh
src/ggml-cuda/cp-async.cuh [new file with mode: 0644]
src/ggml-cuda/fattn-common.cuh
src/ggml-cuda/fattn-mma-f16.cuh
src/ggml-cuda/mma.cuh
src/ggml-cuda/mmq.cuh