From b307de7ef201bfa693f9b2cd54098c60750dc54c Mon Sep 17 00:00:00 2001
From: Zhang Yi <yi3.zhang@intel.com>
Date: Tue, 7 Jan 2025 13:37:55 +0800
Subject: [PATCH] [CPU]fix sdpa test with shapeof

---
 .../src/common/concat_transpose_sdp_transpose.cpp          | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_transpose_sdp_transpose.cpp b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_transpose_sdp_transpose.cpp
index fd4cf76f6847ec..5dbdb2954cbdce 100644
--- a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_transpose_sdp_transpose.cpp
+++ b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/common/concat_transpose_sdp_transpose.cpp
@@ -78,7 +78,7 @@ class ConcatSDPTransposeTestBase : public testing::WithParamInterface<ConcatSDPT
         result << "Prc=" << inType << "_";
         result << "HasShapeOf=" << hasShapeof << "_";
         result << "quantKeyByChannel=" << quantKeyByChannel << "_";
-        result << "groupSize= " << groupSize << "_";
+        result << "groupSize=" << groupSize << "_";
         result << "TransposeOrder=";
         result << "(";
         for (const auto& itr : transposeOrder) {
@@ -92,7 +92,6 @@ class ConcatSDPTransposeTestBase : public testing::WithParamInterface<ConcatSDPT
     void SetUp() override {
         ElementType inType;
         InputShapeAndTransposeOrder inputShapeAndOrders;
-        bool hasShapeOf;
         std::tie(inType, inputShapeAndOrders, hasShapeOf, quantKeyByChannel, keyGroupSize) = this->GetParam();
         std::vector<InputShape>& inputShapes = inputShapeAndOrders.first;
         transposeOrder = inputShapeAndOrders.second;
@@ -372,6 +371,8 @@ TEST_P(ConcatSDPTransposeTest, CompareWithRefs) {
     // Transformation TSShapeOfForward will change:
     // ?->transpose->shapeof ==> ?-->shapeof->gather
     //                            |->transpose
+    size_t expectedGatherCount = hasShapeOf ? 1 : 0;
+    std::cout << "ConcatSDPTEST|" << expectedGatherCount << std::endl;
     CheckNumberOfNodesWithType(compiledModel, "Gather", hasShapeOf ? 1 : 0);
     auto expectedOutputs = run_test(functionRefs);
     CheckNumberOfNodesWithType(compiledModel, "ScaledDotProductAttention", 0);
@@ -467,7 +468,7 @@ INSTANTIATE_TEST_SUITE_P(smoke_ConcatSDPTransposeByChannelTest,
                          ConcatSDPTransposeTest,
                          ::testing::Combine(::testing::Values(ElementType::f32),
                                             ::testing::ValuesIn(shapesWithGreedySearch),
-                                            ::testing::Values(true),
+                                            ::testing::Values(false),
                                             ::testing::Values(true),
                                             ::testing::Values(8)),
                          ConcatSDPTransposeTest::getTestCaseName);