A023-05
Challenges of high-resolution large-ensemble data assimilation on supercomputer Fugaku
Abstract:
As the successor of the K computer, which enabled the first global 870m-mesh atmospheric simulation, the development of Fugaku started in 2014. We have committed to systems-applications co-design to achieve 100 times higher performance of NICAM-LETKF than the K computer. Fugaku has a many-core general-purpose CPU (A64FX) based on the Armv8 instruction set and has a high memory transfer performance comparable to that of GPUs. The system also has a high bandwidth of the data transfer for inter-node communication and file I/O with high resiliency. These features are advantageous for weather/climate models, which has complex, huge codebase and data-intensive computational features. We have optimized the source code using a refactoring method, which we call the "household account" method, and also helped the compiler development about a function of partitioning loops with huge bodies. We enhanced the active usage of the single-precision floating-point calculation not only in the dynamical core and major physical processes in the simulation model but also in the eigenvalue decomposition used in the DA.
We performed a global 1024-member ensemble atmospheric simulation and data assimilation with horizontal resolutions of 14km and 3.5km. For the 3.5km-mesh case, we used the 131,705 nodes of Fugaku and the amount of data passed from the model simulation to the DA system was 1.3 PiB. In this study, we show that data volume and transfer performance are clear bottlenecks in near-future weather/climate simulations. It is more important to promote the "data-centric" design in future modeling studies.