From b0bbef0e89aed5edff5a2151e6578c251211f334 Mon Sep 17 00:00:00 2001 From: Nabil Freij Date: Tue, 2 Jul 2024 17:13:36 -0700 Subject: [PATCH] Tweak for Arm Mac --- pyflct/setup_package.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/pyflct/setup_package.py b/pyflct/setup_package.py index 2195568..31b8bcd 100644 --- a/pyflct/setup_package.py +++ b/pyflct/setup_package.py @@ -1,5 +1,6 @@ import os import sys +import platform import subprocess from glob import glob from collections import defaultdict @@ -27,14 +28,17 @@ def get_extensions(): else: cfg["libraries"].append("m") # WE assume we have brew installed on Mac OS to provide FFTW3 - if sys.platform.lower() == "darwin": + if platform.system().lower() == "darwin": brew_path = ( subprocess.run(["brew", "--prefix"], stdout=subprocess.PIPE).stdout.decode("utf-8").replace("\n", "") ) cfg["include_dirs"].append(f"{brew_path}/include") - if not (sys.platform.lower() == "darwin" and sys.processor().lower() == "arm"): + if platform.system().lower() == "darwin" and platform.processor().lower() == "arm": + os.environ["DYLD_LIBRARY_PATH"] = os.environ.get("DYLD_LIBRARY_PATH", []) + [f"{brew_path}/lib"] + else: # This does not get registered on ARM Mac via Brew cfg["libraries"].append("fftw3") cfg["include_dirs"].append("/usr/include") + cfg["include_dirs"].append("/usr/local/include") cfg["extra_compile_args"].extend(["-O3", "-w", "-fomit-frame-pointer", "-fPIC"]) return [Extension("pyflct._flct", **cfg)]