1 files changed, 2 insertions, 7 deletions
diff --git a/src/PackMatrix.cc b/src/PackMatrix.cc
index ff7b842..c7503dd 100644
--- a/src/PackMatrix.cc
+++ b/src/PackMatrix.cc
@@ -36,8 +36,7 @@ int PackMatrix<PT, inpType, accType>::packedBufferSize(
   if (!cpuinfo_initialize()) {
     throw std::runtime_error("Failed to initialize cpuinfo!");
   }
-  if ((!fbgemmHasAvx512VnniSupport() && !fbgemmHasAvx512Support() &&
-       !fbgemmHasAvx2Support())) {
+  if ((!fbgemmHasAvx512Support() && !fbgemmHasAvx2Support())) {
     assert(0 && "unknown architecure");
   }
 
@@ -47,11 +46,7 @@ int PackMatrix<PT, inpType, accType>::packedBufferSize(
     NCB = params->NCB;
     KCB = params->KCB;
   } else {
-    if (fbgemmHasAvx512VnniSupport()) {
-      MCB = PackingTraits<inpType, accType, inst_set_t::avx512_vnni>::MCB;
-      NCB = PackingTraits<inpType, accType, inst_set_t::avx512_vnni>::NCB;
-      KCB = PackingTraits<inpType, accType, inst_set_t::avx512_vnni>::KCB;
-    } else if (fbgemmHasAvx512Support()) {
+    if (fbgemmHasAvx512Support()) {
       MCB = PackingTraits<inpType, accType, inst_set_t::avx512>::MCB;
       NCB = PackingTraits<inpType, accType, inst_set_t::avx512>::NCB;
       KCB = PackingTraits<inpType, accType, inst_set_t::avx512>::KCB;