sdpa: replace the experimental torch op with cudnn::sdpa_fwd / cudnn::sdpa_bwd - #780

Merged

vedaanta merged 5 commits into

NVIDIA:developfrom

vedaanta:vagarwalla/remove-experimental-sdpa

Aug 31, 2026

Commits on Aug 31, 2026

sdpa: add torch custom ops cudnn::sdpa_fwd / cudnn::sdpa_bwd

vedaanta
and
claude
committed
af96be7
test(sdpa): drop the #613 xfail from test_thd_kv_packed_views

vedaanta
and
claude
committed
df06b39
refactor(sdpa): remove the experimental SDPA torch op

vedaanta
and
claude
committed
a878d8f
docs/test: fold the torch-op docs into Attention.md, move the test under sdpa/

vedaanta
and
claude
committed
ce82916
fix(sdpa): address review findings on the torch ops

vedaanta
and
claude
committed
83e135b