mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-09-04 10:47:38 +02:00
* CUDA: XOR swizzle flash attn K,V smem fp16 tiles Signed-off-by: ynankani <[email protected]> * Fix use 64bit generic pointer instead of 32bit shared pointer Signed-off-by: ynankani <[email protected]> * fix shared memory race in FA on DGX Spark * Handle corener case Signed-off-by: ynankani <[email protected]> * Add swizzle test cases and gate sync for swizzled path only Signed-off-by: ynankani <[email protected]> * gate CUDA PTX Signed-off-by: ynankani <[email protected]> * offset calculation specific for swizzle branch Signed-off-by: ynankani <[email protected]> * Reafctor code Signed-off-by: ynankani <[email protected]> * Refactor FA swizzle ldmatrix if/else into helpers (K row/col, V offset) Signed-off-by: ynankani <[email protected]> * rebase and update test case args Signed-off-by: ynankani <[email protected]> * Allow swizzle for non-pow2 shapes, for which nbatch_2%32==0 Signed-off-by: ynankani <[email protected]> --------- Signed-off-by: ynankani <[email protected]>