diff --git a/ci/cscs-ci-dace-determinism.yml b/ci/cscs-ci-dace-determinism.yml index f9764e2061..6bead3c2e5 100644 --- a/ci/cscs-ci-dace-determinism.yml +++ b/ci/cscs-ci-dace-determinism.yml @@ -64,7 +64,6 @@ build_cscs_amd_rocm: USE_MPI: 0 # TODO(havogt): to workaround the libfabric hook injecting incompatible libraries SLURM_JOB_NUM_NODES: 1 SLURM_TIMELIMIT: 60 - allow_failure: true artifacts: when: always paths: diff --git a/pyproject.toml b/pyproject.toml index f128e45fd0..571b0ecdbc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -98,7 +98,7 @@ dependencies = [ 'click>=8.0.0', 'cmake>=3.22', 'cytoolz>=1.0.1', - 'dace>=2.0.0a5', + 'dace>=2.0.0a6', 'deepdiff>=8.1.0', 'devtools>=0.6', 'factory-boy>=3.3.3', diff --git a/src/gt4py/cartesian/backend/dace_backend.py b/src/gt4py/cartesian/backend/dace_backend.py index 6dacdd08e1..031c8403b5 100644 --- a/src/gt4py/cartesian/backend/dace_backend.py +++ b/src/gt4py/cartesian/backend/dace_backend.py @@ -596,8 +596,6 @@ def keep_line(line: str) -> bool: line = line.strip() if line == '#include "../../include/hash.h"': return False - if line.startswith("DACE_EXPORTED") and line.endswith(");"): - return False if line == "#include ": return False return True diff --git a/src/gt4py/next/program_processors/runners/dace/workflow/common.py b/src/gt4py/next/program_processors/runners/dace/workflow/common.py index 883489e90e..0c108313ce 100644 --- a/src/gt4py/next/program_processors/runners/dace/workflow/common.py +++ b/src/gt4py/next/program_processors/runners/dace/workflow/common.py @@ -83,14 +83,10 @@ def set_dace_config( # creating any further sub-folder to compile the SDFG. dace.Config.set("cache", value="single") - # Workaround to disable detection of the CUDA architecture in DaCe, and instead use the one provided by GT4Py. - # TODO(edopao): revisit this workaround once it is possible to disable GPU detection in DaCe. - # (see https://github.com/spcl/dace/pull/2424) + # Disable detection of the CUDA architecture in DaCe, and instead use the one provided by GT4Py. if device_arch := gtx_compilation_common.get_device_arch(): - dace.Config.set( - "compiler.extra_cmake_args", - value=f"-DLOCAL_CUDA_ARCHITECTURES={device_arch}", - ) + dace.Config.set("compiler.cuda.cuda_arch", value=device_arch) + dace.Config.set("compiler.cuda.hip_arch", value=device_arch) # Prevents the implicit change of Memlets to Maps. Instead they should be handled by # `gt4py.next.program_processors.runners.dace.transfromations.gpu_utils.gt_gpu_transform_non_standard_memlet()`. diff --git a/uv.lock b/uv.lock index 130dde96da..f1e2b6bc26 100644 --- a/uv.lock +++ b/uv.lock @@ -991,14 +991,16 @@ wheels = [ [[package]] name = "dace" -version = "2.0.0a5" +version = "2.0.0a6" source = { registry = "https://pypi.org/simple" } dependencies = [ { name = "astunparse" }, { name = "dill" }, { name = "fparser" }, + { name = "ml-dtypes" }, { name = "networkx" }, { name = "numpy" }, + { name = "ordered-set" }, { name = "packaging" }, { name = "ply" }, { name = "pyreadline", marker = "sys_platform == 'win32' or (extra == 'extra-5-gt4py-cuda12' and extra == 'extra-5-gt4py-jax-cuda13') or (extra == 'extra-5-gt4py-cuda12' and extra == 'extra-5-gt4py-rocm6') or (extra == 'extra-5-gt4py-cuda12' and extra == 'extra-5-gt4py-rocm7') or (extra == 'extra-5-gt4py-jax-cuda13' and extra == 'extra-5-gt4py-rocm6') or (extra == 'extra-5-gt4py-jax-cuda13' and extra == 'extra-5-gt4py-rocm7') or (extra == 'extra-5-gt4py-rocm6' and extra == 'extra-5-gt4py-rocm7')" }, @@ -1006,7 +1008,7 @@ dependencies = [ { name = "sympy" }, { name = "typing-extensions" }, ] -sdist = { url = "https://files.pythonhosted.org/packages/55/b1/6ecb926fad1efcb99afc9f148493f4bd33a788c3c7079b90ad764e7f56f3/dace-2.0.0a5.tar.gz", hash = "sha256:09d9185c9d0dd663ce00c3cfd11d0c9048f80415db7080d09d109fce0e044a02", size = 6185419, upload-time = "2026-07-23T17:57:33.557Z" } +sdist = { url = "https://files.pythonhosted.org/packages/ea/f2/31941fed070b22a94a265e612fb9615080a3d5f788273e37be6eadf57042/dace-2.0.0a6.tar.gz", hash = "sha256:d05884911a5f49bc89bf4b90b1aca2c1dc713541c5e0b5c9039cb7b39de24102", size = 6373222, upload-time = "2026-08-14T10:17:41.117Z" } [[package]] name = "debugpy" @@ -1524,7 +1526,7 @@ requires-dist = [ { name = "cupy-cuda13x", marker = "extra == 'cuda13'", specifier = ">=14.0" }, { name = "cupy-rocm-7-0", marker = "extra == 'rocm7'", specifier = ">=14.0" }, { name = "cytoolz", specifier = ">=1.0.1" }, - { name = "dace", specifier = ">=2.0.0a5" }, + { name = "dace", specifier = ">=2.0.0a6" }, { name = "deepdiff", specifier = ">=8.1.0" }, { name = "devtools", specifier = ">=0.6" }, { name = "factory-boy", specifier = ">=3.3.3" },