From 687cfc64c48afbfd3cae034a30613cab56e8a6bc Mon Sep 17 00:00:00 2001
From: M <mortezaho.1376@gmail.com>
Date: Tue, 29 Oct 2024 05:55:36 -0700
Subject: [PATCH] [CPU][ARM] Fix ARM tests failing because of overflow (#27074)

### Details:
 - Fixes ARM test overflow for Multiple Query SDP.
---
 .../src/nodes/kernels/scaled_attn/mha_single_token.cpp        | 2 +-
 .../subgraph_tests/src/common/concat_multiple_query_sdp.cpp   | 2 +-
 .../functional/shared_tests_instances/skip_tests_config.cpp   | 4 ++--
 3 files changed, 4 insertions(+), 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
index 6b6df3c3181ee0..25ddbb1b4246b1 100644
--- 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/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_multiple_query_sdp.cpp b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_multiple_query_sdp.cpp
index d05e7840562191..d74ab99fb3d5ab 100644
--- a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_multiple_query_sdp.cpp
+++ b/src/plugins/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
index e7c006ab97427f..6edc4f062536d0 100644
--- 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(