Add c++17 to compilation flag

state-spaces · Dec 4, 2023 · 023c25d · 023c25d
1 parent 33dc96c
commit 023c25d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mamba_ssm/__init__.py b/mamba_ssm/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "1.0.0"
+__version__ = "1.0.1"
 
 from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
 from mamba_ssm.modules.mamba_simple import Mamba

diff --git a/setup.py b/setup.py
@@ -134,10 +134,11 @@ def append_nvcc_threads(nvcc_extra_args):
                 "csrc/selective_scan/selective_scan_bwd_bf16_complex.cu",
             ],
             extra_compile_args={
-                "cxx": ["-O3"],
+                "cxx": ["-O3", "-std=c++17"],
                 "nvcc": append_nvcc_threads(
                     [
                         "-O3",
+                        "-std=c++17",
                         "-U__CUDA_NO_HALF_OPERATORS__",
                         "-U__CUDA_NO_HALF_CONVERSIONS__",
                         "-U__CUDA_NO_BFLOAT16_OPERATORS__",