Update on "[Inductor] Pattern matcher support for mutable ops with non-view inputs"

yf225 · yf225 · commit fb2656aa4a0a · 2025-05-06T21:21:11.000-07:00
Fixes the non-view input use case in #152441. Pull-Request-resolved: #152767 cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy chenyang78 kadeng muchulee8 amjames chauhang aakhundov [ghstack-poisoned]
diff --git a/test/inductor/test_pattern_matcher.py b/test/inductor/test_pattern_matcher.py
@@ -1730,6 +1730,26 @@ def f1_replaced(x):
         self.assertEqual(f1_out, f1_replaced_out)
         self.assertEqual(count, 1)
 
+        # Case 2: mutates graph input
+        @torch.compile(fullgraph=True, backend=custom_backend)
+        def f2(x):
+            out = torch.zeros_like(x)
+            foo_inplace(x)
+            bar_out(x, out)
+            return out
+
+        def f2_replaced(x):
+            out = torch.zeros_like(x)
+            foobar_out(x, out)
+            return out
+
+        f2_inp = inp.clone().detach()
+        f2_replaced_inp = inp.clone().detach()
+        f2_out = f2(f2_inp)
+        f2_replaced_out = f2_replaced(f2_replaced_inp)
+        self.assertEqual(f2_inp, f2_replaced_inp)
+        self.assertEqual(f2_out, f2_replaced_out)
+        self.assertEqual(count, 1)
 
 if __name__ == "__main__":
     if IS_LINUX and HAS_GPU:
