diff --git a/python/paddle/fluid/tests/unittests/test_group_norm_op.py b/python/paddle/fluid/tests/unittests/test_group_norm_op.py index 6f3961fc03ebd41566408712697633671d9c01aa..eb69e283a35216e096128edbb57972ccfc74a659 100644 --- a/python/paddle/fluid/tests/unittests/test_group_norm_op.py +++ b/python/paddle/fluid/tests/unittests/test_group_norm_op.py @@ -19,7 +19,7 @@ import numpy as np from operator import mul import paddle.fluid.core as core import paddle.fluid as fluid -from op_test import OpTest +from op_test import OpTest, skip_check_grad_ci from testsuite import create_op @@ -151,6 +151,10 @@ class TestGroupNormOpBigEps3(TestGroupNormOp): self.attrs['epsilon'] = 0.5 +@skip_check_grad_ci( + reason='''This test case is used to ensure whether the gradient checking results between CPU and GPU + are consistent when using the same inputs, thus, it doesn't need to call check_grad.''' +) class TestGroupNormOpLargeData(TestGroupNormOp): def init_test_case(self): self.shape = (2, 32, 64, 64) @@ -190,6 +194,10 @@ class TestGroupNormOpBigEps3_With_NHWC(TestGroupNormOp): self.data_format = "NHWC" +@skip_check_grad_ci( + reason='''This test case is used to ensure whether the gradient checking results between CPU and GPU + are consistent when using the same inputs, thus, it doesn't need to call check_grad.''' +) class TestGroupNormOpLargeData_With_NHWC(TestGroupNormOp): def init_test_case(self): self.shape = (2, 64, 32, 32) # NCHW