Skip to content

Fix stream synchronization in MTMG graph construction#4275

Merged
rapids-bot[bot] merged 7 commits intorapidsai:branch-24.06from
ChuckHastings:mtmg_debugging2
May 23, 2024
Merged

Fix stream synchronization in MTMG graph construction#4275
rapids-bot[bot] merged 7 commits intorapidsai:branch-24.06from
ChuckHastings:mtmg_debugging2

Conversation

@ChuckHastings
Copy link
Copy Markdown
Collaborator

@ChuckHastings ChuckHastings commented Mar 25, 2024

Restructure to pass stream instead of handle and synchronize appropriately.

Closes #4236

@ChuckHastings ChuckHastings added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels May 21, 2024
@ChuckHastings ChuckHastings marked this pull request as ready for review May 21, 2024 20:50
@ChuckHastings ChuckHastings requested a review from a team as a code owner May 21, 2024 20:50
Copy link
Copy Markdown
Contributor

@seunghwak seunghwak left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

* @param use_edge_type Whether or not the edgelist will have edge types
*/
per_device_edgelist_t(cugraph::mtmg::handle_t const& handle,
per_device_edgelist_t(rmm::cuda_stream_view stream_view,
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Something tedious, but I guess our convention is to pass the handle as the first input argument and stream_view as the last input argument. Should we better keep this convention?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, I should have caught that. I will make that change.

@ChuckHastings
Copy link
Copy Markdown
Collaborator Author

/merge

@rapids-bot rapids-bot bot merged commit 0d64b72 into rapidsai:branch-24.06 May 23, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cuGraph improvement Improvement / enhancement to an existing function non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

MTMG Use of streams

2 participants