dpmix/setup.py at master · andrewcron/dpmix · 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
"""
Created on Mar 15, 2012

@author: Jacob Frelinger
"""

from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
from numpy import get_include
from cyarma import include_dir as arma_dir
from cyrand import include_dir as rng_dir

setup(
    name='dpmix',
    version='0.4',
    packages=['dpmix'],
    package_dir={'dpmix': 'src'},
    description='Optimized (and GPU enhanced) fitting of Gaussian Mixture Models',
    maintainer='Jacob Frelinger',
    maintainer_email='jacob.frelinger@duke.edu',
    author='Andrew Cron',
    author_email='andrew.cron@duke.edu',
    url='https://github.com/andrewcron/pycdp',
    requires=[
        'numpy (>=1.6)',
        'scipy (>=0.6)',
        'matplotlib (>=1.0)',
        'cython (>=0.17)',
        'cyarma (==0.2)',
        'cyrand (>=0.2)',
        'mpi4py'
    ],
    package_data={'dpmix': ['cufiles/*.cu']},
    cmdclass={'build_ext': build_ext},
    ext_modules=[
        Extension(
            "dpmix.munkres",
            ["src/munkres.pyx", "src/cpp/Munkres.cpp"],
            include_dirs=[get_include(), 'src/cpp', 'cpp/'],
            language='c++'
        ),
        Extension(
            "dpmix.sampler",
            ["src/sampler_utils.pyx"],
            include_dirs=[
                get_include(),
                arma_dir,
                rng_dir,
                '/usr/include',
                '/usr/local/include',
                '/opt/local/include'
            ],
            library_dirs=['/usr/lib', '/usr/local/lib', '/opt/local/lib'],
            libraries=['armadillo'],
            language='c++',
            extra_compile_args=['-fopenmp'],
            extra_link_args=['-fopenmp', '-lgomp']
        )
    ],
)