1 简介
这篇文章为图神经网络设计了一个基于Multi-GPU的计算框架,主要包括,图顶点和边分块,冗余顶点的过滤,计算与传输的重叠。该框架适用于符合Scatter-ApplyEdge-Gather-ApplyVertex Neural Networks (SAGA-NN)类型的图神经网络。
2 NeuGraph计算框架
从整体上看,NeuGraph主要包括4部分:
- 一个转换引擎能够将GNN转换成chunk粒度的数据流图,就是按照NeuGraph的方式处理数据
- 一个流调度器,能够减少CPU与GPU之间的数据传输量并且尽可能的重叠计算和数据传输
- 一个图扩散引擎,使用一系列的快速扩散kernels加速计算,并且将某些操作融合以便消除多余的内存拷贝(没看懂这个描述)
- 一个数据流运行时
2.1 Graph-Aware Dataflow Translation
将点集均分成 P P