f

Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>

Author: joseph-isaacs

vortex/build.rs

@@ -0,0 +1,30 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: Copyright the Vortex contributors
+
+use std::process::Command;
+
+fn main() {
+    // Declare the cfg so rustc doesn't warn about unexpected cfg.
+    println!("cargo::rustc-check-cfg=cfg(cuda_available)");
+
+    // Only enable CUDA on Linux (matching vortex-cuda's behavior)
+    if cfg!(not(target_os = "linux")) {
+        return;
+    }
+
+    // Check if nvcc is available
+    if !cuda_available() {
+        return;
+    }
+
+    println!("cargo:rustc-cfg=cuda_available");
+
+    // Set rpath for nvCOMP dylib (re-exported by vortex-cuda).
+    #[expect(clippy::expect_used)]
+    let nvcomp_lib = std::env::var("DEP_VORTEX_CUDA_LIB_DIR").expect("must have nvcomp");
+    println!("cargo:rustc-link-arg=-Wl,-rpath,{nvcomp_lib}");
+}
+
+fn cuda_available() -> bool {
+    Command::new("nvcc").arg("--version").output().is_ok()
+}