Add conda recipe for distributed-ucxx package

rapidsai · Oct 23, 2023 · d3bd472 · d3bd472
1 parent 1cf1014
commit d3bd472
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/conda/recipes/ucxx/build_and_install_distributed_ucxx.sh b/conda/recipes/ucxx/build_and_install_distributed_ucxx.sh
@@ -0,0 +1,6 @@
+#!/bin/bash
+
+# SPDX-FileCopyrightText: Copyright (c) 2023, NVIDIA CORPORATION & AFFILIATES.
+# SPDX-License-Identifier: BSD-3-Clause
+
+./build.sh distributed_ucxx
diff --git a/conda/recipes/ucxx/meta.yaml b/conda/recipes/ucxx/meta.yaml
@@ -249,3 +249,40 @@ outputs:
       license_family: BSD
       license_file: ../../../LICENSE
       summary: UCX Python interface built on top of the libucxx C++ implementation
+
+
+  - name: distributed-ucxx
+    version: {{ version }}
+    script: build_and_install_distributed_ucxx.sh
+    build:
+      number: {{ GIT_DESCRIBE_NUMBER }}
+      string: py{{ python }}_{{ date_string }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
+    requirements:
+      build:
+        - sysroot_{{ target_platform }}
+      host:
+        - python
+        - pip
+        - scikit-build>=0.13.1
+        - setuptools
+        - dask
+        - distributed
+        - {{ pin_subpackage('ucxx', exact=True) }}
+      run:
+        - python * *_cpython
+        - packaging
+        - {{ pin_subpackage('ucxx', exact=True) }}
+        - pynvml >=11.4.1
+      run_constrained:
+        - cupy >=9.5.0
+        - numba >=0.57.0
+    test:
+    test:
+      imports:
+        - distributed_ucxx
+    about:
+      home: https://rapids.ai/
+      license: BSD-3-Clause
+      license_family: BSD
+      license_file: ../../../LICENSE
+      summary: UCX communication module for Dask Distributed