Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update heterogeneous sampling algorithm code to use the newly developed heterogeneous sampling primitve. #4922

Merged
merged 18 commits into from
Feb 12, 2025

Conversation

seunghwak
Copy link
Contributor

@seunghwak seunghwak commented Feb 5, 2025

Update heterogeneous sampling application code to use the new heterogeneous sampling primitive.

Breaking as the edge_type_view input parameter is no longer optional (it is required for heterogeneous sampling).

closes #4919

Copy link

copy-pr-bot bot commented Feb 5, 2025

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@seunghwak seunghwak changed the base branch from branch-25.02 to branch-25.04 February 5, 2025 23:31
@seunghwak seunghwak added this to the 25.04 milestone Feb 5, 2025
@seunghwak seunghwak added improvement Improvement / enhancement to an existing function breaking Breaking change labels Feb 7, 2025
@seunghwak seunghwak marked this pull request as ready for review February 7, 2025 16:44
@seunghwak seunghwak requested a review from a team as a code owner February 7, 2025 16:44
@ChuckHastings
Copy link
Collaborator

Missing some copyright year updates.

@@ -994,6 +994,10 @@ struct neighbor_sampling_functor : public cugraph::c_api::abstract_functor {
// otherwised throw an error and suggest the user to call uniform neighbor sample instead

if (num_edge_types_ > 1) {
if (edge_types == nullptr) {
Copy link
Collaborator

Choose a reason for hiding this comment

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

Is there a reason to prefer an explicit if statement rather than CUGRAPH_EXPECTS(edge_types == nullptr,...?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

No, I was lazy. I just copied-and-pasted-and-modified the CUGRAPH_FAIL statement from the code below. I will make an update.

Copy link
Contributor

@jnke2016 jnke2016 left a comment

Choose a reason for hiding this comment

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

Looks good to me

@ChuckHastings
Copy link
Collaborator

/merge

@rapids-bot rapids-bot bot merged commit 549aacd into rapidsai:branch-25.04 Feb 12, 2025
79 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
breaking Breaking change cuGraph improvement Improvement / enhancement to an existing function
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Update heterogeneous sampling algorithm to use new primitive
3 participants