pytorch

0d8e53df - Revert "[Composable API] `replicate`: change to per module call, remove `mark_root_module()` (#89222)"

Revert "[Composable API] `replicate`: change to per module call, remove `mark_root_module()` (#89222)" This reverts commit 65a0dcffd8d387bb8c90216e63fdabb6e33e4e4d. Reverted https://github.com/pytorch/pytorch/pull/89222 on behalf of https://github.com/malfet due to Included unintended submodule updates