Update setup.py

MooncellWiki · Sep 10, 2024 · 86a1b6d · 86a1b6d
1 parent 34604b4
commit 86a1b6d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -52,11 +52,12 @@ def build_extensions(self):
                     native_arg = "-mcpu=apple-m1"
                 else:
                     native_arg = "-mcpu=native"
-            else:
-                native_arg = "-march=native"
 
             for ext in self.extensions:
-                ext.extra_compile_args.extend(["-std=c++20", "-mno-avx2", native_arg])
+                if native_arg:
+                    ext.extra_compile_args.extend(["-std=c++20", native_arg])
+                else:
+                    ext.extra_compile_args.extend(["-std=c++20"])
 
         super().build_extensions()