AstrAI/csrc
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
..
kernels fix: correct gqa_decode_attn reduction + add gqa_prefill_attn 2026-07-06 13:45:18 +08:00
__init__.py feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention 2026-07-06 12:09:58 +08:00
build.py fix: correct gqa_decode_attn reduction + add gqa_prefill_attn 2026-07-06 13:45:18 +08:00