diff --git a/dependencies.yaml b/dependencies.yaml index b16d37f4bb..e1cc919d83 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -209,7 +209,7 @@ dependencies: common: - output_types: [conda] packages: - - &rmm_conda rmm==24.8.*,>=0.0.0a0 + - &rmm_unsuffixed rmm==24.8.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -232,13 +232,17 @@ dependencies: - &cuda_python cuda-python - output_types: [requirements, pyproject] matrices: - - matrix: {cuda: "12.*"} + - matrix: + cuda: "12.*" + cuda_suffixed: "true" packages: - &rmm_cu12 rmm-cu12==24.8.*,>=0.0.0a0 - - matrix: {cuda: "11.*"} + - matrix: + cuda: "11.*" + cuda_suffixed: "true" packages: - &rmm_cu11 rmm-cu11==24.8.*,>=0.0.0a0 - - {matrix: null, packages: [*rmm_conda] } + - {matrix: null, packages: [*rmm_unsuffixed] } checks: common: - output_types: [conda, requirements] @@ -260,7 +264,7 @@ dependencies: - h5py>=3.8.0 - benchmark>=1.8.2 - openblas - - *rmm_conda + - *rmm_unsuffixed nn_bench_python: common: - output_types: [conda] @@ -441,7 +445,7 @@ dependencies: - &numpy numpy>=1.23,<2.0a0 - output_types: [conda] packages: - - *rmm_conda + - *rmm_unsuffixed - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -470,7 +474,7 @@ dependencies: - matrix: {cuda: "11.*"} packages: - *rmm_cu11 - - {matrix: null, packages: [*rmm_conda]} + - {matrix: null, packages: [*rmm_unsuffixed]} run_raft_dask: common: - output_types: [conda, pyproject] @@ -482,8 +486,8 @@ dependencies: - rapids-dask-dependency==24.8.*,>=0.0.0a0 - output_types: conda packages: - - &pylibraft_conda pylibraft==24.8.*,>=0.0.0a0 - - &ucx_py_conda ucx-py==0.39.*,>=0.0.0a0 + - &pylibraft_unsuffixed pylibraft==24.8.*,>=0.0.0a0 + - &ucx_py_unsuffixed ucx-py==0.39.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -493,15 +497,19 @@ dependencies: specific: - output_types: [requirements, pyproject] matrices: - - matrix: {cuda: "12.*"} + - matrix: + cuda: "12.*" + cuda_suffixed: "true" packages: - &pylibraft_cu12 pylibraft-cu12==24.8.*,>=0.0.0a0 - &ucx_py_cu12 ucx-py-cu12==0.39.*,>=0.0.0a0 - - matrix: {cuda: "11.*"} + - matrix: + cuda: "11.*" + cuda_suffixed: "true" packages: - &pylibraft_cu11 pylibraft-cu11==24.8.*,>=0.0.0a0 - &ucx_py_cu11 ucx-py-cu11==0.39.*,>=0.0.0a0 - - {matrix: null, packages: [*pylibraft_conda, *ucx_py_conda]} + - {matrix: null, packages: [*pylibraft_unsuffixed, *ucx_py_unsuffixed]} test_python_common: common: - output_types: [conda, requirements, pyproject] @@ -520,7 +528,7 @@ dependencies: packages: # UCXX is not currently a hard-dependency thus only installed during tests, # this will change in the future. - - &distributed_ucxx_conda distributed-ucxx==0.39.*,>=0.0.0a0 + - &distributed_ucxx_unsuffixed distributed-ucxx==0.39.*,>=0.0.0a0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -529,18 +537,22 @@ dependencies: specific: - output_types: [requirements, pyproject] matrices: - - matrix: {cuda: "12.*"} + - matrix: + cuda: "12.*" + cuda_suffixed: "true" packages: - distributed-ucxx-cu12==0.39.*,>=0.0.0a0 - - matrix: {cuda: "11.*"} + - matrix: + cuda: "11.*" + cuda_suffixed: "true" packages: - distributed-ucxx-cu11==0.39.*,>=0.0.0a0 - - {matrix: null, packages: [*distributed_ucxx_conda]} + - {matrix: null, packages: [*distributed_ucxx_unsuffixed]} depends_on_ucx_build: common: - output_types: conda packages: - - &ucx_conda_build ucx==1.15.0 + - ucx==1.15.0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -549,10 +561,14 @@ dependencies: specific: - output_types: [requirements, pyproject] matrices: - - matrix: {cuda: "12.*"} + - matrix: + cuda: "12.*" + cuda_suffixed: "true" packages: - libucx-cu12==1.15.0 - - matrix: {cuda: "11.*"} + - matrix: + cuda: "11.*" + cuda_suffixed: "true" packages: - libucx-cu11==1.15.0 - matrix: null @@ -562,7 +578,7 @@ dependencies: common: - output_types: conda packages: - - &ucx_conda_run ucx>=1.15.0 + - ucx>=1.15.0 - output_types: requirements packages: # pip recognizes the index as a global option for the requirements.txt file @@ -571,10 +587,14 @@ dependencies: specific: - output_types: [requirements, pyproject] matrices: - - matrix: {cuda: "12.*"} + - matrix: + cuda: "12.*" + cuda_suffixed: "true" packages: - libucx-cu12>=1.15.0 - - matrix: {cuda: "11.*"} + - matrix: + cuda: "11.*" + cuda_suffixed: "true" packages: - libucx-cu11>=1.15.0 - matrix: null diff --git a/python/pylibraft/pyproject.toml b/python/pylibraft/pyproject.toml index 1329a3990b..e32cf5f902 100644 --- a/python/pylibraft/pyproject.toml +++ b/python/pylibraft/pyproject.toml @@ -123,6 +123,7 @@ requires = [ "rmm==24.8.*,>=0.0.0a0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. dependencies-file = "../../dependencies.yaml" +matrix-entry = "cuda_suffixed=true" [tool.pytest.ini_options] filterwarnings = [ diff --git a/python/raft-ann-bench/pyproject.toml b/python/raft-ann-bench/pyproject.toml index 226dc41e40..d22dd567fe 100644 --- a/python/raft-ann-bench/pyproject.toml +++ b/python/raft-ann-bench/pyproject.toml @@ -68,3 +68,4 @@ build-backend = "setuptools.build_meta" requires = [] dependencies-file = "../../dependencies.yaml" commit-files = ["src/raft_ann_bench/GIT_COMMIT"] +matrix-entry = "cuda_suffixed=true" diff --git a/python/raft-dask/pyproject.toml b/python/raft-dask/pyproject.toml index 24ee27ddea..26a1e9ffe2 100644 --- a/python/raft-dask/pyproject.toml +++ b/python/raft-dask/pyproject.toml @@ -125,3 +125,4 @@ requires = [ "ninja", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. dependencies-file = "../../dependencies.yaml" +matrix-entry = "cuda_suffixed=true"