Feature/add driver checkpoint - Githubissues

ai2cm / pace

Implementation of the FV3GFS / SHiELD atmospheric model in Python

https://ai2cm.github.io/pace/

Apache License 2.0

37 stars 12 forks source link

Feature/add driver checkpoint #391

Open elynnwu opened 1 year ago

elynnwu commented 1 year ago

Purpose

This PR adds driver-in/out checkpointer. Currently also supports non orchestrated backends and only tested using numpy.

Code changes:

checkpoint: added checkpoint selector to expose checkpoint type to driver
driver: added checkpointer_config to driver config, defaults to NullCheckpointer
grid and initialization: for serialbox type, add option to read the netcdf file
fv_dynamics: add checkpoint driver-in
update_atmos_state: add checkpoint driver-out
test_checkpoints: add test_driver driver-in/out checkpoint

Infrastructure changes:

driver checkpoint test added
added all folder for c12_6ranks_baroclinic_dycore_microphysics which combines all savepoint netcdf to one folder

Checklist

Before submitting this PR, please make sure:

[ ] You have followed the coding standards guidelines established at Code Review Checklist.
[ ] Docstrings and type hints are added to new and updated routines, as appropriate
[ ] All relevant documentation has been updated or added (e.g. README, CONTRIBUTING docs)
[ ] For each public change and fix in pace-util, HISTORY has been updated
[ ] Unit tests are added or updated for non-stencil code changes