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

[GraphBolt][CUDA] Showcase GPUCachedFeature usage in an example. #6955

Closed
Tracked by #6826
mfbalin opened this issue Jan 16, 2024 · 1 comment · Fixed by #7074
Closed
Tracked by #6826

[GraphBolt][CUDA] Showcase GPUCachedFeature usage in an example. #6955

mfbalin opened this issue Jan 16, 2024 · 1 comment · Fixed by #7074
Assignees

Comments

@mfbalin
Copy link
Collaborator

mfbalin commented Jan 16, 2024

No description provided.

@mfbalin
Copy link
Collaborator Author

mfbalin commented Jan 16, 2024

@frozenbugs Do you think anybody would like to work on adding such an example? We could showcase it after a larger dataset is added to GraphBolt in as part of #6909. We can then showcase the speedups hopefully, however, to show speedups, the feature copy needs to dominate the overall training runtime. But with #6954, the feature copy can be fully overlapped so it is going to be a bit harder to show speedups after it is merged.

When we have a large dataset such as ogbn-papers100M or maybe the mag240M hetero-rgcn.py example, we need to wrap its main TorchBasedFeature with a GPUCachedFeature in #6939.

@github-project-automation github-project-automation bot moved this to 🏠 Backlog in DGL Project Tracker Jan 25, 2024
@frozenbugs frozenbugs moved this from 🏠 Backlog to 📋 Planned in DGL Project Tracker Jan 25, 2024
@frozenbugs frozenbugs added this to the 2023 12.30 Graphbolt milestone Jan 25, 2024
@TristonC TristonC self-assigned this Jan 25, 2024
@github-project-automation github-project-automation bot moved this from 📋 Planned to ✅ Done in DGL Project Tracker Feb 2, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Archived in project
Development

Successfully merging a pull request may close this issue.

3 participants