DeepSpeed
807d1b5d - scripts/check-torchcuda.py: add checking for tensor.is_cuda (#3843)

Commit
2 years ago
scripts/check-torchcuda.py: add checking for tensor.is_cuda (#3843) .cpp files are excluded Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
Parents
Loading