Skip to content

Commit c98f76d

Browse files
committed
minor change
1 parent 6c377fb commit c98f76d

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

python/paddle/fluid/tests/unittests/mkldnn/test_prelu_mkldnn_op.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
import numpy as np
1919
import paddle
2020
import paddle.fluid.core as core
21-
from paddle.fluid.tests.unittests.op_test import OpTest, convert_float_to_uint16
21+
from paddle.fluid.tests.unittests.op_test import OpTest, convert_float_to_uint16, skip_check_grad_ci
2222

2323

2424
def ref_prelu(x, weight, mode):
@@ -109,6 +109,9 @@ def init_attrs(self):
109109

110110
# BF16 TESTS
111111
def create_bf16_test_class(parent):
112+
@skip_check_grad_ci(
113+
reason="[skip shape check] Input(Alpha) must be 1-D and only has one data in 'all' mode"
114+
)
112115
class TestPReluBF16OneDNNOp(parent):
113116
def set_inputs(self, ):
114117
self.inputs = {
@@ -157,7 +160,7 @@ def test_check_grad(self):
157160
pass
158161

159162
# TODO jakpiase, when base class for BF16 oneDNN tests
160-
# will be done, add grad tests
163+
# will be done, add grad BF16 tests
161164
# def test_check_grad(self):
162165
# if core.is_compiled_with_cuda() or not core.supports_bfloat16():
163166
# pass

0 commit comments

Comments
 (0)