DeepSpeed
f69f8840 - Removal of cuda hardcoded string with get_device function (#5351)

Commit
1 year ago
Removal of cuda hardcoded string with get_device function (#5351) In UTs removed 'cuda' string hardcode by replacing with device variable set to get_accelerator().device_name() Co-authored-by: Shaik Raza Sikander <srsikander@habana.ai> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Author
Parents
Loading