- Replace bf16 pre-scale Q loading with direct 32-bit aligned bf16x2 reads - Apply scale in float32 after Q@K^T, before online softmax - Reduces causal max error from 2^-6 to 2^-8 with zero perf cost |
||
|---|---|---|
| .. | ||
| attn_common.cuh | ||
| 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 | ||