From ae637702d94904329157bd6e0324928a8262522c Mon Sep 17 00:00:00 2001
From: Irina Efode <irina.efode@intel.com>
Date: Thu, 20 May 2021 17:11:43 +0300
Subject: [PATCH] [IE TESTS][IE CPU] Fix BF16 threshold in ROI tests (#5703)

---
 .../plugin/cpu/shared_tests_instances/skip_tests_config.cpp  | 5 -----
 .../plugin/cpu/single_layer_tests/psroi_pooling.cpp          | 2 +-
 .../functional/plugin/cpu/single_layer_tests/roialign.cpp    | 2 +-
 3 files changed, 2 insertions(+), 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
index 869f0de1a2d..c994445b095 100644
--- 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
index 517badc6915..5327e72a6a2 100644
--- 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 @@ protected:
         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
index c49f0b5b322..aac49d69274 100644
--- 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 @@ protected:
         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");
     }