GNNBENCH：面向单GPU GNN系统的公平且高效的基准测试
机器学习
2024-04-08 v1 分布式、并行与集群计算
摘要
我们假设，缺乏标准化基准测试导致了社区未注意到的若干基本陷阱，这些陷阱在GNN系统设计和评估中屡屡出现。在本工作中，我们提出GNNBench，一个专注于系统创新的即插即用基准平台。GNNBench提供了交换其囚禁张量数据的新协议，支持System API中的自定义类，并允许自动集成相同的系统模块到多个深度学习框架，如PyTorch和TensorFlow。为说明此类基准框架的重要性，我们集成了多个GNN系统。我们的结果表明，与GNNBench集成有助于我们识别出若干值得社区关注的测量问题。
关键词
引用
@article{arxiv.2404.04118,
title = {GNNBENCH: Fair and Productive Benchmarking for Single-GPU GNN System},
author = {Yidong Gong and Pradeep Kumar},
journal= {arXiv preprint arXiv:2404.04118},
year = {2024}
}