- Convert AttentionParams to a template struct supporting arbitrary types - Rename attn_common.cuh -> attn_common.h (no CUDA-specific code remains) - Include standard headers explicitly in each .cuh instead of via attn_common.cuh - Allow .h files in csrc/ via .gitignore |
||
|---|---|---|
| .. | ||
| attn_common.h | ||
| attn_decode.cu | ||
| attn_decode_split_kv.cuh | ||
| attn_decode_split_kv_mma.cuh | ||
| attn_entry_utils.cuh | ||
| attn_mma_utils.cuh | ||
| attn_prefill.cu | ||
| attn_prefill_split_q.cuh | ||
| attn_prefill_split_q_mma.cuh | ||