From 89e77c2514a7250dcd1847027ee1efc9e4e134e8 Mon Sep 17 00:00:00 2001
From: Hugo Larcher <hugo.larcher@huggingface.co>
Date: Mon, 2 Dec 2024 22:12:26 +0100
Subject: [PATCH] chore: add dev-containers

---
 .devcontainer/Dockerfile_trtllm | 2 +-
 .devcontainer/devcontainer.json | 9 +++------
 2 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/.devcontainer/Dockerfile_trtllm b/.devcontainer/Dockerfile_trtllm
index 9de62bdb..e396dc84 100644
--- a/.devcontainer/Dockerfile_trtllm
+++ b/.devcontainer/Dockerfile_trtllm
@@ -53,7 +53,7 @@ RUN wget "https://download.open-mpi.org/release/open-mpi/v4.1/$OMPI_TARBALL_FILE
 
 # Install TensorRT
 FROM cuda-builder AS trt-builder
-COPY install_tensorrt.sh /opt/install_tensorrt.sh
+COPY backends/trtllm/scripts/install_tensorrt.sh /opt/install_tensorrt.sh
 RUN chmod +x /opt/install_tensorrt.sh && \
     /opt/install_tensorrt.sh
 
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 08acb6b4..8b4eb89c 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -2,9 +2,9 @@
 // README at: https://github.com/devcontainers/templates/tree/main/src/cpp
 {
   "name": "CUDA",
-//  "image": "nvidia/cuda:12.6.0-devel-ubuntu24.04",
   "build": {
-    "dockerfile": "Dockerfile_trtllm"
+    "dockerfile": "Dockerfile_trtllm",
+    "context": ".."
   },
   "remoteEnv": {
     "PATH": "${containerEnv:PATH}:/usr/local/cuda/bin",
@@ -15,8 +15,5 @@
     "jetbrains" : {
       "backend" : "CLion"
     }
-  },
-
-	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
-	// "remoteUser": "root"
+  }
 }