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

Format NAX attention routing predicate

1aee415
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs