DeepSpeed
b4626194 - zero/mics.py: use on_accelerator instead of cuda only (#3806)

Comment changes are shownComment changes are hidden
Commit
2 years ago
zero/mics.py: use on_accelerator instead of cuda only (#3806) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
Parents
  • deepspeed/runtime/zero
    • File
      mics.py