diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index c4c4200129..655b3e99ae 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -19,6 +19,7 @@ dependencies: - dask-cuda==24.6.* - dask-cudf==24.6.* - dask-ml +- distributed-ucxx==0.38.* - doxygen=1.9.1 - gcc_linux-64=11.* - gmock>=1.13.0 diff --git a/conda/environments/all_cuda-122_arch-x86_64.yaml b/conda/environments/all_cuda-122_arch-x86_64.yaml index 89fba07152..b35b378147 100644 --- a/conda/environments/all_cuda-122_arch-x86_64.yaml +++ b/conda/environments/all_cuda-122_arch-x86_64.yaml @@ -21,6 +21,7 @@ dependencies: - dask-cuda==24.6.* - dask-cudf==24.6.* - dask-ml +- distributed-ucxx==0.38.* - doxygen=1.9.1 - gcc_linux-64=11.* - gmock>=1.13.0 diff --git a/dependencies.yaml b/dependencies.yaml index 47df04189e..4ff855f380 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -9,6 +9,7 @@ files: - common_build - cuda - cuda_version + - depends_on_distributed_ucxx - docs - py_build - py_run @@ -54,6 +55,7 @@ files: output: none includes: - cuda_version + - depends_on_distributed_ucxx - py_version - test_cuml - test_python @@ -84,6 +86,7 @@ files: table: project.optional-dependencies key: test includes: + - depends_on_distributed_ucxx - test_python channels: - rapidsai @@ -442,3 +445,25 @@ dependencies: - pandas - *scikit_learn - seaborn + depends_on_distributed_ucxx: + common: + - output_types: conda + 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.38.* + - output_types: requirements + packages: + # pip recognizes the index as a global option for the requirements.txt file + - --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.*"} + packages: + - distributed-ucxx-cu12==0.38.* + - matrix: {cuda: "11.*"} + packages: + - distributed-ucxx-cu11==0.38.* + - {matrix: null, packages: [*distributed_ucxx_conda]} diff --git a/python/pyproject.toml b/python/pyproject.toml index 56bf55f5c8..7413154ea0 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -82,6 +82,7 @@ classifiers = [ test = [ "dask-glm==0.3.0", "dask-ml", + "distributed-ucxx==0.38.*", "hdbscan @ git+https://github.com/scikit-learn-contrib/hdbscan.git@master", "hypothesis>=6.0,<7", "nltk",