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

This commit is contained in:
Irina Efode 2021-05-20 17:11:43 +03:00 committed by GitHub
parent 370617d909
commit ae637702d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 2 additions and 7 deletions

View File

@ -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.*)",
};

View File

@ -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");
}

View File

@ -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");
}