huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-19 15:52:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

use std::env::const::ARCH

Browse Source
This commit is contained in:
Morgan Funtowicz 2024-07-25 10:44:42 +00:00
parent fcbf2fc1ac
commit 69a5804e51
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backends/trtllm/build.rs
View File

@ -1,4 +1,5 @@
use std::env; use std::env;
use std::env::consts::ARCH;
use std::path::{absolute, PathBuf}; use std::path::{absolute, PathBuf};
use cxx_build::CFG; use cxx_build::CFG;
@ -126,10 +127,7 @@ fn main() {
}); });
// NCCL is slightly trickier because it might not have a pkgconfig installed // NCCL is slightly trickier because it might not have a pkgconfig installed
let nccl_library_path = NCCL_ROOT_DIR.unwrap_or(&format!( let nccl_library_path = NCCL_ROOT_DIR.unwrap_or(&format!("/usr/local/{}-linux-gnu", ARCH));
"/usr/local/{}-linux-gnu",
env::var("CARGO_CFG_TARGET_ARCH").unwrap()
));
println!(r"cargo:rustc-link-search=native={}", nccl_library_path); println!(r"cargo:rustc-link-search=native={}", nccl_library_path);
println!("cargo:rustc-link-lib=dylib=nccl"); println!("cargo:rustc-link-lib=dylib=nccl");