Skip to content
This repository has been archived by the owner on Nov 25, 2024. It is now read-only.

Forward-merge branch-24.06 into branch-24.08 #178

Merged
merged 1 commit into from
May 29, 2024
Merged

Conversation

rapids-bot[bot]
Copy link

@rapids-bot rapids-bot bot commented May 29, 2024

Forward-merge triggered by push to branch-24.06 that creates a PR to keep branch-24.08 up-to-date. If this PR is unable to be immediately merged due to conflicts, it will remain open for the team to manually merge. See forward-merger docs for more info.

This PR introduces support for distributed graph sampling (via NCCL backend). The initial implementation focuses on the uniform neighbor sampler.
We are going to extend it to support other samplers in future.

Highlights:
- Distributed Graph Storage: Now, the graph structure (represented by `row_ptr` and `col_indx` tensors) can be stored as wholememory arrays in a distributed fashion with even distribution across ranks (support both `cpu` and `cuda` storage type).
- Distributed Sampling: The sampling process leverages the existing wholegraph gather function to collect the sampled nodes and edges across all ranks.
- Uniform Neighbor Sampler Support: Currently, only the uniform neighbor sampler is supported.

cc. @linhu-nv @dongxuy04 @BradReesWork  @nvcastet @TristonC

Authors:
  - Chang Liu (https://github.com/chang-l)

Approvers:
  - https://github.com/linhu-nv
  - Brad Rees (https://github.com/BradReesWork)

URL: #171
@GPUtester GPUtester merged commit 00365ee into branch-24.08 May 29, 2024
Copy link
Author

rapids-bot bot commented May 29, 2024

SUCCESS - forward-merge complete.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants