Fixes the bug when total_nnz is > integer limit (#3766)

e7ad4c9c · sanchit-misra · GitHub · 4b182b90 · e7ad4c9c
Unverified Commit e7ad4c9c authored Feb 24, 2022 by sanchit-misra Committed by GitHub Feb 23, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/array/cpu/spmm_blocking_libxsmm.h src/array/cpu/spmm_blocking_libxsmm.h +1 -1

No files found.
--- a/src/array/cpu/spmm_blocking_libxsmm.h
+++ b/src/array/cpu/spmm_blocking_libxsmm.h
@@ -454,7 +454,7 @@ void SpMMRedopCsrOpt(
  const IdType K = csr.num_cols;
  const IdType* indptr = csr.indptr.Ptr<IdType>();
  CHECK_NOTNULL(indptr);
-  const int total_nnz = indptr[M];
+  const IdType total_nnz = indptr[M];
  if (M <= 0 || K <= 0 || N <= 0 || total_nnz <= 0) return;

  const double avg_degree = total_nnz * 1.0 / M;