Skip to content

[PyTorch] Add head-parallel FA4 backward for cuDNN CP attention - #3510

Open
bzantium wants to merge 3 commits into
NVIDIA:mainfrom
bzantium:feat/mixed-cp-fa4-backward
Open

[PyTorch] Add head-parallel FA4 backward for cuDNN CP attention#3510
bzantium wants to merge 3 commits into
NVIDIA:mainfrom
bzantium:feat/mixed-cp-fa4-backward

Keep wide-index CP tests small by default

b517bbf
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Sep 12, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO