|
5 | 5 | import numpy |
6 | 6 | numpy_path = os.path.join(numpy.__path__[0], 'core', 'include') |
7 | 7 | # use additional compiler flags: "-ffast-math" "-g" |
| 8 | + |
| 9 | +# FIXME proper win/unix lib/includes. |
| 10 | +from distutils.sysconfig import get_python_inc |
| 11 | +incdir = os.path.join(get_python_inc(plat_specific=1)) |
| 12 | +#unix_include_path = "/usr/include" |
| 13 | +#win_include_path = "C:\Python26\include" |
| 14 | +#include_path = win_include_path |
| 15 | + |
| 16 | +win_lib_path = "C:\Python26\PCbuild" |
| 17 | +unix_lib_path = "/usr/lib" |
| 18 | +lib_path = win_lib_path |
| 19 | + |
8 | 20 | setup( |
9 | 21 | name = "bolt", |
10 | 22 | license = "MIT", |
11 | | - ext_modules = [Extension("bolt/trainer/sgd", ["bolt/trainer/sgd.c"], |
12 | | - include_dirs=['/usr/include/',numpy_path], |
| 23 | + ext_modules = [ |
| 24 | + Extension("bolt.trainer.sgd", ["bolt/trainer/sgd.c"], |
| 25 | + include_dirs=[incdir,numpy_path], |
13 | 26 | extra_link_args=["-O3"], |
14 | | - library_dirs=['/usr/lib',], |
15 | | - extra_compile_args=["-O3","-g"],), |
16 | | - Extension("bolt/trainer/avgperceptron", ["bolt/trainer/avgperceptron.c"], |
17 | | - include_dirs=['/usr/include/',numpy_path], |
| 27 | + library_dirs=[lib_path,], |
| 28 | + extra_compile_args=["-O3","-g"] |
| 29 | + ), |
| 30 | + Extension("bolt.trainer.avgperceptron", ["bolt/trainer/avgperceptron.c"], |
| 31 | + include_dirs=[incdir,numpy_path], |
18 | 32 | extra_link_args=["-O3"], |
19 | | - library_dirs=['/usr/lib',],#'/usr/local/lib',], |
20 | | - extra_compile_args=["-O3","-g"],), |
21 | | - Extension("bolt/trainer/maxent", ["bolt/trainer/maxent.c"], |
22 | | - include_dirs=['/usr/include/',numpy_path], |
| 33 | + library_dirs=[lib_path,], |
| 34 | + extra_compile_args=["-O3","-g"] |
| 35 | + ), |
| 36 | + Extension("bolt.trainer.maxent", ["bolt/trainer/maxent.c"], |
| 37 | + include_dirs=[incdir,numpy_path], |
23 | 38 | extra_link_args=["-O3"], |
24 | | - library_dirs=['/usr/lib',],#'/usr/local/lib',], |
25 | | - extra_compile_args=["-O3","-g"],), |
| 39 | + library_dirs=[lib_path,], |
| 40 | + extra_compile_args=["-O3","-g"] |
| 41 | + ), |
26 | 42 | ], |
27 | | - version = "1.2", |
| 43 | + version = "1.4", |
28 | 44 | description="Bolt Online Learning Toolbox", |
29 | 45 | author='Peter Prettenhofer', |
30 | 46 | |
|
36 | 52 | problems encountered in natural language processing |
37 | 53 | and information retrieval. """ |
38 | 54 | ) |
| 55 | + |
| 56 | + |
0 commit comments