From a2cc6a036f0b299f5d9a6c2627e96be1158b4306 Mon Sep 17 00:00:00 2001
From: River Li <river.li@intel.com>
Date: Tue, 12 Dec 2023 22:13:22 +0800
Subject: [PATCH] [CPU Test] use internal_properties for snippets_mode (#21549)

---
 .../functional/single_layer_tests/classes/eltwise.cpp    | 5 +++--
 .../functional/subgraph_tests/src/conv_sum_broadcast.cpp | 8 ++++----
 .../tests/functional/subgraph_tests/src/mha.cpp          | 9 +++++----
 .../tests/functional/subgraph_tests/src/ngram.cpp        | 5 +++--
 4 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/src/plugins/intel_cpu/tests/functional/single_layer_tests/classes/eltwise.cpp b/src/plugins/intel_cpu/tests/functional/single_layer_tests/classes/eltwise.cpp
index 633cc1f7df6390..83d518bafa907b 100644
--- a/src/plugins/intel_cpu/tests/functional/single_layer_tests/classes/eltwise.cpp
+++ b/src/plugins/intel_cpu/tests/functional/single_layer_tests/classes/eltwise.cpp
@@ -6,6 +6,7 @@
 
 #include "common_test_utils/node_builders/eltwise.hpp"
 #include "gtest/gtest.h"
+#include "internal_properties.hpp"
 #include "openvino/core/type/element_type.hpp"
 #include "openvino/runtime/properties.hpp"
 #include "test_utils/cpu_test_utils.hpp"
@@ -150,9 +151,9 @@ void EltwiseLayerCPUTest::SetUp() {
 #endif
 
     if (enforceSnippets) {
-        configuration.insert({"SNIPPETS_MODE", "IGNORE_CALLBACK"});
+        configuration.insert(ov::intel_cpu::snippets_mode(ov::intel_cpu::SnippetsMode::IGNORE_CALLBACK));
     } else {
-        configuration.insert({"SNIPPETS_MODE", "DISABLE"});
+        configuration.insert(ov::intel_cpu::snippets_mode(ov::intel_cpu::SnippetsMode::DISABLE));
     }
     ov::ParameterVector parameters{std::make_shared<ov::op::v0::Parameter>(netType, inputDynamicShapes.front())};
     std::shared_ptr<ov::Node> secondaryInput;
diff --git a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/conv_sum_broadcast.cpp b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/conv_sum_broadcast.cpp
index 3d6b89db51fe8e..320229a22a1778 100644
--- a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/conv_sum_broadcast.cpp
+++ b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/conv_sum_broadcast.cpp
@@ -2,16 +2,16 @@
 // SPDX-License-Identifier: Apache-2.0
 //
 
-#include "common_test_utils/node_builders/convolution.hpp"
 #include "common_test_utils/node_builders/activation.hpp"
+#include "common_test_utils/node_builders/convolution.hpp"
 #include "cpp_interfaces/interface/ie_internal_plugin_config.hpp"
+#include "internal_properties.hpp"
 #include "ov_models/builders.hpp"
 #include "ov_models/utils/ov_helpers.hpp"
 #include "ov_ops/type_relaxed.hpp"
 #include "shared_test_classes/base/ov_subgraph.hpp"
 #include "test_utils/convolution_params.hpp"
 #include "test_utils/fusing_test_utils.hpp"
-#include "internal_properties.hpp"
 
 #include <regex>
 
@@ -132,8 +132,8 @@ class ConvSumInPlaceTest : public testing::WithParamInterface<convSumBroadcastPa
         function = makeNgraphFunction(getNetType(), inputParams, sum, "ConvolutionSumBroadcast");
 
         targetDevice = ov::test::utils::DEVICE_CPU;
-        if (!configuration.count("SNIPPETS_MODE")) {
-            configuration.insert({"SNIPPETS_MODE", "DISABLE"});
+        if (!configuration.count(ov::intel_cpu::snippets_mode.name())) {
+            configuration.insert(ov::intel_cpu::snippets_mode(ov::intel_cpu::SnippetsMode::DISABLE));
         }
     }
 
diff --git a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/mha.cpp b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/mha.cpp
index 66bcb7baf5b744..1655e58cbe67a1 100644
--- a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/mha.cpp
+++ b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/mha.cpp
@@ -5,6 +5,7 @@
 #include "common_test_utils/common_utils.hpp"
 #include "common_test_utils/ov_tensor_utils.hpp"
 #include "functional_test_utils/skip_tests_config.hpp"
+#include "internal_properties.hpp"
 #include "ov_models/builders.hpp"
 #include "shared_test_classes/base/ov_subgraph.hpp"
 #include "test_utils/cpu_test_utils.hpp"
@@ -235,8 +236,8 @@ class MHATest : public testing::WithParamInterface<MHATuple>, virtual public Sub
         // Snippets MHA tokenization has limitations to avoid performance degradations. These limitations depend on
         // target machine. Just for testing, we disable these limitations to allow Snippets to tokenize pattern on all
         // machines for validation.
-        if (!configuration.count("SNIPPETS_MODE")) {
-            configuration.insert({"SNIPPETS_MODE", "IGNORE_CALLBACK"});
+        if (!configuration.count(ov::intel_cpu::snippets_mode.name())) {
+            configuration.insert(ov::intel_cpu::snippets_mode(ov::intel_cpu::SnippetsMode::IGNORE_CALLBACK));
         }
     }
 };
@@ -617,8 +618,8 @@ class MHAQuantTest : public testing::WithParamInterface<MHATuple>,
         // Snippets MHA tokenization has limitations to avoid performance degradations. These limitations depend on
         // target machine. Just for testing, we disable these limitations to allow Snippets to tokenize pattern on all
         // machines for validation.
-        if (!configuration.count("SNIPPETS_MODE")) {
-            configuration.insert({"SNIPPETS_MODE", "IGNORE_CALLBACK"});
+        if (!configuration.count(ov::intel_cpu::snippets_mode.name())) {
+            configuration.insert(ov::intel_cpu::snippets_mode(ov::intel_cpu::SnippetsMode::IGNORE_CALLBACK));
         }
     }
 };
diff --git a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/ngram.cpp b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/ngram.cpp
index 865268c500a31b..1634d2a62a42b7 100644
--- a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/ngram.cpp
+++ b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/ngram.cpp
@@ -4,6 +4,7 @@
 
 #include "common_test_utils/common_utils.hpp"
 #include "common_test_utils/ov_tensor_utils.hpp"
+#include "internal_properties.hpp"
 #include "shared_test_classes/base/ov_subgraph.hpp"
 #include "test_utils/cpu_test_utils.hpp"
 
@@ -201,8 +202,8 @@ class NgramCPUTest : public testing::WithParamInterface<NgramTestParams>,
         init_input_shapes(inputShapes);
         function = initNgram(inputDynamicShapes, data_et, idces_et, k);
 
-        if (!configuration.count("SNIPPETS_MODE")) {
-            configuration.insert({"SNIPPETS_MODE", "DISABLE"});
+        if (!configuration.count(ov::intel_cpu::snippets_mode.name())) {
+            configuration.insert(ov::intel_cpu::snippets_mode(ov::intel_cpu::SnippetsMode::DISABLE));
         }
     }
 };