Skip to content

Commit

Permalink
[FORK][FIX] IP weights compression: max bcast blocking computation
Browse files Browse the repository at this point in the history
[FORK][FEATURE] InnerProduct primitive: squashed weight decompression
  • Loading branch information
dmitry-gorokhov committed Jan 20, 2025
1 parent c7ecd8f commit 1789b1e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/cpu/x64/brgemm/brgemm_utils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -208,7 +208,8 @@ int calculate_max_bcast_block(brgemm_desc_t *brg, const int adj_ld_block2) {
max_reg_count
= nstl::min(max_reg_count, max_isa_regs - max_bcst_regs - 5);

const int postops_regs = brg->attr()
// For dynamic quantization case it is more performant to maximize the amount of accumulators
const int postops_regs = brg->attr() && !brg->with_src_dyn_quant
? injector::aux_vec_count(
brg->attr()->post_ops_, brg->isa_impl, true)
: 0;
Expand Down

0 comments on commit 1789b1e

Please sign in to comment.