frozenleaves/oneDNN
1
0
Fork 0
mirror of https://github.com/uxlfoundation/oneDNN.git synced 2025-10-20 18:43:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

cpu: x64: matmul: enable GEMV code path to brgemm matmul for avx512

Browse Source
This commit is contained in:
Simonov, Alexander
2025-10-16 04:31:03 -07:00
parent d2b59266e1
commit e3c08d0784
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/cpu/x64/matmul/brgemm_matmul_utils.cpp
View File

@ -417,7 +417,8 @@ bool is_gemv_applicable(const brgemm_matmul_conf_t &bgmmc,
if (bgmmc.with_reduce) return false;
// BRGEMV currently supports only f32 and AVX2.
if (utils::one_of(false, bm_conf_utils.is_f32(), bgmmc.isa == avx2))
if (utils::one_of(false, bm_conf_utils.is_f32(),
bgmmc.isa == avx2 || bgmmc.isa == avx512_core))
return false;
if (utils::one_of(format_tag::undef, bm_conf_utils.get_gemv_A_tag(A_md),
@ -1439,6 +1440,9 @@ status_t init_brgemm_matmul_conf(cpu_isa_t isa, brgemm_matmul_conf_t &bgmmc,
bgmmc.is_gemv
= is_gemv_applicable(bgmmc, bm_conf_utils, src_md, weights_md);
VCONDCHECK_BG(IMPLICATION(bgmmc.is_gemv, isa == avx2),
"Fall back to the AVX2 implementation for the GEMV code path");
if (!bgmmc.is_gemv && bm_conf_utils.is_f32() && bgmmc.isa == avx2
&& (bgmmc.N == 1 || bgmmc.M == 1)) {
// The brgemm matmul implementation for avx2 and f32 data type has