Skip to content

Commit

Permalink
Add missing cuda prefix
Browse files Browse the repository at this point in the history
Signed-off-by: Adrien Gallouët <[email protected]>
  • Loading branch information
angt committed Feb 7, 2025
1 parent 4841f71 commit b6cfa0f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion backends/llamacpp/build.rs
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,8 @@ fn main() {
if version.len() > 2 {
version.pop();
}
pkg_config::Config::new().probe(&version.join(".")).unwrap();
let cuda_version = format!("cuda-{}", version.join("."));
pkg_config::Config::new().probe(&cuda_version).unwrap();
}
pkg_config::Config::new().probe("llama").unwrap();

Expand Down

0 comments on commit b6cfa0f

Please sign in to comment.