Skip to content

Commit 5ac74f8

Browse files
authored
disable gespmm (dmlc#2371)
1 parent 12a193f commit 5ac74f8

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

src/array/cuda/spmm.cu

+1-8
Original file line numberDiff line numberDiff line change
@@ -244,14 +244,7 @@ void SpMMCsr(const std::string& op, const std::string& reduce,
244244
bool use_efeat = op != "copy_lhs";
245245

246246
if (reduce == "sum") {
247-
if ((!use_efeat || is_scalar_efeat) && feat_len > 64) { // ge-spmm
248-
if (use_efeat && !IsNullArray(csr.data)) // reorder edge data
249-
efeat = IndexSelect(efeat, csr.data);
250-
SWITCH_OP(op, Op, {
251-
cuda::GESpMMCsr<IdType, DType, Op>(
252-
csr, ufeat, efeat, out, feat_len);
253-
});
254-
} else if (sizeof(IdType) == 4 && op == "copy_lhs") { // cusparse
247+
if (sizeof(IdType) == 4 && op == "copy_lhs") { // cusparse
255248
int64_t x_length = 1;
256249
for (int i = 1; i < ufeat->ndim; ++i)
257250
x_length *= ufeat->shape[i];

0 commit comments

Comments
 (0)