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