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

62351d8f · Ilia Taraban · GitHub · e1f663f8 · 62351d8f
Unverified Commit 62351d8f authored Jul 10, 2023 by Ilia Taraban Committed by GitHub Jul 10, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

src/array/cpu/spmm.h src/array/cpu/spmm.h +4 -4

No files found.
--- a/src/array/cpu/spmm.h
+++ b/src/array/cpu/spmm.h
@@ -267,9 +267,9 @@ 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) ||
+  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>(