microsoft
/
CNTK
зеркало из
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

disable fp16 test for poolWithSequenceAxis (#3810)

This commit is contained in:
liqunfu 2020-03-31 08:55:14 -07:00 коммит произвёл GitHub
Родитель e1467a79dc
Коммит e939648002
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 6 добавлений и 4 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

10
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])