depot/third_party/nixpkgs/pkgs/by-name/ab/abpoa/simd-arch.patch
Default email 159e378cbb Project import generated by Copybara.
GitOrigin-RevId: c04d5652cfa9742b1d519688f65d1bbccea9eb7e
2024-09-19 17:19:46 +03:00

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):