[CPU][ARM] Fix ARM tests failing because of overflow (openvinotoolkit…

…#27074) ### Details: - Fixes ARM test overflow for Multiple Query SDP.
CuriousPanCake · Nov 6, 2024 · 687cfc6 · 687cfc6
1 parent 191db21
commit 687cfc6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/mha_single_token.cpp b/src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/mha_single_token.cpp
@@ -1148,7 +1148,7 @@ void mha_single_token(const ov::intel_cpu::PlainTensor& query,
                                                                            past_v_scale_zp,
                                                                            head_sum);
         } else {
-            OPENVINO_THROW("Unsupported precision: ", query.get_precision());
+            OPENVINO_THROW("Unsupported precision: ", present_key.get_precision());
         }
 #else
         if (present_key.get_precision() == ov::element::u8) {

diff --git a/...intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_multiple_query_sdp.cpp b/...intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_multiple_query_sdp.cpp
@@ -238,7 +238,7 @@ class ConcatMultiQuerySDPTest : public testing::WithParamInterface<ConcatMultiQu
                 inputs.insert({param, t});
             } else if (param->get_element_type() == element::f16) {
                 ov::Tensor t{ov::element::f16, shape};
-                strided_iota(static_cast<ov::float16*>(t.data()), t.get_size(), val, 0.1f);
+                strided_iota(static_cast<ov::float16*>(t.data()), t.get_size(), val, 0.0f);
                 inputs.insert({param, t});
             } else {
                 ov::Tensor t{ov::element::bf16, shape};

diff --git a/src/plugins/intel_cpu/tests/functional/shared_tests_instances/skip_tests_config.cpp b/src/plugins/intel_cpu/tests/functional/shared_tests_instances/skip_tests_config.cpp
@@ -358,8 +358,7 @@ std::vector<std::string> disabledTestPatterns() {
     retVector.emplace_back(R"(smoke_VariableState/OVInferRequestVariableStateTest.*)");
     // Issue: 141705
     retVector.emplace_back(R"(.*smoke_arm_Deconv_2D_Planar_FP16/DeconvolutionLayerCPUTest.*INFERENCE_PRECISION_HINT=f16.*)");
-    // Issue: 154882
-    retVector.emplace_back(R"(.*ConcatMultiQuerySDPTest.*f16.*)");
+    retVector.emplace_back(R"(.*ConcatMultiQuerySDPTest.*u8.*)");
 #endif
 
 #if defined(OPENVINO_ARCH_ARM)
@@ -539,6 +538,7 @@ std::vector<std::string> disabledTestPatterns() {
         // Skip fp16 tests for paltforms that don't support fp16 precision
         retVector.emplace_back(R"(.*INFERENCE_PRECISION_HINT=(F|f)16.*)");
         retVector.emplace_back(R"(.*Prc=f16.*)");
+        retVector.emplace_back(R"(.*ConcatMultiQuerySDPTest.*f16.*HasShapeOf=1.*)");
     } else {
         // Issue 117407
         retVector.emplace_back(