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

huggingface · Jan 22, 2025 · e6cd499 · e6cd499
1 parent 77db839
commit e6cd499
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/candle-flash-attn/build.rs b/candle-flash-attn/build.rs
@@ -88,6 +88,12 @@ fn main() -> Result<()> {
         .arg("--use_fast_math")
         .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");
     builder.build_lib(out_file);