pytorch

3fb6e119 - [PT-D][TP] Fix the module registration in TP API (#93412)

[PT-D][TP] Fix the module registration in TP API (#93412) Pull Request resolved: https://github.com/pytorch/pytorch/pull/93412 Approved by: https://github.com/XilunWu