[PyTorch] Support cudagraph recomputation - #2518

Merged

ksivaman merged 5 commits into

NVIDIA:mainfrom

buptzyb:graph_backward

Dec 31, 2025

Commits on Dec 14, 2025

replace autograd.grad with autograd.backward
buptzyb
committed
7bad846

Commits on Dec 16, 2025

[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
4ee6870
get/set graphable rng state
buptzyb
committed
1dfd60b
fix lint
buptzyb
committed
de93920

Commits on Dec 31, 2025

Merge branch 'main' into graph_backward
ksivaman
authored
fd6ae4f