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

[FEA] Support Edge ID Lookup in PyLibcuGraph #4687

Merged
merged 8 commits into from
Oct 21, 2024

Conversation

alexbarghi-nv
Copy link
Member

@alexbarghi-nv alexbarghi-nv commented Oct 1, 2024

Support Edge ID lookup in pylibcugraph. Also fixes some bugs in the C API (i.e. lookup table not being cleaned up correctly, container being incorrectly dereferenced as graph).

Verified in rapidsai/cugraph-gnn#50

@alexbarghi-nv alexbarghi-nv marked this pull request as ready for review October 10, 2024 15:45
@alexbarghi-nv alexbarghi-nv requested review from a team as code owners October 10, 2024 15:45
Copy link
Collaborator

@ChuckHastings ChuckHastings left a comment

Choose a reason for hiding this comment

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

Thanks for catching these!

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
Contributor

@rlratzel rlratzel left a comment

Choose a reason for hiding this comment

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

The Python/Cython changes look fine, just a couple of questions/requests:

  • Would it be better to be more explicit and replace the word "container" with "map", since (I'm assuming) it has to be a map?
  • Can you add a smoketest of some kind for the PLC updates? This would also be great for users looking for an example of how to use the new APIs.

@alexbarghi-nv
Copy link
Member Author

  • Would it be better to be more explicit and replace the word "container" with "map", since (I'm assuming) it has to be a map?

We're calling it EdgeIdLookupTable at the Python level. That should be ok, right?

Copy link
Contributor

@rlratzel rlratzel left a comment

Choose a reason for hiding this comment

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

Looks great, thanks @alexbarghi-nv !

@alexbarghi-nv
Copy link
Member Author

/merge

@rapids-bot rapids-bot bot merged commit 27f8ce1 into rapidsai:branch-24.12 Oct 21, 2024
132 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CMake cuGraph feature request New feature or request non-breaking Non-breaking change python
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants