pytorch

c09d84d3 - [Reland] Add should_traverse_fn to torch.fx.node.map_aggregate (#81695)

[Reland] Add should_traverse_fn to torch.fx.node.map_aggregate (#81695) Test Plan: CI Differential Revision: D37956824 Pull Request resolved: https://github.com/pytorch/pytorch/pull/81695 Approved by: https://github.com/jamesr66a