Dgl graph store#2046
Dgl graph store#2046rapids-bot[bot] merged 9 commits intorapidsai:branch-22.04from BradReesWork:dgl-graph-store
Conversation
Codecov Report
@@ Coverage Diff @@
## branch-22.04 #2046 +/- ##
===============================================
Coverage ? 73.27%
===============================================
Files ? 154
Lines ? 9947
Branches ? 0
===============================================
Hits ? 7289
Misses ? 2658
Partials ? 0 Continue to review full report at Codecov.
|
|
@gpucibot merge |
|
Hi Brad, Thank you, |
wangxiaoyunNV
left a comment
There was a problem hiding this comment.
I have a concern about CuGraphStore class, it should have a to_device function or parameter in the sampling functions.
Thank you!
| max_depth=length, use_padding=True) | ||
|
|
||
| return p, w, s | ||
|
|
There was a problem hiding this comment.
Can we also have a function like async_fetch(self, ids, device): in the cuGraphStore class? so we can send the sampled graphs to different GPUs.
Thank you!
new GraphStore for DGL intergration Code will need to be added to DGL Authors: - Brad Rees (https://github.com/BradReesWork) Approvers: - Rick Ratzel (https://github.com/rlratzel) URL: rapidsai/cugraph#2046
new GraphStore for DGL intergration
Code will need to be added to DGL