Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
6862e372
Unverified
Commit
6862e372
authored
May 24, 2023
by
Andrzej Kotłowski
Committed by
GitHub
May 24, 2023
Browse files
Upgrade libxsmm (#5725)
Co-authored-by:
Rhett Ying
<
85214957+Rhett-Ying@users.noreply.github.com
>
parent
3d179846
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
4 deletions
+4
-4
CMakeLists.txt
CMakeLists.txt
+1
-1
src/array/cpu/spmm_blocking_libxsmm.h
src/array/cpu/spmm_blocking_libxsmm.h
+2
-2
third_party/libxsmm
third_party/libxsmm
+1
-1
No files found.
CMakeLists.txt
View file @
6862e372
...
...
@@ -222,7 +222,7 @@ if((NOT MSVC) AND USE_LIBXSMM)
)
endif
(
REBUILD_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
)
if
(
NOT MSVC
)
...
...
src/array/cpu/spmm_blocking_libxsmm.h
View file @
6862e372
...
...
@@ -260,12 +260,12 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel(
kernel
=
libxsmm_dispatch_meltw_opreduce_vecs_idx
(
N
,
&
_ld
,
&
_ld
,
LIBXSMM_DATATYPE_F32
,
LIBXSMM_DATATYPE_F32
,
(
sizeof
(
IdType
)
==
8
)
?
LIBXSMM_DATATYPE_I64
:
LIBXSMM_DATATYPE_I32
,
opredop_flags
);
opredop_flags
,
0
);
}
else
{
// assume bf16
kernel
=
libxsmm_dispatch_meltw_opreduce_vecs_idx
(
N
,
&
_ld
,
&
_ld
,
LIBXSMM_DATATYPE_BF16
,
LIBXSMM_DATATYPE_BF16
,
(
sizeof
(
IdType
)
==
8
)
?
LIBXSMM_DATATYPE_I64
:
LIBXSMM_DATATYPE_I32
,
opredop_flags
);
opredop_flags
,
0
);
}
if
(
kernel
==
nullptr
)
{
...
...
libxsmm
@
80090603
Compare
fa687556
...
80090603
Subproject commit
fa687556130b6298430f1c0555a14cf79ab6101
c
Subproject commit
80090603e43f6ddc870cc42e1403dd0af07744c
c
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment