论文标题
GSUITE:用于GPU的图形神经网络推断的灵活和框架独立的基准套件
gSuite: A Flexible and Framework Independent Benchmark Suite for Graph Neural Network Inference on GPUs
论文作者
论文摘要
随着图形神经网络(GNN)的兴趣正在增长,基准测试和性能表征研究的重要性正在增加。到目前为止,我们已经看到许多研究研究并介绍了GNN的性能和计算效率。但是,到目前为止所做的工作是使用一些高级GNN框架进行的。尽管这些框架提供了易用性,但它们对其他现有库的依赖性太多。实施细节和依赖项的层次使GNN模型的性能分析变得复杂,该模型建立在这些框架之上,尤其是在使用建筑模拟器时。此外,在先前的表征研究中通常忽略了GNN计算的不同方法,并且仅评估了一种常见的计算模型之一。基于我们观察到的这些缺点和需求,我们开发了一个独立的框架,支持多功能计算模型的基准套件,易于配置,并且可以与架构模拟器一起使用,而无需付出额外的努力。 我们称为GSUITE的基准套件仅利用硬件供应商的库,因此它独立于任何其他框架。 GSUITE可以使用当代GPU剖面和架构GPU模拟器对GNN推断进行详细的性能表征研究。为了说明我们新基准套件的好处,我们使用一组具有各种数据集的知名GNN模型进行了详细的特征研究。在真实的GPU卡和定时详细播放的GPU模拟器上运行GSUITE。我们还暗示了计算模型对性能的影响。我们使用几个评估指标来严格测量GNN计算的性能。
As the interest to Graph Neural Networks (GNNs) is growing, the importance of benchmarking and performance characterization studies of GNNs is increasing. So far, we have seen many studies that investigate and present the performance and computational efficiency of GNNs. However, the work done so far has been carried out using a few high-level GNN frameworks. Although these frameworks provide ease of use, they contain too many dependencies to other existing libraries. The layers of implementation details and the dependencies complicate the performance analysis of GNN models that are built on top of these frameworks, especially while using architectural simulators. Furthermore, different approaches on GNN computation are generally overlooked in prior characterization studies, and merely one of the common computational models is evaluated. Based on these shortcomings and needs that we observed, we developed a benchmark suite that is framework independent, supporting versatile computational models, easily configurable and can be used with architectural simulators without additional effort. Our benchmark suite, which we call gSuite, makes use of only hardware vendor's libraries and therefore it is independent of any other frameworks. gSuite enables performing detailed performance characterization studies on GNN Inference using both contemporary GPU profilers and architectural GPU simulators. To illustrate the benefits of our new benchmark suite, we perform a detailed characterization study with a set of well-known GNN models with various datasets; running gSuite both on a real GPU card and a timing-detailed GPU simulator. We also implicate the effect of computational models on performance. We use several evaluation metrics to rigorously measure the performance of GNN computation.