-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
27 lines (24 loc) · 889 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
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
import os.path
# Adjust to point to your SRILM build directory
SRILM_DIR = "/opt/srilm"
# Adjust to match your architecture -- if unsure, build SRILM and then see
# what subdirectories you have in SRILM_DIR/lib/.
SRILM_ARCH = "i686-m64"
SRILM_INCLUDE_DIR = os.path.join(SRILM_DIR, "include")
SRILM_LIB_DIR = os.path.join(SRILM_DIR, "lib", SRILM_ARCH)
setup(
cmdclass = {'build_ext': build_ext},
ext_modules = [
Extension("srilm",
["srilm.pyx"],
language="c++",
include_dirs=[SRILM_INCLUDE_DIR],
libraries=["oolm", "dstruct", "misc"],
extra_compile_args=['-fopenmp', '-fPIC'],
extra_link_args=["-L" + SRILM_LIB_DIR, '-lgomp', '-lz'],
)
],
)