-
Allow accelerator to instantiate the device (#5255)
frostbyte_neo 发布于
2024-08-15 15:01:27 +00:00 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下载附件