Skip to content

Commit b6048f0

Browse files
committed
Use TARGET instead of HOST
1 parent 7e13673 commit b6048f0

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

llama-cpp-sys-2/build.rs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,12 +232,16 @@ fn main() {
232232
.to_string(),
233233
);
234234

235+
let bindgen_target = env::var("TARGET").or_else(|_| {
236+
env::var("HOST")
237+
}).expect("Failed to get TARGET or HOST environment variable");
238+
235239
// Bindings
236240
let bindings = bindgen::Builder::default()
237241
.header("wrapper.h")
238242
.clang_arg(format!("-I{}", llama_src.join("include").display()))
239243
.clang_arg(format!("-I{}", llama_src.join("ggml/include").display()))
240-
.clang_arg(format!("--target={}", env::var("HOST").unwrap()))
244+
.clang_arg(format!("--target={}", bindgen_target))
241245
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
242246
.derive_partialeq(true)
243247
.allowlist_function("ggml_.*")

0 commit comments

Comments
 (0)