Skip to content

Bump GT4Py version (includes dace fix) #785

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

Open
wants to merge 20 commits into
base: main
Choose a base branch
from
Open

Conversation

edopao
Copy link
Contributor

@edopao edopao commented Jun 20, 2025

The new gt4py version icon4py_staging_20250627 contains fixes for the dace backend. However, the dace backend is not yet enabled in CI. The main reasons are the following:

  • The tracer-advection datatest on GPU times out (obs. not used in solve_nonhydro). This is could be some issue in the gpu-specific SDFG transformations, to be investigated.
  • The hash of the translation stage is not stable (same for dace and gtfn), which results in multiple lowering in datatest of the same stencil, although not needed, and long CI jobs. This happens for compute_perturbed_quantities_and_interpolation and interpolate_rho_theta_v_to_half_levels_and_compute_pressure_buoyancy_acceleration.
  • The CI config files need to be updated, in a separate PR, to be consistent with other CI jobs.

Additionally, this PR disables the tests on orchestrated diffusion granule because dace orchestration is currently not compatible with precompiled programs.

@edopao edopao force-pushed the update_gt4py_version branch 2 times, most recently from 591121a to cd6e3f1 Compare June 23, 2025 04:24
@edopao edopao force-pushed the update_gt4py_version branch from cd6e3f1 to 371163b Compare June 23, 2025 07:04
@edopao edopao force-pushed the update_gt4py_version branch from c6e4930 to bd2ab18 Compare June 23, 2025 08:13
@edopao edopao force-pushed the update_gt4py_version branch from c8f1123 to b61933c Compare June 24, 2025 11:46
@edopao
Copy link
Contributor Author

edopao commented Jun 25, 2025

cscs-ci run dace

@edopao
Copy link
Contributor Author

edopao commented Jun 27, 2025

cscs-ci run dace

Copy link

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.

@edopao
Copy link
Contributor Author

edopao commented Jun 27, 2025

cscs-ci run dace

@edopao
Copy link
Contributor Author

edopao commented Jun 27, 2025

cscs-ci run default

@edopao edopao marked this pull request as ready for review June 27, 2025 19:59
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.

1 participant