From 1179fdff7e546a52b1da8f261deedf53e08c304f Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Thu, 9 Jan 2025 22:17:22 +0100
Subject: [PATCH] misc(ci): detect gha build

---
 Dockerfile_trtllm        | 1 +
 backends/trtllm/build.rs | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm
index 0b13f2f8..76d9091c 100644
--- a/Dockerfile_trtllm
+++ b/Dockerfile_trtllm
@@ -92,6 +92,7 @@ ENV USE_LLD_LINKER=ON
 ENV CUDA_ARCH_LIST=${cuda_arch_list}
 ENV CMAKE_CXX_COMPILER_LAUNCHER=sccache
 ENV RUSTC_WRAPPER=sccache
+ENV IS_GHA_BUILD=${is_gha_build}
 
 COPY . .
 COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt
diff --git a/backends/trtllm/build.rs b/backends/trtllm/build.rs
index 4d907954..1a229940 100644
--- a/backends/trtllm/build.rs
+++ b/backends/trtllm/build.rs
@@ -113,7 +113,7 @@ fn build_backend(is_debug: bool, opt_level: &str, out_dir: &PathBuf) -> (PathBuf
         )
         .define("TGI_TRTLLM_BACKEND_TRT_ROOT", tensorrt_path);
 
-    if is_debug {
+    if is_debug || *IS_GHA_BUILD {
         config.define("TGI_TRTLLM_BACKEND_BUILD_TESTS", "ON");
     }
 
@@ -209,7 +209,7 @@ fn build_ffi_layer(deps_folder: &PathBuf, is_debug: bool) {
 fn main() {
     // Misc variables
     let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
-    let build_profile = env::var("PROFILE").unwrap();
+    let build_profile = env::var("DEBUG").unwrap();
     let (is_debug, opt_level) = match build_profile.as_ref() {
         "debug" => (true, "0"),
         "dev" => (true, "0"),