Datacenter-Scale Analysis and Optimization of GPU Machine Learning Workloads

IEEE Micro - Special Issue on Commercial Products


We present a system to collectively optimize efficiency in a very large scale deployment of GPU servers for machine learning workloads at Facebook. Our system (a) measures and stores system-wide efficiency metrics for every executed workflow, (b) aggregates data from across the execution stack to identify optimization opportunities that maximize fleet-wide efficiency improvements, (c) provides periodic and on-demand whole-system profiling for workflows and, (d) automatically analyzes traces for common anti-patterns. We present each component of the stack and show case studies demonstrating the use of the tools to significantly improve performance. To our knowledge, our system is the most complete and effective solution for identifying and addressing efficiency problems in datacenter-scale GPU deployments.

Featured Publications