PR #1254 Activation checkpointing improvements

Rename PA_TO_cpu

tjruwase committed 4 years ago

Code cleanup

tjruwase committed 4 years ago

Revert accidental change

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

Merge branch 'master' into olruwase/improve_activation_checkpoint

tjruwase committed 4 years ago

DeepSpeed Activation checkpointing improvements #1254 Merged