From aa986d8577bacf3bdb49db0c9b6d8d5d16d06584 Mon Sep 17 00:00:00 2001 From: dmitrygo Date: Thu, 29 Aug 2024 10:16:00 +0400 Subject: [PATCH] [FORK][FIX] Fixed debug assert in jit_io_helper_t [FORK][FEATURE] Enable avx2 jit reorder for bf16 data type --- src/cpu/x64/utils/jit_io_helper.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cpu/x64/utils/jit_io_helper.cpp b/src/cpu/x64/utils/jit_io_helper.cpp index a8527362639..70a44cb0cdf 100644 --- a/src/cpu/x64/utils/jit_io_helper.cpp +++ b/src/cpu/x64/utils/jit_io_helper.cpp @@ -202,9 +202,8 @@ bool jit_io_helper_t::is_data_type_supported(const data_type_t dt) { case data_type::u8: case data_type::s8: return true; case data_type::bf16: - return is_superset(isa_, avx512_core) || isa_ == avx2_vnni_2; case data_type::f16: - return is_superset(isa_, avx512_core_fp16) || isa_ == avx2_vnni_2; + return is_superset(isa_, avx2); case data_type::f8_e4m3: case data_type::f8_e5m2: return is_superset(isa_, avx512_core_amx); default: assert(!"Unsupported data type"); @@ -837,6 +836,7 @@ void jit_io_helper_t::store_bf16( assert(bf16_supported_ && "Unsupported data type."); assert((src_vmm.isZMM() || src_vmm.isYMM()) && "Store operation for bf16 is not supported for Xmms."); + assert(is_superset(isa_, avx512_core) || isa_ == avx2_vnni_2); const auto &cvt_lower_vmm = typename vreg_traits::Vmm_lower_t(src_vmm.getIdx());