Update on "[Inductor] Pattern matcher support for mutable ops with no…

…n-view inputs" 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]
pytorch · yf225 · May 4, 2025 · May 4, 2025 · May 4, 2025 · May 4, 2025
commit 69fd7e1c348aa49d0112f7edf3b8d0ba35c47443
diff --git a/test/inductor/test_pattern_matcher.py b/test/inductor/test_pattern_matcher.py
@@ -1726,11 +1726,11 @@ def f1_replaced(x):
         f1_replaced_inp = inp.clone().detach()
         f1_out = f1(f1_inp)
         f1_replaced_out = f1_replaced(f1_replaced_inp)
+        self.assertEqual(count, 1)
         self.assertEqual(f1_inp, f1_replaced_inp)
         self.assertEqual(f1_out, f1_replaced_out)
-        self.assertEqual(count, 1)
 
-        # Case 2: mutates graph input (not supported yet)
+        # Case 2: mutates graph input
         @torch.compile(fullgraph=True, backend=custom_backend)
         def f2(x):
             out = torch.zeros_like(x)
@@ -1747,9 +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)
+        self.assertEqual(count, 1)
         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: