huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix candle-flash-attn build on Windows (msvc) (#2734)

Browse Source
This commit is contained in:
Sergei Grebnov
2025-01-22 13:19:48 -08:00
committed by GitHub
parent 77db8396d0
commit e6cd499e98
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-flash-attn/build.rs
View File

@ -88,6 +88,12 @@ fn main() -> Result<()> {
.arg("--use_fast_math") .arg("--use_fast_math")
.arg("--verbose"); .arg("--verbose");
if let Ok(target) = std::env::var("TARGET") {
if target.contains("msvc") {
builder = builder.arg("-D_USE_MATH_DEFINES");
}
}
let out_file = build_dir.join("libflashattention.a"); let out_file = build_dir.join("libflashattention.a");
builder.build_lib(out_file); builder.build_lib(out_file);