ghostflow-cuda 0.1.0

CUDA backend for GhostFlow ML framework
Documentation 
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
31
32
33
34
35
36
37
38

//! Build script for CUDA compilation

fn main() {
    println!("cargo:rerun-if-changed=src/optimized_kernels.cu");
    
    // Check if CUDA is available
    #[cfg(feature = "cuda")]
    {
        // Find CUDA toolkit
        let cuda_path = std::env::var("CUDA_PATH")
            .or_else(|_| std::env::var("CUDA_HOME"))
            .unwrap_or_else(|_| "/usr/local/cuda".to_string());
        
        println!("cargo:rustc-link-search=native={}/lib64", cuda_path);
        println!("cargo:rustc-link-lib=cudart");
        println!("cargo:rustc-link-lib=cublas");
        
        // Compile CUDA kernels using nvcc
        cc::Build::new()
            .cuda(true)
            .flag("-arch=sm_70") // Volta and newer
            .flag("-gencode=arch=compute_70,code=sm_70")
            .flag("-gencode=arch=compute_75,code=sm_75")
            .flag("-gencode=arch=compute_80,code=sm_80")
            .flag("-gencode=arch=compute_86,code=sm_86")
            .flag("--use_fast_math")
            .flag("-O3")
            .file("src/optimized_kernels.cu")
            .compile("ghostflow_cuda_kernels");
        
        println!("cargo:warning=Compiled CUDA kernels successfully");
    }
    
    #[cfg(not(feature = "cuda"))]
    {
        println!("cargo:warning=Building without CUDA support - GPU operations will use CPU fallback");
    }
}