-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
33 lines (27 loc) · 1.02 KB
/
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
#!/usr/bin/env python3
import os
import platform
from Cython.Build import cythonize
from distutils.extension import Extension
from distutils.core import setup
import numpy
source_files = ["src/matching.pyx", "src/helper.c"]
if platform.system() == "Linux":
exts = Extension(name='matching',
sources=source_files,
include_dirs=[numpy.get_include()],
extra_compile_args=['-fopenmp', "-O3",
],
extra_link_args=['-fopenmp'],
language="c",)
elif platform.system() == "Darwin":
os.environ['CC'] = 'gcc-13'
exts = Extension(name='matching',
sources=source_files,
include_dirs=[numpy.get_include()],
extra_compile_args=['-fopenmp', "-O3",
],
extra_link_args=['-fopenmp'],
language="c",)
setup(name='matching',
ext_modules=cythonize(exts))