Add support for CUDA FFI.

zmwangx · Apr 11, 2024 · c684bec · c684bec
1 parent 3a00222
commit c684bec
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/build.rs b/build.rs
@@ -646,7 +646,7 @@ fn main() {
     let statik = env::var("CARGO_FEATURE_STATIC").is_ok();
     let ffmpeg_major_version: u32 = env!("CARGO_PKG_VERSION_MAJOR").parse().unwrap();
 
-    let include_paths: Vec<PathBuf> = if env::var("CARGO_FEATURE_BUILD").is_ok() {
+    let mut include_paths: Vec<PathBuf> = if env::var("CARGO_FEATURE_BUILD").is_ok() {
         println!(
             "cargo:rustc-link-search=native={}",
             search().join("lib").to_string_lossy()
@@ -766,6 +766,14 @@ fn main() {
             .include_paths
     };
 
+    include_paths.extend(
+        pkg_config::Config::new()
+            .statik(statik)
+            .probe("cuda")
+            .unwrap()
+            .include_paths
+    );
+
     if statik && cfg!(target_os = "macos") {
         let frameworks = vec![
             "AppKit",
@@ -1260,6 +1268,7 @@ fn main() {
         .header(search_include(&include_paths, "libavutil/hash.h"))
         .header(search_include(&include_paths, "libavutil/hmac.h"))
         .header(search_include(&include_paths, "libavutil/hwcontext.h"))
+        .header(search_include(&include_paths, "libavutil/hwcontext_cuda.h"))
         .header(search_include(&include_paths, "libavutil/imgutils.h"))
         .header(search_include(&include_paths, "libavutil/lfg.h"))
         .header(search_include(&include_paths, "libavutil/log.h"))