Update on "[invoke_subgraph] Run missing graph passes recursively"

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy chenyang78 kadeng muchulee8 amjames chauhang aakhundov [ghstack-poisoned]
pytorch · anijain2305 · May 2, 2025 · May 2, 2025 · May 3, 2025 · May 4, 2025
commit cbf8ffc1462d518fcee0563580e884bb0fac5ae7
diff --git a/torch/_inductor/fx_passes/post_grad.py b/torch/_inductor/fx_passes/post_grad.py
@@ -1207,14 +1207,9 @@ def view_to_reshape(gm):
     ):
         nd.target = torch.ops.aten.reshape.default
 
-    subgraph_names: OrderedSet[str] = OrderedSet()
-    for node in sorted(gm.graph.find_nodes(op="get_attr")):
-        attr_name = node.target
-        if "." not in attr_name and attr_name not in subgraph_names:
-            sub_mod = getattr(gm, attr_name)
-            if isinstance(sub_mod, torch.fx.GraphModule):
-                subgraph_names.add(attr_name)
-                view_to_reshape(sub_mod)
+    for child_mod in gm.children():
+        if isinstance(child_mod, torch.fx.GraphModule):
+            view_to_reshape(child_mod)
 
 
 def should_prefer_unfused_addmm(match):