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

) Fixes #102169 Pull Request resolved: #103011 Approved by: https://github.com/Chillee
pytorch · Sep 12, 2023 · e066056 · e066056
1 parent 063a626
commit e066056
Showing 1 changed file with 3 additions and 0 deletions.
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