xla
Add optimization barrier to enable the gradient checkpointing
#3482

Merged

Add optimization barrier to enable the gradient checkpointing #3482

JackCaoG merged 2 commits into master from optimization_barrier

Add optimization barrier

15790eff

JackCaoG force pushed from f1654d0e to 4b0ada73 4 years ago

Make optimization_barrier take vector and add test for correctness

cf19c0c6

JackCaoG force pushed from 4b0ada73 to cf19c0c6 4 years ago

JackCaoG merged 2c2f554a into master 4 years ago

JackCaoG deleted the optimization_barrier branch 4 years ago

JackCaoG added gradient_checkpointing

Reviewers

No reviews

Assignees

No one assigned

Labels

gradient_checkpointing

Milestone

No milestone