Skip to content

Commit e939648

Browse files
authored
disable fp16 test for poolWithSequenceAxis (microsoft#3810)
1 parent e1467a7 commit e939648

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

bindings/python/cntk/tests/onnx_op_test.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -426,8 +426,9 @@ def test_AveragePool(tmpdir, dtype, device_id):
426426
#AveragePool
427427
@pytest.mark.parametrize("dtype", DType_Config)
428428
def test_AveragePoolWithSequenceAxis(tmpdir, dtype, device_id):
429-
if device_id == -1 and dtype == np.float16:
430-
pytest.skip('Test is skipped on CPU with float16 data')
429+
if dtype == np.float16:
430+
# CI reporting "FP16 convolution is only supported via cuDNN." on GPU with float16 data
431+
pytest.skip('Test is skipped with float16 data')
431432
device = cntk_device(device_id)
432433
with C.default_options(dtype=dtype):
433434
img = np.reshape(np.arange(16, dtype = dtype), [1, 4, 4])
@@ -1342,8 +1343,9 @@ def test_MaxPool(tmpdir, dtype, device_id):
13421343
#MaxPool
13431344
@pytest.mark.parametrize("dtype", DType_Config)
13441345
def test_MaxPoolWithSequenceAxis(tmpdir, dtype, device_id):
1345-
if device_id == -1 and dtype == np.float16:
1346-
pytest.skip('Test is skipped on CPU with float16 data')
1346+
if dtype == np.float16:
1347+
# CI reporting "FP16 convolution is only supported via cuDNN." on GPU with float16 data
1348+
pytest.skip('Test is skipped with float16 data')
13471349
device = cntk_device(device_id)
13481350
with C.default_options(dtype=dtype):
13491351
img = np.reshape(np.arange(16, dtype = dtype), [1, 4, 4])

0 commit comments

Comments
 (0)