Skip to content

Add a fused full-attention path for head_dim 256 on NAX devices#3842

Open
wyanzhao wants to merge 2 commits into
ml-explore:mainfrom
wyanzhao:nax-256-dsplit
Open

Add a fused full-attention path for head_dim 256 on NAX devices#3842
wyanzhao wants to merge 2 commits into
ml-explore:mainfrom
wyanzhao:nax-256-dsplit

Commits

Commits on Jul 13, 2026