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

Sampling modifications to support PyG and DGL options #3696

Merged

Conversation

ChuckHastings
Copy link
Collaborator

Modify sampling to add options for managing inclusion of sources on different hops.

Closes #3667
Closes #3665
Closes #3664

@ChuckHastings ChuckHastings self-assigned this Jul 6, 2023
@ChuckHastings ChuckHastings added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Jul 6, 2023
@ChuckHastings ChuckHastings added this to the 23.08 milestone Jul 6, 2023
@ChuckHastings ChuckHastings marked this pull request as ready for review July 6, 2023 13:26
@ChuckHastings ChuckHastings requested review from a team as code owners July 6, 2023 13:26
@alexbarghi-nv
Copy link
Member

This looks good so far, I will start testing it now.

Copy link
Member

@alexbarghi-nv alexbarghi-nv left a comment

Choose a reason for hiding this comment

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

Everything appears to be fully functional for both SG and MG. I now have a PR up for the changes to cuGraph-Python and pylibcugraph (#3699)

Copy link
Contributor

@naimnv naimnv left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
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.

Review part 1.

Copy link
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.

Review Part (2/2)

@alexbarghi-nv alexbarghi-nv self-requested a review July 7, 2023 20:37
Copy link
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.

Copy link
Member

@alexbarghi-nv alexbarghi-nv left a comment

Choose a reason for hiding this comment

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

👍

Copy link
Member

@VibhuJawa VibhuJawa left a comment

Choose a reason for hiding this comment

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

Tested the PR on a few datasets yesterday and everything seems to work as expected.

@ChuckHastings
Copy link
Collaborator Author

/merge

@rapids-bot rapids-bot bot merged commit 1692e4c into rapidsai:branch-23.08 Jul 18, 2023
rapids-bot bot pushed a commit that referenced this pull request Jul 19, 2023
Updates pylibcugraph and cuGraph-Python to support the new flags at the C++/C API layer (`unique_sources`, `carry_over_sources`, `deduplicate_sources`.  Adds appropriate tests.  Fixes a couple cuGraph-Python bugs that the new tests uncovered as well, and properly exposes `return_hops`.

Merge after #3696

Authors:
  - Alex Barghi (https://github.com/alexbarghi-nv)
  - Vibhu Jawa (https://github.com/VibhuJawa)
  - Chuck Hastings (https://github.com/ChuckHastings)
  - Brad Rees (https://github.com/BradReesWork)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)
  - Chuck Hastings (https://github.com/ChuckHastings)

URL: #3699
@ChuckHastings ChuckHastings deleted the sampling_updates_for_23.08 branch September 27, 2023 21:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement / enhancement to an existing function non-breaking Non-breaking change
Projects
None yet
6 participants