AstrAI/csrc/kernels
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
..
gqa_decode_attn.cu fix: correct gqa_decode_attn reduction + add gqa_prefill_attn 2026-07-06 13:45:18 +08:00
gqa_prefill_attn.cu fix: correct gqa_decode_attn reduction + add gqa_prefill_attn 2026-07-06 13:45:18 +08:00