AstrAI

Commit Graph

Author	SHA1	Message	Date
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
ViperEkura	579b8c3129	fix: correct gqa_decode_attn reduction + add gqa_prefill_attn - gqa_decode_attn: rewrite to per-KV-head, K in smem - gqa_prefill_attn: new kernel for Q_len > 1 with GQA	2026-07-06 13:45:18 +08:00

Author

SHA1

Message

Date

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

ViperEkura

579b8c3129

fix: correct gqa_decode_attn reduction + add gqa_prefill_attn

- gqa_decode_attn: rewrite to per-KV-head, K in smem
- gqa_prefill_attn: new kernel for Q_len > 1 with GQA

2026-07-06 13:45:18 +08:00

2 Commits