Apr, 2024

GNNBENCH: 公平高效的单 GPU 图神经网络系统基准测试

TL;DR通过 GNNBench 基准平台,我们提出了一种新的协议来交换捕获的张量数据,支持系统 API 中的自定义类,并允许将同一系统模块自动集成到诸多深度学习框架中,如 PyTorch 和 TensorFlow,这帮助我们发现了几个需要关注的测量问题。