Skip to content

Commit 055b6cd

Browse files
committed
update
1 parent 7039f63 commit 055b6cd

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

setup.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -61,9 +61,11 @@ def get_extensions():
6161
print('Compiling without OpenMP...')
6262

6363
# Compile for mac arm64
64-
if (sys.platform == 'darwin' and platform.machine() == 'arm64'):
65-
extra_compile_args['cxx'] += ['-arch', 'arm64']
66-
extra_link_args += ['-arch', 'arm64']
64+
if sys.platform == 'darwin':
65+
extra_compile_args['cxx'] += ['-D_LIBCPP_DISABLE_AVAILABILITY']
66+
if platform.machine == 'arm64':
67+
extra_compile_args['cxx'] += ['-arch', 'arm64']
68+
extra_link_args += ['-arch', 'arm64']
6769

6870
if suffix == 'cuda':
6971
define_macros += [('WITH_CUDA', None)]

0 commit comments

Comments
 (0)