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
