diff --git a/conda/conda-build/meta.yaml b/conda/conda-build/meta.yaml index 6edab7b99..fe1deacba 100644 --- a/conda/conda-build/meta.yaml +++ b/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]