Handle CUDA_NVCC_FLAGS in flash attn v3

EricLBuehler · Feb 25, 2025 · 8d3ea29 · 8d3ea29
1 parent 3ecc604
commit 8d3ea29
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/candle-flash-attn-v3/build.rs b/candle-flash-attn-v3/build.rs
@@ -212,8 +212,8 @@ fn main() -> Result<()> {
 
                 // https://github.com/EricLBuehler/mistral.rs/issues/286
                 if let Some(cuda_nvcc_flags_env) = CUDA_NVCC_FLAGS {
-                    command = command.arg("--compiler-options");
-                    command = command.arg(cuda_nvcc_flags_env);
+                    command.arg("--compiler-options");
+                    command.arg(cuda_nvcc_flags_env);
                 }
 
                 let output = command