GATv2 Efficiency Evaluation on artnet exp
5.45Latency speedup (fwd)CSR fused kernels
Evaluation Results
| Method | Links | ||||
|---|---|---|---|---|---|
| CSR fused kernelsHeads (H)=2, Head Dimension (D)=642026.05 | 5.45 | 2.04 | 6.9 | 5.34 | |
| CSR fused kernelsHeads (H)=8, Head Dimension (D)=1282026.05 | 4.73 | 1.83 | 10.38 | 6.37 | |
| PyGHeads (H)=2, Head Dimension (D)=642026.05 | 0.39 | 0.51 | 0.59 | 0.72 | |
| PyGHeads (H)=8, Head Dimension (D)=1282026.05 | 0.36 | 0.47 | 0.57 | 0.66 |