DeepSpeed
eb07d41f - Allow accelerator to instantiate the device (#5255)

Commit
1 year ago
Allow accelerator to instantiate the device (#5255) when instantiating torch.device for HPU it cannot be fed with HPU:1 annotation, but only "HPU". moving the logic to accelerator will allow to solve this issue, with single line change. --------- Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Joe Mayer <114769929+jomayeri@users.noreply.github.com>
Author
Parents
Loading