DeepSpeed
b4626194
- zero/mics.py: use on_accelerator instead of cuda only (#3806)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
2 years ago
zero/mics.py: use on_accelerator instead of cuda only (#3806) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
References
#3806 - zero/mics.py: use on_accelerator instead of cuda only
Author
guoyejun
Parents
f8551b43
Files
1
deepspeed/runtime/zero
mics.py
Loading