Commit Graph

15 Commits

Author SHA1 Message Date
183292d919 O normalize using tmem_ptr base (same as epilogue) + CUTLASS sub-tile pattern 2026-05-22 19:18:16 +00:00
365e8f53af O normalize with full layout (no sub-tiling), Repetition(64) 2026-05-22 19:17:03 +00:00
7bc94e610a Disable ALL O copies to verify baseline 2026-05-22 19:14:33 +00:00
590d6e9fba Disable O rescale too for NO-OP test 2026-05-22 19:13:43 +00:00
51cec1405d DEBUG: O load+store NO-OP to verify TMEM copy correctness 2026-05-22 19:12:43 +00:00
e3e2668192 Re-enable O rescale + normalize with corr_tile_size=32 2026-05-22 19:11:17 +00:00
d77c965646 Disable O rescale + normalize, verify softmax P only 2026-05-22 19:09:33 +00:00
dcc64dd14d FIX: O sub-tile count should be HEAD_DIM/corr_tile_size, not 128/corr_tile_size 2026-05-22 19:06:52 +00:00
48b24ba005 Full pipeline: O rescale + final normalize with CUTLASS sub-tile approach 2026-05-22 19:05:59 +00:00
a85894df89 Test softmax P vs unnormalized reference (no O normalize) 2026-05-22 19:05:27 +00:00
c0b39fc2bf O normalize using CUTLASS reference sub-tile approach 2026-05-22 19:04:37 +00:00
3dbda0eebb Fix O normalize: use 2D register tensor indexing 2026-05-22 18:59:15 +00:00
6b61d5274c Add O normalization with sub-tile TMEM read-modify-write 2026-05-22 18:58:35 +00:00
b936c6220d Simplify: softmax P only, no O rescale/normalize yet 2026-05-22 18:57:00 +00:00
e2fad84205 Real softmax test built on working identity diag 2026-05-22 18:55:31 +00:00