pytorch

6ca991ca - [Composable API] Add fully_shard debug function to print sharded tree structure, module names and managed param fqns (#99133)

[Composable API] Add fully_shard debug function to print sharded tree structure, module names and managed param fqns (#99133) Adding a fully_shard debug function to print sharded tree structure like following format, return module names and their managed parameter fqns as well.  Pull Request resolved: https://github.com/pytorch/pytorch/pull/99133 Approved by: https://github.com/rohan-varma