category2vec/setup.py at master · rakuten-nlp/category2vec · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from distutils.core import setup, Extension
import distutils.ccompiler
from Cython.Build import cythonize
import numpy as np
import os
import sys
import settings

def readSettings():
    macros = [('MAX_SENTENCE_LEN', settings.max_sentence_len),
              ('EXP_TABLE_SIZE',settings.exp_table_size),
              ('MAX_EXP',settings.max_exp),
              ('MAX_DIMENSION',settings.max_dimension),
              ('EPSILON',settings.epsilon),
              ('EPSILON_D',settings.epsilon_d),
              ('RHO',settings.rho),
              ('ADAM_L',settings.adam_l),
              ('ADAM_B1',settings.adam_b1),
              ('ADAM_B2',settings.adam_b2),
              ('ADAM_A',settings.adam_a)
    ]
    ext_inc = []
    ext_libs = []
    ext_c_comp_args = []
    ext_cpp_comp_args = []
    if settings.use_double:
        macros.append(('USE_DOUBLE', 1))
    if settings.use_blas:
        macros.append(('USE_BLAS', 1))
        ext_inc.extend(settings.blas_include)
        ext_libs.extend(settings.blas_libs)
    if settings.fast_sqrt:
        macros.append(('FAST_SQRT', 1))
    if settings.use_avx:
        macros.append(('USE_AVX', 1))
        ext_c_comp_args.append("-mavx")
        ext_cpp_comp_args.append("-mavx")
    if not settings.use_clang:
        ext_cpp_comp_args.append("-std=c++0x")
    if settings.force_gcc:
        os.environ["CC"] = "gcc"
        os.environ["CXX"] = "g++"

    return macros, ext_inc, ext_libs, ext_c_comp_args, ext_cpp_comp_args

model_dir = os.path.dirname(__file__) or os.getcwd()
includes = [model_dir, np.get_include()]
c_comp_args = []
cpp_comp_args = []

macros, ext_inc, ext_libs, ext_c_comp_args, ext_cpp_comp_args = readSettings()
includes.extend(ext_inc)
libs = ext_libs
c_comp_args.extend(ext_c_comp_args)
cpp_comp_args.extend(ext_cpp_comp_args)

extensions = [
    Extension(
        "word2vec_inner",
        define_macros = macros,
        sources=["word2vec_inner.pyx", "options.c"],
        include_dirs = includes,
        libraries=libs,
        extra_compile_args=c_comp_args,
    ),
    Extension(
        "cat2vec_bind",
        define_macros = macros,
        sources=["cat2vec_bind.pyx", "cat2vec_calc.cpp", "options.c"],
        include_dirs = includes,
        libraries=libs,
        language="c++",
        extra_compile_args=cpp_comp_args,
    ),
    Extension(
        "sent2vec_bind",
        define_macros = macros,
        sources=["sent2vec_bind.pyx", "sent2vec_calc.cpp", "options.c"],
        include_dirs = includes,
        libraries=libs,
        language="c++",
        extra_compile_args=cpp_comp_args,
    ),
]

setup(
    ext_modules = cythonize(extensions),
 )