Skip to content

Commit

Permalink
FIX style fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
dantegd committed Jan 28, 2025
1 parent 8e58d29 commit c75cd87
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 2 deletions.
3 changes: 2 additions & 1 deletion cpp/include/raft/linalg/detail/lanczos.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -1004,7 +1004,8 @@ int computeSmallestEigenvectors(
index_type_t n = A.nrows_;

// Check that parameters are valid
RAFT_EXPECTS(nEigVecs > 0 && static_cast<uint64_t>(nEigVecs) <= n, "Invalid number of eigenvectors.");
RAFT_EXPECTS(nEigVecs > 0 && static_cast<uint64_t>(nEigVecs) <= n,
"Invalid number of eigenvectors.");
RAFT_EXPECTS(restartIter > 0, "Invalid restartIter.");
RAFT_EXPECTS(tol > 0, "Invalid tolerance.");
RAFT_EXPECTS(maxIter >= nEigVecs, "Invalid maxIter.");
Expand Down
10 changes: 9 additions & 1 deletion cpp/include/raft/sparse/linalg/detail/spectral.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,15 @@ void fit_embedding(raft::resources const& handle,
rmm::device_uvector<int> src_offsets(n + 1, stream);
rmm::device_uvector<int> dst_cols(nnz, stream);
rmm::device_uvector<T> dst_vals(nnz, stream);
convert::coo_to_csr(handle, rows, cols, vals, static_cast<int>(nnz), static_cast<int>(n), src_offsets.data(), dst_cols.data(), dst_vals.data());
convert::coo_to_csr(handle,
rows,
cols,
vals,
static_cast<int>(nnz),
static_cast<int>(n),
src_offsets.data(),
dst_cols.data(),
dst_vals.data());

rmm::device_uvector<T> eigVals(n_components + 1, stream);
rmm::device_uvector<T> eigVecs(n * (n_components + 1), stream);
Expand Down

0 comments on commit c75cd87

Please sign in to comment.