pytorch

ca89e794 - [SPMD][Easy] switch to tree_map_only to simplify code (#99547)

[SPMD][Easy] switch to tree_map_only to simplify code (#99547) Pull Request resolved: https://github.com/pytorch/pytorch/pull/99547 Approved by: https://github.com/fegin