diff --git a/candle-flash-attn/build.rs b/candle-flash-attn/build.rs
index d52ab92f..773c5638 100644
--- a/candle-flash-attn/build.rs
+++ b/candle-flash-attn/build.rs
@@ -88,6 +88,7 @@ fn main() -> Result<()> {
             .map(|(cu_file, obj_file)| {
                 let mut command = std::process::Command::new("nvcc");
                 command
+                    .arg("-std=c++17")
                     .arg(format!("--gpu-architecture=sm_{compute_cap}"))
                     .arg("-c")
                     .args(["-o", obj_file.to_str().unwrap()])