DeepSpeed
Activation checkpointing for non-tensor arguments and return values
#741

Merged

Activation checkpointing for non-tensor arguments and return values #741

tjruwase merged 6 commits into master from olruwase/non_tensor_activation_checkpoint

Activation checkpoint support for non tensor input/output

97484a97

tjruwase requested a review from

ShadenSmith 4 years ago

tjruwase requested a review from

samyam 4 years ago

tjruwase requested a review from

arashashari 4 years ago

tjruwase requested a review from

awan-10 4 years ago

tjruwase requested a review from

cli99 4 years ago

tjruwase requested a review from

conglongli 4 years ago

tjruwase requested a review from

eltonzheng 4 years ago

tjruwase requested a review from

jeffra 4 years ago

tjruwase requested a review from

minjiaz 4 years ago

tjruwase requested a review from

niumanar 4 years ago

tjruwase requested a review from

RezaYazdaniAminabadi 4 years ago

Format fixes

e291fbdc

Merge branch 'master' into olruwase/non_tensor_activation_checkpoint

7b422dcc

ShadenSmith approved these changes on 2021-02-12

Merge branch 'master' into olruwase/non_tensor_activation_checkpoint

9d9d4cd3

Address PR comments; Add ordering edge case tests

ddbae126

Merge branch 'olruwase/non_tensor_activation_checkpoint' of github.co…

aba1be28

tjruwase merged ec8b1cb0 into master 4 years ago

Reviewers

ShadenSmith

samyam

arashashari

awan-10

cli99

conglongli

eltonzheng

jeffra

minjiaz

niumanar

RezaYazdaniAminabadi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

DeepSpeed Activation checkpointing for non-tensor arguments and return values #741 Merged

Activation checkpointing for non-tensor arguments and return values #741

DeepSpeed
Activation checkpointing for non-tensor arguments and return values
#741

Merged