AstrAI/csrc/kernels
ViperEkura bcdd93e0eb 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
2026-07-06 16:14:55 +08:00
..
gqa_decode_attn.cu feat: split kernel defs from bindings, add prefill tiled kernel and pure C tests 2026-07-06 16:14:55 +08:00
gqa_decode_attn.cuh feat: split kernel defs from bindings, add prefill tiled kernel and pure C tests 2026-07-06 16:14:55 +08:00
gqa_prefill_attn.cu feat: split kernel defs from bindings, add prefill tiled kernel and pure C tests 2026-07-06 16:14:55 +08:00
gqa_prefill_attn.cuh feat: split kernel defs from bindings, add prefill tiled kernel and pure C tests 2026-07-06 16:14:55 +08:00