DeepSpeed
4f477328 - [NPU] replace 'cuda' with get_accelerator().device_name() (#5095)

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.
Author
Parents
Loading