DeepSpeed
9aeba94a - Avoid deprecation warnings in `CHECK_CUDA` (#3854)

Commit
2 years ago
Avoid deprecation warnings in `CHECK_CUDA` (#3854) The `type()` function is deprecated and `is_cuda()` can be used since about forever. This avoids MANY warnings when compiling extensions. Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
Author
Parents
Loading