pytorch

2cf70321 - [Composable API][Easy] Fix some follow-ups (#90471)

[Composable API][Easy] Fix some follow-ups (#90471) Pull Request resolved: https://github.com/pytorch/pytorch/pull/90471 Approved by: https://github.com/mrshenli