2 papers
cs.DC2024
A Unified CPU-GPU Protocol for GNN Training
Yi-Chien Lin, Gangda Deng, Viktor Prasanna
Training a Graph Neural Network (GNN) model on large-scale graphs involves a high volume of data communication and computations. While state-of-the-art CPUs and GPUs feature high c…
cs.DC2024
ARGO: An Auto-Tuning Runtime System for Scalable GNN Training on Multi-Core Processor
Yi-Chien Lin, Yuyang Chen, Sameh Gobriel +3
As Graph Neural Networks (GNNs) become popular, libraries like PyTorch-Geometric (PyG) and Deep Graph Library (DGL) are proposed; these libraries have emerged as the de facto stand…