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

Add LocalGraphStore for distributed training #7451

Merged
merged 12 commits into from
May 27, 2023

Conversation

ZhengHongming888
Copy link
Contributor

This code belongs to the part of the whole distributed training for PyG.

This class extends from GraphStore and use the store dict to save the graph topology and also use edge_ids dict to save the global eids.

We also include the unit test under /test folder to show how the graph edge_index/ edge_ids will be saving/getting and verified.

Any comments please let us know. thanks

@codecov
Copy link

codecov bot commented May 27, 2023

Codecov Report

Merging #7451 (7ccdd2d) into master (8770c93) will increase coverage by 0.01%.
The diff coverage is 100.00%.

❗ Current head 7ccdd2d differs from pull request most recent head 640815d. Consider uploading reports for the commit 640815d to get more accurate results

@@            Coverage Diff             @@
##           master    #7451      +/-   ##
==========================================
+ Coverage   91.42%   91.43%   +0.01%     
==========================================
  Files         443      445       +2     
  Lines       24656    24689      +33     
==========================================
+ Hits        22542    22575      +33     
  Misses       2114     2114              
Impacted Files Coverage Δ
torch_geometric/distributed/__init__.py 100.00% <100.00%> (ø)
torch_geometric/distributed/local_graph_store.py 100.00% <100.00%> (ø)

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@rusty1s rusty1s changed the title add data/local_graph_store.py & its test code /test Add LocalGraphStore for distributed training May 27, 2023
@rusty1s rusty1s enabled auto-merge (squash) May 27, 2023 19:55
@rusty1s rusty1s merged commit d64fa43 into pyg-team:master May 27, 2023
12 checks passed
@ZhengHongming888
Copy link
Contributor Author

Thanks Matthias!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants