From a38f960440bf3ea84092ba32133bf00ca840db2e Mon Sep 17 00:00:00 2001 From: John Kirkham Date: Tue, 3 May 2022 20:36:20 -0700 Subject: [PATCH] Require UCX 1.12.1+ --- conda/environments/raft_dev_cuda11.0.yml | 1 + conda/environments/raft_dev_cuda11.2.yml | 1 + conda/environments/raft_dev_cuda11.4.yml | 1 + conda/environments/raft_dev_cuda11.5.yml | 1 + conda/recipes/pyraft/conda_build_config.yaml | 2 ++ conda/recipes/pyraft/meta.yaml | 2 ++ 6 files changed, 8 insertions(+) create mode 100644 conda/recipes/pyraft/conda_build_config.yaml diff --git a/conda/environments/raft_dev_cuda11.0.yml b/conda/environments/raft_dev_cuda11.0.yml index 8bcf9abca2..2a1f8113d6 100644 --- a/conda/environments/raft_dev_cuda11.0.yml +++ b/conda/environments/raft_dev_cuda11.0.yml @@ -13,6 +13,7 @@ dependencies: - rapids-doc-env=22.02.* - rmm=22.02.* - dask-cuda=22.02.* +- ucx>=1.12.1 - ucx-py=0.26.* - ucx-proc=*=gpu - doxygen>=1.8.20 diff --git a/conda/environments/raft_dev_cuda11.2.yml b/conda/environments/raft_dev_cuda11.2.yml index e8cf33c8f8..ec4adc0c9e 100644 --- a/conda/environments/raft_dev_cuda11.2.yml +++ b/conda/environments/raft_dev_cuda11.2.yml @@ -13,6 +13,7 @@ dependencies: - rapids-doc-env=22.02.* - rmm=22.02.* - dask-cuda=22.02.* +- ucx>=1.12.1 - ucx-py=0.26.* - ucx-proc=*=gpu - doxygen>=1.8.20 diff --git a/conda/environments/raft_dev_cuda11.4.yml b/conda/environments/raft_dev_cuda11.4.yml index 2d7ec1bfed..ce2bd54635 100644 --- a/conda/environments/raft_dev_cuda11.4.yml +++ b/conda/environments/raft_dev_cuda11.4.yml @@ -13,6 +13,7 @@ dependencies: - rapids-doc-env=22.02.* - rmm=22.02.* - dask-cuda=22.02.* +- ucx>=1.12.1 - ucx-py=0.26.* - ucx-proc=*=gpu - doxygen>=1.8.20 diff --git a/conda/environments/raft_dev_cuda11.5.yml b/conda/environments/raft_dev_cuda11.5.yml index ad5cb9630d..6b8ca74db9 100644 --- a/conda/environments/raft_dev_cuda11.5.yml +++ b/conda/environments/raft_dev_cuda11.5.yml @@ -14,6 +14,7 @@ dependencies: - rapids-doc-env=22.02.* - rmm=22.02.* - dask-cuda=22.02.* +- ucx>=1.12.1 - ucx-py=0.26.* - ucx-proc=*=gpu - doxygen>=1.8.20 diff --git a/conda/recipes/pyraft/conda_build_config.yaml b/conda/recipes/pyraft/conda_build_config.yaml new file mode 100644 index 0000000000..c542be2d20 --- /dev/null +++ b/conda/recipes/pyraft/conda_build_config.yaml @@ -0,0 +1,2 @@ +ucx_version: + - "1.12.1" diff --git a/conda/recipes/pyraft/meta.yaml b/conda/recipes/pyraft/meta.yaml index de81f8c755..ab9dc28fda 100644 --- a/conda/recipes/pyraft/meta.yaml +++ b/conda/recipes/pyraft/meta.yaml @@ -34,6 +34,7 @@ requirements: - cudatoolkit {{ cuda_version }}.* - cuda-python >=11.5,<12.0 - nccl>=2.9.9 + - ucx {{ ucx_version }} - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu run: @@ -42,6 +43,7 @@ requirements: - libraft-headers {{ version }} - nccl>=2.9.9 - rmm {{ minor_version }} + - ucx >={{ ucx_version }} - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu - dask>=2022.03.0