DeepSpeed
4f477328
- [NPU] replace 'cuda' with get_accelerator().device_name() (#5095)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[NPU] replace 'cuda' with get_accelerator().device_name() (#5095) Replace 'cuda' with `get_accelerator().device_name()` to support other accelerators. I searched the whole repo trying to fix same issue and this seems to be the only one.
References
#5095 - [NPU] replace 'cuda' with get_accelerator().device_name()
Author
minchao-sun
Parents
b42a4706
Loading