-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
35 lines (29 loc) · 762 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import platform
from setuptools import Extension, setup
from Cython.Build import build_ext
if platform.system() == "Windows":
compile_args = ["/openmp"]
elif platform.system() == "Darwin":
compile_args = ["-Xpreprocessor", "-fopenmp"]
else:
compile_args = ["-fopenmp", "-std=c++11"]
ext_modules = [
Extension(
"fastcore._sim",
["fastcore/_sim.pyx"],
extra_compile_args=compile_args,
extra_link_args=compile_args,
),
Extension(
"fastcore._dag",
["fastcore/_dag.pyx"],
extra_compile_args=compile_args,
extra_link_args=compile_args,
)
]
setup(
name='fastcore',
cmdclass={'build_ext': build_ext},
install_requires=["numpy"],
ext_modules=ext_modules,
)