zisk-asm-runner 1.1.0-alpha

Runtime for managing the ZisK assembly emulator process and its shared-memory I/O
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use std::path::Path;
use std::process::Command;

/// Detect CUDA presence using the same probe cascade as `mem-cpp/build.rs`:
/// nvcc on PATH, `/usr/local/cuda/bin/nvcc`, `/opt/cuda/bin/nvcc`. Returns
/// false on macOS or under `cpu-only`.
fn detect_gpu() -> bool {
    if cfg!(feature = "cpu-only") {
        return false;
    }
    if cfg!(target_os = "macos") {
        return false;
    }
    if Command::new("nvcc").arg("--version").output().map(|o| o.status.success()).unwrap_or(false) {
        return true;
    }
    for candidate in ["/usr/local/cuda/bin/nvcc", "/opt/cuda/bin/nvcc"] {
        if Path::new(candidate).exists() {
            return true;
        }
    }
    false
}

fn main() {
    if detect_gpu() {
        println!("cargo:rustc-cfg=gpu");
    }
    println!("cargo::rustc-check-cfg=cfg(gpu)");
}