Enhancing High-Throughput GPU Random Walks Through Multi-Task Concurrency Orchestration
Random walk is a powerful tool for large-scale graph learning, but its high computational demand presents a challenge. While GPUs can accelerate random walk tasks, current frameworks fail to fully utilize GPU parallelism due to memory-to-compute bandwidth imbalance. In this article, CoWalker, an efficient GPU framework, is proposed to facilitate concurrent execution of random walks for high overall throughput. CoWalker features three novel designs. First, it incorporates a multi-level execution model that effectively orchestrates diverse walk tasks and reduces GPU stalls based on multiple graph characteristics. Second, it collaboratively manages graph data and streaming multiprocessors to minimize memory access interference and maximize core utilization under concurrent tasks. Finally, a multi-dimensional scheduler selects compatible random walk task combinations based on memory footprints to achieve maximum throughput. CoWalker significantly improves throughput over state-of-the-art baselines by mitigating concurrency overheads and effectively harnessing GPU parallelism. Our extensive evaluations on real-world workloads demonstrate that CoWalker achieves 2.75× higher overall system throughput compared with commercial tools and 1.56× over the SOTA academic system.
Authors
- Minyi Guo (ORCID: https://orcid.org/0000-0003-0034-2302)
- Shixuan Sun (ORCID: https://orcid.org/0000-0003-4060-9438)
- Xiaofeng Hou (ORCID: https://orcid.org/0000-0003-4372-7851)
- Junyi Mei (ORCID: https://orcid.org/0009-0008-1956-1242)
- Pengyu Wang (ORCID: https://orcid.org/0000-0002-3704-1530)
- Chao Li (ORCID: https://orcid.org/0000-0001-6218-4659)
- Jing Wang (ORCID: https://orcid.org/0000-0003-0547-6780)
- Cheng Xu (ORCID: https://orcid.org/0000-0002-4062-3558)
- Baoping Hao (ORCID: https://orcid.org/0009-0000-4936-1534)
Institutions
- Shanghai Jiao Tong University (CN)
- Alibaba Group (China) (CN)
Publication Details
- Journal
- ACM Transactions on Architecture and Code Optimization
- Published
- 2025-01-10
- DOI
- https://doi.org/10.1145/3711820
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China