Files
nvfp4-megamoe-kernel/dsv4
biondizzle 40dd56eac2 KV-1: Fix shared memory corruption in block_reduce
block_reduce_sum/max write to smem[0..n_warps-1] but we passed &s_amax
(single float). For 128 threads / 4 warps, this wrote 4 floats starting
at &s_amax, corrupting adjacent shared variables (s_inv_rms, s_vals).

Fix: use s_scratch[8] array (4 for sum, 4 for max) with proper sizing.
2026-06-02 09:49:12 +00:00
..