[CINN] reopen cumsum prim+cinn unittest (#54494)

thisjiang · web-flow · commit 2e4ac0195de9 · 2023-06-09T17:07:36.000+08:00
diff --git a/test/legacy_test/test_cumsum_op.py b/test/legacy_test/test_cumsum_op.py
@@ -142,7 +142,7 @@ def init_dtype(self):
         self.dtype = self.dtype_ = np.float64
 
     def set_enable_cinn(self):
-        self.enable_cinn = True
+        pass
 
     def set_attrs_input_output(self):
         self.attrs = {'axis': 2}
@@ -241,7 +241,7 @@ def init_dtype(self):
         self.dtype = self.dtype_ = np.float64
 
     def set_enable_cinn(self):
-        self.enable_cinn = True
+        pass
 
     def set_attrs_input_output(self):
         self.attrs = {'axis': 2, 'exclusive': True}
@@ -314,8 +314,6 @@ def setUp(self):
         self.python_api = cumsum_wrapper
         self.public_python_api = paddle.cumsum
         self.init_dtype()
-        # TODO(thisjiang): set `True` after reduce+cast at shape=[4, 5, 20, 20], dim=[2]'s fusion bug has fixed
-        self.enable_cinn = False
         self.attrs = {'axis': 2, "exclusive": True}
         self.x = np.random.random((4, 5, 20)).astype(self.dtype)
         self.out = np.concatenate(
@@ -381,7 +379,7 @@ def init_dtype(self):
         self.dtype = self.dtype_ = np.float64
 
     def set_enable_cinn(self):
-        self.enable_cinn = True
+        pass
 
 
 def create_test_fp16_class(parent, max_relative_error=1e-2):
@@ -390,8 +388,7 @@ def init_dtype(self):
             self.dtype = self.dtype_ = np.float16
 
         def set_enable_cinn(self):
-            # TODO(thisjiang): set `pass` after reduce+cast at shape=[4, 5, 20, 20], dim=[2]'s fusion bug has fixed
-            self.enable_cinn = False
+            pass
 
         def test_check_output(self):
             self.check_output()
