From eaf0cf096f596bc9de60a8a54015902244ded2af Mon Sep 17 00:00:00 2001 From: liqun fu Date: Tue, 31 Mar 2020 08:50:09 -0700 Subject: [PATCH] disable fp16 test for poolWithSequenceAxis --- bindings/python/cntk/tests/onnx_op_test.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/bindings/python/cntk/tests/onnx_op_test.py b/bindings/python/cntk/tests/onnx_op_test.py index ff3f0bc544c8..ad4503e0b8df 100644 --- a/bindings/python/cntk/tests/onnx_op_test.py +++ b/bindings/python/cntk/tests/onnx_op_test.py @@ -426,8 +426,9 @@ def test_AveragePool(tmpdir, dtype, device_id): #AveragePool @pytest.mark.parametrize("dtype", DType_Config) def test_AveragePoolWithSequenceAxis(tmpdir, dtype, device_id): - if device_id == -1 and dtype == np.float16: - pytest.skip('Test is skipped on CPU with float16 data') + if dtype == np.float16: + # CI reporting "FP16 convolution is only supported via cuDNN." on GPU with float16 data + pytest.skip('Test is skipped with float16 data') device = cntk_device(device_id) with C.default_options(dtype=dtype): img = np.reshape(np.arange(16, dtype = dtype), [1, 4, 4]) @@ -1342,8 +1343,9 @@ def test_MaxPool(tmpdir, dtype, device_id): #MaxPool @pytest.mark.parametrize("dtype", DType_Config) def test_MaxPoolWithSequenceAxis(tmpdir, dtype, device_id): - if device_id == -1 and dtype == np.float16: - pytest.skip('Test is skipped on CPU with float16 data') + if dtype == np.float16: + # CI reporting "FP16 convolution is only supported via cuDNN." on GPU with float16 data + pytest.skip('Test is skipped with float16 data') device = cntk_device(device_id) with C.default_options(dtype=dtype): img = np.reshape(np.arange(16, dtype = dtype), [1, 4, 4])