From b6cfa0fbc07cc2b0d5e97490438757da42addf8e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adrien=20Gallou=C3=ABt?= <angt@huggingface.co>
Date: Fri, 7 Feb 2025 11:48:16 +0000
Subject: [PATCH] Add missing cuda prefix
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Adrien Gallouët <angt@huggingface.co>
---
 backends/llamacpp/build.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/backends/llamacpp/build.rs b/backends/llamacpp/build.rs
index b554694b9b2..19b1987d947 100644
--- a/backends/llamacpp/build.rs
+++ b/backends/llamacpp/build.rs
@@ -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();