DeepSpeed
807d1b5d
- scripts/check-torchcuda.py: add checking for tensor.is_cuda (#3843)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
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>
References
#3843 - scripts/check-torchcuda.py: add checking for tensor.is_cuda
Author
guoyejun
Parents
2ded2ff0
Loading