[IE TESTS][IE CPU] Fix BF16 threshold in ROI tests (openvinotoolkit#5703

)
yekruglov · Jun 7, 2021 · 6cf9cce · 6cf9cce
1 parent d79b27d
commit 6cf9cce
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp
@@ -74,11 +74,6 @@ std::vector<std::string> disabledTestPatterns() {
         // TODO: 54718 Accuracy mismatch
         R"(.*GroupDeconv_2D_DW_BF16.*K\(3\.3\)_S\(1\.1\).*primitive=jit_avx512_dw.*)",
         R"(.*GroupDeconv_2D_DW_BF16.*K\(3\.3\)_S\(2\.2\).*primitive=jit_avx512_dw.*)",
-        // TODO: iefode: fix BF16 tests (PR5624)
-        R"(.*smoke_ROIAlignLayoutTest.*BF16.*)",
-        R"(.*smoke_PSROIPoolingAverageLayoutTest.*BF16.*)",
-        R"(.*smoke_PSROIPoolingBilinearLayoutTest.*BF16.*)",
-        R"(.*smoke_ROIAlignLayoutTest.*BF16.*)",
         // reference doesn't cover I8, U8 cases. Issue: 55842
         R"(.*Gather7LayerTest.*netPRC=I8.*)",
     };

diff --git a/inference-engine/tests/functional/plugin/cpu/single_layer_tests/psroi_pooling.cpp b/inference-engine/tests/functional/plugin/cpu/single_layer_tests/psroi_pooling.cpp
@@ -94,7 +94,7 @@ class PSROIPoolingLayerCPUTest : public testing::WithParamInterface<PSROIPooling
         psroi->get_rt_info() = getCPUInfo();
         selectedType = getPrimitiveType() + "_" + inPrc.name();
 
-        threshold = 0.001f;
+        threshold = 1e-2;
         const ngraph::ResultVector results{std::make_shared<ngraph::opset3::Result>(psroi)};
         function = std::make_shared<ngraph::Function>(results, params, "PSROIPooling");
     }

diff --git a/inference-engine/tests/functional/plugin/cpu/single_layer_tests/roialign.cpp b/inference-engine/tests/functional/plugin/cpu/single_layer_tests/roialign.cpp
@@ -95,7 +95,7 @@ class ROIAlignLayerCPUTest : public testing::WithParamInterface<ROIAlignLayerCPU
         roialign->get_rt_info() = getCPUInfo();
         selectedType = std::string("unknown_") + inPrc.name();
 
-        threshold = 0.001f;
+        threshold = 1e-2;
         const ngraph::ResultVector results{std::make_shared<ngraph::opset3::Result>(roialign)};
         function = std::make_shared<ngraph::Function>(results, params, "ROIAlign");
     }