Address shape recalc concern

openvinotoolkit · Jan 13, 2025 · 4aa19d8 · 4aa19d8
1 parent 1962d8c
commit 4aa19d8
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
@@ -176,8 +176,9 @@ class BrgemmBlocking : public snippets::lowered::pass::RangedPass, public Brgemm
         int64_t M, N, K;
         std::tie(M, N, K) = get_brgemm_dimensions(brgemm_expr);
         ss << M << "," << N << "," << K << ",";
-        int64_t m_block, n_block, k_block;
-        std::tie(m_block, n_block, k_block) = get_blocking_params(brgemm_expr);
+        int64_t m_block = brgemm_expr->get_input_port_descriptor(0)->get_subtensor().front();
+        int64_t n_block = brgemm_expr->get_input_port_descriptor(1)->get_subtensor().back();
+        int64_t k_block = brgemm_expr->get_input_port_descriptor(0)->get_subtensor().back();
         ss << m_block << "," << n_block << "," << k_block;
         ss << '\n';
         std::ofstream csv_file(csv_path, std::ios_base::app);