DeepSpeed
1c9e5ef2 - Check overflow on device without host synchronization for each tensor (#5115)
Commit
View On GitHub
Comment changes are shownComment changes are hidden
Commit
1 year ago
Check overflow on device without host synchronization for each tensor (#5115) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
BacharL BacharL
Parents
fac1df99
  • deepspeed/runtime/zero
    • File
      stage_1_and_2.py
FAQ Terms Privacy Refunds Impressum