From fcbf2fc1ac32e52d2d4f8fec0d5dee658fef1cf6 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <morgan@huggingface.co>
Date: Thu, 25 Jul 2024 10:36:55 +0000
Subject: [PATCH] fix envvar CARGO_CFG_TARGET_ARCH set at runtime vs compile
 time

---
 backends/trtllm/build.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backends/trtllm/build.rs b/backends/trtllm/build.rs
index c8890394..c698c825 100644
--- a/backends/trtllm/build.rs
+++ b/backends/trtllm/build.rs
@@ -128,7 +128,7 @@ fn main() {
     // NCCL is slightly trickier because it might not have a pkgconfig installed
     let nccl_library_path = NCCL_ROOT_DIR.unwrap_or(&format!(
         "/usr/local/{}-linux-gnu",
-        env!("CARGO_CFG_TARGET_ARCH")
+        env::var("CARGO_CFG_TARGET_ARCH").unwrap()
     ));
     println!(r"cargo:rustc-link-search=native={}", nccl_library_path);
     println!("cargo:rustc-link-lib=dylib=nccl");