Skip to content

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

Open
wyanzhao wants to merge 4 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 4 commits into
ml-explore:mainfrom
wyanzhao:nax-256-dsplit

Merge branch 'main' into nax-256-dsplit

31f1555
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs