From bf58e72e5776a831bc65882e97e38f64626d5dec Mon Sep 17 00:00:00 2001 From: David Eberius Date: Tue, 15 Oct 2024 17:25:28 -0400 Subject: [PATCH] cpu: x64: rnn: Fixed a bug that forced the usage of avx_512_core even when avx_512_core_vnni is available. --- src/cpu/x64/rnn/rnn_brgemm_utils.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cpu/x64/rnn/rnn_brgemm_utils.cpp b/src/cpu/x64/rnn/rnn_brgemm_utils.cpp index dc3864bd3d0..39107ab2d9e 100644 --- a/src/cpu/x64/rnn/rnn_brgemm_utils.cpp +++ b/src/cpu/x64/rnn/rnn_brgemm_utils.cpp @@ -85,7 +85,7 @@ x64::cpu_isa_t brgemm_calc_isa( if (rnn.is_cell_dt_int8()) { return utils::map(true, x64::isa_undef, mayiuse(avx512_core_vnni), - avx512_core, mayiuse(avx512_core), avx512_core); + avx512_core_vnni, mayiuse(avx512_core), avx512_core); } else if (rnn.is_cell_dt_bf16()) { return x64::avx512_core_bf16; } else if (rnn.is_cell_dt_f16()) {