From 5a317ffad730846be175c5a1af64bbaf4c002e16 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Sun, 26 Jan 2025 15:21:38 +0100
Subject: [PATCH] backend(trtllm): inject ompi_version build arg in dependent
 step

---
 Dockerfile_trtllm | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm
index 148d29e0..6538996a 100644
--- 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 \