ViperEkura/AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
1f0be382ad9032add7d1a980e8f734b69f27b1c4
AstrAI/csrc/kernels
T
History
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_split_kv_mma.cuh
refactor: extract load_q_mma_frags template, unify comment style
2026-07-15 19:07:18 +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.cu
refactor: stride-based attn interface with layout and causal mask
2026-07-14 21:34:42 +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_split_kv_mma.cuh
refactor: extract load_q_mma_frags template, unify comment style
2026-07-15 19:07:18 +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.cu
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
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.cu
refactor: stride-based attn interface with layout and causal mask
2026-07-14 21:34:42 +08:00