xla
Add Checkpoint api to use optimization barrier
#3524

Merged

Add Checkpoint api to use optimization barrier #3524

JackCaoG merged 3 commits into master from checkpoint_api

Add PyTorch/XLA version of checkpoint with optimization barrier

a8c7b902

Add checkpoint code

e8c8bdb9

update test

9274644e

JackCaoG force pushed from f16c2435 to 9274644e 3 years ago

JackCaoG requested a review from

miladm 3 years ago

JackCaoG merged 16498c34 into master 3 years ago

JackCaoG deleted the checkpoint_api branch 3 years ago

JackCaoG added gradient_checkpointing

Reviewers

miladm

Assignees

No one assigned

Labels

gradient_checkpointing

Milestone

No milestone