-
Notifications
You must be signed in to change notification settings - Fork 7
Muphys bug fix: drivers run_graupel_only and run_full_muphys #877
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Conversation
…merge' into muphys_bug_fix
35d058f to
3752592
Compare
...s/muphys/src/icon4py/model/atmosphere/subgrid_scale_physics/muphys/driver/run_full_muphys.py
Outdated
Show resolved
Hide resolved
Clean up the graupel_only driver, and create an integration test to run through pytest. Co-authored-by: Will Sawyer <wsawyer@cscs.ch>
Co-authored-by: Will Sawyer <vectorflux@gmail.com> Co-authored-by: Will Sawyer <wsawyer@cscs.ch>
| return testing_defs.get_test_data_root_path() / "full_muphys_data" / experiment.name | ||
|
|
||
|
|
||
| @dataclasses.dataclass(frozen=True) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
refactor utility functions to be shared between full_muphys and graupel_only tests
| @classmethod | ||
| def allocate(cls, allocator: gtx_typing.FieldBufferAllocationUtil, domain: gtx.Domain): | ||
| zeros = functools.partial(gtx.zeros, domain=domain, allocator=allocator) | ||
| # TODO +1 size fields? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
make these fields horizontal only
|
Mandatory Tests Please make sure you run these tests via comment before you merge!
Optional Tests To run benchmarks you can use:
To run tests and benchmarks with the DaCe backend you can use:
To run test levels ignored by the default test suite (mostly simple datatest for static fields computations) you can use:
For more detailed information please look at CI in the EXCLAIM universe. |
This PR addresses minor deficiencies in the graupel and saturation_adjustment GT4Py programs, which might influence performance, but more importantly in the run_graupel_only.py and run_full_muphys.py drivers which call them. These were restructured to improve performance, in particular by performing a set-up of both programs in the initialization step.