Skip to content

[JAX] [PyT] [Common] Enable D=256 BWD cuDNN fused attn for Blackwell CC 10.x - #3056

Merged
KshitijLakhani merged 34 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/feat/enable-headdim256-bwd-sm100
Jul 31, 2026
Merged

[JAX] [PyT] [Common] Enable D=256 BWD cuDNN fused attn for Blackwell CC 10.x #3056
KshitijLakhani merged 34 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/feat/enable-headdim256-bwd-sm100

Use is_training in DPA test helper

bca22ba
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jul 31, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO