DeepSpeed
scripts/check-torchcuda.py: add checking for tensor.is_cuda
#3843
Merged

Loading