WattGPU: Predicting Inference Power and Latency on Unseen GPUs and LLMs
Summary
WattGPU introduces two predictive models for mean GPU power draw and Inter-Token Latency (ITL) during Large Language Model (LLM) inference. This approach leverages only publicly available LLM metadata and GPU specifications, eliminating the need for hardware access or profiling and enabling generalization to unseen NVIDIA server-grade GPUs and LLMs. Evaluated on 42 open-source LLMs (0.1B--27B parameters) and 8 GPUs, the power draw model achieved a median absolute percentage error (MAPE) of ≤3.4% for offline and ≤13.5% for server scenarios on unseen GPUs. The latency model achieved ≤8.5% in server mode, both maintaining strong GPU ranking correlations (Kendall τ≥0.76). WattGPU reduces MAPE by approximately 4× on unseen LLM-GPU combinations for server scenarios compared to standard baselines. Its data and code are publicly available at https://github.com/maufadel/wattgpu.
Key takeaway
For AI Architects and Machine Learning Engineers optimizing LLM deployments, WattGPU offers a critical tool to predict inference power and latency on unseen GPUs and LLMs. You can now make informed hardware selection decisions and optimize resource allocation without the time-consuming and costly process of exhaustive hardware profiling. This enables more efficient data center operations and reduced energy consumption for your LLM workloads.
Key insights
WattGPU predicts LLM inference power and latency on new hardware without profiling, using only public metadata.
Principles
- Public metadata enables cross-hardware prediction.
- Generalization to unseen hardware is achievable.
- Predictive models outperform physical baselines.
Method
WattGPU employs two predictive models for mean GPU power draw and Inter-Token Latency, trained on public LLM metadata and GPU specifications, validated via leave-one-GPU/LLM-out cross-validation.
In practice
- Optimize LLM-GPU matching without profiling.
- Estimate energy consumption for new deployments.
- Compare GPU efficiency for LLM workloads.
Topics
- LLM Inference
- GPU Performance Prediction
- Energy Efficiency
- Data Center Optimization
- NVIDIA GPUs
- Predictive Modeling
Code references
Best for: MLOps Engineer, Research Scientist, CTO, AI Scientist, Machine Learning Engineer, AI Architect
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.