AstrAI/csrc/kernels
ViperEkura 1f0be382ad refactor: extract load_q_mma_frags template, unify comment style
- Add load_q_mma_frags<KD>() shared template in attn_mma_utils.cuh
- Replace ~15 duplicated Q-load lines in 3 MMA kernels
- Unify section header comment style to // ---- Section ----
- Remove duplicate separator line in attn_mma_utils.cuh
2026-07-15 19:07:18 +08:00
..
attn_common.h refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_decode.cu refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_decode_split_kv.cuh refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_decode_split_kv_mma.cuh refactor: extract load_q_mma_frags template, unify comment style 2026-07-15 19:07:18 +08:00
attn_entry_utils.cuh refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_mma_utils.cuh refactor: extract load_q_mma_frags template, unify comment style 2026-07-15 19:07:18 +08:00
attn_paged_decode.cu refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_paged_decode_split_kv.cuh refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_paged_decode_split_kv_mma.cuh refactor: extract load_q_mma_frags template, unify comment style 2026-07-15 19:07:18 +08:00
attn_prefill.cu refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_prefill_split_q.cuh refactor: stride-based attn interface with layout and causal mask 2026-07-14 21:34:42 +08:00
attn_prefill_split_q_mma.cuh refactor: extract load_q_mma_frags template, unify comment style 2026-07-15 19:07:18 +08:00