update setup.py

yahoojapan · Jan 28, 2025 · 81580e5 · 81580e5
1 parent 918c9b6
commit 81580e5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -114,15 +114,19 @@
             'include_dirs': ['/usr/local/include',
                              pybind11.get_include(True),
                              pybind11.get_include(False)],
-            'extra_compile_args': ['-std=c++11', '-Ofast', '-march=x86-64', '-DNDEBUG'],
+            'extra_compile_args': ['-std=c++11', '-Ofast',
+                                   '-march=x86_64' if platform.processor() == 'x86-64' else '',
+                                   '-DNDEBUG'],
             'sources': ['src/ngtpy.cpp']
         }
     elif static_library_avx2 or shared_library_avx2:
         params = {
             'include_dirs': ['/usr/local/include',
                              pybind11.get_include(True),
                              pybind11.get_include(False)],
-            'extra_compile_args': ['-std=c++11', '-Ofast', '-march=haswell', '-DNDEBUG'],
+            'extra_compile_args': ['-std=c++11', '-Ofast',
+                                   '-march=haswell' if platform.processor() == 'x86-64' else '',
+                                   '-DNDEBUG'],
             'sources': ['src/ngtpy_avx2.cpp']
         }
     else: