Skip to content

CUDA backend: keep wide attention heads that are not a power of 2 off the Triton SDPA kernel - #24427

Open
shoumikhin wants to merge 3 commits into
pytorch:mainfrom
shoumikhin:cuda-sdpa-wide-head
Open

CUDA backend: keep wide attention heads that are not a power of 2 off the Triton SDPA kernel#24427
shoumikhin wants to merge 3 commits into
pytorch:mainfrom
shoumikhin:cuda-sdpa-wide-head

Commits

  1. Commits on Nov 2, 2026