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
5ac74f86
Unverified
Commit
5ac74f86
authored
Nov 26, 2020
by
Zihao Ye
Committed by
GitHub
Nov 26, 2020
Browse files
disable gespmm (#2371)
parent
12a193f2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
8 deletions
+1
-8
src/array/cuda/spmm.cu
src/array/cuda/spmm.cu
+1
-8
No files found.
src/array/cuda/spmm.cu
View file @
5ac74f86
...
...
@@ -244,14 +244,7 @@ void SpMMCsr(const std::string& op, const std::string& reduce,
bool
use_efeat
=
op
!=
"copy_lhs"
;
if
(
reduce
==
"sum"
)
{
if
((
!
use_efeat
||
is_scalar_efeat
)
&&
feat_len
>
64
)
{
// ge-spmm
if
(
use_efeat
&&
!
IsNullArray
(
csr
.
data
))
// reorder edge data
efeat
=
IndexSelect
(
efeat
,
csr
.
data
);
SWITCH_OP
(
op
,
Op
,
{
cuda
::
GESpMMCsr
<
IdType
,
DType
,
Op
>
(
csr
,
ufeat
,
efeat
,
out
,
feat_len
);
});
}
else
if
(
sizeof
(
IdType
)
==
4
&&
op
==
"copy_lhs"
)
{
// cusparse
if
(
sizeof
(
IdType
)
==
4
&&
op
==
"copy_lhs"
)
{
// cusparse
int64_t
x_length
=
1
;
for
(
int
i
=
1
;
i
<
ufeat
->
ndim
;
++
i
)
x_length
*=
ufeat
->
shape
[
i
];
...
...
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