pytorch

0af3203c - fix torchrun script for custom device (#105443)

fix torchrun script for custom device (#105443) Fixes #ISSUE_NUMBER as the title,add torchrun support for custom device Pull Request resolved: https://github.com/pytorch/pytorch/pull/105443 Approved by: https://github.com/kumpera