Skip to content

Commit

Permalink
Move CUDA_NVCC_FLAGS to last
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Feb 25, 2025
1 parent dd81c1a commit 45c9163
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions candle-flash-mla/build.rs
Original file line number Diff line number Diff line change
Expand Up @@ -148,15 +148,15 @@ fn main() -> Result<()> {
command.args(["-ccbin", ccbin_path]);
}

// Add the source file
command.arg(input);

// https://github.com/EricLBuehler/mistral.rs/issues/286
if let Some(cuda_nvcc_flags_env) = CUDA_NVCC_FLAGS {
command.arg("--compiler-options");
command.arg(cuda_nvcc_flags_env);
}

// Add the source file
command.arg(input);

let output = command
.spawn()
.with_context(|| format!("Failed to spawn nvcc for {input:?}"))?
Expand Down

0 comments on commit 45c9163

Please sign in to comment.