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:
+1
-1
@@ -11,6 +11,7 @@
|
||||
!csrc/**/*.py
|
||||
|
||||
!csrc/**/*.cu
|
||||
!csrc/**/*.cuh
|
||||
|
||||
!scripts/**/*.sh
|
||||
|
||||
@@ -32,4 +33,3 @@
|
||||
|
||||
# Allow build files
|
||||
!/setup.py
|
||||
!/AGENTS.md
|
||||
|
||||
Reference in New Issue
Block a user