Understanding GNN Computational Graph: A Coordinated Computation, IO, and Memory Perspective

Hengrui Zhang, Zhongming Yu, Guohao Dai +4

Graph Neural Networks (GNNs) have been widely used in various domains, and GNNs with sophisticated computational graph lead to higher latency and larger memory consumption. Optimiz…