-
Notifications
You must be signed in to change notification settings - Fork 28
/
build.py
43 lines (37 loc) · 1.11 KB
/
build.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
36
37
38
39
40
41
42
43
import os
# See if Cython is installed
try:
from Cython.Build import cythonize
# Do nothing if Cython is not available
except ImportError:
# Got to provide this function. Otherwise, poetry will fail
def build(setup_kwargs):
pass
# Cython is installed. Compile
else:
from setuptools import Extension
from setuptools.dist import Distribution
from distutils.command.build_ext import build_ext
# This function will be executed in setup.py:
def build(setup_kwargs):
# The file you want to compile
extensions = ["dingo/volestipy.pyx"]
# gcc arguments hack: enable optimizations
os.environ["CFLAGS"] = [
"-std=c++17",
"-O3",
"-DBOOST_NO_AUTO_PTR",
"-ldl",
"-lm",
]
# Build
setup_kwargs.update(
{
"ext_modules": cythonize(
extensions,
language_level=3,
compiler_directives={"linetrace": True},
),
"cmdclass": {"build_ext": build_ext},
}
)