论文阅读:NeuGraph: Parallel Deep Neural Network Computation on Large Graphs

NeuGraph是一个针对大型图神经网络的计算框架,利用Multi-GPU并行处理,通过图感知数据流转换、选择性调度、管道调度优化计算效率,减少数据传输并重叠计算。其核心包括图分块、过滤冗余顶点、计算与传输优化,以及图传播引擎,旨在提高GNN模型的性能。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1 简介

这篇文章为图神经网络设计了一个基于Multi-GPU的计算框架,主要包括,图顶点和边分块,冗余顶点的过滤,计算与传输的重叠。该框架适用于符合Scatter-ApplyEdge-Gather-ApplyVertex Neural Networks (SAGA-NN)类型的图神经网络。
在这里插入图片描述

2 NeuGraph计算框架

从整体上看,NeuGraph主要包括4部分:

  1. 一个转换引擎能够将GNN转换成chunk粒度的数据流图,就是按照NeuGraph的方式处理数据
  2. 一个流调度器,能够减少CPU与GPU之间的数据传输量并且尽可能的重叠计算和数据传输
  3. 一个图扩散引擎,使用一系列的快速扩散kernels加速计算,并且将某些操作融合以便消除多余的内存拷贝(没看懂这个描述)
  4. 一个数据流运行时

2.1 Graph-Aware Dataflow Translation

将点集均分成 P P

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值