[NVIDIA][CuTe,Fwd/Bwd,sm120] Attention dropout (#2436), forward + backward - #2657

Open

johnnynunez wants to merge 8 commits into

Dao-AILab:mainfrom

johnnynunez:feat/cute-dropout-2436-full

Commits on Jun 28, 2026

[CuTe,Fwd,sm120] Fix use_tma_O crash on SM120 (issue #2649)
johnnynunez
committed
e1e9ed0
[CuTe,Bwd,sm120] Enable SM120 backward: undefined dQ_single_wg + nulled softmax_scale
johnnynunez
committed
e3d39a3
[CuTe,Fwd,sm120] Fix use_tma_O crash on SM120 (issue #2649)
johnnynunez
committed
aa53725
[CuTe,Bwd,sm120] Fix SM120 backward: undefined dQ_single_wg + nulled softmax_scale
johnnynunez
committed
ef1958b
[CuTe,Fwd/Bwd,sm120] Dropout support (#2436) for forward + backward
johnnynunez
committed
5a54626

Commits on Jul 7, 2026