diff --git a/python/paddle/fluid/tests/unittests/test_mean_op.py b/python/paddle/fluid/tests/unittests/test_mean_op.py index cbc426a4f02c5f48a4bdf102ffad844bff9c69cf..5c1d2c23b18f64ccad3ddc0a2e9dda7497eec9d2 100644 --- a/python/paddle/fluid/tests/unittests/test_mean_op.py +++ b/python/paddle/fluid/tests/unittests/test_mean_op.py @@ -176,7 +176,7 @@ class TestReduceMeanOp(OpTest): pass def if_enable_cinn(self): - pass + self.enable_cinn = False def test_check_output(self): if self.dtype != 'float16': @@ -281,6 +281,9 @@ class TestReduceMeanOpShape6D(TestReduceMeanOp): def set_attrs(self): self.shape = [2, 3, 4, 5, 6, 7] + def if_enable_cinn(self): + self.enable_cinn = False + class TestReduceMeanOpShape6DBF16(TestReduceMeanBF16Op): def set_attrs(self):