build.rs

fn main() {
    #[cfg(feature = "cuda")]
    println!("cargo::rustc-link-search=native=/usr/lib/x86_64-linux-gnu/");
    #[cfg(feature = "cuda")]
    println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
    #[cfg(feature = "cuda")]
    println!("cargo:rustc-link-lib=cudart");
}