Add C++17 arg to compiler, since C++17 features are used, fixes windows build

2022-10-03 13:43:51 -04:00 · 2022-10-03 13:43:51 -04:00 · 9b1b011bf6
commit 9b1b011bf6
parent 9d7fd5b6e7
1 changed files with 2 additions and 1 deletions
--- 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",