Articles | Volume 16, issue 9
https://doi.org/10.5194/gmd-16-2719-2023
https://doi.org/10.5194/gmd-16-2719-2023
Development and technical paper
 | Highlight paper
 | 
17 May 2023
Development and technical paper | Highlight paper |  | 17 May 2023

Pace v0.2: a Python-based performance-portable atmospheric model

Johann Dahm, Eddie Davis, Florian Deconinck, Oliver Elbert, Rhea George, Jeremy McGibbon, Tobias Wicky, Elynn Wu, Christopher Kung, Tal Ben-Nun, Lucas Harris, Linus Groner, and Oliver Fuhrer

Related authors

Earth Virtualization Engines (EVE)
Bjorn Stevens, Stefan Adami, Tariq Ali, Hartwig Anzt, Zafer Aslan, Sabine Attinger, Jaana Bäck, Johanna Baehr, Peter Bauer, Natacha Bernier, Bob Bishop, Hendryk Bockelmann, Sandrine Bony, Guy Brasseur, David N. Bresch, Sean Breyer, Gilbert Brunet, Pier Luigi Buttigieg, Junji Cao, Christelle Castet, Yafang Cheng, Ayantika Dey Choudhury, Deborah Coen, Susanne Crewell, Atish Dabholkar, Qing Dai, Francisco Doblas-Reyes, Dale Durran, Ayoub El Gaidi, Charlie Ewen, Eleftheria Exarchou, Veronika Eyring, Florencia Falkinhoff, David Farrell, Piers M. Forster, Ariane Frassoni, Claudia Frauen, Oliver Fuhrer, Shahzad Gani, Edwin Gerber, Debra Goldfarb, Jens Grieger, Nicolas Gruber, Wilco Hazeleger, Rolf Herken, Chris Hewitt, Torsten Hoefler, Huang-Hsiung Hsu, Daniela Jacob, Alexandra Jahn, Christian Jakob, Thomas Jung, Christopher Kadow, In-Sik Kang, Sarah Kang, Karthik Kashinath, Katharina Kleinen-von Königslöw, Daniel Klocke, Uta Kloenne, Milan Klöwer, Chihiro Kodama, Stefan Kollet, Tobias Kölling, Jenni Kontkanen, Steve Kopp, Michal Koran, Markku Kulmala, Hanna Lappalainen, Fakhria Latifi, Bryan Lawrence, June Yi Lee, Quentin Lejeun, Christian Lessig, Chao Li, Thomas Lippert, Jürg Luterbacher, Pekka Manninen, Jochem Marotzke, Satoshi Matsouoka, Charlotte Merchant, Peter Messmer, Gero Michel, Kristel Michielsen, Tomoki Miyakawa, Jens Müller, Ramsha Munir, Sandeep Narayanasetti, Ousmane Ndiaye, Carlos Nobre, Achim Oberg, Riko Oki, Tuba Özkan-Haller, Tim Palmer, Stan Posey, Andreas Prein, Odessa Primus, Mike Pritchard, Julie Pullen, Dian Putrasahan, Johannes Quaas, Krishnan Raghavan, Venkatachalam Ramaswamy, Markus Rapp, Florian Rauser, Markus Reichstein, Aromar Revi, Sonakshi Saluja, Masaki Satoh, Vera Schemann, Sebastian Schemm, Christina Schnadt Poberaj, Thomas Schulthess, Cath Senior, Jagadish Shukla, Manmeet Singh, Julia Slingo, Adam Sobel, Silvina Solman, Jenna Spitzer, Philip Stier, Thomas Stocker, Sarah Strock, Hang Su, Petteri Taalas, John Taylor, Susann Tegtmeier, Georg Teutsch, Adrian Tompkins, Uwe Ulbrich, Pier-Luigi Vidale, Chien-Ming Wu, Hao Xu, Najibullah Zaki, Laure Zanna, Tianjun Zhou, and Florian Ziemen
Earth Syst. Sci. Data, 16, 2113–2122, https://doi.org/10.5194/essd-16-2113-2024,https://doi.org/10.5194/essd-16-2113-2024, 2024
Short summary
DCMIP2016: the tropical cyclone test case
Justin L. Willson, Kevin A. Reed, Christiane Jablonowski, James Kent, Peter H. Lauritzen, Ramachandran Nair, Mark A. Taylor, Paul A. Ullrich, Colin M. Zarzycki, David M. Hall, Don Dazlich, Ross Heikes, Celal Konor, David Randall, Thomas Dubos, Yann Meurdesoif, Xi Chen, Lucas Harris, Christian Kühnlein, Vivian Lee, Abdessamad Qaddouri, Claude Girard, Marco Giorgetta, Daniel Reinert, Hiroaki Miura, Tomoki Ohno, and Ryuji Yoshida
Geosci. Model Dev., 17, 2493–2507, https://doi.org/10.5194/gmd-17-2493-2024,https://doi.org/10.5194/gmd-17-2493-2024, 2024
Short summary
Exploring hail and lightning diagnostics over the Alpine-Adriatic region in a km-scale climate model
Ruoyi Cui, Nikolina Ban, Marie-Estelle Demory, Raffael Aellig, Oliver Fuhrer, Jonas Jucker, Xavier Lapillonne, and Christoph Schär
Weather Clim. Dynam., 4, 905–926, https://doi.org/10.5194/wcd-4-905-2023,https://doi.org/10.5194/wcd-4-905-2023, 2023
Short summary
Multiple same-level and telescoping nesting in GFDL's dynamical core
Joseph Mouallem, Lucas Harris, and Rusty Benson
Geosci. Model Dev., 15, 4355–4371, https://doi.org/10.5194/gmd-15-4355-2022,https://doi.org/10.5194/gmd-15-4355-2022, 2022
Short summary
Enhancing the accessibility of unified modeling systems: GFDL System for High-resolution prediction on Earth-to-Local Domains (SHiELD) v2021b in a container
Kai-Yuan Cheng, Lucas M. Harris, and Yong Qiang Sun
Geosci. Model Dev., 15, 1097–1105, https://doi.org/10.5194/gmd-15-1097-2022,https://doi.org/10.5194/gmd-15-1097-2022, 2022
Short summary

Related subject area

Climate and Earth system modeling
Impact of ocean vertical-mixing parameterization on Arctic sea ice and upper-ocean properties using the NEMO-SI3 model
Sofia Allende, Anne Marie Treguier, Camille Lique, Clément de Boyer Montégut, François Massonnet, Thierry Fichefet, and Antoine Barthélemy
Geosci. Model Dev., 17, 7445–7466, https://doi.org/10.5194/gmd-17-7445-2024,https://doi.org/10.5194/gmd-17-7445-2024, 2024
Short summary
Bridging the gap: a new module for human water use in the Community Earth System Model version 2.2.1
Sabin I. Taranu, David M. Lawrence, Yoshihide Wada, Ting Tang, Erik Kluzek, Sam Rabin, Yi Yao, Steven J. De Hertog, Inne Vanderkelen, and Wim Thiery
Geosci. Model Dev., 17, 7365–7399, https://doi.org/10.5194/gmd-17-7365-2024,https://doi.org/10.5194/gmd-17-7365-2024, 2024
Short summary
A new lightning scheme in the Canadian Atmospheric Model (CanAM5.1): implementation, evaluation, and projections of lightning and fire in future climates
Cynthia Whaley, Montana Etten-Bohm, Courtney Schumacher, Ayodeji Akingunola, Vivek Arora, Jason Cole, Michael Lazare, David Plummer, Knut von Salzen, and Barbara Winter
Geosci. Model Dev., 17, 7141–7155, https://doi.org/10.5194/gmd-17-7141-2024,https://doi.org/10.5194/gmd-17-7141-2024, 2024
Short summary
Methane dynamics in the Baltic Sea: investigating concentration, flux, and isotopic composition patterns using the coupled physical–biogeochemical model BALTSEM-CH4 v1.0
Erik Gustafsson, Bo G. Gustafsson, Martijn Hermans, Christoph Humborg, and Christian Stranne
Geosci. Model Dev., 17, 7157–7179, https://doi.org/10.5194/gmd-17-7157-2024,https://doi.org/10.5194/gmd-17-7157-2024, 2024
Short summary
Split-explicit external mode solver in the finite volume sea ice–ocean model FESOM2
Tridib Banerjee, Patrick Scholz, Sergey Danilov, Knut Klingbeil, and Dmitry Sidorenko
Geosci. Model Dev., 17, 7051–7065, https://doi.org/10.5194/gmd-17-7051-2024,https://doi.org/10.5194/gmd-17-7051-2024, 2024
Short summary

Cited articles

Abadi, M., Agarwal, A., Barham, P., Brevdo, E., Chen, Z., Citro, C., Corrado, G. S., Davis, A., Dean, J., Devin, M., Ghemawat, S., Goodfellow, I., Harp, A., Irving, G., Isard, M., Jia, Y., Jozefowicz, R., Kaiser, L., Kudlur, M., Levenberg, J., Mané, D., Monga, R., Moore, S., Murray, D., Olah, C., Schuster, M., Shlens, J., Steiner, B., Sutskever, I., Talwar, K., Tucker, P., Vanhoucke, V., Vasudevan, V., Viégas, F., Vinyals, O., Warden, P., Wattenberg, M., Wicke, M., Yu, Y., and Zheng, X.: TensorFlow: Large-Scale Machine Learning on Heterogeneous Systems, Zenodo [software], https://doi.org/10.5281/zenodo.4724125, 2015. a
Adams, S., Ford, R., Hambley, M., Hobson, J., Kavčič, I., Maynard, C., Melvin, T., Müller, E., Mullerworth, S., Porter, A., Rezny, M., Shipway, B., and Wong, R.: LFRic: Meeting the challenges of scalability and performance portability in Weather and Climate models, J. Parallel Distr. Com., 132, 383–396, https://doi.org/10.1016/j.jpdc.2019.02.007, 2019. a
Alnaes, M. S., Blechta, J., Hake, J., Johansson, A., Kehlet, B., Logg, A., Richardson, C., Ring, J., and Rognes, M. E., and Wells, G. N.: The FEniCS Project Version 1.5, Archive of Numerical Software [software], https://doi.org/10.11588/ans.2015.100.20553, 2015. a
Augier, P., Bolz-Tereick, C. F., Guelton, S., and Mohanan, A. V.: Reducing the ecological impact of computing through education and Python compilers, Nature Astronomy, 5, 334–335, https://doi.org/10.1038/s41550-021-01342-y, 2021. a
Behnel, S., Bradshaw, R., Citro, C., Dalcin, L., Seljebotn, D. S., and Smith, K.: Cython: The Best of Both Worlds, Comput. Sci. Eng., 13, 31–39, https://doi.org/10.1109/MCSE.2010.118, 2011. a, b
Download
Executive editor
Achieving both performance and portability in a whole dynamical core implemented in a high-productivity language such as Python is an eye-opening result which rebuts some widely held assumptions in the geoscientific modelling community. This is a paper which everyone who writes geoscientific models should read.
Short summary
It is hard for scientists to write code which is efficient on different kinds of supercomputers. Python is popular for its user-friendliness. We converted a Fortran code, simulating Earth's atmosphere, into Python. This new code auto-converts to a faster language for processors or graphic cards. Our code runs 3.5–4 times faster on graphic cards than the original on processors in a specific supercomputer system.