CRIUgpu: Transparent Checkpointing of GPU-Accelerated Workloads

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Stoyanov, Radostin, Spišaková, Viktória, Ramos, Jesus, Gurfinkel, Steven, Vagin, Andrei, Reber, Adrian, Armour, Wesley, Bruno, Rodrigo
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912242569576448
author Stoyanov, Radostin
Spišaková, Viktória
Ramos, Jesus
Gurfinkel, Steven
Vagin, Andrei
Reber, Adrian
Armour, Wesley
Bruno, Rodrigo
author_facet Stoyanov, Radostin
Spišaková, Viktória
Ramos, Jesus
Gurfinkel, Steven
Vagin, Andrei
Reber, Adrian
Armour, Wesley
Bruno, Rodrigo
contents Deep learning training at scale is resource-intensive and time-consuming, often running across hundreds or thousands of GPUs for weeks or months. Efficient checkpointing is crucial for running these workloads, especially in multi-tenant environments where compute resources are shared, and job preemptions or interruptions are common. However, transparent and unified GPU snapshots are particularly challenging because of the hardware architecture differences between CPU and GPU, including memory subsystems, dynamic parallelism, and thread synchronization. State-of-the-art GPU checkpointing techniques typically leverage mechanisms that intercept, log, and replay device API calls. However, this approach adds performance overhead and requires hardware-specific implementation that is difficult to test, maintain, and integrate with existing container platforms. In this paper, we present CRIUgpu - a novel approach for transparent checkpointing of GPU-accelerated workloads that builds on recently introduced driver capabilities, enabling support for CUDA and ROCm applications. Our evaluation results show that CRIUgpu works with a variety of deep learning and high-performance computing workloads running across multiple GPUs, completely eliminating steady-state performance overheads, and significantly reducing recovery times compared to state-of-the-art transparent GPU checkpointing mechanisms.
format Preprint
id arxiv_https___arxiv_org_abs_2502_16631
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CRIUgpu: Transparent Checkpointing of GPU-Accelerated Workloads
Stoyanov, Radostin
Spišaková, Viktória
Ramos, Jesus
Gurfinkel, Steven
Vagin, Andrei
Reber, Adrian
Armour, Wesley
Bruno, Rodrigo
Distributed, Parallel, and Cluster Computing
Deep learning training at scale is resource-intensive and time-consuming, often running across hundreds or thousands of GPUs for weeks or months. Efficient checkpointing is crucial for running these workloads, especially in multi-tenant environments where compute resources are shared, and job preemptions or interruptions are common. However, transparent and unified GPU snapshots are particularly challenging because of the hardware architecture differences between CPU and GPU, including memory subsystems, dynamic parallelism, and thread synchronization. State-of-the-art GPU checkpointing techniques typically leverage mechanisms that intercept, log, and replay device API calls. However, this approach adds performance overhead and requires hardware-specific implementation that is difficult to test, maintain, and integrate with existing container platforms. In this paper, we present CRIUgpu - a novel approach for transparent checkpointing of GPU-accelerated workloads that builds on recently introduced driver capabilities, enabling support for CUDA and ROCm applications. Our evaluation results show that CRIUgpu works with a variety of deep learning and high-performance computing workloads running across multiple GPUs, completely eliminating steady-state performance overheads, and significantly reducing recovery times compared to state-of-the-art transparent GPU checkpointing mechanisms.
title CRIUgpu: Transparent Checkpointing of GPU-Accelerated Workloads
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2502.16631