red-candle 1.0.0.pre.3 → 1.0.0.pre.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 375ffec5c2293170bdb0e78ffe2584c8a963968d3226b601d12ddcc040c256ec
-  data.tar.gz: ea2490c59eca2480728afb7169fc3a24f4270e63491648c1d09fd111707dcb7e
+  metadata.gz: 53283961925b76f6fdc39634e01d3d08f58805584488053f7e82fbd484b96b21
+  data.tar.gz: 91d3dae0c3c1f6686980708fc898132178454712fe9adf2a79d59f75038c21f1
 SHA512:
-  metadata.gz: 9011c72cbe2a065917f26774862bb1f82489e6d5354376cb616df7cea18646ca60ccd5e68282f39b23f89eead196ab41ce70cdaaca224f97aef61914559f653f
-  data.tar.gz: 4a48101bdcffeecd1bc687dacc7eee7dc42ddbb417b05cce57b97abad83cefe90d965be5c0a8256b8f5c9ba90d80f1159c5392097558acf93a3095913a68a2ad
+  metadata.gz: 809ab126883fc6a0b706f46b34922ccbbb5eb18b6b153b703404d17bdd2626a0bafe5cebbdd9168389c44f7506b4f0dabd2b9ed7a6ad6348b7406a899e3e5484
+  data.tar.gz: 1d908b031b207bf43e65d81c3f8ac34cd1451cdd38d32476a30e83c23b34c6148a2aa7e235b804ad47041b0d781b1ca36585c2c83b0fa7cd1e9ba26d7cbfc377

data/ext/candle/build.rs

@@ -0,0 +1,116 @@
+use std::env;
+use std::path::Path;
+
+fn main() {
+    // Register our custom cfg flags with rustc
+    println!("cargo::rustc-check-cfg=cfg(force_cpu)");
+    println!("cargo::rustc-check-cfg=cfg(has_cuda)");
+    println!("cargo::rustc-check-cfg=cfg(has_metal)");
+    println!("cargo::rustc-check-cfg=cfg(has_mkl)");
+    println!("cargo::rustc-check-cfg=cfg(has_accelerate)");
+    
+    println!("cargo:rerun-if-changed=build.rs");
+    println!("cargo:rerun-if-env-changed=CANDLE_FORCE_CPU");
+    println!("cargo:rerun-if-env-changed=CANDLE_CUDA_PATH");
+    println!("cargo:rerun-if-env-changed=CUDA_ROOT");
+    println!("cargo:rerun-if-env-changed=CUDA_PATH");
+    println!("cargo:rerun-if-env-changed=CANDLE_FEATURES");
+    println!("cargo:rerun-if-env-changed=CANDLE_ENABLE_CUDA");
+
+    // Check if we should force CPU only
+    if env::var("CANDLE_FORCE_CPU").is_ok() {
+        println!("cargo:rustc-cfg=force_cpu");
+        println!("cargo:warning=CANDLE_FORCE_CPU is set, disabling all acceleration");
+        return;
+    }
+
+    // Detect CUDA availability
+    let cuda_available = detect_cuda();
+    let cuda_enabled = env::var("CANDLE_ENABLE_CUDA").is_ok();
+    
+    if cuda_available && cuda_enabled {
+        println!("cargo:rustc-cfg=has_cuda");
+        println!("cargo:warning=CUDA detected and enabled via CANDLE_ENABLE_CUDA");
+    } else if cuda_available && !cuda_enabled {
+        println!("cargo:warning=CUDA detected but not enabled. To enable CUDA support (coming soon), set CANDLE_ENABLE_CUDA=1");
+    }
+
+    // Detect Metal availability (macOS only)
+    #[cfg(target_os = "macos")]
+    {
+        println!("cargo:rustc-cfg=has_metal");
+        println!("cargo:warning=Metal detected (macOS), Metal acceleration will be available");
+    }
+
+    // Detect MKL availability
+    if detect_mkl() {
+        println!("cargo:rustc-cfg=has_mkl");
+        println!("cargo:warning=Intel MKL detected, MKL acceleration will be available");
+    }
+
+    // Detect Accelerate framework (macOS)
+    #[cfg(target_os = "macos")]
+    {
+        println!("cargo:rustc-cfg=has_accelerate");
+        println!("cargo:warning=Accelerate framework detected (macOS)");
+    }
+}
+
+fn detect_cuda() -> bool {
+    // Check environment variables first
+    if env::var("CANDLE_CUDA_PATH").is_ok() {
+        return true;
+    }
+
+    if env::var("CUDA_ROOT").is_ok() || env::var("CUDA_PATH").is_ok() {
+        return true;
+    }
+
+    // Check common CUDA installation paths
+    let cuda_paths = [
+        "/usr/local/cuda",
+        "/opt/cuda",
+        "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA",
+        "C:\\CUDA",
+    ];
+
+    for path in &cuda_paths {
+        if Path::new(path).exists() {
+            return true;
+        }
+    }
+
+    // Check if nvcc is in PATH
+    if let Ok(path_var) = env::var("PATH") {
+        for path in env::split_paths(&path_var) {
+            if path.join("nvcc").exists() || path.join("nvcc.exe").exists() {
+                return true;
+            }
+        }
+    }
+
+    false
+}
+
+fn detect_mkl() -> bool {
+    // Check environment variables
+    if env::var("MKLROOT").is_ok() || env::var("MKL_ROOT").is_ok() {
+        return true;
+    }
+
+    // Check common MKL installation paths
+    let mkl_paths = [
+        "/opt/intel/mkl",
+        "/opt/intel/oneapi/mkl/latest",
+        "C:\\Program Files (x86)\\Intel\\oneAPI\\mkl\\latest",
+        "C:\\Program Files\\Intel\\oneAPI\\mkl\\latest",
+    ];
+
+    for path in &mkl_paths {
+        if Path::new(path).exists() {
+            return true;
+        }
+    }
+
+    false
+}

data/lib/candle/version.rb

@@ -1,3 +1,3 @@
 module Candle
-  VERSION = "1.0.0.pre.3"
+  VERSION = "1.0.0.pre.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: red-candle
 version: !ruby/object:Gem::Version
-  version: 1.0.0.pre.3
+  version: 1.0.0.pre.4
 platform: ruby
 authors:
 - Christopher Petersen
@@ -43,6 +43,7 @@ files:
 - bin/console
 - bin/setup
 - ext/candle/Cargo.toml
+- ext/candle/build.rs
 - ext/candle/extconf.rb
 - ext/candle/rustfmt.toml
 - ext/candle/src/lib.rs