From fdb51f294ece31c9befd9c9ef15a5714c79a2722 Mon Sep 17 00:00:00 2001 From: Ryo Suzuki Date: Fri, 4 Oct 2024 17:32:50 +0100 Subject: [PATCH] cpu: aarch64: brgemm: Fix unimplemented conditions for brgemm (#2148) --- src/cpu/aarch64/brgemm/brgemm_utils.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cpu/aarch64/brgemm/brgemm_utils.cpp b/src/cpu/aarch64/brgemm/brgemm_utils.cpp index 24e109584a8..c517d9a0856 100644 --- a/src/cpu/aarch64/brgemm/brgemm_utils.cpp +++ b/src/cpu/aarch64/brgemm/brgemm_utils.cpp @@ -50,14 +50,14 @@ impl::data_type_t get_accum_datatype(brgemm_t *brg) { status_t init_kernel_datatype( brgemm_t *brg, impl::data_type_t dt_a, impl::data_type_t dt_b) { - if (dt_a != data_type::undef && dt_b != data_type::undef) + if (!(dt_a != data_type::undef && dt_b != data_type::undef)) return status::unimplemented; brg->is_int8 = utils::one_of(dt_a, data_type::u8, data_type::s8) && utils::one_of(dt_b, data_type::u8, data_type::s8); brg->is_bf16 = (dt_a == data_type::bf16) && (dt_b == data_type::bf16); brg->is_f32 = (dt_a == data_type::f32) && (dt_b == data_type::f32); brg->is_f16 = utils::one_of(data_type::f16, dt_a, dt_b); - if (brg->is_int8 || brg->is_bf16 || brg->is_f32 || brg->is_f16) + if (!(brg->is_int8 || brg->is_bf16 || brg->is_f32 || brg->is_f16)) return status::unimplemented; return status::success; }