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

9b1b011b · Eric Engelhart · 9d7fd5b6 · 9b1b011b
Commit 9b1b011b authored Oct 03, 2022 by Eric Engelhart
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

setup.py setup.py +2 -1

No files found.
--- 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",