AstrAI/csrc/kernels
ViperEkura 6adc221c10 refactor: extract shared MMA utils into gqa_mma_utils.cuh
- Move mma16816, ld2, pk2, pkb, ldmatrix_x4/x2/x2_trans to shared header
- gqa_prefill_attn_mma.cuh and gqa_decode_attn_mma.cuh both include it
2026-07-07 23:01:15 +08:00
..
gqa_common.cuh fix: correct prefill mask index, unify GQA kernel interface 2026-07-06 17:21:23 +08:00
gqa_decode_attn.cu feat: MMA head-packing decode kernel with scalar fallback dispatch 2026-07-07 22:56:02 +08:00
gqa_decode_attn.cuh fix: correct prefill mask index, unify GQA kernel interface 2026-07-06 17:21:23 +08:00
gqa_decode_attn_mma.cuh refactor: extract shared MMA utils into gqa_mma_utils.cuh 2026-07-07 23:01:15 +08:00
gqa_mma_utils.cuh refactor: extract shared MMA utils into gqa_mma_utils.cuh 2026-07-07 23:01:15 +08:00
gqa_prefill_attn.cu feat: ldmatrix + smem padding for mma prefill kernel 2026-07-06 20:55:22 +08:00
gqa_prefill_attn.cuh feat: tensor-core mma prefill with build-time dispatch 2026-07-06 20:33:24 +08:00
gqa_prefill_attn_mma.cuh refactor: extract shared MMA utils into gqa_mma_utils.cuh 2026-07-07 23:01:15 +08:00