diff options
| author | emka <mitja@kleider.name> | 2023-12-03 17:01:16 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-03 16:01:16 +0000 |
| commit | 8418154ee0cade8b9aebc22750c7717eb273b65d (patch) | |
| tree | e0e6af789e8e42488a886b85f63c009569bb6032 /candle-examples | |
| parent | 99b7273b0322b06d2dc3f9408312d39cc407c56a (diff) | |
| download | candle-8418154ee0cade8b9aebc22750c7717eb273b65d.tar.gz candle-8418154ee0cade8b9aebc22750c7717eb273b65d.tar.bz2 candle-8418154ee0cade8b9aebc22750c7717eb273b65d.zip | |
Add nvcc ccbin support to examples (#1401)
Diffstat (limited to 'candle-examples')
| -rw-r--r-- | candle-examples/build.rs | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/candle-examples/build.rs b/candle-examples/build.rs index b3c17cc3..0af3a6a4 100644 --- a/candle-examples/build.rs +++ b/candle-examples/build.rs @@ -32,6 +32,8 @@ impl KernelDirectories { if should_compile { #[cfg(feature = "cuda")] { + let ccbin_env = std::env::var("CANDLE_NVCC_CCBIN"); + println!("cargo:rerun-if-env-changed=CANDLE_NVCC_CCBIN"); let mut command = std::process::Command::new("nvcc"); let out_dir = ptx_file.parent().context("no parent for ptx file")?; let include_dirs: Vec<String> = @@ -44,6 +46,11 @@ impl KernelDirectories { .arg(format!("-I/{}", self.kernel_dir)) .args(include_dirs) .arg(cu_file); + if let Ok(ccbin_path) = &ccbin_env { + command + .arg("-allow-unsupported-compiler") + .args(["-ccbin", ccbin_path]); + } let output = command .spawn() .context("failed spawning nvcc")? |