From 4aa19d812960cd2c3036cf07483edf0669ae9521 Mon Sep 17 00:00:00 2001 From: Arseniy Obolenskiy Date: Mon, 13 Jan 2025 14:09:46 +0100 Subject: [PATCH] Address shape recalc concern --- .../include/snippets/lowered/pass/brgemm_blocking.hpp | 5 +++-- 1 file changed, 3 insertions(+), 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 index 147b75c00c7636..d3a7451569a284 100644 --- 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);