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

Commit

2 years ago

zero/mics.py: use on_accelerator instead of cuda only (#3806) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

References

Author

guoyejun

Parents