Update on "Add reinplacing pass for scatters + incremental fake tensor updating"

Chillee · Chillee · commit 2298b6d58a02 · 2023-08-25T14:51:50.000-07:00
mutation for params)

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 ipiszy ngimel yf225 chenyang78 kadeng muchulee8 aakhundov anijain2305

[ghstack-poisoned]
diff --git a/torch/_inductor/fx_passes/post_grad.py b/torch/_inductor/fx_passes/post_grad.py
@@ -601,6 +601,8 @@ def remove_noop_ops(graph: torch.fx.Graph):
                 src = node.args[src_index]
             else:
                 src = src_index(node.args)
+            if not isinstance(src, torch.fx.Node):
+                continue
             # See fx_passes/README.md for a discussion of why this is
             # necessary.
             if (
