SpMM with GCN operator Performance on ogbn-arxiv
2.63Forward Latency SpeedupTC-GNN
Evaluation Results
| Method | Links | ||||
|---|---|---|---|---|---|
| TC-GNNhidden dim=5122026.05 | 2.63 | — | 1.93 | — | |
| cuSPARSEhidden dim=5122026.05 | 2.3 | 0.97 | 0.49 | 0.54 | |
| cuSPARSE + A˜⊤hidden dim=5122026.05 | 2.27 | 1.64 | 0.23 | 0.29 | |
| Ours (WSB/TC)hidden dim=5122026.05 | 1.56 | 1.41 | 1.49 | 0.99 | |
| FuseGNNhidden dim=5122026.05 | 0.27 | 1.44 | 0.21 | 0.2 | |
| PyGhidden dim=5122026.05 | 0.2 | 0.18 | 0.01 | 0.01 |