fix: setup platform check on Apple Silicon

iamDecode · Apr 25, 2023 · 83519d9 · 83519d9
1 parent c1e6e47
commit 83519d9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -44,9 +44,9 @@
 # Note that -O3 may sometimes cause mysterious problems, so we limit ourselves to -O2.
 
 # Modules involving numerical computations
-if platform.system() == 'Darwin' and platform.machine() == 'aarch64' or 'universal2' in os.environ.get('CIBW_ARCHS', ''):  # Apple Silicon
-  extra_compile_args_math_optimized = ['-O2']
-  extra_compile_args_math_debug = ['-O0', '-g']
+if platform.system() == 'Darwin' and platform.machine() == 'aarch64' or 'universal2' or 'arm64' in os.environ.get('CIBW_ARCHS', ''):  # Apple Silicon
+  extra_compile_args_math_optimized = ['-O2', '-march=native']
+  extra_compile_args_math_debug = ['-O0', '-march=native', '-g']
 else:
   extra_compile_args_math_optimized = ['-mtune=native', '-march=native', '-O2', '-msse', '-msse2', '-mfma', '-mfpmath=sse']
   extra_compile_args_math_debug = ['-mtune=native', '-march=native', '-O0', '-g']