tests/unit/test_activation_checkpointing.py · ec8b1cb0a0a5752bba029da4bdc91616c0f5bec7 · OpenDAS / deepspeed

Olatunji Ruwase authored Feb 12, 2021

* Activation checkpoint support for non tensor input/output

* Format fixes

* Address PR comments; Add ordering edge case tests

ec8b1cb0

test_activation_checkpointing.py 8.03 KB