pytorch

f4edbaa6 - [PT-D] Use process group of the partial tensor so sub pg comm will be enabled during reshard

[PT-D] Use process group of the partial tensor so sub pg comm will be enabled during reshard Pull Request resolved: https://github.com/pytorch/pytorch/pull/79357 Approved by: https://github.com/wanchaol