diff options
Diffstat (limited to 'candle-kernels/build.rs')
| -rw-r--r-- | candle-kernels/build.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-kernels/build.rs b/candle-kernels/build.rs index 3c8e96a9..58518412 100644 --- a/candle-kernels/build.rs +++ b/candle-kernels/build.rs @@ -184,6 +184,7 @@ mod cuda { let mut command = std::process::Command::new("nvcc"); command.arg(format!("--gpu-architecture=sm_{compute_cap}")) .arg("--ptx") + .arg("--expt-relaxed-constexpr") .args(["--default-stream", "per-thread"]) .args(["--output-directory", &out_dir]) // Flash attention only |