[Misc] Disable LibXSMM SpMM Cmp for AVX2 platforms (#5962)

dmlc · Jul 10, 2023 · 62351d8 · 62351d8
1 parent e1f663f
commit 62351d8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/array/cpu/spmm.h b/src/array/cpu/spmm.h
@@ -267,10 +267,10 @@ void SpMMCmpCsr(
 #if !defined(_WIN32)
 #ifdef USE_LIBXSMM
   int cpu_id = libxsmm_cpuid_x86();
-  const bool no_libxsmm =
-      bcast.use_bcast || std::is_same<DType, double>::value ||
-      (std::is_same<DType, BFloat16>::value && cpu_id < LIBXSMM_X86_AVX512) ||
-      !dgl::runtime::Config::Global()->IsLibxsmmAvailable();
+  const bool no_libxsmm = bcast.use_bcast ||
+                          std::is_same<DType, double>::value ||
+                          cpu_id < LIBXSMM_X86_AVX512 ||
+                          !dgl::runtime::Config::Global()->IsLibxsmmAvailable();
   if (!no_libxsmm) {
     SpMMCmpCsrLibxsmm<IdType, DType, Op, Cmp>(
         bcast, csr, ufeat, efeat, out, argu, arge);