Skip to content

Fused SDPA full-attention for head_dim=192/256 (revival of #3293)#3660

Open
hojin12312 wants to merge 5 commits into
ml-explore:mainfrom
hojin12312:sdpa-256-revival
Open

Fused SDPA full-attention for head_dim=192/256 (revival of #3293)#3660
hojin12312 wants to merge 5 commits into
ml-explore:mainfrom
hojin12312:sdpa-256-revival

test: cover large head-dim SDPA routes

c9f8a15
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs