Address shape recalc concern

openvinotoolkit · Jan 13, 2025 · cfaf273 · cfaf273
1 parent 1962d8c
commit cfaf273
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/common/snippets/include/snippets/lowered/pass/brgemm_blocking.hpp b/src/common/snippets/include/snippets/lowered/pass/brgemm_blocking.hpp
@@ -111,8 +111,9 @@ class BrgemmBlocking : public snippets::lowered::pass::RangedPass, public Brgemm
                 continue;
             OPENVINO_ASSERT(!blocking_loop_exists(loop_manager, brgemm_expr),
                             "Brgemm mustn't be covered in loops before blocking pass");
-            size_t m_block, n_block, k_block;
-            std::tie(m_block, n_block, k_block) = get_blocking_params(brgemm_expr);
+            size_t m_block = brgemm_expr->get_input_port_descriptor(0)->get_subtensor().front();
+            size_t n_block = brgemm_expr->get_input_port_descriptor(1)->get_subtensor().back();
+            size_t k_block = brgemm_expr->get_input_port_descriptor(0)->get_subtensor().back();
             modified = mark_blocking_loops(linear_ir, expr_it, m_block, n_block, k_block);
 #ifdef SNIPPETS_DEBUG_CAPS
             dump_params_to_csv(brgemm_expr, linear_ir);