The rapid development of AI technologies has significantly intensified the energy consumption challenge faced by AI clusters. Dynamic Voltage and Frequency Scaling (DVFS) stands as a crucial approach for managing power consumption in GPUs and NPUs. However, conventional DVFS strategies often assume uniform performance across computational units, overlooking the heterogeneity in performance and power caused by chip manufacturing variability. This variability leads to imbalanced task execution times across different processors, consequently reducing overall energy efficiency. Traditional task allocation strategies also fall short in leveraging the energy efficiency characteristics of processors, limiting the optimization potential of AI clusters. To tackle these challenges, we propose the VAFSA framework. VAFSA employs a variability-aware DVFS strategy that enables GPU/NPUs to adjust their operating frequencies to ensure synchronized computation times. VAFSA also features a task allocation strategy that prioritizes selecting processors with higher energy efficiency to further reduce energy consumption. Our simulation based on Ascend NPUs and NVIDIA GPUs demonstrates that VAFSA can reduce energy consumption of AI workloads by 3.5% to 44% in diverse scenarios.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Improving the Energy Efficiency of AI Clusters Through Variability-Aware Frequency Scaling and Task Allocation

  • Yijia Zhang,
  • Dongxiang Zhang,
  • Bingqiang Wang,
  • Qiang Wang,
  • Shixun Zhang

摘要

The rapid development of AI technologies has significantly intensified the energy consumption challenge faced by AI clusters. Dynamic Voltage and Frequency Scaling (DVFS) stands as a crucial approach for managing power consumption in GPUs and NPUs. However, conventional DVFS strategies often assume uniform performance across computational units, overlooking the heterogeneity in performance and power caused by chip manufacturing variability. This variability leads to imbalanced task execution times across different processors, consequently reducing overall energy efficiency. Traditional task allocation strategies also fall short in leveraging the energy efficiency characteristics of processors, limiting the optimization potential of AI clusters. To tackle these challenges, we propose the VAFSA framework. VAFSA employs a variability-aware DVFS strategy that enables GPU/NPUs to adjust their operating frequencies to ensure synchronized computation times. VAFSA also features a task allocation strategy that prioritizes selecting processors with higher energy efficiency to further reduce energy consumption. Our simulation based on Ascend NPUs and NVIDIA GPUs demonstrates that VAFSA can reduce energy consumption of AI workloads by 3.5% to 44% in diverse scenarios.