[CPU]fix sdpa test with shapeof

openvinotoolkit · Jan 7, 2025 · b307de7 · b307de7
1 parent bd217a3
commit b307de7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/..._cpu/tests/functional/custom/subgraph_tests/src/common/concat_transpose_sdp_transpose.cpp b/..._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);