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

Improvements for Papers100m single gpu and single node multi gpu examples (Cugraph, GATConv, better default hyperparams, eval on all ranks) #8173

Merged
merged 260 commits into from
Mar 27, 2024

Conversation

puririshi98
Copy link
Contributor

@puririshi98 puririshi98 commented Oct 10, 2023

Introduces improvements to NVIDIA GPU examples:

  • Adds support for cuGraph data loading
  • Adds GAT in addition to GCN
  • Enables evaluation on all ranks
  • Improves default hyperparams

~45% test acc w/ latest hyperparams on 8 gpus

latest cugraph single node multigpu works but drops degree 0 nodes in neighborloading causing lower test accuracy. @alexbarghi-nv will fix in 24.04 RAPIDS release. in general i think once cugraph 24.02 is out we can merge this PR and the test accuracy will automatically go up when using cugraph once users upgrade to 24.04 cugraph or newer

@puririshi98 puririshi98 requested a review from rusty1s October 10, 2023 19:36
@puririshi98 puririshi98 self-assigned this Oct 10, 2023
@puririshi98 puririshi98 requested a review from wsad1 as a code owner October 10, 2023 19:36
pre-commit-ci bot and others added 22 commits March 6, 2024 23:07
@puririshi98
Copy link
Contributor Author

@rusty1s lmk if anything needed to merge

@rusty1s
Copy link
Member

rusty1s commented Mar 25, 2024

On it.

@puririshi98 puririshi98 merged commit 870179f into master Mar 27, 2024
14 of 15 checks passed
@puririshi98 puririshi98 deleted the cugraph-paper100m branch March 27, 2024 15:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants