File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
python/paddle/fluid/tests/unittests/mkldnn Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 1818import numpy as np
1919import paddle
2020import 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
2424def ref_prelu (x , weight , mode ):
@@ -109,6 +109,9 @@ def init_attrs(self):
109109
110110# BF16 TESTS
111111def 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
You can’t perform that action at this time.
0 commit comments