From 067d7f1d0d4a81bc938ded151546d17189da1ecb Mon Sep 17 00:00:00 2001 From: lidanqing Date: Wed, 23 Dec 2020 12:20:51 +0100 Subject: [PATCH] fix conv2d int8 windows UT (#29528) --- .../fluid/tests/unittests/mkldnn/test_conv2d_int8_mkldnn_op.py | 2 ++ tools/windows/run_unittests.sh | 1 - 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_int8_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_int8_mkldnn_op.py index 88f1fb7fd2..c8cc04cb5a 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_int8_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_int8_mkldnn_op.py @@ -28,6 +28,8 @@ def conv2d_forward_refer(input, filter, group, conv_param): return out +@unittest.skipIf(not core.supports_bfloat16(), + "place does not support oneDNN INT8") class TestConv2DInt8Op(TestConv2DOp): def setUp(self): self.op_type = "conv2d" diff --git a/tools/windows/run_unittests.sh b/tools/windows/run_unittests.sh index 7ad9ce4346..0ed9d01d99 100644 --- a/tools/windows/run_unittests.sh +++ b/tools/windows/run_unittests.sh @@ -100,7 +100,6 @@ diable_wingpu_test="^test_analysis_predictor$|\ ^test_print_op$|\ ^test_py_func_op$|\ ^test_weight_decay$|\ -^test_conv2d_int8_mkldnn_op$|\ ^test_crypto$|\ ^test_callbacks$|\ ^test_program_prune_backward$|\ -- GitLab