utils.py: reset_peak_memory_stats when empty cache
#2803

Merged

deepspeed/runtime/utils.py: reset_peak_memory_stats when empty cache #2803

tjruwase merged 5 commits into deepspeedai:master from guoyejun:yjguo/reset_peak_memory_stats

guoyejun requested a review from

jeffra 2 years ago

guoyejun requested a review from

tjruwase 2 years ago

tjruwase approved these changes on 2023-02-08

deepspeed/runtime/utils.py: reset_peak_memory_stats when empty cache

68c90e30

guoyejun force pushed from 21aab87c to 68c90e30 2 years ago

Merge branch 'master' into yjguo/reset_peak_memory_stats

5d306692

Merge branch 'master' into yjguo/reset_peak_memory_stats

ee864d4c

Merge branch 'master' into yjguo/reset_peak_memory_stats

9ed3db3a

Merge branch 'master' into yjguo/reset_peak_memory_stats

0fd67018

tjruwase merged 6eca037c into master 2 years ago

Reviewers

tjruwase

jeffra

Assignees

No one assigned

Labels

None yet

Milestone

No milestone