DeepSpeed
d24d3de9 - Samyamr/cpu memory bloat fix zero (#233)

Commit
5 years ago
Samyamr/cpu memory bloat fix zero (#233) * Fix for CPU memory Bloating Issue caused by pyorch backward graph creation in allgather. Fixed by calling detach on tensors before calling all_gather * Fix for CPU memory Bloating Issue caused by pyorch backward graph creation in allgather. Fixed by calling detach on tensors before calling all_gather * Fix for CPU memory Bloating Issue caused by pyorch backward graph creation in allgather. Fixed by calling detach on tensors before calling all_gather
Author
Parents
Loading