Skip to content

Fix SDPA vmap with GQA/MQA shapes (n_heads != n_kv_heads)

14700f3
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Open

Fix SDPA vmap with GQA/MQA shapes (n_heads != n_kv_heads) #3385

Fix SDPA vmap with GQA/MQA shapes (n_heads != n_kv_heads)
14700f3
Select commit
Loading
Failed to load commit list.

Annotations

1 warning
Check Lint
succeeded Apr 9, 2026 in 22s