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 d4b26dababb..55772b5938a 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 @@ -165,8 +165,6 @@ std::vector disabledTestPatterns() { // Failure happened on win and macos for current seeds. R"(.*CTCLossLayerTest.*CMR=1.*)", R"(.*CTCLossLayerCPUTest.*ctcMergeRepeated=1.*)", - // Issue: 72151 - R"(.*smoke_ROIAlignLayoutTest.*)", }; #define FIX_62820 0 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 cd7864c7fe8..a8587e46f1e 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 @@ -94,17 +94,17 @@ protected: auto coordsTensorData = static_cast(coordsTensor.data()); for (size_t i = 0; i < coordsTensor.get_size(); i += 4) { coordsTensorData[i] = 1.f; - coordsTensorData[i] = 1.f; - coordsTensorData[i] = 19.f; - coordsTensorData[i] = 19.f; + coordsTensorData[i + 1] = 1.f; + coordsTensorData[i + 2] = 19.f; + coordsTensorData[i + 3] = 19.f; } } else if (coordsET == ElementType::bf16) { auto coordsTensorData = static_cast(coordsTensor.data()); for (size_t i = 0; i < coordsTensor.get_size(); i += 4) { coordsTensorData[i] = static_cast(ngraph::bfloat16(1.f).to_bits()); - coordsTensorData[i] = static_cast(ngraph::bfloat16(1.f).to_bits()); - coordsTensorData[i] = static_cast(ngraph::bfloat16(19.f).to_bits()); - coordsTensorData[i] = static_cast(ngraph::bfloat16(19.f).to_bits()); + coordsTensorData[i + 1] = static_cast(ngraph::bfloat16(1.f).to_bits()); + coordsTensorData[i + 2] = static_cast(ngraph::bfloat16(19.f).to_bits()); + coordsTensorData[i + 3] = static_cast(ngraph::bfloat16(19.f).to_bits()); } } else { IE_THROW() << "roi align. Unsupported precision: " << coordsET; @@ -112,13 +112,10 @@ protected: auto roisIdxTensor = ov::runtime::Tensor{ funcInputs[2].get_element_type(), targetInputStaticShapes[2] }; auto roisIdxTensorData = static_cast(roisIdxTensor.data()); - if (roisIdxTensor.get_size() == 1) { - roisIdxTensorData[0] = 1; - } else if (roisIdxTensor.get_size() == 2) { - roisIdxTensorData[0] = 0; - roisIdxTensorData[1] = 1; - } else { - IE_THROW() << "Unexpected roiIdx size: " << roisIdxTensor.get_size(); + std::int32_t batchIdx = 0; + for (int i = 0; i < roisIdxTensor.get_size(); i++) { + roisIdxTensorData[i] = batchIdx; + batchIdx = (batchIdx + 1) % targetInputStaticShapes[0][0]; } inputs.insert({ funcInputs[0].get_node_shared_ptr(), data_tensor });