Skip to content

Commit 1789b1e

Browse files
[FORK][FIX] IP weights compression: max bcast blocking computation
[FORK][FEATURE] InnerProduct primitive: squashed weight decompression
1 parent c7ecd8f commit 1789b1e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/cpu/x64/brgemm/brgemm_utils.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,8 @@ int calculate_max_bcast_block(brgemm_desc_t *brg, const int adj_ld_block2) {
208208
max_reg_count
209209
= nstl::min(max_reg_count, max_isa_regs - max_bcst_regs - 5);
210210

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

0 commit comments

Comments
 (0)