Skip to content

Conversation

@vectorflux
Copy link
Contributor

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.

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)
Copy link
Contributor

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?
Copy link
Contributor

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

@github-actions
Copy link

github-actions bot commented Dec 9, 2025

Mandatory Tests

Please make sure you run these tests via comment before you merge!

  • cscs-ci run default

Optional Tests

To run benchmarks you can use:

  • cscs-ci run benchmark-bencher

To run tests and benchmarks with the DaCe backend you can use:

  • cscs-ci run dace

To run test levels ignored by the default test suite (mostly simple datatest for static fields computations) you can use:

  • cscs-ci run extra

For more detailed information please look at CI in the EXCLAIM universe.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants