diff --git a/cpp/include/raft/sparse/linalg/degree.cuh b/cpp/include/raft/sparse/linalg/degree.cuh index 9ce7321de0..dde811ee2d 100644 --- a/cpp/include/raft/sparse/linalg/degree.cuh +++ b/cpp/include/raft/sparse/linalg/degree.cuh @@ -68,7 +68,7 @@ template void coo_degree_scalar( const int* rows, const T* vals, nnz_type nnz, T scalar, outT* results, cudaStream_t stream = 0) { - detail::coo_degree_scalar<64>(rows, vals, (uint64_t)nnz, scalar, results, stream); + detail::coo_degree_scalar<64>(rows, vals, nnz, scalar, results, stream); } /** diff --git a/cpp/include/raft/sparse/linalg/detail/symmetrize.cuh b/cpp/include/raft/sparse/linalg/detail/symmetrize.cuh index 0fb3b53962..b248de855d 100644 --- a/cpp/include/raft/sparse/linalg/detail/symmetrize.cuh +++ b/cpp/include/raft/sparse/linalg/detail/symmetrize.cuh @@ -104,7 +104,7 @@ RAFT_KERNEL coo_symmetrize_kernel(nnz_t* row_ind, // Note that if we did find a match, we don't need to // compute `res` on it here because it will be computed // in a different thread. - if (!found_match && cur_val != 0.0) { + if (!found_match && vals[idx] != 0.0) { orows[out_start_idx + row_nnz] = cur_col; ocols[out_start_idx + row_nnz] = cur_row; ovals[out_start_idx + row_nnz] = res;