We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e052fde commit 71d3b06Copy full SHA for 71d3b06
python/paddle/fluid/tests/unittests/test_conv2d_op.py
@@ -172,9 +172,9 @@ def test_check_grad_no_input(self):
172
173
def create_test_cudnn_bf16_class(parent):
174
@unittest.skipIf(
175
- not core.is_compiled_with_cuda() or core.cudnn_version() < 8100,
176
- "core is not compiled with CUDA and cudnn version need larger than 8.1.0"
177
- )
+ not core.is_compiled_with_cuda() or
+ not core.is_bfloat16_supported(core.CUDAPlace(0)),
+ "core is not compiled with CUDA and do not support bfloat16")
178
class TestConv2DCUDNNBF16(parent):
179
def get_numeric_grad(self, place, check_name):
180
scope = core.Scope()
0 commit comments