Update meta.yaml

conda/conda-build/meta.yaml

@@ -15,6 +15,11 @@
   {# We need to have a default value for the initial pass over the recipe #}
   {% set debug_build_bool = false %}
 {% endif %}
+{% if ucx_configured == "true" %}
+  {% set ucx_configured_bool = true %}
+{% else %}
+  {% set ucx_configured_bool = false %}
+{% endif %}
 {% set default_env_var = '' %}
 {% if build_number is defined %}
     {# do nothing if defined #}
@@ -53,6 +58,12 @@
 {% set debug_tag='' %}
 {% endif %}
 
+{% if ucx_configured_bool %}
+{% set ucx_tag='_ucx' %}
+{% else %}
+{% set ucx_tag='' %}
+{% endif %}
+
 package:
   name: {{ name|lower }}
   version: {{ version }}
@@ -72,7 +83,8 @@ build:
   missing_dso_whitelist:
     -  '*libcuda.so*'
 
-  string: "cuda{{ cuda_major }}_py{{ py_version }}{{ git_describe_hash }}_{{ PKG_BUILDNUM }}{{ cpu_tag }}{{ debug_tag }}"
+  string: "cuda{{ cuda_major }}_py{{ py_version }}{{ git_describe_hash }}_{{ PKG_BUILDNUM }}{{ ucx_tag }}{{ cpu_tag }}{{ debug_tag }}"
+
 
   script_env:
     - SCCACHE_BUCKET
@@ -141,6 +153,9 @@ requirements:
     - libcurand-dev
     - libcusolver-dev
 {% endif %}
+{% if ucx_configured_bool %}
+    - ucx
+{% endif %}
 
   # Runtime python dependencies
   run:
@@ -155,6 +170,9 @@ requirements:
     - cuda-version >={{ cuda_version }},<{{ cuda_major+1 }}
     - nccl
 {% endif %}
+{% if ucx_configured_bool %}
+    - ucx
+{% endif %}
 
   run_constrained:
     - __glibc >=2.17  # [linux]