Abstract:Earth System Models (ESMs) are intricate models employed for simulating the Earth's climate, typically constructed from distinct independent components dedicated to simulate specific natural phenomena (such as atmosphere and ocean dynamics, atmospheric chemistry, land and ocean biosphere, etc.). In order to capture the interactions between these processes, ESMs utilize coupling libraries, which oversee the synchronization and field exchanges among independent developed codes typically operating in parallel as a Multi-Program, Multi-Data (MPMD) application. The performance achieved depends on the coupling approach, as well as on the number of parallel resources and scalability properties of each component. Determining the appropriate number of resources to use for each component in coupled ESMs is crucial for efficient utilization of the High Performance Computing (HPC) infrastructures used in climate modelling. However, this task traditionally involves manual testing of multiple process allocations by trial and error, requiring significant time investment from researchers. Thus, making the process more error-prone, and often resulting in a loss in application performance due to the complexity of the task. This paper introduces the automatic load-balance tool (auto-lb), a methodology and tool for determining the resource allocation to each component within coupled ESMs, aimed at improving the application's performance. Notably, this methodology is automatic and does not require expertise in HPC to improve the performance achieved by coupled ESMs. This is accomplished by minimizing the load-imbalance: reducing each constituent's execution cost (core-hours), as well as minimizing the core-hours wasted resulting from the synchronizations between them, without penalizing the execution speed of the entire model. This optimization is achieved regardless of the scalability properties of each constituent and the complexity of their dependencies during the coupling. To achieve this, we designed a new performance metric called "Fittingness" to assess the performance of coupled execution evaluating the trade-off between the parallel efficiency and application throughput. This metric is intended for scenarios where optimality can depend on various criteria and constraints. Aiming for maximum speed might not be desirable if it leads to a decrease in parallel efficiency and, therefore, increasing the computational costs of simulation. The methodology was tested across multiple experiments using the widely recognized European ESM, EC-Earth3. The results were compared with real operational configurations, such as those used for the Coupled Model Intercomparison Project Phase 6 (CMIP6) and for the European Climate Prediction Project (EUCP), and validated on different HPC platforms. All of them suggest that the current approaches lead to performance loss, and that auto-lb can achieve better results in both, execution speed and reduction of the core-hours needed. When comparing to the EC-Earth standard-resolution CPMIP6 runs, we achieved a configuration 4.7 % faster while also reducing the core-hours required by 1.3 %. Likewise, when compared to the EC-Earth high-resolution EUCP runs, the method presented showed an improvement of 34 % in the speed, with a 6.7 % reduction in the core-hours consumed.

A Deep Learning-Based Consistency Test Approach for Earth System Models on Heterogeneous Many-Core Systems

Characterizing uncertainties of Earth system modeling with heterogeneous many-core architecture computing

Optimizing high-resolution Community Earth System Model on a heterogeneous many-core supercomputing platform

Toward Earth system modeling with resolved clouds and ocean submesoscales on heterogeneous many-core HPCs

A Deep Learning Earth System Model for Stable and Efficient Simulation of the Current Climate

HiRM: Hierarchical Resource Management for Earth System Models on Many-Core Clusters

Hirm: Hierarchical Resource Management for Coupled Climate Models on Many-Core Clusters

Kilometer-Level Coupled Modeling Using 40 Million Cores: An Eight-Year Journey of Model Development

Load-Adaptive Continuous Coupled-Simulation Ensembles with DataStorm and Chameleon

Model calibration using ESEm v1.0.0 -- an open, scalable Earth System Emulator

Evaluating Statistical Consistency in the Ocean Model Component of the Community Earth System Model (pycect V2.0)

Reducing Time and Computing Costs in EC-Earth: An Automatic Load-Balancing Approach for Coupled ESMs

An online ensemble coupled data assimilation capability for the Community Earth System Model: system design and evaluation

Iterative Integration of Deep Learning in Hybrid Earth Surface System Modelling

CESMTuner: An Auto-tuning Framework for the Community Earth System Model

DiffESM: Conditional Emulation of Temperature and Precipitation in Earth System Models with 3D Diffusion Models

An Automatic Performance Model-Based Scheduling Tool for Coupled Climate System Models

The Ensemble Consistency Test: From CESM to MPAS and Beyond

Developing Intelligent Earth System Models: an AI Framework for Replacing Sub-Modules Based on Incremental Learning and Its Application

Improving the Parallelism of CESM on GPU