Skip to content

Commit

Permalink
Fixing style
Browse files Browse the repository at this point in the history
  • Loading branch information
cjnolet committed Jan 31, 2022
1 parent d7b54a9 commit 2b2d363
Show file tree
Hide file tree
Showing 3 changed files with 1,262 additions and 1,164 deletions.
28 changes: 24 additions & 4 deletions cpp/include/raft/sparse/convert/detail/dense.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -98,13 +98,33 @@ void csr_to_dense(cusparseHandle_t handle,
RAFT_CUSPARSE_TRY(cusparseSetMatType(out_mat, CUSPARSE_MATRIX_TYPE_GENERAL));

size_t buffer_size;
RAFT_CUSPARSE_TRY(raft::sparse::cusparsecsr2dense_buffersize(
handle, nrows, ncols, nnz, out_mat, csr_data, csr_indptr, csr_indices, out, lda, &buffer_size, stream));
RAFT_CUSPARSE_TRY(raft::sparse::cusparsecsr2dense_buffersize(handle,
nrows,
ncols,
nnz,
out_mat,
csr_data,
csr_indptr,
csr_indices,
out,
lda,
&buffer_size,
stream));

rmm::device_uvector<char> buffer(buffer_size, stream);

RAFT_CUSPARSE_TRY(raft::sparse::cusparsecsr2dense(
handle, nrows, ncols, nnz, out_mat, csr_data, csr_indptr, csr_indices, out, lda, buffer.data(), stream));
RAFT_CUSPARSE_TRY(raft::sparse::cusparsecsr2dense(handle,
nrows,
ncols,
nnz,
out_mat,
csr_data,
csr_indptr,
csr_indices,
out,
lda,
buffer.data(),
stream));

RAFT_CUSPARSE_TRY_NO_THROW(cusparseDestroyMatDescr(out_mat));

Expand Down
Loading

0 comments on commit 2b2d363

Please sign in to comment.