Skip to content

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

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

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

Revert "Test the checkpoint RNG guard without Philox state access"

5ea84ec
Select commit
Loading
Failed to load commit list.