[CPU][ARM] Fix ARM tests failing because of overflow

openvinotoolkit · Oct 16, 2024 · 6a4ed9a · 6a4ed9a
1 parent 9486b7d
commit 6a4ed9a
Show file tree

Hide file tree

Showing 3 changed files with 3 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
@@ -1143,7 +1143,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)