backend(trtllm): link against CUDA 12.8

huggingface · Feb 6, 2025 · 393a7ef · 393a7ef
1 parent d5fc057
commit 393a7ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backends/trtllm/build.rs b/backends/trtllm/build.rs
@@ -7,7 +7,7 @@ use std::sync::LazyLock;
 
 const ADDITIONAL_BACKEND_LINK_LIBRARIES: [&str; 1] = ["spdlog"];
 const CUDA_ARCH_LIST: Option<&str> = option_env!("CUDA_ARCH_LIST");
-const CUDA_REQUIRED_VERSION: &str = "12.6";
+const CUDA_REQUIRED_VERSION: &str = "12.8";
 const MPI_REQUIRED_VERSION: &str = "4.1";
 const INSTALL_PREFIX: Option<&str> = option_env!("CMAKE_INSTALL_PREFIX");
 const TENSORRT_ROOT_DIR: Option<&str> = option_env!("TENSORRT_ROOT_DIR");