[Snippets][CPU] Make EnforcePrecision pass arch independent (#32315)

### Tickets:
 - N/A

Author: aobolensk

src/plugins/intel_cpu/src/nodes/subgraph.cpp

@@ -46,7 +46,9 @@
 
 #    include "emitters/snippets/x64/cpu_generator.hpp"
 #    include "executors/x64/subgraph.hpp"
+#    include "snippets/op/brgemm.hpp"
 #    include "snippets/pass/matmul_to_brgemm.hpp"
+#    include "transformations/snippets/x64/op/brgemm_utils.hpp"
 #elif defined(OPENVINO_ARCH_ARM64)
 #    include <cpu/aarch64/cpu_isa_traits.hpp>
 
@@ -74,9 +76,9 @@
 #    include "snippets/lowered/pass/init_loops.hpp"
 #    include "snippets/lowered/pass/insert_buffers.hpp"
 #    include "snippets/lowered/pass/insert_loops.hpp"
+#    include "transformations/snippets/common/pass/enforce_precision.hpp"
 #    include "transformations/snippets/x64/pass/brgemm_to_brgemm_cpu.hpp"
 #    include "transformations/snippets/x64/pass/eliminate_brgemm_copy_b.hpp"
-#    include "transformations/snippets/x64/pass/enforce_precision.hpp"
 #    include "transformations/snippets/x64/pass/fuse_brgemm_cpu_postops.hpp"
 #    include "transformations/snippets/x64/pass/lowered/adjust_brgemm_copy_b_loop_ports.hpp"
 #    include "transformations/snippets/x64/pass/lowered/brgemm_cpu_blocking.hpp"
@@ -551,7 +553,19 @@ Subgraph::DataFlowPasses Subgraph::getDataFlowPasses() {
                                                ov::snippets::pass::MatMulToBrgemm,
                                                pass::EnforcePrecision,
                                                element::f32,
-                                               context->getConfig().inferencePrecision);
+                                               context->getConfig().inferencePrecision,
+                                               [](const std::shared_ptr<ov::Node>& op) {
+                                                   std::set<std::vector<ov::element::Type>> types;
+                                                   if (ov::is_type<ov::snippets::op::Brgemm>(op)) {
+                                                       if (ov::intel_cpu::brgemm_utils::is_fp16_supported()) {
+                                                           types.insert({ov::element::f16, ov::element::f16});
+                                                       }
+                                                       if (ov::intel_cpu::brgemm_utils::is_bf16_supported()) {
+                                                           types.insert({ov::element::bf16, ov::element::bf16});
+                                                       }
+                                                   }
+                                                   return types;
+                                               });
     }
 
     SNIPPETS_REGISTER_PASS_RELATIVE_X86_64(Place::Before,

src/plugins/intel_cpu/src/transformations/snippets/x64/pass/enforce_precision.cpp renamed to src/plugins/intel_cpu/src/transformations/snippets/common/pass/enforce_precision.cpp

@@ -21,10 +21,8 @@
 #include "openvino/itt.hpp"
 #include "ov_ops/type_relaxed.hpp"
 #include "snippets/itt.hpp"
-#include "snippets/op/brgemm.hpp"
 #include "snippets/op/convert_saturation.hpp"
 #include "snippets/pass/propagate_precision.hpp"
-#include "transformations/snippets/x64/op/brgemm_utils.hpp"
 #include "transformations/utils/utils.hpp"
 #include "utils/general_utils.h"
 
@@ -37,8 +35,8 @@ EnforcePrecision::EnforcePrecision(
         get_supported_precisions)
     : source(source),
       target(target),
-      get_supported_precisions(get_supported_precisions == nullptr ? get_supported_precisions_default
-                                                                   : get_supported_precisions) {
+      get_supported_precisions(get_supported_precisions) {
+    OPENVINO_ASSERT(get_supported_precisions != nullptr, "get_supported_precisions callback is not set");
     OPENVINO_ASSERT(source != target, "source and target precisions have to be different");
 }
 
@@ -132,17 +130,3 @@ bool EnforcePrecision::run_on_model(const std::shared_ptr<ov::Model>& m) {
 
     return was_updated;
 }
-
-std::set<std::vector<ov::element::Type>> EnforcePrecision::get_supported_precisions_default(
-    const std::shared_ptr<ov::Node>& op) noexcept {
-    std::set<std::vector<ov::element::Type>> types;
-    if (ov::is_type<snippets::op::Brgemm>(op)) {
-        if (brgemm_utils::is_fp16_supported()) {
-            types.insert({element::f16, element::f16});
-        }
-        if (brgemm_utils::is_bf16_supported()) {
-            types.insert({element::bf16, element::bf16});
-        }
-    }
-    return types;
-}

src/plugins/intel_cpu/src/transformations/snippets/x64/pass/enforce_precision.hpp renamed to src/plugins/intel_cpu/src/transformations/snippets/common/pass/enforce_precision.hpp

@@ -23,14 +23,11 @@ class EnforcePrecision : public ov::pass::ModelPass {
     EnforcePrecision(element::Type source,
                      element::Type target,
                      const std::function<std::set<std::vector<element::Type>>(const std::shared_ptr<ov::Node>& op)>&
-                         get_supported_precisions = nullptr);
+                         get_supported_precisions);
 
     bool run_on_model(const std::shared_ptr<ov::Model>& m) override;
 
 private:
-    static std::set<std::vector<element::Type>> get_supported_precisions_default(
-        const std::shared_ptr<ov::Node>& op) noexcept;
-
     const element::Type source;
     const element::Type target;
     const std::function<std::set<std::vector<element::Type>>(const std::shared_ptr<ov::Node>& op)>

src/plugins/intel_cpu/tests/unit/snippets_transformations/x64/enforce_precision.cpp renamed to src/plugins/intel_cpu/tests/unit/snippets_transformations/common/enforce_precision.cpp

@@ -10,8 +10,8 @@
 #include <gtest/gtest.h>
 
 #include "openvino/core/type/element_type.hpp"
-#include "transformations/snippets/x64/pass/enforce_precision.hpp"
 #include "common_test_utils/common_utils.hpp"
+#include "transformations/snippets/common/pass/enforce_precision.hpp"
 #include "two_binary_ops.hpp"
 
 namespace ov {