-
Notifications
You must be signed in to change notification settings - Fork 304
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
Replace PropertyGraph in cugraph-PyG with FeatureStore #3159
Conversation
… with the server, added tests which use/test the new builtin test extensions.
…s (via rapids-pytest-benchmark plugin) easier to read, changes to allow RemoteGraph to use either a server-side Graph or PropertyGraph.
…r LocalCUDACluster for MG service, re-wrote start list generation to work with both SG and MG, added server extension for generating start list, changed scale from 24 to 23 to prevent OOM, added separate function for creating MG graphs, added debug print to show result sizes, fixes to cugraph MG Graph for incorrect column names, added util to cugraph for starting and stopping dask client.
…2-initial_service_benchmarks
…2-initial_service_benchmarks
…art so it fails faster, including the stdout/stderr output in the exception raised or to console depending on how it exited.
…chmarks for different libraries.
python/cugraph-pyg/cugraph_pyg/tests/mg/test_mg_cugraph_sampler.py
Outdated
Show resolved
Hide resolved
Co-authored-by: Vibhu Jawa <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Minor review on using cuda-array-interface
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Noticed one minor item which need not hold up approval.
/merge |
Codecov ReportBase: 55.22% // Head: 55.31% // Increases project coverage by
Additional details and impacted files@@ Coverage Diff @@
## branch-23.02 #3159 +/- ##
================================================
+ Coverage 55.22% 55.31% +0.08%
================================================
Files 148 148
Lines 9543 9553 +10
================================================
+ Hits 5270 5284 +14
+ Misses 4273 4269 -4
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. ☔ View full report at Codecov. |
Closes #3152 Depends on #3148 and #3159 This PR does support multiple trainers, even though there is no example for it. Authors: - Alex Barghi (https://github.com/alexbarghi-nv) - Chuck Hastings (https://github.com/ChuckHastings) - Vibhu Jawa (https://github.com/VibhuJawa) Approvers: - Vibhu Jawa (https://github.com/VibhuJawa) - Rick Ratzel (https://github.com/rlratzel) URL: #3170
Replaces PropertyGraph in cugraph-PyG with the new FeatureStore, which is the preferred solution for storing tensor features.
Closes #3151
Closes #3079