Skip to content

Commit

Permalink
Update setup.py
Browse files Browse the repository at this point in the history
  • Loading branch information
bab2min authored Aug 5, 2024
1 parent ba9540a commit eebb1a4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,14 +199,14 @@ def build_extension(self, ext):
arch_levels = ['none', 'sse2', 'avx2']
elif platform.system() == 'Darwin':
if tomotopy_cpu_arch == 'arm64':
arch_levels = ['aarch64']
arch_levels = ['arm64']
elif build_for_manyplatform:
arch_levels = ['sse2', 'avx', 'avx2']
else:
arch_levels = ['sse2']
elif build_for_manyplatform:
if tomotopy_cpu_arch in ('arm64', 'aarch64'):
arch_levels = ['aarch64']
arch_levels = ['arm64']
else:
arch_levels = ['native']

Expand Down Expand Up @@ -237,7 +237,7 @@ def build_extension(self, ext):

for arch in arch_levels:
if len(arch_levels) > 1:
if arch in ('none', 'native', 'aarch64'):
if arch in ('none', 'native', 'arm64'):
module_name = '_tomotopy_none'
else:
module_name = '_tomotopy_' + arch
Expand Down

0 comments on commit eebb1a4

Please sign in to comment.