ytopt: Autotuning Scientific Applications for Energy Efficiency at Large Scales

Xingfu Wu, Prasanna Balaprakash, Michael Kruse, Jaehoon Koo, Brice Videau +5 more
3/28/2023
cs.DCcs.LGcs.PF

No official code found. Best implementation: ytopt-team/ytopt (55 stars, BSD-2-Clause, PyTorch, last commit 2026-04-30).

Abstract

As we enter the exascale computing era, efficiently utilizing power and optimizing the performance of scientific applications under power and energy constraints has become critical and challenging. We propose a low-overhead autotuning framework to autotune performance and energy for various hybrid MPI/OpenMP scientific applications at large scales and to explore the tradeoffs between application runtime and power/energy for energy efficient application execution, then use this framework to autotune four ECP proxy applications -- XSBench, AMG, SWFFT, and SW4lite. Our approach uses Bayesian optimization with a Random Forest surrogate model to effectively search parameter spaces with up to 6 million different configurations on two large-scale production systems, Theta at Argonne National Laboratory and Summit at Oak Ridge National Laboratory. The experimental results show that our autotuning framework at large scales has low overhead and achieves good scalability. Using the proposed autotuning framework to identify the best configurations, we achieve up to 91.59% performance improvement, up to 21.2% energy savings, and up to 37.84% EDP improvement on up to 4,096 nodes.

View on arXivView PDF

Code Implementations(1)

ytopt-team/ytoptCommunity100%
PyTorchBSD-2-Clause

ytopt: machine-learning-based autotuning and hyperparameter optimization framework using Bayesian Optimization

5518C, C++Jan 15, 20195 months agoBSD-2-Clause
asynchronous-and-dynamic-task-managementasynchronous-searchautotuneautotuningbayesian-optimization+5 more

Cite this paper

@article{wu2023ytopt,
  title  = {ytopt: Autotuning Scientific Applications for Energy Efficiency at Large Scales},
  author = {Xingfu Wu and Prasanna Balaprakash and Michael Kruse and Jaehoon Koo and Brice Videau and Paul Hovland and Valerie Taylor and Brad Geltz and Siddhartha Jana and Mary Hall},
  year   = {2023},
  eprint = {2303.16245},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2303.16245v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2303.16245.svg)](https://codeofpaper.com/paper/2303.16245)

Discussion