Merge pull request #194 from bab2min/dev_neon

fixed a bug generating wrong arm64 binary
bab2min · Jan 22, 2023 · 933aeae · 933aeae
2 parents ec8b828 + 890ce21
commit 933aeae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -64,9 +64,11 @@
     if tomotopy_cpu_arch == 'arm64':
         arch_levels = {'': ''}
         cargs += ['-DTOMOTOPY_ISA=arm64', '-arch', 'arm64']
+        largs += ['-arch', 'arm64']
     else:
         if 'many' not in os.environ.get('AUDITWHEEL_PLAT', ''): arch_levels = {'':'-msse2'}
         cargs += ['-arch', 'x86_64']
+        largs += ['-arch', 'x86_64']
 elif 'many' in os.environ.get('AUDITWHEEL_PLAT', ''):
     cargs = ['-std=c++1y', '-O3', '-fpermissive', '-g0', '-Wno-unused-variable', '-Wno-switch']
 else: