Skip to content

Enable fused SDPA vector kernel for asymmetric Q/V head dims (192, 128)#3637

Open
yohann-bearzi wants to merge 2 commits into
ml-explore:mainfrom
yohann-bearzi:sdpa-asym-headdim
Open

Enable fused SDPA vector kernel for asymmetric Q/V head dims (192, 128)#3637
yohann-bearzi wants to merge 2 commits into
ml-explore:mainfrom
yohann-bearzi:sdpa-asym-headdim

Add MiMo asymmetric-head SDPA benchmark (192/128 decode, fused vs fal…

dc9a1cb
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs