CUDA: generalized (mma) FA, add Volta support (llama/17505)

* CUDA: generalized (mma) FA, add Volta support

* use struct for MMA FA kernel config

---------

Co-authored-by: Aman Gupta <aman>
This commit is contained in:
Johannes Gäßler
2025-12-12 17:53:16 +02:00
committed by Georgi Gerganov
co-authored by Aman Gupta
parent 4a00f2e3a4
commit 7adbcafb6c
10 changed files with 966 additions and 759 deletions
+1 -1
View File
@@ -2279,7 +2279,7 @@ extern "C" {
float stop,
float step);
#define GGML_KQ_MASK_PAD 64
#define GGML_KQ_MASK_PAD 1
// q: [n_embd_k, n_batch, n_head, ne3 ]
// k: [n_embd_k, n_kv, n_head_kv, ne3 ]