sdpa: add torch custom ops cudnn::sdpa_fwd / cudnn::sdpa_bwd - #517

Closed

vedaanta wants to merge 2 commits into

NVIDIA:developfrom

vedaanta:vagarwalla/sdpa-ex-torch-ops

Commits on Aug 27, 2026

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

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

vedaanta
and
claude
committed
4af94b0