Skip to content

[PyTorch] Preserve CUDA RNG state during checkpoint recompute - #3835

Open
losewayy wants to merge 2 commits into
NVIDIA:mainfrom
losewayy:fix/cudarngstates-tracker-checkpoint-dropout
Open

[PyTorch] Preserve CUDA RNG state during checkpoint recompute#3835
losewayy wants to merge 2 commits into
NVIDIA:mainfrom
losewayy:fix/cudarngstates-tracker-checkpoint-dropout

[PyTorch] Test CUDA RNG restoration after checkpoint backward

613a449
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Nov 8, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO