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

yf225 · yf225 · commit c4d23b8858e6 · 2025-05-08T14:55:23.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
@@ -1675,11 +1675,11 @@ def foobar_out(x: torch.Tensor, out: torch.Tensor) -> None:
         def mutable_ops_pattern(x, out):
             foo_inplace(x)
             bar_out(x, out)
-            return out
+            return x, out
 
         def mutable_ops_replacement(x, out):
             foobar_out(x, out)
-            return out
+            return x, out
 
         inp = torch.randn(3)
 
@@ -1747,8 +1747,9 @@ def f2_replaced(x):
         f2_replaced_inp = inp.clone().detach()
         f2_out = f2(f2_inp)
         f2_replaced_out = f2_replaced(f2_replaced_inp)
-        with self.assertRaisesRegex(AssertionError, "Pattern matcher does not yet support mutable ops that mutate graph input"):
-            self.assertEqual(f2_out, f2_replaced_out)
+        self.assertEqual(f1_inp, f1_replaced_inp)
+        self.assertEqual(f2_out, f2_replaced_out)
+        self.assertEqual(count, 1)
 
 if __name__ == "__main__":
     if IS_LINUX and HAS_GPU:
