From ebcfd96d9444968a164556ceaae9d215d6336b40 Mon Sep 17 00:00:00 2001
From: Chengxu Yang <yangchengxu@bytedance.com>
Date: Tue, 15 Aug 2023 22:29:34 +0800
Subject: [PATCH] add c++17 flags (#452)

---
 candle-flash-attn/build.rs | 1 +
 1 file changed, 1 insertion(+)

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()])