Skip to content

Setup e2e integration test for graph store homogeneous inference #433

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.

Sign up for GitHub

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

Jump to bottom
Merged
kmontemayor2-sc merged 19 commits into from
Jan 21, 2026
Merged

Setup e2e integration test for graph store homogeneous inference #433

kmontemayor2-sc merged 19 commits into from
Jan 21, 2026

Conversation

@kmontemayor2-sc
Copy link
Collaborator

@kmontemayor2-sc kmontemayor2-sc commented Jan 13, 2026 •
edited
Loading

Scope of work done

Please focus your reviews on the code under examples/link_prediction/graph_store. Everything else should be reviewed in other prs (mainly #432 )

Specifically, examples/link_prediction/graph_store/homogeneous_inference.py is the most important file here.

As a follow up to this I will make it so users can customize the server code.

The main differences between this code and the "colocated" mode (which should be documented in the docstring) are:

  1. Calling init_compute_node to setup the entire heterogeneous training cluster
  2. Using GraphStoreInfo to capture the cluster topology
  3. Using RemoteDistDataset instead of DistDataset as a pointer to the DistDataset stored on the storage cluster.

Where is the documentation for this feature?: N/A

Did you add automated tests or write a test plan?

Updated Changelog.md? NO

Ready for code review?: NO

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 13, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 13, 2026 •
edited
Loading

GiGL Automation

@ 18:04:16UTC : 🔄 E2E Test started.

@ 18:24:09UTC : ❌ Workflow failed.
Please check the logs for more details.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 13, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 13, 2026

GiGL Automation

@ 18:25:14UTC : 🔄 E2E Test started.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 13, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 13, 2026

GiGL Automation

@ 21:17:12UTC : 🔄 E2E Test started.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 14, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 14, 2026 •
edited
Loading

GiGL Automation

@ 21:27:12UTC : 🔄 E2E Test started.

@ 22:21:44UTC : ❌ Workflow failed.
Please check the logs for more details.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 14, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 14, 2026 •
edited
Loading

GiGL Automation

@ 23:48:00UTC : 🔄 E2E Test started.

@ 24:51:24UTC : ✅ Workflow completed successfully.

kmontemayor2-sc
kmontemayor2-sc commented Jan 15, 2026
View reviewed changes
@kmontemayor2-sc kmontemayor2-sc changed the title Kmonte/gs homo e2e inf Setup e2e integration test for graph store homogeneous inference Jan 15, 2026
@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 16, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 16, 2026 •
edited
Loading

GiGL Automation

@ 01:37:59UTC : 🔄 E2E Test started.

@ 03:05:51UTC : ✅ Workflow completed successfully.

svij-sc
svij-sc reviewed Jan 17, 2026
View reviewed changes
Copy link
Collaborator

@svij-sc svij-sc left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Amazing!
Really excited to see this go in 😄

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 20, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 20, 2026 •
edited
Loading

GiGL Automation

@ 18:09:54UTC : 🔄 E2E Test started.

@ 19:30:12UTC : ❌ Workflow failed.
Please check the logs for more details.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 20, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 20, 2026 •
edited
Loading

GiGL Automation

@ 19:41:53UTC : 🔄 E2E Test started.

@ 21:05:18UTC : ❌ Workflow failed.
Please check the logs for more details.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 20, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 20, 2026 •
edited
Loading

GiGL Automation

@ 21:14:23UTC : 🔄 E2E Test started.

@ 22:40:21UTC : ❌ Workflow failed.
Please check the logs for more details.

mkolodner-sc
mkolodner-sc approved these changes Jan 20, 2026
View reviewed changes
Copy link
Collaborator

@mkolodner-sc mkolodner-sc left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks a lot Kyle!

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 20, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 20, 2026 •
edited
Loading

GiGL Automation

@ 22:51:05UTC : 🔄 E2E Test started.

@ 24:15:14UTC : ❌ Workflow failed.
Please check the logs for more details.

@kmontemayor2-sc
Copy link
Collaborator Author

kmontemayor2-sc commented Jan 21, 2026

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Jan 21, 2026 •
edited
Loading

GiGL Automation

@ 24:31:34UTC : 🔄 E2E Test started.

@ 01:55:46UTC : ✅ Workflow completed successfully.

@kmontemayor2-sc kmontemayor2-sc marked this pull request as ready for review January 21, 2026 16:57
@kmontemayor2-sc kmontemayor2-sc added this pull request to the merge queue Jan 21, 2026
Merged via the queue into main with commit 33b109d Jan 21, 2026
6 checks passed
@kmontemayor2-sc kmontemayor2-sc deleted the kmonte/gs-homo-e2e-inf branch January 21, 2026 18:24
zfan3-sc pushed a commit that referenced this pull request Jan 21, 2026
@kmontemayor2-sc @kmonte @zfan3-sc
Setup e2e integration test for graph store homogeneous inference (#433)
ea16ba2 
Co-authored-by: kmontemayor <kyle.e.montemayor@gmail.com>
zfan3-sc pushed a commit that referenced this pull request Jan 21, 2026
@kmontemayor2-sc @kmonte @zfan3-sc
Setup e2e integration test for graph store homogeneous inference (#433)
b2734df 
Co-authored-by: kmontemayor <kyle.e.montemayor@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@svij-sc svij-sc svij-sc approved these changes

@mkolodner-sc mkolodner-sc mkolodner-sc approved these changes

@nshah-sc nshah-sc Awaiting requested review from nshah-sc nshah-sc is a code owner

@yliu2-sc yliu2-sc Awaiting requested review from yliu2-sc yliu2-sc is a code owner

@xgao4-sc xgao4-sc Awaiting requested review from xgao4-sc xgao4-sc is a code owner

@zfan3-sc zfan3-sc Awaiting requested review from zfan3-sc zfan3-sc is a code owner

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

5 participants

@kmontemayor2-sc @svij-sc @mkolodner-sc @kmonte