pytorch

65a0dcff - [Composable API] `replicate`: change to per module call, remove `mark_root_module()` (#89222)

[Composable API] `replicate`: change to per module call, remove `mark_root_module()` (#89222) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89222 Approved by: https://github.com/zhaojuanmao