159e378cbb
GitOrigin-RevId: c04d5652cfa9742b1d519688f65d1bbccea9eb7e
27 lines
1.1 KiB
Diff
27 lines
1.1 KiB
Diff
diff --git a/setup.py b/setup.py
|
|
index c6fc019..49fb3c8 100644
|
|
--- a/setup.py
|
|
+++ b/setup.py
|
|
@@ -14,11 +14,12 @@ machine_arch = os.popen("uname -m").readlines()[0].rsplit()[0]
|
|
|
|
if machine_system == "Darwin":
|
|
# note: see https://github.com/pypa/wheel/issues/406
|
|
- simd_flag = ['-march=native', '-D__AVX2__', '-mmacosx-version-min=10.9']
|
|
if machine_arch in ["aarch64", "arm64"]:
|
|
+ simd_flag = ['-march=armv8-a+simd', '-D__AVX2__', '-mmacosx-version-min=10.9']
|
|
os.environ['_PYTHON_HOST_PLATFORM'] = "macosx-10.9-arm64"
|
|
os.environ['ARCHFLAGS'] = "-arch arm64"
|
|
else: # x86_64
|
|
+ simd_flag = ['-msse2', '-mmacosx-version-min=10.9']
|
|
os.environ['_PYTHON_HOST_PLATFORM'] = "macosx-10.9-x86_64"
|
|
os.environ['ARCHFLAGS'] = "-arch x86_64"
|
|
else:
|
|
@@ -27,7 +28,7 @@ else:
|
|
elif machine_arch in ["aarch32"]:
|
|
simd_flag = ['-march=armv8-a+simd', '-mfpu=auto -D__AVX2__']
|
|
else: # x86_64
|
|
- simd_flag=['-march=native']
|
|
+ simd_flag=[]
|
|
if os.getenv('SSE4', False):
|
|
simd_flag=['-msse4.1']
|
|
elif os.getenv('SSE2', False):
|