A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 4, 10:00 UTC
[ROCm][CI] Upgraded UCX and RIXL (#41210)
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Andreas Karatzas committed
May 1, 2026 at 02:40 UTC
32964e770041fa4124f98c66efb3ff721ca608b6
1 file changed
+3
-2
docker/Dockerfile.rocm
+3
-2
@@ -124,9 +124,9 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/vllm/v1 /vllm_v1
124
125
# RIXL/UCX build stages
126
FROM base AS build_rixl
127
-ARG RIXL_BRANCH="bf4a7214"
127
+ARG RIXL_BRANCH="39be1de8"
128
ARG RIXL_REPO="https://github.com/ROCm/RIXL.git"
129
-ARG UCX_BRANCH="7009d7a1"
129
+ARG UCX_BRANCH="bfb51733"
130
ARG UCX_REPO="https://github.com/openucx/ucx.git"
131
ENV ROCM_PATH=/opt/rocm
132
ENV UCX_HOME=/usr/local/ucx
@@ -192,6 +192,7 @@ RUN cd /opt/rixl && \
192
sed -i "s/--exclude 'libamdhip64\*'/--exclude 'libamdhip64*' --exclude 'libcore*' --exclude 'libpull*'/" \
193
contrib/build-wheel.sh && \
194
mkdir -p /app/install && \
195
+ _ucx_install_dir=${UCX_HOME} \
196
./contrib/build-wheel.sh \
197
--output-dir /app/install \
198
--rocm-dir ${ROCM_PATH} \