mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
[None][infra] Update ucx to 1.20.x (#9977)
Signed-off-by: Yoray Zack <yorayz@nvidia.com> Signed-off-by: Yoray Zack <62789610+zackyoray@users.noreply.github.com>
This commit is contained in:
parent
4f75a31a45
commit
63e7a2fa70
@ -1,7 +1,8 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
set -ex
|
set -ex
|
||||||
|
|
||||||
UCX_VERSION="v1.19.x"
|
UCX_VERSION="v1.20.x"
|
||||||
|
UCX_COMMIT="f656dbdf93e72e60b5d6ca78b9e3d9e744e789bd"
|
||||||
UCX_INSTALL_PATH="/usr/local/ucx/"
|
UCX_INSTALL_PATH="/usr/local/ucx/"
|
||||||
CUDA_PATH="/usr/local/cuda"
|
CUDA_PATH="/usr/local/cuda"
|
||||||
UCX_REPO="https://github.com/openucx/ucx.git"
|
UCX_REPO="https://github.com/openucx/ucx.git"
|
||||||
@ -9,7 +10,10 @@ UCX_REPO="https://github.com/openucx/ucx.git"
|
|||||||
mkdir -p /third-party-source
|
mkdir -p /third-party-source
|
||||||
|
|
||||||
rm -rf ${UCX_INSTALL_PATH}
|
rm -rf ${UCX_INSTALL_PATH}
|
||||||
git clone --depth 1 -b ${UCX_VERSION} ${UCX_REPO}
|
git clone -b ${UCX_VERSION} ${UCX_REPO}
|
||||||
|
cd ucx
|
||||||
|
git checkout ${UCX_COMMIT}
|
||||||
|
cd ..
|
||||||
tar -czf /third-party-source/ucx-${UCX_VERSION}.tar.gz ucx
|
tar -czf /third-party-source/ucx-${UCX_VERSION}.tar.gz ucx
|
||||||
cd ucx
|
cd ucx
|
||||||
./autogen.sh
|
./autogen.sh
|
||||||
|
|||||||
@ -13,7 +13,7 @@
|
|||||||
# images are adopted from PostMerge pipelines, the abbreviated commit hash is used instead.
|
# images are adopted from PostMerge pipelines, the abbreviated commit hash is used instead.
|
||||||
IMAGE_NAME=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm
|
IMAGE_NAME=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm
|
||||||
|
|
||||||
LLM_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-x86_64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512121105-9707
|
LLM_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-x86_64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512151112-9977
|
||||||
LLM_SBSA_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-aarch64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512121105-9707
|
LLM_SBSA_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-aarch64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512151112-9977
|
||||||
LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py310-trt10.13.3.9-skip-tritondevel-202512121105-9707
|
LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py310-trt10.13.3.9-skip-tritondevel-202512151112-9977
|
||||||
LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py312-trt10.13.3.9-skip-tritondevel-202512121105-9707
|
LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py312-trt10.13.3.9-skip-tritondevel-202512151112-9977
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user