RT Journal Article
T1 Solving the task scheduling and GPU reconfiguration problem on MIG devices via deep reinforcement learning
A1 Villarrubia Elvira, Jorge
A1 Costero Valero, Luis María
A1 Igual Peña, Francisco Daniel
A1 Olcoz Herrero, Katzalin
AB Recent advances in dynamic GPU partitioning, such as NVIDIA's Multi-Instance GPU (MIG) technology, have enhanced resource utilization by enabling task co-execution without contention. However, existing MIG schedulers remain limited to static or task-agnostic methods that sacrifice optimality for tractability. This paper presents a Deep Reinforcement Learning framework that seeks to minimize the completion time of a task queue by holistically addressing the dimensions of the problem: task molding, GPU reconfiguration and execution order. To manage the vast solution space, we apply optimizations such as discrete and canonical representation of states, unification of equivalent configurations, action masking, or promoting the exploration of reconfigurations; this offers insights for similar resource management scenarios. The proposed models are extensively evaluated with widely used benchmarks of the Rodinia and Altis suites, and synthetic workloads generated to emulate a wide range of plausible real situations. The final model improves to the state-of-the-art, especially in workloads that clearly contradict the assumptions of previous proposals, achieving a difference of less than 20% to the optimum. Additionally, two different approaches to the problem are faced (offline vs. online), discussing their theoretical advantages and disadvantages, and evaluating them experimentally for the final model.
PB Elsevier
SN 0167-739X
YR 2026
FD 2026-03
LK https://hdl.handle.net/20.500.14352/126143
UL https://hdl.handle.net/20.500.14352/126143
LA eng
NO Álvarez-Domínguez, Álvaro, et al. «No Black Holes from Light». Physical Review Letters, vol. 133, n.º 4, julio de 2024, p. 041401. DOI.org (Crossref), https://doi.org/10.1103/PhysRevLett.133.041401.
NO © 2025 The Author(s).
NO Ministerio de Ciencia e Innovación (España)
NO Agencia Estatal de Investigación
NO European Commission
DS Docta Complutense
RD 14 abr 2026