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

[RESEARCH] Investigate Renumbering as Part of Uniform Neighbor Sample Call #3668

Closed
3 tasks done
alexbarghi-nv opened this issue Jun 22, 2023 · 0 comments · Fixed by #3734
Closed
3 tasks done

[RESEARCH] Investigate Renumbering as Part of Uniform Neighbor Sample Call #3668

alexbarghi-nv opened this issue Jun 22, 2023 · 0 comments · Fixed by #3734
Assignees
Labels
feature request New feature or request

Comments

@alexbarghi-nv
Copy link
Member

alexbarghi-nv commented Jun 22, 2023

It should be possible in our C++ sampling code to renumber the final edgelist to conform with PyG numbering requirements (vertices are numbered in order of appearance). This would be done in parallel over all GPUs, yielding a significant performance boost over creating the MFGs during the loading process.

The only catch is that uniform_neighbor_sample would need to also output the original vertex ids so that the features can be retrieved during the loading process.

This would remove the need for a separate create_mfg C++ function to be called at loading time.

Tasks

  1. seunghwak
  2. ChuckHastings
  3. alexbarghi-nv
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

3 participants