From 9b1b011bf6fa369bebe9e8fbab4049451b75b931 Mon Sep 17 00:00:00 2001
From: Eric Engelhart <bob80333@gmail.com>
Date: Mon, 3 Oct 2022 13:43:51 -0400
Subject: [PATCH] Add C++17 arg to compiler, since C++17 features are used,
 fixes windows build

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index eabcf06..ec1415a 100644
--- a/setup.py
+++ b/setup.py
@@ -125,10 +125,11 @@ ext_modules.append(
             "csrc/flash_attn/src/fmha_block_dgrad_fp16_kernel_loop.sm80.cu",
         ],
         extra_compile_args={
-            "cxx": ["-O3"] + generator_flag,
+            "cxx": ["-O3", "-std=c++17"] + generator_flag,
             "nvcc": append_nvcc_threads(
                 [
                     "-O3",
+                    "-std=c++17",
                     "-U__CUDA_NO_HALF_OPERATORS__",
                     "-U__CUDA_NO_HALF_CONVERSIONS__",
                     "--expt-relaxed-constexpr",