Skip to content

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
@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.

5 participants