geos_wrapper

GeosDycoreWrapper

Provides an interface for the Geos model to access the Pace dycore. Takes numpy arrays as inputs, returns a dictionary of numpy arrays as outputs

StencilBackendCompilerOverride

Override the Pace global stencil JIT to allow for 9-rank build on any setup.

This is a workaround that requires to know exactly when build is happening. Using this as a context manager, we leverage the DaCe build system to override the name and build the 9 codepaths required- while every other rank wait.

This should be removed when we refactor the GT JIT to distribute building much more efficiently