-
Notifications
You must be signed in to change notification settings - Fork 315
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
Update heterogeneous sampling algorithm code to use the newly developed heterogeneous sampling primitve. #4922
Conversation
…to enh_hetero_sample
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. |
…to enh_hetero_sample
Missing some copyright year updates. |
cpp/src/c_api/neighbor_sampling.cpp
Outdated
@@ -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) { |
There was a problem hiding this comment.
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,...
?
There was a problem hiding this comment.
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.
There was a problem hiding this 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
/merge |
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