Unverified Commit 6862e372 authored by Andrzej Kotłowski's avatar Andrzej Kotłowski Committed by GitHub
Browse files

Upgrade libxsmm (#5725)


Co-authored-by: default avatarRhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
parent 3d179846
...@@ -222,7 +222,7 @@ if((NOT MSVC) AND USE_LIBXSMM) ...@@ -222,7 +222,7 @@ if((NOT MSVC) AND USE_LIBXSMM)
) )
endif(REBUILD_LIBXSMM) endif(REBUILD_LIBXSMM)
add_dependencies(dgl libxsmm) add_dependencies(dgl libxsmm)
list(APPEND DGL_LINKER_LIBS -L${CMAKE_SOURCE_DIR}/third_party/libxsmm/lib/ xsmm) list(APPEND DGL_LINKER_LIBS -L${CMAKE_SOURCE_DIR}/third_party/libxsmm/lib/ xsmm.a)
endif((NOT MSVC) AND USE_LIBXSMM) endif((NOT MSVC) AND USE_LIBXSMM)
if(NOT MSVC) if(NOT MSVC)
......
...@@ -260,12 +260,12 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel( ...@@ -260,12 +260,12 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel(
kernel = libxsmm_dispatch_meltw_opreduce_vecs_idx( kernel = libxsmm_dispatch_meltw_opreduce_vecs_idx(
N, &_ld, &_ld, LIBXSMM_DATATYPE_F32, LIBXSMM_DATATYPE_F32, N, &_ld, &_ld, LIBXSMM_DATATYPE_F32, LIBXSMM_DATATYPE_F32,
(sizeof(IdType) == 8) ? LIBXSMM_DATATYPE_I64 : LIBXSMM_DATATYPE_I32, (sizeof(IdType) == 8) ? LIBXSMM_DATATYPE_I64 : LIBXSMM_DATATYPE_I32,
opredop_flags); opredop_flags, 0);
} else { // assume bf16 } else { // assume bf16
kernel = libxsmm_dispatch_meltw_opreduce_vecs_idx( kernel = libxsmm_dispatch_meltw_opreduce_vecs_idx(
N, &_ld, &_ld, LIBXSMM_DATATYPE_BF16, LIBXSMM_DATATYPE_BF16, N, &_ld, &_ld, LIBXSMM_DATATYPE_BF16, LIBXSMM_DATATYPE_BF16,
(sizeof(IdType) == 8) ? LIBXSMM_DATATYPE_I64 : LIBXSMM_DATATYPE_I32, (sizeof(IdType) == 8) ? LIBXSMM_DATATYPE_I64 : LIBXSMM_DATATYPE_I32,
opredop_flags); opredop_flags, 0);
} }
if (kernel == nullptr) { if (kernel == nullptr) {
......
Subproject commit fa687556130b6298430f1c0555a14cf79ab6101c Subproject commit 80090603e43f6ddc870cc42e1403dd0af07744cc
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