[inductor] [test] Fix test_add_complex4 op_count to match cpp_wrapper

karthickai · pytorchmergebot · commit 3fd238de60ab · 2025-05-09T15:35:36.000Z
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -1411,9 +1411,10 @@ def fn(a, b):
             )
             _, code = run_and_get_code(fn, x, y)
             code = " ".join(code)
-            self.assertEqual(
-                code.count("view_dtype" if config.cpp_wrapper else "aten.view"), 9
-            )
+            if config.cpp_wrapper:
+                self.assertEqual(code.count("view_dtype"), 3)
+            else:
+                self.assertEqual(code.count("aten.view"), 9)
 
     def test_add_complex5(self):
         def fn(a, b, alpha):
