Characterizing Concurrency Mechanisms for NVIDIA GPUs under Deep Learning Workloads
Abstract
We investigate the performance of the concurrency mechanisms available on NVIDIA's new Ampere GPU microarchitecture under deep learning training and inference workloads. In contrast to previous studies that treat the GPU as a black box, we examine scheduling at the microarchitectural level. We find that the lack of fine-grained preemption mechanisms, robust task prioritization options, and contention-aware thread block placement policies limits the effectiveness of NVIDIA's concurrency mechanisms. In summary, the sequential nature of deep learning workloads and their fluctuating resource requirements and kernel runtimes make executing such workloads while maintaining consistently high utilization and low, predictable turnaround times difficult on current NVIDIA hardware.
- Publication:
-
arXiv e-prints
- Pub Date:
- October 2021
- DOI:
- 10.48550/arXiv.2110.00459
- arXiv:
- arXiv:2110.00459
- Bibcode:
- 2021arXiv211000459G
- Keywords:
-
- Computer Science - Distributed;
- Parallel;
- and Cluster Computing;
- Computer Science - Hardware Architecture;
- Computer Science - Machine Learning
- E-Print:
- To Appear in the 39th International Symposium on Computer Performance, Modeling, Measurements and Evaluation (Performance 21)