fix 'Node' object is not iterable in functorch.compile.minifier (#103011)

makslevental · pytorchmergebot · commit e066056414ae · 2023-09-12T23:47:40.000Z
Fixes #102169 Pull Request resolved: #103011 Approved by: https://github.com/Chillee
diff --git a/torch/_functorch/fx_minifier.py b/torch/_functorch/fx_minifier.py
@@ -261,6 +261,9 @@ def remove_outputs(cur_graph, cur_inps, granularity):
                 output = node
                 break
 
+        if isinstance(output.args[0], fx.Node):
+            return None
+
         output_args = sorted(output.args[0], key=lambda x: x.idx if isinstance(x, fx.Node) else int(1e9))
         if len(output_args) == 1:
             return None
