AGILE: Lightweight and Efficient Asynchronous GPU-SSD Integration

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Yang, Zhuoping, Zhuang, Jinming, Chen, Xingzhen, Jones, Alex K., Zhou, Peipei
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909752549703680
author Yang, Zhuoping
Zhuang, Jinming
Chen, Xingzhen
Jones, Alex K.
Zhou, Peipei
author_facet Yang, Zhuoping
Zhuang, Jinming
Chen, Xingzhen
Jones, Alex K.
Zhou, Peipei
contents GPUs are critical for compute-intensive applications, yet emerging workloads such as recommender systems, graph analytics, and data analytics often exceed GPU memory capacity. Existing solutions allow GPUs to use CPU DRAM or SSDs as external memory, and the GPU-centric approach enables GPU threads to directly issue NVMe requests, further avoiding CPU intervention. However, current GPU-centric approaches adopt synchronous I/O, forcing threads to stall during long communication delays. We propose AGILE, a lightweight asynchronous GPU-centric I/O library that eliminates deadlock risks and integrates a flexible HBM-based software cache. AGILE overlaps computation and I/O, improving performance by up to 1.88$\times$ across workloads with diverse computation-to-communication ratios. Compared to BaM on DLRM, AGILE achieves up to 1.75$\times$ speedup through efficient design and overlapping; on graph applications, AGILE reduces software cache overhead by up to 3.12$\times$ and NVMe I/O overhead by up to 2.85$\times$; AGILE also lowers per-thread register usage by up to 1.32$\times$.
format Preprint
id arxiv_https___arxiv_org_abs_2504_19365
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AGILE: Lightweight and Efficient Asynchronous GPU-SSD Integration
Yang, Zhuoping
Zhuang, Jinming
Chen, Xingzhen
Jones, Alex K.
Zhou, Peipei
Distributed, Parallel, and Cluster Computing
GPUs are critical for compute-intensive applications, yet emerging workloads such as recommender systems, graph analytics, and data analytics often exceed GPU memory capacity. Existing solutions allow GPUs to use CPU DRAM or SSDs as external memory, and the GPU-centric approach enables GPU threads to directly issue NVMe requests, further avoiding CPU intervention. However, current GPU-centric approaches adopt synchronous I/O, forcing threads to stall during long communication delays. We propose AGILE, a lightweight asynchronous GPU-centric I/O library that eliminates deadlock risks and integrates a flexible HBM-based software cache. AGILE overlaps computation and I/O, improving performance by up to 1.88$\times$ across workloads with diverse computation-to-communication ratios. Compared to BaM on DLRM, AGILE achieves up to 1.75$\times$ speedup through efficient design and overlapping; on graph applications, AGILE reduces software cache overhead by up to 3.12$\times$ and NVMe I/O overhead by up to 2.85$\times$; AGILE also lowers per-thread register usage by up to 1.32$\times$.
title AGILE: Lightweight and Efficient Asynchronous GPU-SSD Integration
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2504.19365