diff options
Diffstat (limited to 'src/PackAWithRowOffset.cc')
| -rw-r--r-- | src/PackAWithRowOffset.cc | 14 |
1 files changed, 3 insertions, 11 deletions
diff --git a/src/PackAWithRowOffset.cc b/src/PackAWithRowOffset.cc index e84c67b..733bf5c 100644 --- a/src/PackAWithRowOffset.cc +++ b/src/PackAWithRowOffset.cc @@ -39,8 +39,7 @@ PackAWithRowOffset<T, accT>::PackAWithRowOffset( if (!cpuinfo_initialize()) { throw std::runtime_error("Failed to initialize cpuinfo!"); } - if ((!fbgemmHasAvx512VnniSupport() && !fbgemmHasAvx512Support() && - !fbgemmHasAvx2Support())) { + if ((!fbgemmHasAvx512Support() && !fbgemmHasAvx2Support())) { assert(0 && "unknown architecure"); } @@ -49,12 +48,7 @@ PackAWithRowOffset<T, accT>::PackAWithRowOffset( BaseType::bcol_ = params->KCB; row_interleave_B_ = params->ROW_INTERLEAVE; } else { - if (fbgemmHasAvx512VnniSupport()) { - BaseType::brow_ = PackingTraits<T, accT, inst_set_t::avx512_vnni>::MCB; - BaseType::bcol_ = PackingTraits<T, accT, inst_set_t::avx512_vnni>::KCB; - row_interleave_B_ = - PackingTraits<T, accT, inst_set_t::avx512_vnni>::ROW_INTERLEAVE; - } else if (fbgemmHasAvx512Support()) { + if (fbgemmHasAvx512Support()) { BaseType::brow_ = PackingTraits<T, accT, inst_set_t::avx512>::MCB; BaseType::bcol_ = PackingTraits<T, accT, inst_set_t::avx512>::KCB; row_interleave_B_ = @@ -195,9 +189,7 @@ int PackAWithRowOffset<T, accT>::rowOffsetBufferSize( if (params) { return params->MCB; } else { - if (fbgemmHasAvx512VnniSupport()) { - return PackingTraits<T, accT, inst_set_t::avx512_vnni>::MCB; - } else if (fbgemmHasAvx512Support()) { + if (fbgemmHasAvx512Support()) { return PackingTraits<T, accT, inst_set_t::avx512>::MCB; } else if (fbgemmHasAvx2Support()) { return PackingTraits<T, accT, inst_set_t::avx2>::MCB; |