MMK: A Hybrid Scheduling Framework for Fine-Grained GPU Sharing for Deep Learning Applications
Abstract
With the rapid growth of deep learning applications and their increasing demand for computational resources, GPU acceleration has become critical for supporting large-scale, compute-intensive deep learning tasks. To improve GPU utilization, emerging GPU sharing technologies such as Multi-Instance GPU (MIG) and Multi-Process Service (MPS) have been widely employed. However, MIG suffers from inefficient resource allocation while MPS introduces performance interference. Although integrating MIG with MPS further enhances GPU sharing capability, this approach still requires complex configuration and coarse-grained scheduling for dynamic workloads, making it ineffective in handling variations in resource demand for online jobs and offline jobs. To address these challenges, we propose MMK , a multi-level GPU sharing system that integrates MIG, MPS, and kernel-level scheduling for online and offline jobs. First, we design a hybrid scheduler to efficiently configure MIG and MPS resources for dynamic job demands. Second, we develop a kernel scheduler to implement fine-grained scheduling strategies that dynamically optimize kernel execution, thereby improving system throughput and reducing resource contention. Extensive experiments demonstrate that compared with the state-of-the-art framework, we reduce the average job completion time and makespan by 28% and 32%, respectively, and increase system throughput by 35%.
Fields of Study & Subjects
Parallel Computing and Optimization Techniques
Cloud Computing and Resource Management
Big Data and Digital Economy
Publication details
- Journal / source
- ACM Transactions on Architecture and Code Optimization
- Publisher
- Association for Computing Machinery (ACM)
- Volume
- 23
- Issue
- 3
- Pages
- 1–25
- Publication date
- 26 June 2026
- Language
- EN