backend(trtllm): inject ompi_version build arg in dependent step

huggingface · Jan 26, 2025 · 5a317ff · 5a317ff
1 parent c632f8a
commit 5a317ff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm
@@ -38,6 +38,7 @@ ENV TENSORRT_INSTALL_PREFIX=/usr/local/tensorrt
 FROM cuda-builder AS mpi-builder
 WORKDIR /opt/src/mpi
 
+ARG ompi_version
 ENV OMPI_VERSION=${ompi_version}
 ENV OMPI_TARBALL_FILENAME=openmpi-${OMPI_VERSION}.tar.bz2
 ADD --checksum=sha256:54a33cb7ad81ff0976f15a6cc8003c3922f0f3d8ceed14e1813ef3603f22cd34 \