The documents distributed by this server have been provided by the contributing authors as a means to ensure timely dissemination of scholarly and technical work on a noncommercial basis. Copyright and all rights therein are maintained by the authors or by other copyright holders, notwithstanding that they have offered their works here electronically. It is understood that all persons copying this information will adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.
Publications of SPCL
|T. Hoefler, L. Groner, T. Ben-Nun, T. Wicky:|
|Weather and Climate Simulations in Python using GT4Py and DaCe|
(Presentation - presented in Virtual, Jan. 2022, )
AbstractWe will discuss how Python, a productive high-level language, can be used to develop and accelerate real weather models. To achieve that, we first describe the principles of data-centric Python – a subset of Python and NumPy that can be compiled and transformed via dataflow analysis. We then extend the model to weather and climate modeling with GT4Py: a high-productivity embedded DSL that allows meteorologists and practitioners to write stencils with ease. GT4Py optimizes each stencil to its full extent, and the Data-Centric Python framework (DaCe) can optimize full weather models with multiple GT4Py stencils and NumPy primitives in tandem. We demonstrate GT4Py and DaCe on the FV3 dynamical core, showing that it increases productivity in authoring climate models, as well as portability and performance, running on CPUs and GPUs up to tenfold faster than the optimized FORTRAN implementation.
Recorded talk (best effort)