Feat([FA4][CUTE DSL]) Add head_dim=256 support (forward + backward) - #2412

Merged

Johnsonms merged 2 commits into

Dao-AILab:mainfrom

wangsiyu:head_dim_256

Apr 23, 2026

Commits on Apr 20, 2026

[Feat] Support flash-attention head_dim 256 in CuteDSL
wangsiyu
committed
34ef830

Commits on Apr 21, 2026

Fix ruff lint errors in head_dim=256 changes
Johnsonms
committed
c1e5e8c