Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "77241c48af6e31d9f7fabba63079ec62ac01be5a"
Unverified Commit 62351d8f authored by Ilia Taraban's avatar Ilia Taraban Committed by GitHub
Browse files

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

parent e1f663f8
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/array/cpu/spmm.h
View file @ 62351d8f
...@@ -267,9 +267,9 @@ void SpMMCmpCsr( ...@@ -267,9 +267,9 @@ void SpMMCmpCsr(
#if !defined(_WIN32) #if !defined(_WIN32)
#ifdef USE_LIBXSMM #ifdef USE_LIBXSMM
int cpu_id = libxsmm_cpuid_x86(); int cpu_id = libxsmm_cpuid_x86();
const bool no_libxsmm = const bool no_libxsmm = bcast.use_bcast ||
bcast.use_bcast || std::is_same<DType, double>::value || std::is_same<DType, double>::value ||
(std::is_same<DType, BFloat16>::value && cpu_id < LIBXSMM_X86_AVX512) || cpu_id < LIBXSMM_X86_AVX512 ||
!dgl::runtime::Config::Global()->IsLibxsmmAvailable(); !dgl::runtime::Config::Global()->IsLibxsmmAvailable();
if (!no_libxsmm) { if (!no_libxsmm) {
SpMMCmpCsrLibxsmm<IdType, DType, Op, Cmp>( SpMMCmpCsrLibxsmm<IdType, DType, Op, Cmp>(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment