Skip to content

Use NCCL for Dask kNN and DBSCAN#7574

Merged
rapids-bot[bot] merged 9 commits intorapidsai:mainfrom
viclafargue:dask-nn-with-nccl-comms
Jan 15, 2026
Merged

Use NCCL for Dask kNN and DBSCAN#7574
rapids-bot[bot] merged 9 commits intorapidsai:mainfrom
viclafargue:dask-nn-with-nccl-comms

Conversation

@viclafargue
Copy link
Copy Markdown
Contributor

@viclafargue viclafargue commented Dec 8, 2025

Closes #7573 and #7530

This PR updates Dask kNN and DBSCAN to ensure that device-to-device communications are performed through NCCL instead of UCX as a temporary workaround for issues currently experienced with the RAFT communicator.

@viclafargue viclafargue changed the title Use NCCL for Dask kNN Use NCCL for Dask kNN and DBSCAN Dec 8, 2025
@viclafargue viclafargue added bug Something isn't working non-breaking Non-breaking change labels Dec 8, 2025
@viclafargue
Copy link
Copy Markdown
Contributor Author

/merge

@rapids-bot rapids-bot Bot merged commit 4ef41fe into rapidsai:main Jan 15, 2026
117 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working CUDA/C++ non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Faulty comms in Dask kNN [BUG] NearestNeighbors.kneighbors() hangs indefinitely with UCX protocol

3 participants