Build with CUDA 12

.ci_support/linux_64_c_compiler_version12cdt_namecos7cuda_compilercuda-nvcccuda_compiler_version12.0cxx_compiler_version12.yaml

@@ -0,0 +1,33 @@
+c_compiler:
+- gcc
+c_compiler_version:
+- '12'
+cdt_name:
+- cos7
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- cuda-nvcc
+cuda_compiler_version:
+- '12.0'
+cxx_compiler:
+- gxx
+cxx_compiler_version:
+- '12'
+docker_image:
+- quay.io/condaforge/linux-anvil-cos7-x86_64
+rdma_core:
+- '28'
+target_platform:
+- linux-64
+ucx:
+- 1.14.0
+zip_keys:
+- - c_compiler_version
+  - cxx_compiler_version
+  - cuda_compiler
+  - cuda_compiler_version
+  - cdt_name
+  - docker_image

.ci_support/linux_aarch64_c_compiler_version10cuda_compilercuda-nvcccuda_compiler_version12.0cxx_compiler_version10.yaml

@@ -0,0 +1,35 @@
+BUILD:
+- aarch64-conda_cos7-linux-gnu
+c_compiler:
+- gcc
+c_compiler_version:
+- '10'
+cdt_arch:
+- aarch64
+cdt_name:
+- cos7
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- cuda-nvcc
+cuda_compiler_version:
+- '12.0'
+cxx_compiler:
+- gxx
+cxx_compiler_version:
+- '10'
+docker_image:
+- quay.io/condaforge/linux-anvil-aarch64
+target_platform:
+- linux-aarch64
+ucx:
+- 1.14.0
+zip_keys:
+- - c_compiler_version
+  - cxx_compiler_version
+  - cuda_compiler
+  - cuda_compiler_version
+  - cdt_name
+  - docker_image

.ci_support/linux_ppc64le_c_compiler_version10cuda_compilercuda-nvcccuda_compiler_version12.0cxx_compiler_version10.yaml

@@ -0,0 +1,31 @@
+c_compiler:
+- gcc
+c_compiler_version:
+- '10'
+cdt_name:
+- cos7
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- cuda-nvcc
+cuda_compiler_version:
+- '12.0'
+cxx_compiler:
+- gxx
+cxx_compiler_version:
+- '10'
+docker_image:
+- quay.io/condaforge/linux-anvil-ppc64le
+target_platform:
+- linux-ppc64le
+ucx:
+- 1.14.0
+zip_keys:
+- - c_compiler_version
+  - cxx_compiler_version
+  - cuda_compiler
+  - cuda_compiler_version
+  - cdt_name
+  - docker_image

.ci_support/migrations/cuda120.yaml

@@ -0,0 +1,39 @@
+migrator_ts: 1682985063
+__migrator:
+  kind:
+    version
+  use_local: true  # TODO: Remove this line after completing testing
+  migration_number:
+    1
+  build_number:
+    1
+  paused: false
+  override_cbc_keys:
+    - cuda_compiler_stub
+  operation: key_add
+  primary_key: cuda_compiler_version
+  commit_message: "Rebuild for CUDA 12"
+
+cuda_compiler:                 # [linux64 or win]
+  - cuda-nvcc                  # [linux64 or win]
+
+cuda_compiler_version:         # [linux64 or win]
+  - 12.0                       # [linux64 or win]
+
+c_compiler_version:            # [linux64]
+  - 12                         # [linux64]
+
+cxx_compiler_version:          # [linux64]
+  - 12                         # [linux64]
+
+fortran_compiler_version:      # [linux64]
+  - 12                         # [linux64]
+
+cudnn:                         # [linux64 or win]
+  - 8                          # [linux64 or win]
+
+cdt_name:                      # [linux64]
+  - cos7                       # [linux64]
+
+docker_image:                                      # [os.environ.get("BUILD_PLATFORM", "").startswith("linux")]
+  - quay.io/condaforge/linux-anvil-cos7-x86_64     # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]

.ci_support/migrations/cuda120_ppc64le_aarch64.yaml

@@ -0,0 +1,48 @@
+migrator_ts: 1683090168
+__migrator:
+  kind:
+    version
+  use_local: true  # TODO: Remove this line after completing testing
+  migration_number:
+    1
+  build_number:
+    1
+  paused: false
+  override_cbc_keys:
+    - cuda_compiler_stub
+  operation: key_add
+  check_solvable: false
+  primary_key: cuda_compiler_version
+  wait_for_migrators:
+    - aarch64 and ppc64le addition
+  exclude_pinned_pkgs: False
+  commit_message: "Rebuild for cuda 12 on ppc64le and aarch64"
+
+arm_variant_type:              # [aarch64]
+  - sbsa                       # [aarch64]
+
+c_compiler_version:            # [ppc64le or aarch64]
+  - 10                         # [ppc64le or aarch64]
+cxx_compiler_version:          # [ppc64le or aarch64]
+  - 10                         # [ppc64le or aarch64]
+fortran_compiler_version:      # [ppc64le or aarch64]
+  - 10                         # [ppc64le or aarch64]
+
+cuda_compiler:                 # [ppc64le or aarch64]
+  - cuda-nvcc                  # [ppc64le or aarch64]
+cuda_compiler_version:         # [ppc64le or aarch64]
+  - 12.0                       # [ppc64le or aarch64]
+
+cudnn:                  # [ppc64le or aarch64]
+  - 8                   # [ppc64le or aarch64]
+
+cdt_name:  # [ppc64le or aarch64]
+  - cos7   # [ppc64le or aarch64]
+
+docker_image:                                           # [os.environ.get("BUILD_PLATFORM", "").startswith("linux") and (ppc64le or aarch64)]
+   # case: native compilation (build == target)
+   - quay.io/condaforge/linux-anvil-ppc64le             # [ppc64le and os.environ.get("BUILD_PLATFORM") == "linux-ppc64le"]
+   - quay.io/condaforge/linux-anvil-aarch64             # [aarch64 and os.environ.get("BUILD_PLATFORM") == "linux-aarch64"]
+   # case: cross-compilation (build != target)
+   - quay.io/condaforge/linux-anvil-comp7               # [ppc64le and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+   - quay.io/condaforge/linux-anvil-comp7               # [aarch64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]

recipe/install_ucx.sh

@@ -3,8 +3,17 @@
 set -xeuo pipefail
 
 EXTRA_ARGS=""
-if [ "${cuda_compiler_version}" != "None" ]; then
-    EXTRA_ARGS="${EXTRA_ARGS} --with-cuda=${CUDA_HOME}"
+
+if [[ "${cuda_compiler_version}" =~ 12.* ]]; then
+  EXTRA_ARGS="${EXTRA_ARGS} --with-cuda=${PREFIX}"
+
+  [[ ${target_platform} == "linux-64" ]] && targetsDir="targets/x86_64-linux"
+  [[ ${target_platform} == "linux-ppc64le" ]] && targetsDir="targets/ppc64le-linux"
+  [[ ${target_platform} == "linux-aarch64" ]] && targetsDir="targets/sbsa-linux"
+
+  export CFLAGS="${CFLAGS} -I${BUILD_PREFIX}/${targetsDir}/include -L${BUILD_PREFIX}/${targetsDir}/lib/stubs"
+elif [[ "${cuda_compiler_version}" != "None" ]]; then
+  EXTRA_ARGS="${EXTRA_ARGS} --with-cuda=${CUDA_HOME}"
 fi
 
 if [ "${cuda_compiler_version}" != "10.2" ] && [ "${target_platform}" == "linux-64" ]; then

recipe/meta.yaml

@@ -1,5 +1,5 @@
 {% set ucx_version = "1.14.1" %}
-{% set number = "1" %}
+{% set number = "2" %}
 
 
 package:
@@ -26,21 +26,24 @@ outputs:
       build:
         - {{ compiler("c") }}
         - {{ compiler("cxx") }}
-        - {{ compiler("cuda") }}         # [cuda_compiler_version != "None"]
+        - {{ compiler("cuda") }}
         - {{ cdt("libnl") }}             # [cdt_name == "cos6"]
         - {{ cdt("libnl3") }}            # [cdt_name == "cos7"]
+        - cuda-nvml-dev ={{ cuda_compiler_version }}       # [cuda_compiler_version == "12.0"]
         - automake
         - autoconf
         - libtool
         - make
         - pkg-config
       host:
+        - cuda-nvml-dev ={{ cuda_compiler_version }}       # [cuda_compiler_version == "12.0"]
         - libnuma
         - rdma-core  # [linux64 and cuda_compiler_version != "10.2"]
       run:
       run_constrained:
-        - cudatoolkit {{ cuda_compiler_version }}|{{ cuda_compiler_version }}.*  # [cuda_compiler_version not in ("None", "11.2")]
+        - cudatoolkit {{ cuda_compiler_version }}|{{ cuda_compiler_version }}.*  # [cuda_compiler_version not in ("None", "11.2", "12.0")]
         - cudatoolkit >=11.2,<12                                                 # [cuda_compiler_version == "11.2"]
+
     script: install_ucx.sh
     test:
       commands: