Skip to content

Commit 2bf7ba7

Browse files
authored
Fixed setup mac script (#278)
Fixed setup mac script
2 parents 9892068 + 18da101 commit 2bf7ba7

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

extensions/setup_mac.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,9 @@
33
import os
44
import re
55
import glob
6+
import sys
67

8+
import pybind11
79
from pybind11.setup_helpers import Pybind11Extension
810
from setuptools import find_packages, setup
911
from setuptools.command.build_ext import build_ext
@@ -19,10 +21,10 @@
1921
# extension modules
2022
ext_modules = [
2123
Pybind11Extension("brainpylib/cpu_ops",
22-
sources=["lib/cpu_ops.cc"] + glob.glob("lib/*_cpu.cc"),
24+
sources=glob.glob("lib/cpu_*.cc"),
2325
cxx_std=11,
24-
# extra_link_args=["-rpath", "/Users/ztqakita/miniforge3/lib"], # m1
25-
extra_link_args=["-rpath", "/Users/ztqakita/opt/miniconda3/lib"], # intel
26+
# extra_link_args=["-rpath", os.environ["CONDA_PREFIX"] + "/lib"],
27+
extra_link_args=["-rpath", re.sub('/lib/.*', '/lib', sys.path[1])],
2628
define_macros=[('VERSION_INFO', __version__)]),
2729
]
2830

0 commit comments

Comments
 (0)