Integrating ytopt and libEnsemble to Autotune OpenMC
CoRR(2024)
摘要
ytopt is a Python machine-learning-based autotuning software package
developed within the ECP PROTEAS-TUNE project. The ytopt software adopts an
asynchronous search framework that consists of sampling a small number of input
parameter configurations and progressively fitting a surrogate model over the
input-output space until exhausting the user-defined maximum number of
evaluations or the wall-clock time. libEnsemble is a Python toolkit for
coordinating workflows of asynchronous and dynamic ensembles of calculations
across massively parallel resources developed within the ECP PETSc/TAO project.
libEnsemble helps users take advantage of massively parallel resources to solve
design, decision, and inference problems and expands the class of problems that
can benefit from increased parallelism. In this paper we present our
methodology and framework to integrate ytopt and libEnsemble to take advantage
of massively parallel resources to accelerate the autotuning process.
Specifically, we focus on using the proposed framework to autotune the ECP
ExaSMR application OpenMC, an open source Monte Carlo particle transport code.
OpenMC has seven tunable parameters some of which have large ranges such as the
number of particles in-flight, which is in the range of 100,000 to 8 million,
with its default setting of 1 million. Setting the proper combination of these
parameter values to achieve the best performance is extremely time-consuming.
Therefore, we apply the proposed framework to autotune the MPI/OpenMP offload
version of OpenMC based on a user-defined metric such as the figure of merit
(FoM) (particles/s) or energy efficiency energy-delay product (EDF) on the OLCF
Frontier TDS system Crusher. The experimental results show that we achieve
improvement up to 29.49
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要