From 2e4ac0195de9302df6d591d67efb7e002f34f7bd Mon Sep 17 00:00:00 2001 From: jiangcheng Date: Fri, 9 Jun 2023 17:07:36 +0800 Subject: [PATCH] [CINN] reopen cumsum prim+cinn unittest (#54494) --- test/legacy_test/test_cumsum_op.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/test/legacy_test/test_cumsum_op.py b/test/legacy_test/test_cumsum_op.py index 60d5855fb0a..0b9ac537f3a 100644 --- a/test/legacy_test/test_cumsum_op.py +++ b/test/legacy_test/test_cumsum_op.py @@ -142,7 +142,7 @@ class TestSumOp1(OpTest): 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 @@ class TestSumOpExclusive1(OpTest): 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 @@ class TestSumOpExclusiveFP16(OpTest): 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 @@ class TestSumOpReverseExclusive(OpTest): 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 create_test_fp16_class(parent, max_relative_error=1e-2): 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() -- GitLab