Release: NDSL 2026.01.00

.github/workflows/create-cache.yaml

@@ -23,14 +23,14 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v5
+        uses: actions/checkout@v6
 
       - name: Setup Python 3.11
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: '3.11'
 
-      - uses: actions/cache@v4
+      - uses: actions/cache@v5
         id: cache
         with:
           path: ~/.cache/pre-commit

.github/workflows/docs_build.yaml

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

.github/workflows/docs_deploy.yaml

@@ -16,15 +16,15 @@ jobs:
 
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v5
+        uses: actions/checkout@v6
 
       - name: Configure Git Credentials
         run: |
           git config user.name github-actions[bot]
           git config user.email github-actions[bot]@users.noreply.github.com
 
       - name: Setup python
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: 3.11
 

.github/workflows/lint.yaml

@@ -18,16 +18,16 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v5
+        uses: actions/checkout@v6
 
       - name: Setup Python 3.11
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: '3.11'
 
       # 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.
-      - uses: actions/cache/restore@v4
+      - uses: actions/cache/restore@v5
         with:
           path: ~/.cache/pre-commit
           key: pre-commit_${{ env.pythonLocation }}_${{ hashFiles('.pre-commit-config.yaml') }}

.github/workflows/unit_tests.yaml

@@ -19,12 +19,12 @@ jobs:
     steps:
 
       - name: Checkout repository
-        uses: actions/checkout@v5
+        uses: actions/checkout@v6
         with:
           submodules: 'recursive'
 
       - name: Setup Python 3.11
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: '3.11'
 

examples/NDSL/01_gt4py_basics.ipynb

@@ -66,14 +66,14 @@
     "shape = (nx, ny, nz)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "arr = np.indices(shape, dtype=float).sum(\n",
     "    axis=0\n",
     ")  # Value of each entry is sum of the I and J index at each point\n",
     "\n",
-    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend)"
+    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend)"
    ]
   },
   {
@@ -199,7 +199,7 @@
    "outputs": [],
    "source": [
     "qty_out = Quantity(\n",
-    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "print(\"Plotting values of qty_in at K = 0\")\n",
@@ -212,7 +212,7 @@
     "qty_out.plot_k_level(0)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "print(\"Resetting qty_out to zero...\")\n",
@@ -224,7 +224,7 @@
     "qty_out.plot_k_level(0)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "print(\"Resetting qty_out to zero...\")\n",
@@ -238,7 +238,7 @@
     "qty_out.plot_k_level(1)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "print(\"Resetting qty_out to zero...\")\n",
     "print(\"Plotting values of qty_in at K = 0\")\n",
@@ -251,7 +251,7 @@
     "qty_out.plot_k_level(0)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros([nx, ny, nz]), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "print(\"Resetting qty_out to zero...\")\n",
     "print(\"Plotting values of qty_out at K = 0\")\n",
@@ -294,13 +294,13 @@
     "shape = (nx + 2 * nhalo, ny + 2 * nhalo, nz)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "arr = np.indices(shape, dtype=float).sum(\n",
     "    axis=0\n",
     ")  # Value of each entry is sum of the I and J index at each point\n",
-    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend)\n",
+    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend)\n",
     "\n",
     "print(\"Plotting values of qty_in at K = 0\")\n",
     "qty_in.plot_k_level(0)\n",
@@ -344,7 +344,7 @@
     "\n",
     "print(\"Resetting qty_out to zeros\")\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "print(\"Executing 'copy_downward' with origin=(1, 1, 0), domain=(nx, ny, nz-1)\")\n",
@@ -401,13 +401,13 @@
     "shape = (nx + 2 * nhalo, ny + 2 * nhalo, nz)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "arr = np.indices(shape, dtype=float).sum(\n",
     "    axis=0\n",
     ")  # Value of each entry is sum of the I and J index at each point\n",
-    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend)\n",
+    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend)\n",
     "\n",
     "\n",
     "@stencil(backend=backend)\n",
@@ -444,13 +444,13 @@
    "outputs": [],
    "source": [
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "arr = np.indices(shape, dtype=float).sum(\n",
     "    axis=0\n",
     ")  # Value of each entry is sum of the I and J index at each point\n",
-    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend)\n",
+    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend)\n",
     "\n",
     "print(\"Plotting values of qty_in at K = 0\")\n",
     "qty_in.plot_k_level(0)\n",
@@ -525,13 +525,13 @@
     "shape = (nx + 2 * nhalo, ny + 2 * nhalo, nz)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "arr = np.indices(shape, dtype=float).sum(\n",
     "    axis=0\n",
     ")  # Value of each entry is sum of the I and J index at each point\n",
-    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend)\n",
+    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend)\n",
     "\n",
     "print(\"Plotting values of qty_in at K = 0\")\n",
     "qty_in.plot_k_level(0)\n",
@@ -546,7 +546,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },

examples/NDSL/02_NDSL_basics.ipynb

@@ -133,15 +133,15 @@
     "shape = (nx + 2 * nhalo, ny + 2 * nhalo, nz)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "\n",
     "arr = np.indices(shape, dtype=float).sum(\n",
     "    axis=0\n",
     ")  # Value of each entry is sum of the I and J index at each point\n",
     "\n",
-    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend)\n",
+    "qty_in = Quantity(data=arr, dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend)\n",
     "\n",
     "print(\"Plotting qty_in at K = 0\")\n",
     "qty_in.plot_k_level(0)\n",
@@ -224,7 +224,7 @@
     "copy_field_offset = CopyFieldOffset(stencil_factory)\n",
     "\n",
     "qty_out = Quantity(\n",
-    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", gt4py_backend=backend\n",
+    "    data=np.zeros(shape), dims=[\"I\", \"J\", \"K\"], units=\"m\", backend=backend\n",
     ")\n",
     "\n",
     "print(\"Initialize qty_out to zeros\")"

mkdocs.yml

@@ -46,7 +46,6 @@ nav:
           - "decomposition": docstrings/comm/decomposition.md
           - "local_comm": docstrings/comm/local_comm.md
           - "mpi": docstrings/comm/mpi.md
-          - "null_comm": docstrings/comm/null_comm.md
           - "partitioner": docstrings/comm/partitioner.md
       - debug:
           - "config": docstrings/debug/config.md

ndsl/__init__.py

@@ -3,7 +3,6 @@
 from .comm.communicator import CubedSphereCommunicator, TileCommunicator
 from .comm.local_comm import LocalComm
 from .comm.mpi import MPIComm
-from .comm.null_comm import NullComm
 from .comm.partitioner import CubedSpherePartitioner, TilePartitioner
 from .constants import ConstantVersions
 from .dsl.caches.codepath import FV3CodePath
@@ -26,7 +25,7 @@
 from .performance.collector import NullPerformanceCollector, PerformanceCollector
 from .performance.profiler import NullProfiler, Profiler
 from .performance.report import Experiment, Report, TimeReport
-from .quantity import Local, Quantity, State
+from .quantity import Local, LocalState, Quantity, State
 from .quantity.field_bundle import FieldBundle, FieldBundleType  # Break circular import
 from .types import Allocator
 from .utils import MetaEnumStr
@@ -38,7 +37,6 @@
     "TileCommunicator",
     "LocalComm",
     "MPIComm",
-    "NullComm",
     "CubedSpherePartitioner",
     "TilePartitioner",
     "ConstantVersions",
@@ -81,6 +79,7 @@
     "Allocator",
     "MetaEnumStr",
     "State",
+    "LocalState",
     "NDSLRuntime",
     "Local",
 ]

ndsl/boilerplate.py

@@ -1,3 +1,5 @@
+import warnings
+
 from ndsl import (
     CompilationConfig,
     DaceConfig,
@@ -56,9 +58,11 @@ def _get_factories(
     if topology == "tile":
         mpi_comm = MPIComm()
         if mpi_comm.Get_size() != 1:
-            raise ValueError(
+            warnings.warn(
                 "Single tile topology requested with an MPI communicator of size "
-                f"{mpi_comm.Get_size()} > 1. Re-configure MPI to run on only one rank."
+                f"{mpi_comm.Get_size()} > 1. Halo exchange will _not_ be done on the cube-sphere.",
+                category=UserWarning,
+                stacklevel=2,
             )
 
         partitioner = TilePartitioner((1, 1))
@@ -69,6 +73,7 @@ def _get_factories(
             n_halo=nhalo,
             layout=partitioner.layout,
             tile_partitioner=partitioner,
+            backend=backend,
         )
         comm = TileCommunicator(comm=mpi_comm, partitioner=partitioner)
     else:
@@ -82,7 +87,13 @@ def _get_factories(
 
 
 def get_factories_single_tile_orchestrated(
-    nx: int, ny: int, nz: int, nhalo: int, backend: str = "dace:cpu"
+    nx: int,
+    ny: int,
+    nz: int,
+    nhalo: int,
+    backend: str = "dace:cpu",
+    *,
+    orchestration_mode: DaCeOrchestration | None = None,
 ) -> tuple[StencilFactory, QuantityFactory]:
     """Build the pair of (StencilFactory, QuantityFactory) for orchestrated code on a single tile topology."""
 
@@ -95,7 +106,7 @@ def get_factories_single_tile_orchestrated(
         nz=nz,
         nhalo=nhalo,
         backend=backend,
-        orchestration=DaCeOrchestration.BuildAndRun,
+        orchestration=orchestration_mode or DaCeOrchestration.BuildAndRun,
         topology="tile",
     )