[Update] NDSL 2026.02.00 breaking change & update to DaCe V2

.github/workflows/create_cache.yml

@@ -25,10 +25,10 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v6
 
-      - name: Setup Python 3.11
+      - name: Setup Python 3.12
         uses: actions/setup-python@v6
         with:
-          python-version: '3.11'
+          python-version: '3.12'
 
       - uses: actions/cache@v5
         id: cache

.github/workflows/lint.yaml

@@ -17,10 +17,10 @@ jobs:
         with:
           submodules: 'recursive'
 
-      - name: Step Python 3.11
+      - name: Step Python 3.12
         uses: actions/setup-python@v6
         with:
-          python-version: '3.11'
+          python-version: '3.12'
 
       # Only restore (don't save) caches on PRs. New caches created from PRs won't be
       # accessible from other PRs, see workflows/create-cache.yaml.

.github/workflows/main_unit_tests.yaml

@@ -34,7 +34,7 @@ jobs:
       run:
         shell: bash -el {0}
     env:
-      python_default: '3.11'
+      python_default: '3.12'
 
     steps:
       - name: "External trigger: Checkout pace/develop"
@@ -44,7 +44,6 @@ jobs:
           submodules: 'recursive'
           repository: NOAA-GFDL/pace
           path: pace
-          ref: develop
 
       - name: 'Setup Python ${{ inputs.python_version && inputs.python_version || env.python_default }}'
         uses: actions/setup-python@v6

.jenkins/driver_configs/baroclinic_c192_54ranks.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: dace:gpu
+    backend: st:dace:gpu:KJI
     rebuild: false
     validate_args: true
     format_source: false

.jenkins/driver_configs/baroclinic_c192_6ranks.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: dace:gpu
+    backend: st:dace:gpu:KJI
     rebuild: false
     validate_args: true
     format_source: false

.jenkins/driver_configs/baroclinic_c48_6ranks_dycore_only.yaml

@@ -2,7 +2,7 @@ dycore_only: true
 disable_step_physics: true
 stencil_config:
   compilation_config:
-    backend: gt:gpu
+    backend: st:gt:gpu:KJI
     rebuild: false
     validate_args: true
     format_source: false

.jenkins/driver_configs/baroclinic_c48_6ranks_dycore_only_serialbox.yaml

@@ -2,7 +2,7 @@ dycore_only: true
 disable_step_physics: true
 stencil_config:
   compilation_config:
-    backend: gt:gpu
+    backend: st:gt:gpu:KJI
     rebuild: false
     validate_args: true
     format_source: false

docs/util/communication.rst

@@ -68,7 +68,7 @@ To see how the boundary and other objects operate, we will need some data to ope
     >>> import numpy as np
     >>> quantity = pace.util.Quantity(
     ...     data=np.zeros((6, 6)),
-    ...     dims=[pace.util.X_DIM, pace.util.Y_DIM],
+    ...     dims=[pace.util.I_DIM, pace.util.J_DIM],
     ...     units="m",
     ...     origin=(1, 1),
     ...     extent=(4, 4),

examples/configs/analytic_test.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_comm_read.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_comm_write.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_dp.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_explicit_physics.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_from_serialbox.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: gtc:numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_null_comm.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false
@@ -94,3 +94,8 @@ physics_config:
   hydrostatic: false
   nwat: 6
   do_qa: true
+
+grid_config:
+  type: generated
+  config:
+    eta_file: './eta79.nc'

examples/configs/baroclinic_c12_orch_cpu.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: dace:cpu
+    backend: st:dace:cpu:KIJ
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_read_restart_fortran.yml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c12_write_restart.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c48_6ranks_serialbox_test.yaml

@@ -2,7 +2,7 @@ dycore_only: true
 disable_step_physics: true
 stencil_config:
   compilation_config:
-    backend: gt:gpu
+    backend: st:gt:gpu:KJI
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/baroclinic_c48_no_out.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/test_external_C12_1x1.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/test_external_C12_2x2.yaml

@@ -1,6 +1,6 @@
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/tropical_read_restart_fortran.yml

@@ -2,7 +2,7 @@ dycore_only: true
 disable_step_physics: true
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/configs/tropicalcyclone_c128.yaml

@@ -2,7 +2,7 @@ dycore_only: true
 disable_step_physics: true
 stencil_config:
   compilation_config:
-    backend: numpy
+    backend: st:numpy:cpu:IJK
     rebuild: false
     validate_args: true
     format_source: false

examples/notebooks/functions.py

@@ -25,7 +25,8 @@
     SubtileGridSizer,
     TilePartitioner,
 )
-from ndsl.constants import RADIUS, X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM, Z_DIM
+from ndsl.config import Backend, backend_python
+from ndsl.constants import I_DIM, I_INTERFACE_DIM, J_DIM, J_INTERFACE_DIM, K_DIM, RADIUS
 from ndsl.grid import (
     AngleGridData,
     ContravariantGridData,
@@ -62,7 +63,7 @@ def init_quantity(
     grid: VariableGrid,
     dims: VariableDims = VariableDims.XYZ,
     units: str = "",
-    backend: str = "numpy",
+    backend: Backend = backend_python,
 ) -> Quantity:
     """
     Use: output = init_quantity(dimensions, grid, dims, units)
@@ -102,7 +103,7 @@ def init_quantity(
     if grid == VariableGrid.CellCenters:
         variable = Quantity(
             data=empty,
-            dims=(X_DIM, Y_DIM, Z_DIM)[:skip_z],
+            dims=(I_DIM, J_DIM, K_DIM)[:skip_z],
             units=units,
             origin=(nhalo, nhalo, 0)[:skip_z],
             extent=(nx, ny, nz)[:skip_z],
@@ -112,7 +113,7 @@ def init_quantity(
     if grid == VariableGrid.CellCorners:
         variable = Quantity(
             data=empty,
-            dims=(X_INTERFACE_DIM, Y_INTERFACE_DIM, Z_DIM)[:skip_z],
+            dims=(I_INTERFACE_DIM, J_INTERFACE_DIM, K_DIM)[:skip_z],
             units=units,
             origin=(nhalo, nhalo, 0)[:skip_z],
             extent=(nx + 1, ny + 1, nz)[:skip_z],
@@ -122,7 +123,7 @@ def init_quantity(
     elif grid == VariableGrid.StaggeredInX:
         variable = Quantity(
             data=empty,
-            dims=(X_INTERFACE_DIM, Y_DIM, Z_DIM)[:skip_z],
+            dims=(I_INTERFACE_DIM, J_DIM, K_DIM)[:skip_z],
             units=units,
             origin=(nhalo, nhalo, 0)[:skip_z],
             extent=(nx + 1, ny, nz)[:skip_z],
@@ -132,7 +133,7 @@ def init_quantity(
     elif grid == VariableGrid.StaggeredInY:
         variable = Quantity(
             data=empty,
-            dims=(X_DIM, Y_INTERFACE_DIM, Z_DIM)[:skip_z],
+            dims=(I_DIM, J_INTERFACE_DIM, K_DIM)[:skip_z],
             units=units,
             origin=(nhalo, nhalo, 0)[:skip_z],
             extent=(nx, ny + 1, nz)[:skip_z],
@@ -310,13 +311,13 @@ def configure_domain(
 
 def configure_stencil(
     domain_configuration: Dict[str, Any],
-    backend: str = "numpy",
+    backend: Backend = backend_python,
     single_layer: bool = True,
 ) -> Dict[str, Any]:
     """
     Use:
     stencil_configuration = configure_stencil(
-        domain_configuration, backend="numpy", single_layer=True)
+        domain_configuration, backend=Backend("st:numpy:cpu:IJK"), single_layer=True)
 
     Inputs:
     - domain configuration (Dict) from configure_domain()
@@ -646,20 +647,20 @@ def calculate_winds_from_streamfunction_grid(
     if isinstance(u_grid.data, np.ndarray) and isinstance(v_grid.data, np.ndarray):
         if grid == GridType.AGrid:
             if not (
-                u_grid.metadata.dims == (X_DIM, Y_DIM, Z_DIM)
-                and v_grid.metadata.dims == (X_DIM, Y_DIM, Z_DIM)
+                u_grid.metadata.dims == (I_DIM, J_DIM, K_DIM)
+                and v_grid.metadata.dims == (I_DIM, J_DIM, K_DIM)
             ):
                 print("Incorrect wind input dimensions for A-grid.")
         elif grid == GridType.CGrid:
             if not (
-                u_grid.metadata.dims == (X_DIM, Y_INTERFACE_DIM, Z_DIM)
-                and v_grid.metadata.dims == (X_INTERFACE_DIM, Y_DIM, Z_DIM)
+                u_grid.metadata.dims == (I_DIM, J_INTERFACE_DIM, K_DIM)
+                and v_grid.metadata.dims == (I_INTERFACE_DIM, J_DIM, K_DIM)
             ):
                 print("Incorrect wind input dimensions for C-grid.")
         elif grid == GridType.DGrid:
             if not (
-                u_grid.metadata.dims == (X_INTERFACE_DIM, Y_DIM, Z_DIM)
-                and v_grid.metadata.dims == (X_DIM, Y_INTERFACE_DIM, Z_DIM)
+                u_grid.metadata.dims == (I_INTERFACE_DIM, J_DIM, K_DIM)
+                and v_grid.metadata.dims == (I_DIM, J_INTERFACE_DIM, K_DIM)
             ):
                 print("Incorrect wind input dimensions for D-grid.")
     else: