Files
nvfp4-megamoe-kernel/dsv4/kernels/attention
biondizzle 8bc8b21470 D1.3: Full correction_epilog with TMA store, normalize in reg before SMEM write
One-way trip: TMEM->reg (normalize) ->SMEM->GMEM
Replicates epilogue_tma_store logic with normalize step added
Uses CUTLASS helpers for correct layout handling
2026-05-23 20:54:23 +00:00
..
2026-05-23 20:08:31 +00:00