From 30efd395259a69e4d4337b32c8c32066f925d234 Mon Sep 17 00:00:00 2001 From: Brandon Zhang Date: Sun, 16 Oct 2022 23:43:53 +0800 Subject: [PATCH] Update setup_mac.py --- extensions/setup_mac.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/extensions/setup_mac.py b/extensions/setup_mac.py index 1450ee46a..8bfc9c0a3 100644 --- a/extensions/setup_mac.py +++ b/extensions/setup_mac.py @@ -3,7 +3,9 @@ import os import re import glob +import sys +import pybind11 from pybind11.setup_helpers import Pybind11Extension from setuptools import find_packages, setup from setuptools.command.build_ext import build_ext @@ -19,10 +21,10 @@ # extension modules ext_modules = [ Pybind11Extension("brainpylib/cpu_ops", - sources=["lib/cpu_ops.cc"] + glob.glob("lib/*_cpu.cc"), + sources=glob.glob("lib/cpu_*.cc"), cxx_std=11, - # extra_link_args=["-rpath", "/Users/ztqakita/miniforge3/lib"], # m1 - extra_link_args=["-rpath", "/Users/ztqakita/opt/miniconda3/lib"], # intel + # extra_link_args=["-rpath", os.environ["CONDA_PREFIX"] + "/lib"], + extra_link_args=["-rpath", re.sub('/lib/.*', '/lib', sys.path[1])], define_macros=[('VERSION_INFO', __version__)]), ]