pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-04 19:44:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[DistGB] exclude edges right after sampling (#7442)

Browse Source
This commit is contained in:
Rhett Ying
2024-06-03 10:16:44 +08:00
committed by GitHub
parent 70e631249c
commit 0fee9be329
3 changed files with 48 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/distributed/rgcn/README.md
View File

@@ -237,9 +237,11 @@ Compared to `DGL`, `GraphBolt` partitions are reduced to **72%** for `ogbn-mag`.
#### ogbn-mag
Compared to `DGL`, sampling with `GraphBolt` is reduced to **22%** for `ogbn-mag`.
Compared to `DGL`, sampling with `GraphBolt` is reduced to **15%**. As for the overhead of `exclude`, it's about **5%** in this test. This number could be higher if larger `fanout` or `batch size` is applied.
| Data Formats | Mean Sampling Time Per Iteration(50 iters in total, slowest rank)(seconds) |
| ------------ | ----------------------------------------------------------- |
| DGL | 7.49 |
| GraphBolt | 1.63 |
The time shown below is the mean sampling time per iteration(60 iters in total, slowest rank). Unit: seconds
| Data Formats | No Exclude | Exclude |
| ------------ | ---------- | ------- |
| DGL | 6.50 | 6.86 |
| GraphBolt | 0.95 | 1.00 |