diff --git a/ci/run_benchmarks.sh b/ci/run_benchmarks.sh index 37ac6bdf..b2c8dd36 100755 --- a/ci/run_benchmarks.sh +++ b/ci/run_benchmarks.sh @@ -6,13 +6,6 @@ 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/conda/environments/all_cuda-130_arch-aarch64.yaml b/conda/environments/all_cuda-130_arch-aarch64.yaml index e6b70cff..0eb9b209 100644 --- a/conda/environments/all_cuda-130_arch-aarch64.yaml +++ b/conda/environments/all_cuda-130_arch-aarch64.yaml @@ -10,6 +10,8 @@ dependencies: - cuda-nvcc-impl - cuda-nvrtc - cuda-version=13.0 +- cudf==25.10.*,>=0.0.0a0 +- dask-cudf==25.10.*,>=0.0.0a0 - distributed-ucxx==0.46.*,>=0.0.0a0 - kvikio==25.10.*,>=0.0.0a0 - numba-cuda>=0.19.1,<0.20.0a0 diff --git a/conda/environments/all_cuda-130_arch-x86_64.yaml b/conda/environments/all_cuda-130_arch-x86_64.yaml index 5cb15cc6..a9dfa77a 100644 --- a/conda/environments/all_cuda-130_arch-x86_64.yaml +++ b/conda/environments/all_cuda-130_arch-x86_64.yaml @@ -10,6 +10,8 @@ dependencies: - cuda-nvcc-impl - cuda-nvrtc - cuda-version=13.0 +- cudf==25.10.*,>=0.0.0a0 +- dask-cudf==25.10.*,>=0.0.0a0 - distributed-ucxx==0.46.*,>=0.0.0a0 - kvikio==25.10.*,>=0.0.0a0 - numba-cuda>=0.19.1,<0.20.0a0 diff --git a/dependencies.yaml b/dependencies.yaml index 886f54d3..ca8fc71e 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -179,10 +179,8 @@ dependencies: 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 + - 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 @@ -194,30 +192,14 @@ 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==25.10.*,>=0.0.0a0 + - dask-cudf==25.10.*,>=0.0.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: - cuda: "12.*" - packages: - # 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: depends_on_dask_cuda: common: - output_types: conda diff --git a/pyproject.toml b/pyproject.toml index 48dfa08d..6ebd852c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -60,6 +60,8 @@ cu12 = [ "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 = [ + "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", "numba-cuda[cu13]>=0.19.1,<0.20.0a0",