repacking

openvinotoolkit · Feb 11, 2025 · d1ef8ae · d1ef8ae
1 parent aac89b6
commit d1ef8ae
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/plugins/intel_cpu/src/emitters/snippets/x64/kernel_executors/brgemm_base.cpp b/src/plugins/intel_cpu/src/emitters/snippets/x64/kernel_executors/brgemm_base.cpp
@@ -257,12 +257,17 @@ void BrgemmBaseKernelExecutor::update_config(const ov::snippets::lowered::Expres
     auto LDB = DIM_CAST(snippets::utils::get_dim_stride(expr->get_input_port(1)));
 
     if (is_type<ov::intel_cpu::BrgemmCPU>(expr->get_node())) {
-    } else if (is_type<ov::intel_cpu::GemmCPU>(expr->get_node())) {
-        const auto& brgemm_node = as_type_ptr<ov::intel_cpu::GemmCPU>(expr->get_node());
+        const auto& brgemm_node = as_type_ptr<ov::intel_cpu::BrgemmCPU>(expr->get_node());
         // In case of data repacking LDB is chosen in accordance with repacking buffer size
         if (with_repacking(brgemm_node->get_type())) {
             LDB = DIM_CAST(brgemm_utils::repacking::compute_repacked_n_dim(LDB, brgemm_node->get_input_element_type(1)));
         }
+    } else if (is_type<ov::intel_cpu::GemmCPU>(expr->get_node())) {
+        const auto& gemm_node = as_type_ptr<ov::intel_cpu::GemmCPU>(expr->get_node());
+        // In case of data repacking LDB is chosen in accordance with repacking buffer size
+        if (with_repacking(gemm_node->get_type())) {
+            LDB = DIM_CAST(brgemm_utils::repacking::compute_repacked_n_dim(LDB, gemm_node->get_input_element_type(1)));
+        }
     } else {
         OV_CPU_JIT_EMITTER_ASSERT(false, "Got invalid node type in update_config");
     }

diff --git a/src/plugins/intel_cpu/src/transformations/snippets/x64/pass/lowered/build_brgemm.cpp b/src/plugins/intel_cpu/src/transformations/snippets/x64/pass/lowered/build_brgemm.cpp
@@ -36,7 +36,7 @@ bool pass::BuildBrgemm::run(snippets::lowered::LinearIR& linear_ir,
     for (auto expr_it = begin; expr_it != end; expr_it++) {
         const auto& expr = *expr_it;
         const auto gemm_node = ov::as_type_ptr<GemmCPU>(expr->get_node());
-        if (!gemm_node || gemm_node->is_dynamic() || with_repacking(gemm_node->get_type())) {
+        if (!gemm_node || gemm_node->is_dynamic()) {
             continue;
         }
         const auto& loop_manager = linear_ir.get_loop_manager();