Skip to content

Commit

Permalink
remove extra tensor setting
Browse files Browse the repository at this point in the history
  • Loading branch information
aobolensk committed Feb 3, 2025
1 parent 1c9c44e commit 1f34bbf
Showing 1 changed file with 1 addition and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ bool pass::BuildBrgemm::run(snippets::lowered::LinearIR& linear_ir,
// Get innermost loop info
// TODO: check K-loop
const auto& inner_loop_info = loop_manager->get_loop_info<snippets::lowered::UnifiedLoopInfo>(loop_ids.front());
if (inner_loop_info->get_work_amount() % inner_loop_info->get_increment() != 0) {
if (inner_loop_info->is_dynamic() || inner_loop_info->get_work_amount() % inner_loop_info->get_increment() != 0) {
continue;
}
auto iter_count = inner_loop_info->get_work_amount() / inner_loop_info->get_increment();
Expand All @@ -85,9 +85,6 @@ bool pass::BuildBrgemm::run(snippets::lowered::LinearIR& linear_ir,

const auto& updated_expr = *expr_it;
updated_expr->set_live_regs(std::move(live_regs));
updated_expr->get_input_port_descriptor(0)->set_subtensor(in0_subtensor);
updated_expr->get_input_port_descriptor(1)->set_subtensor(in1_subtensor);
updated_expr->get_output_port_descriptor(0)->set_subtensor(out_subtensor);

modified |= true;
}
Expand Down

0 comments on commit 1f34bbf

Please sign in to comment.