Skip to content

Commit

Permalink
Swap blocks and threads_per_block in compute_graph_laplacian (#…
Browse files Browse the repository at this point in the history
…2597)

These were incorrectly ordered before, leading to errors on larger data
in `cuml.UMAP`.

Fixes rapidsai/cuml#6370.
  • Loading branch information
jcrist authored Feb 28, 2025
1 parent 0b244c4 commit d77fec7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cpp/include/raft/sparse/linalg/detail/laplacian.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ auto compute_graph_laplacian(
auto static constexpr const threads_per_block = 256;
auto blocks = std::min(int((dim + threads_per_block - 1) / threads_per_block), 65535);
auto stream = resource::get_cuda_stream(res);
detail::compute_graph_laplacian_kernel<<<threads_per_block, blocks, 0, stream>>>(
detail::compute_graph_laplacian_kernel<<<blocks, threads_per_block, 0, stream>>>(
result.get_elements().data(),
result_structure.get_indices().data(),
result_structure.get_indptr().data(),
Expand Down

0 comments on commit d77fec7

Please sign in to comment.