diff --git a/.github/workflows/build-ucxx.yaml b/.github/workflows/build-ucxx.yaml index 6070740c..a69d5428 100644 --- a/.github/workflows/build-ucxx.yaml +++ b/.github/workflows/build-ucxx.yaml @@ -54,7 +54,7 @@ concurrency: jobs: conda-python-build: secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@cuda13.0 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index a82e2498..4b04bbb9 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -34,7 +34,7 @@ concurrency: jobs: conda-python-build: secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@cuda13.0 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -45,7 +45,7 @@ jobs: if: github.ref_type == 'branch' needs: [conda-python-build] secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@cuda13.0 with: arch: "amd64" branch: ${{ inputs.branch }} @@ -58,7 +58,7 @@ jobs: upload-conda: needs: [conda-python-build] secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-upload-packages.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-upload-packages.yaml@cuda13.0 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -66,7 +66,7 @@ jobs: sha: ${{ inputs.sha }} wheel-build: secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@cuda13.0 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} @@ -83,7 +83,7 @@ jobs: wheel-publish: needs: wheel-build secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@cuda13.0 with: build_type: ${{ inputs.build_type || 'branch' }} branch: ${{ inputs.branch }} diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index b3008cb9..730723df 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -17,7 +17,7 @@ jobs: - wheel-tests - telemetry-setup secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@cuda13.0 with: needs: ${{ toJSON(needs) }} telemetry-setup: @@ -34,27 +34,27 @@ jobs: checks: secrets: inherit needs: telemetry-setup - uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@cuda13.0 with: ignored_pr_jobs: telemetry-summarize conda-python-build: needs: checks secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@cuda13.0 with: build_type: pull-request script: ci/build_python.sh conda-python-tests: needs: conda-python-build secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@cuda13.0 with: build_type: pull-request script: ci/test_python.sh docs-build: needs: conda-python-build secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@cuda13.0 with: build_type: pull-request node_type: "gpu-l4-latest-1" @@ -64,7 +64,7 @@ jobs: wheel-build: needs: checks secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@cuda13.0 with: build_type: pull-request # Package is pure Python and only ever requires one build. @@ -78,7 +78,7 @@ jobs: wheel-tests: needs: wheel-build secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@cuda13.0 with: build_type: pull-request script: "ci/test_wheel.sh" diff --git a/.github/workflows/test-ucxx.yaml b/.github/workflows/test-ucxx.yaml index 7092afc9..fe566806 100644 --- a/.github/workflows/test-ucxx.yaml +++ b/.github/workflows/test-ucxx.yaml @@ -45,7 +45,7 @@ on: jobs: conda-python-ucxx-tests: secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@cuda13.0 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index da13bb9a..6dffc31e 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -25,7 +25,7 @@ on: jobs: conda-python-tests: secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@cuda13.0 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} @@ -34,7 +34,7 @@ jobs: sha: ${{ inputs.sha }} wheel-tests: secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@cuda13.0 with: build_type: ${{ inputs.build_type }} branch: ${{ inputs.branch }} diff --git a/.github/workflows/trigger-breaking-change-alert.yaml b/.github/workflows/trigger-breaking-change-alert.yaml index 48bf37af..72751d07 100644 --- a/.github/workflows/trigger-breaking-change-alert.yaml +++ b/.github/workflows/trigger-breaking-change-alert.yaml @@ -12,7 +12,7 @@ jobs: trigger-notifier: if: contains(github.event.pull_request.labels.*.name, 'breaking') secrets: inherit - uses: rapidsai/shared-workflows/.github/workflows/breaking-change-alert.yaml@branch-25.10 + uses: rapidsai/shared-workflows/.github/workflows/breaking-change-alert.yaml@cuda13.0 with: sender_login: ${{ github.event.sender.login }} sender_avatar: ${{ github.event.sender.avatar_url }} diff --git a/ci/run_benchmarks.sh b/ci/run_benchmarks.sh index b2c8dd36..37ac6bdf 100755 --- a/ci/run_benchmarks.sh +++ b/ci/run_benchmarks.sh @@ -6,6 +6,13 @@ set -euo pipefail # Support invoking run_benchmarks.sh outside the script directory cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../dask_cuda +# TODO: run cudf benchmarks unconditionally once there are CUDA 13 cudf packages +# ref: https://github.com/rapidsai/dask-cuda/pull/1536#issuecomment-321247489 +if [[ ! $(python -c "import cudf") ]]; then + echo "skipping benchmarks: cudf packages not available" + exit 0 +fi + python benchmarks/local_cudf_shuffle.py \ --partition-size="1 KiB" \ -d 0 \ diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh index d4bc26f7..e51691ce 100755 --- a/ci/test_wheel.sh +++ b/ci/test_wheel.sh @@ -5,6 +5,9 @@ set -eou pipefail source rapids-init-pip +RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}" +export RAPIDS_CUDA_MAJOR + DASK_CUDA_WHEELHOUSE=$(RAPIDS_PY_WHEEL_NAME="dask-cuda" RAPIDS_PY_WHEEL_PURE="1" rapids-download-wheels-from-github python) # Install cuda-suffixed dependencies b/c while `dask-cuda` has no cuda suffix, the test dependencies do @@ -16,7 +19,11 @@ rapids-dependency-file-generator \ rapids-logger "Installing test dependencies" # echo to expand wildcard -rapids-pip-retry install -v --prefer-binary -r /tmp/requirements-test.txt "$(echo "${DASK_CUDA_WHEELHOUSE}"/dask_cuda*.whl)" +rapids-pip-retry install \ + -v \ + --prefer-binary \ + -r /tmp/requirements-test.txt \ + "$(echo "${DASK_CUDA_WHEELHOUSE}"/dask_cuda*.whl)[cu${RAPIDS_CUDA_MAJOR}]" EXITCODE=0 # shellcheck disable=SC2317 diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index 5556d926..036544fa 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -14,8 +14,7 @@ dependencies: - dask-cudf==25.10.*,>=0.0.0a0 - distributed-ucxx==0.46.*,>=0.0.0a0 - kvikio==25.10.*,>=0.0.0a0 -- numactl-devel-cos7-aarch64 -- numba-cuda>=0.19.0,<0.20.0a0 +- numba-cuda>=0.19.1,<0.20.0a0 - numpy>=1.23,<3.0a0 - numpydoc>=1.1.0 - pandas>=1.3 @@ -32,6 +31,5 @@ dependencies: - sphinx-click>=2.7.1 - sphinx-rtd-theme>=0.5.1 - ucx-py==0.46.*,>=0.0.0a0 -- ucxx==0.46.*,>=0.0.0a0 - zict>=2.0.0 name: all_cuda-129_arch-aarch64 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index 2567ea4f..cbe4f090 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -14,8 +14,7 @@ dependencies: - dask-cudf==25.10.*,>=0.0.0a0 - distributed-ucxx==0.46.*,>=0.0.0a0 - kvikio==25.10.*,>=0.0.0a0 -- numactl-devel-cos7-x86_64 -- numba-cuda>=0.19.0,<0.20.0a0 +- numba-cuda>=0.19.1,<0.20.0a0 - numpy>=1.23,<3.0a0 - numpydoc>=1.1.0 - pandas>=1.3 @@ -32,6 +31,5 @@ dependencies: - sphinx-click>=2.7.1 - sphinx-rtd-theme>=0.5.1 - ucx-py==0.46.*,>=0.0.0a0 -- ucxx==0.46.*,>=0.0.0a0 - zict>=2.0.0 name: all_cuda-129_arch-x86_64 diff --git a/conda/environments/all_cuda-130_arch-aarch64.yaml b/conda/environments/all_cuda-130_arch-aarch64.yaml new file mode 100644 index 00000000..e6b70cff --- /dev/null +++ b/conda/environments/all_cuda-130_arch-aarch64.yaml @@ -0,0 +1,33 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai +- rapidsai-nightly +- conda-forge +dependencies: +- click >=8.1 +- cuda-core==0.3.* +- cuda-nvcc-impl +- cuda-nvrtc +- cuda-version=13.0 +- distributed-ucxx==0.46.*,>=0.0.0a0 +- kvikio==25.10.*,>=0.0.0a0 +- numba-cuda>=0.19.1,<0.20.0a0 +- numpy>=1.23,<3.0a0 +- numpydoc>=1.1.0 +- pandas>=1.3 +- pre-commit +- pynvml>=12.0.0,<13.0.0a0 +- pytest +- pytest-cov +- pytest-timeout +- python>=3.10,<3.14 +- rapids-build-backend>=0.4.0,<0.5.0dev0 +- rapids-dask-dependency==25.10.*,>=0.0.0a0 +- setuptools>=64.0.0 +- sphinx +- sphinx-click>=2.7.1 +- sphinx-rtd-theme>=0.5.1 +- ucx-py==0.46.*,>=0.0.0a0 +- zict>=2.0.0 +name: all_cuda-130_arch-aarch64 diff --git a/conda/environments/all_cuda-130_arch-x86_64.yaml b/conda/environments/all_cuda-130_arch-x86_64.yaml new file mode 100644 index 00000000..5cb15cc6 --- /dev/null +++ b/conda/environments/all_cuda-130_arch-x86_64.yaml @@ -0,0 +1,33 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai +- rapidsai-nightly +- conda-forge +dependencies: +- click >=8.1 +- cuda-core==0.3.* +- cuda-nvcc-impl +- cuda-nvrtc +- cuda-version=13.0 +- distributed-ucxx==0.46.*,>=0.0.0a0 +- kvikio==25.10.*,>=0.0.0a0 +- numba-cuda>=0.19.1,<0.20.0a0 +- numpy>=1.23,<3.0a0 +- numpydoc>=1.1.0 +- pandas>=1.3 +- pre-commit +- pynvml>=12.0.0,<13.0.0a0 +- pytest +- pytest-cov +- pytest-timeout +- python>=3.10,<3.14 +- rapids-build-backend>=0.4.0,<0.5.0dev0 +- rapids-dask-dependency==25.10.*,>=0.0.0a0 +- setuptools>=64.0.0 +- sphinx +- sphinx-click>=2.7.1 +- sphinx-rtd-theme>=0.5.1 +- ucx-py==0.46.*,>=0.0.0a0 +- zict>=2.0.0 +name: all_cuda-130_arch-x86_64 diff --git a/conda/recipes/dask-cuda/recipe.yaml b/conda/recipes/dask-cuda/recipe.yaml index 9225837e..6e158e71 100644 --- a/conda/recipes/dask-cuda/recipe.yaml +++ b/conda/recipes/dask-cuda/recipe.yaml @@ -37,7 +37,7 @@ requirements: - python - click >=8.1 - numba >=0.60.0,<0.62.0a0 - - numba-cuda >=0.19.0,<0.20.0a0 + - numba-cuda >=0.19.1,<0.20.0a0 - numpy >=1.23,<3.0a0 - pandas >=1.3 - pynvml >=12.0.0,<13.0.0a0 diff --git a/dask_cuda/tests/test_cudf_builtin_spilling.py b/dask_cuda/tests/test_cudf_builtin_spilling.py index 80b1d482..e7a4e474 100644 --- a/dask_cuda/tests/test_cudf_builtin_spilling.py +++ b/dask_cuda/tests/test_cudf_builtin_spilling.py @@ -1,3 +1,5 @@ +# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. +# SPDX-License-Identifier: Apache-2.0 import pytest from distributed.sizeof import safe_sizeof @@ -6,6 +8,7 @@ from dask_cuda.is_spillable_object import is_spillable_object from dask_cuda.proxify_host_file import ProxifyHostFile +cudf = pytest.importorskip("cudf") cupy = pytest.importorskip("cupy") pandas = pytest.importorskip("pandas") @@ -14,7 +17,6 @@ reason="Current version of cudf doesn't support built-in spilling", ) -import cudf # noqa: E402 from cudf.core.buffer.spill_manager import ( # noqa: E402 SpillManager, get_global_manager, diff --git a/dask_cuda/tests/test_spill.py b/dask_cuda/tests/test_spill.py index daa4c5bc..8169d852 100644 --- a/dask_cuda/tests/test_spill.py +++ b/dask_cuda/tests/test_spill.py @@ -15,10 +15,10 @@ from distributed.utils import Deadline from distributed.utils_test import gen_cluster, gen_test, loop # noqa: F401 -import dask_cudf +dask_cudf = pytest.importorskip("dask_cudf") -from dask_cuda import LocalCUDACluster, utils -from dask_cuda.utils_test import IncreasedCloseTimeoutNanny +from dask_cuda import LocalCUDACluster, utils # noqa: E402 +from dask_cuda.utils_test import IncreasedCloseTimeoutNanny # noqa: E402 if not utils.has_device_memory_resource(): pytest.skip( diff --git a/dependencies.yaml b/dependencies.yaml index 7c6edfd7..886f54d3 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -3,7 +3,7 @@ files: all: output: conda matrix: - cuda: ["12.9"] + cuda: ["12.9", "13.0"] arch: [x86_64, aarch64] includes: - build_python @@ -42,6 +42,22 @@ files: table: build-system includes: - build_python + py_cu12: + output: pyproject + pyproject_dir: . + extras: + table: project.optional-dependencies + key: cu12 + includes: + - extra_cu12 + py_cu13: + output: pyproject + pyproject_dir: . + extras: + table: project.optional-dependencies + key: cu13 + includes: + - extra_cu13 py_run: output: pyproject pyproject_dir: . @@ -92,15 +108,16 @@ dependencies: cuda: "12.9" packages: - cuda-version=12.9 - cuda: - specific: - - output_types: conda - matrices: - matrix: - cuda: "12.*" + cuda: "13.0" packages: - - cuda-nvcc-impl - - cuda-nvrtc + - cuda-version=13.0 + cuda: + common: + - output_types: conda + packages: + - cuda-nvcc-impl + - cuda-nvrtc develop: common: - output_types: [conda, requirements] @@ -148,6 +165,28 @@ dependencies: - pynvml>=12.0.0,<13.0.0a0 - rapids-dask-dependency==25.10.*,>=0.0.0a0 - zict>=2.0.0 + extra_cu12: + common: + - output_types: [pyproject] + packages: + - cudf-cu12==25.10.*,>=0.0.0a0 + - dask-cudf-cu12==25.10.*,>=0.0.0a0 + - distributed-ucxx-cu12==0.46.*,>=0.0.0a0 + - kvikio-cu12==25.10.*,>=0.0.0a0 + - ucx-py-cu12==0.46.*,>=0.0.0a0 + - numba-cuda[cu12]>=0.19.1,<0.20.0a0 + extra_cu13: + common: + - output_types: [pyproject] + packages: + # TODO: add 'cudf' and 'dask-cudf' back to this dependency list once there are CUDA 13 packages for those + # ref: https://github.com/rapidsai/dask-cuda/pull/1536#issuecomment-3212474898 + # - cudf-cu13==25.10.*,>=0.0.0a0 + # - dask-cudf-cu13==25.10.*,>=0.0.0a0 + - distributed-ucxx-cu13==0.46.*,>=0.0.0a0 + - kvikio-cu13==25.10.*,>=0.0.0a0 + - ucx-py-cu13==0.46.*,>=0.0.0a0 + - numba-cuda[cu13]>=0.19.1,<0.20.0a0 test_python: common: - output_types: [conda, requirements, pyproject] @@ -155,47 +194,30 @@ dependencies: - pytest - pytest-cov - pytest-timeout + - output_types: [requirements, pyproject] + packages: + # ensure dependencies of 'dask.dataframe' are available + # ref: https://github.com/rapidsai/dask-cuda/pull/1536#discussion_r2298528510 + # TODO: remove this once there are CUDA 13 'dask-cudf' packages + - 'dask[dataframe]' - output_types: [conda] packages: - - &cudf_unsuffixed cudf==25.10.*,>=0.0.0a0 - - &dask_cudf_unsuffixed dask-cudf==25.10.*,>=0.0.0a0 - - &distributed_ucxx_unsuffixed distributed-ucxx==0.46.*,>=0.0.0a0 - - &kvikio_unsuffixed kvikio==25.10.*,>=0.0.0a0 - - &ucx_py_unsuffixed ucx-py==0.46.*,>=0.0.0a0 - - ucxx==0.46.*,>=0.0.0a0 - - &numba_cuda numba-cuda>=0.19.0,<0.20.0a0 - + - distributed-ucxx==0.46.*,>=0.0.0a0 + - kvikio==25.10.*,>=0.0.0a0 + - ucx-py==0.46.*,>=0.0.0a0 + - numba-cuda>=0.19.1,<0.20.0a0 specific: - - output_types: conda - matrices: - - matrix: - arch: x86_64 - packages: - - numactl-devel-cos7-x86_64 - - matrix: - arch: aarch64 - packages: - - numactl-devel-cos7-aarch64 - - output_types: [requirements, pyproject] + - output_types: [conda] matrices: - matrix: cuda: "12.*" - cuda_suffixed: "true" packages: - - cudf-cu12==25.10.*,>=0.0.0a0 - - dask-cudf-cu12==25.10.*,>=0.0.0a0 - - distributed-ucxx-cu12==0.46.*,>=0.0.0a0 - - kvikio-cu12==25.10.*,>=0.0.0a0 - - ucx-py-cu12==0.46.*,>=0.0.0a0 - - &numba_cuda_cu12 numba-cuda[cu12]>=0.19.0,<0.20.0a0 + # TODO: add 'cudf' and 'dask-cudf' back to the 'common' list above once there are CUDA 13 packages for those + # ref: https://github.com/rapidsai/dask-cuda/pull/1536#issuecomment-3212474898 + - cudf==25.10.*,>=0.0.0a0 + - dask-cudf==25.10.*,>=0.0.0a0 - matrix: packages: - - *cudf_unsuffixed - - *dask_cudf_unsuffixed - - *distributed_ucxx_unsuffixed - - *kvikio_unsuffixed - - *ucx_py_unsuffixed - - *numba_cuda_cu12 depends_on_dask_cuda: common: - output_types: conda diff --git a/docs/source/install.rst b/docs/source/install.rst index b665bd88..8f647353 100644 --- a/docs/source/install.rst +++ b/docs/source/install.rst @@ -16,7 +16,7 @@ To install the latest version of Dask-CUDA along with the CUDA Toolkit: .. code-block:: bash - conda install -c rapidsai -c conda-forge -c nvidia dask-cuda cuda-version=12.9 + conda install -c rapidsai -c conda-forge -c nvidia dask-cuda cuda-version=13.0 Pip --- @@ -28,6 +28,18 @@ To install the latest version of Dask-CUDA: python -m pip install dask-cuda +`dask-cuda` also provides extras to ensure a consistent set of optional dependencies are installed based on the CUDA version you're using. + +.. code-block:: bash + + # CUDA 13 + python -m pip install 'dask-cuda[cu13]' + +.. code-block:: bash + + # CUDA 12 + python -m pip install 'dask-cuda[cu12]' + Source ------ diff --git a/pyproject.toml b/pyproject.toml index cad84bfd..48dfa08d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -51,6 +51,20 @@ docs = [ "sphinx-click>=2.7.1", "sphinx-rtd-theme>=0.5.1", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`. +cu12 = [ + "cudf-cu12==25.10.*,>=0.0.0a0", + "dask-cudf-cu12==25.10.*,>=0.0.0a0", + "distributed-ucxx-cu12==0.46.*,>=0.0.0a0", + "kvikio-cu12==25.10.*,>=0.0.0a0", + "numba-cuda[cu12]>=0.19.1,<0.20.0a0", + "ucx-py-cu12==0.46.*,>=0.0.0a0", +] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`. +cu13 = [ + "distributed-ucxx-cu13==0.46.*,>=0.0.0a0", + "kvikio-cu13==25.10.*,>=0.0.0a0", + "numba-cuda[cu13]>=0.19.1,<0.20.0a0", + "ucx-py-cu13==0.46.*,>=0.0.0a0", +] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`. [project.urls] Homepage = "https://github.com/rapidsai/dask-cuda"