From ef0031182ec1cbb86da405f60d43187260b6b2e6 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Wed, 23 Oct 2024 15:48:17 +0200
Subject: [PATCH] chore(docker): add mpi to ld_library_path

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

diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm
index 6d3c26d6..2a0ae049 100644
--- a/Dockerfile_trtllm
+++ b/Dockerfile_trtllm
@@ -89,7 +89,7 @@ RUN apt update && apt install -y python3-minimal python3-dev && \
 
 WORKDIR /usr/local/tgi/bin
 
-ENV LD_LIBRARY_PATH="/usr/local/tgi/lib:/usr/local/tensorrt/lib:/usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH"
+ENV LD_LIBRARY_PATH="/usr/local/tgi/lib:/usr/local/mpi/lib:/usr/local/tensorrt/lib:/usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH"
 ENV TOKENIZERS_PARALLELISM=false
 ENV OMPI_MCA_plm_rsh_agent=""