diff --git a/conda/recipes/dask-cuda/recipe.yaml b/conda/recipes/dask-cuda/recipe.yaml index 47def929..c3c75878 100644 --- a/conda/recipes/dask-cuda/recipe.yaml +++ b/conda/recipes/dask-cuda/recipe.yaml @@ -36,6 +36,7 @@ requirements: run: - python - click >=8.1 + - cuda-core =0.3 - numba >=0.60.0,<0.62.0a0 - numba-cuda >=0.19.1,<0.20.0a0 - numpy >=1.23,<3.0a0 diff --git a/dependencies.yaml b/dependencies.yaml index 37139893..ea529a0f 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -160,6 +160,7 @@ dependencies: packages: - click >=8.1 - cuda-core==0.3.* + - numba-cuda>=0.19.1,<0.20.0a0 - numpy>=1.23,<3.0a0 # 'nvidia-ml-py' provides the 'pynvml' module, since v12.560.30 # ref: https://github.com/conda-forge/nvidia-ml-py-feedstock/pull/24 @@ -171,19 +172,13 @@ dependencies: 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 + - cuda-core[cu12]==0.3.* - numba-cuda[cu12]>=0.19.1,<0.20.0a0 extra_cu13: common: - output_types: [pyproject] packages: - - 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 + - cuda-core[cu13]==0.3.* - numba-cuda[cu13]>=0.19.1,<0.20.0a0 test_python: common: @@ -194,11 +189,35 @@ dependencies: - pytest-timeout - 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 - - numba-cuda>=0.19.1,<0.20.0a0 + - &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 + specific: + - output_types: [requirements, pyproject] + 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 + - matrix: + cuda: "13.*" + cuda_suffixed: "true" + packages: + - 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 + - matrix: + packages: + - *cudf_unsuffixed + - *dask_cudf_unsuffixed + - *distributed_ucxx_unsuffixed + - *kvikio_unsuffixed depends_on_dask_cuda: common: - output_types: conda diff --git a/pyproject.toml b/pyproject.toml index c00c8e81..bab5eaf5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,6 +19,7 @@ requires-python = ">=3.10" dependencies = [ "click >=8.1", "cuda-core==0.3.*", + "numba-cuda>=0.19.1,<0.20.0a0", "numpy>=1.23,<3.0a0", "nvidia-ml-py>=12.560.30", "pandas>=1.3", @@ -52,17 +53,11 @@ docs = [ "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", + "cuda-core[cu12]==0.3.*", "numba-cuda[cu12]>=0.19.1,<0.20.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", + "cuda-core[cu13]==0.3.*", "numba-cuda[cu13]>=0.19.1,<0.20.0a0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.