From c81d1dbbe0a6d8725f05634020ab6b71ffa25a54 Mon Sep 17 00:00:00 2001
From: snordmann <snordmann@nvidia.com>
Date: Tue, 19 Mar 2024 16:53:50 +0200
Subject: [PATCH] TL/MLX5: add tls in cmake config

---
 .ci/scripts/build_ucc.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.ci/scripts/build_ucc.sh b/.ci/scripts/build_ucc.sh
index 58bb7ffdcb..2c570f808e 100755
--- a/.ci/scripts/build_ucc.sh
+++ b/.ci/scripts/build_ucc.sh
@@ -8,7 +8,8 @@ cd "${UCC_SRC_DIR}"
 mkdir -p "${UCC_SRC_DIR}/build"
 cd "${UCC_SRC_DIR}/build"
 "${UCC_SRC_DIR}/configure" --with-ucx="${UCX_INSTALL_DIR}" --with-cuda="${CUDA_HOME}" \
-    --prefix="${UCC_INSTALL_DIR}" --enable-gtest --with-mpi
+    --prefix="${UCC_INSTALL_DIR}" --enable-gtest --with-mpi \
+    --with-tls=cuda,nccl,self,sharp,shm,ucp,mlx5
 make -j install
 echo "${UCC_INSTALL_DIR}/lib" > /etc/ld.so.conf.d/ucc.conf
 ldconfig