TL/MLX5: add tls in cmake config

openucx · Mar 19, 2024 · c81d1db · c81d1db
1 parent df699b0
commit c81d1db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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