pytorch

de0dee30 - [PT-D][3/N] Sync TP API change to Pytorch (#89535)

[PT-D][3/N] Sync TP API change to Pytorch (#89535) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89535 Approved by: https://github.com/wanchaol