feat: split kernel defs from bindings, add prefill tiled kernel and pure C tests

- Split .cuh/.cu for gqa_decode_attn and gqa_prefill_attn
- gqa_prefill_attn: tiled shared-memory K/V, fused load, compute-opt, mask support
- Add pure C tests under csrc/tests/ for fast nvcc-only iteration
- Update .gitignore for build artifacts
This commit is contained in:
2026-07-06 16:14:55 +08:00
parent 579b8c3129
commit bcdd93e0eb
7 changed files with 438 additions and 173 deletions
+1 -1
View File
@@ -11,6 +11,7 @@
!csrc/**/*.py
!csrc/**/*.cu
!csrc/**/*.cuh
!scripts/**/*.sh
@@ -32,4 +33,3 @@
# Allow build files
!/setup.py
!/AGENTS.md