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

yf225 · yf225 · commit 9f64b015e2a6 · 2025-05-06T22:01:56.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,7 +1730,7 @@ def f1_replaced(x):
         self.assertEqual(f1_out, f1_replaced_out)
         self.assertEqual(count, 1)
 
-        # Case 2: mutates graph input
+        # Case 2: mutates graph input (not supported yet)
         @torch.compile(fullgraph=True, backend=custom_backend)
         def f2(x):
             out = torch.zeros_like(x)
@@ -1747,9 +1747,8 @@ def f2_replaced(x):
         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)
+        with self.assertRaisesRegex(AssertionError, "Pattern matcher does not yet support mutable ops that mutate graph input"):
+            self.assertEqual(f2_out, f2_replaced_out)
 
 if __name__ == "__main__":
     if IS_LINUX and HAS_GPU:
