Apply all replacements on backward graph args during inductor codegen. (#155469)

laithsakka · pytorchmergebot · commit a205e8fd7357 · 2025-06-10T08:56:18.000Z
Summary: temporary mitigation for #155468 Test Plan: NA Rollback Plan: Differential Revision: D76096355 Pull Request resolved: #155469 Approved by: https://github.com/bobrenjc93
diff --git a/torch/_inductor/graph.py b/torch/_inductor/graph.py
@@ -1060,7 +1060,12 @@ def placeholder(
         example = super().placeholder(target, args, kwargs)  # type: ignore[arg-type]
         target = self.qualify_name(target)
         if isinstance(example, SymTypes):
-            expr = _get_placeholder_expr(example.node)
+            # TODO fix partitioning issue and re-enable for backward
+            # https://github.com/pytorch/pytorch/issues/155468.
+            if not V.graph.is_backward:
+                expr = _get_placeholder_expr(example.node)
+            else:
+                expr = example.node.expr
             self.graph_inputs[target] = expr
             self.graph_input_names.append(target)
             return expr
