DeepSpeed
zero/mics.py: use on_accelerator instead of cuda only
#3806
Merged

Loading