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

NVTX Profiler changes #8

Open
wants to merge 1 commit into
base: zaristei/rag_remote_backend
Choose a base branch
from

Conversation

zaristei
Copy link
Owner

Subset of this PR, containing the addition of NVTX profiling capabilities and examples

puririshi98 added a commit to pyg-team/pytorch_geometric that referenced this pull request Nov 26, 2024
…xamples) (#9666)

Follow up to [PR
9597](#9597). Includes
multiple changes related to LLM+GNN experiments and scaling up to a
remote backend. Including:

- LargeGraphIndexer for building a large knowledge graph locally from
multiple samples in an arbitrary dataset
- Remote Backend Loader and examples for deploying a Retrieval algorithm
to a third party backend FeatureStore or GraphStore
- NVTX profiling tools for nsys users
- Quality of Life improvements and benchmarking scripts for G-Retriever.

Updates using these for WebQSP will be moved to a seperate PR

UPDATE:
PR is being broken up into smaller PRs. These can be previewed here:

- zaristei#6
- zaristei#7
- zaristei#8

---------

Co-authored-by: Zack Aristei <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Zachary Aristei <[email protected]>
Co-authored-by: Rishi Puri <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant