Skip to content

Commit

Permalink
reduce duplication, removed unused things in dependencies.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
@jameslamb
jameslamb committed Dec 16, 2024
1 parent 3720d8e commit 4db939e
Showing 1 changed file with 60 additions and 96 deletions.
156 changes: 60 additions & 96 deletions dependencies.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,22 +6,22 @@ files:
cuda: ["11.8", "12.5"]
arch: [x86_64, aarch64]
includes:
- rapids_build
- build_pylibraft
- checks
- cuda
- cuda_version
- depends_on_cuda_python
- depends_on_cupy
- depends_on_distributed_ucxx
- depends_on_rmm
- develop
- checks
- test_libraft
- docs
- rapids_build_setuptools
- rapids_build
- rapids_build_skbuild
- run_raft_dask
- run_pylibraft
- test_python_common
- run_raft_dask
- test_libraft
- test_pylibraft
- test_python_common
test_cpp:
output: none
includes:
Expand All @@ -31,10 +31,10 @@ files:
output: none
includes:
- cuda_version
- depends_on_cupy
- py_version
- test_python_common
- test_pylibraft
- depends_on_cupy
- test_python_common
checks:
output: none
includes:
Expand Down Expand Up @@ -62,15 +62,18 @@ files:
table: tool.rapids-build-backend
key: requires
includes:
- depends_on_cuda_python
- depends_on_rmm
- rapids_build
- build_pylibraft
py_run_pylibraft:
output: pyproject
pyproject_dir: python/pylibraft
extras:
table: project
includes:
- cuda_wheels
- depends_on_cuda_python
- depends_on_rmm
- run_pylibraft
py_test_pylibraft:
output: pyproject
Expand All @@ -79,9 +82,9 @@ files:
table: project.optional-dependencies
key: test
includes:
- test_python_common
- test_pylibraft
- depends_on_cupy
- test_pylibraft
- test_python_common
py_build_raft_dask:
output: pyproject
pyproject_dir: python/raft-dask
Expand All @@ -96,16 +99,16 @@ files:
table: tool.rapids-build-backend
key: requires
includes:
- rapids_build
- depends_on_ucx_build
- rapids_build
py_run_raft_dask:
output: pyproject
pyproject_dir: python/raft-dask
extras:
table: project
includes:
- run_raft_dask
- depends_on_distributed_ucxx
- run_raft_dask
py_test_raft_dask:
output: pyproject
pyproject_dir: python/raft-dask
Expand All @@ -125,7 +128,7 @@ dependencies:
common:
- output_types: [conda, requirements, pyproject]
packages:
- &rapids_build_backend rapids-build-backend>=0.3.0,<0.4.0.dev0
- rapids-build-backend>=0.3.0,<0.4.0.dev0
- output_types: [conda]
packages:
- scikit-build-core>=0.10.0
Expand Down Expand Up @@ -180,44 +183,6 @@ dependencies:
- matrix: {cuda: "11.2", arch: aarch64}
packages: [nvcc_linux-aarch64=11.2]

build_pylibraft:
common:
- output_types: [conda]
packages:
- &rmm_unsuffixed rmm==25.2.*,>=0.0.0a0
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
# This index is needed for rmm-cu{11,12}.
- --extra-index-url=https://pypi.nvidia.com
- --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
specific:
- output_types: [conda, requirements, pyproject]
matrices:
- matrix:
cuda: "12.*"
packages:
- &cuda_python12 cuda-python>=12.6.2,<13.0a0
- matrix:
cuda: "11.*"
packages:
- &cuda_python11 cuda-python>=11.8.5,<12.0a0
- matrix:
packages:
- &cuda_python cuda-python
- output_types: [requirements, pyproject]
matrices:
- matrix:
cuda: "12.*"
cuda_suffixed: "true"
packages:
- &rmm_cu12 rmm-cu12==25.2.*,>=0.0.0a0
- matrix:
cuda: "11.*"
cuda_suffixed: "true"
packages:
- &rmm_cu11 rmm-cu11==25.2.*,>=0.0.0a0
- {matrix: null, packages: [*rmm_unsuffixed] }
checks:
common:
- output_types: [conda, requirements]
Expand Down Expand Up @@ -398,13 +363,6 @@ dependencies:
- recommonmark
- sphinx-copybutton
- sphinx-markdown-tables
rapids_build_setuptools:
common:
- output_types: [requirements, pyproject]
packages:
- wheel
- setuptools
- *rapids_build_backend
py_version:
specific:
- output_types: conda
Expand All @@ -429,42 +387,6 @@ dependencies:
- output_types: [conda, pyproject]
packages:
- numpy>=1.23,<3.0a0
- output_types: [conda]
packages:
- *rmm_unsuffixed
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
# This index is needed for cudf and rmm.
- --extra-index-url=https://pypi.nvidia.com
- --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
specific:
- output_types: [conda, requirements, pyproject]
matrices:
- matrix:
cuda: "12.*"
packages:
- *cuda_python12
- matrix:
cuda: "11.*"
packages:
- *cuda_python11
- matrix:
packages:
- *cuda_python
- output_types: [requirements, pyproject]
matrices:
- matrix:
cuda: "12.*"
cuda_suffixed: "true"
packages:
- *rmm_cu12
- matrix:
cuda: "11.*"
cuda_suffixed: "true"
packages:
- *rmm_cu11
- {matrix: null, packages: [*rmm_unsuffixed]}
run_raft_dask:
common:
- output_types: [conda, pyproject]
Expand Down Expand Up @@ -511,6 +433,21 @@ dependencies:
packages:
- scikit-learn
- scipy
depends_on_cuda_python:
specific:
- output_types: [conda, requirements, pyproject]
matrices:
- matrix:
cuda: "12.*"
packages:
- cuda-python>=12.6.2,<13.0a0
- matrix:
cuda: "11.*"
packages:
- cuda-python>=11.8.5,<12.0a0
- matrix:
packages:
- cuda-python
depends_on_distributed_ucxx:
common:
- output_types: conda
Expand All @@ -537,6 +474,33 @@ dependencies:
packages:
- distributed-ucxx-cu11==0.42.*,>=0.0.0a0
- {matrix: null, packages: [*distributed_ucxx_unsuffixed]}
depends_on_rmm:
common:
- output_types: conda
packages:
- &rmm_unsuffixed rmm==25.2.*,>=0.0.0a0
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
# This index is needed for rmm-cu{11,12}.
- --extra-index-url=https://pypi.nvidia.com
- --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
specific:
- output_types: [requirements, pyproject]
matrices:
- matrix:
cuda: "12.*"
cuda_suffixed: "true"
packages:
- rmm-cu12==25.2.*,>=0.0.0a0
- matrix:
cuda: "11.*"
cuda_suffixed: "true"
packages:
- rmm-cu11==25.2.*,>=0.0.0a0
- matrix:
packages:
- *rmm_unsuffixed
depends_on_ucx_build:
common:
- output_types: conda
Expand Down

0 comments on commit 4db939e

Please sign in to comment.