libgpuarray/setup.py at master · ejls/libgpuarray · 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
89
90
91
92
93
94
import sys

have_cython = False

try:
    import Cython
    if Cython.__version__ < '0.19':
        raise Exception('cython is too old or not installed '
                        '(at least 0.19 required)')
    from Cython.Build import cythonize
    have_cython = True
except Exception:
    # for devel version
    raise
    def cythonize(arg):
        return arg

# clang gives an error if passed -mno-fused-madd
# (and I don't even understand why it's passed in the first place)
if sys.platform =='darwin':
    from distutils import sysconfig, ccompiler
    sysconfig_customize_compiler = sysconfig.customize_compiler
    def customize_compiler(compiler):
        sysconfig_customize_compiler(compiler)
        if sys.platform == 'darwin':
            while '-mno-fused-madd' in compiler.compiler:
                compiler.compiler.remove('-mno-fused-madd')
            while '-mno-fused-madd' in compiler.compiler_so:
                compiler.compiler_so.remove('-mno-fused-madd')
            while '-mno-fused-madd' in compiler.linker_so:
                compiler.linker_so.remove('-mno-fused-madd')
    sysconfig.customize_compiler = customize_compiler
    ccompiler.customize_compiler = customize_compiler

try:
    from setuptools import setup, Extension as _Extension

    # setuptools is stupid and rewrites "sources" to change '.pyx' to '.c'
    # if it can't find Pyrex (and in recent versions, Cython).
    #
    # This is a really stupid thing to do behind the users's back (since
    # it breaks development builds) especially with no way of disabling it
    # short of the hack below.
    class Extension(_Extension):
        def __init__(self, *args, **kwargs):
            save_sources = kwargs.get('sources', None)
            _Extension.__init__(self, *args, **kwargs)
            self.sources = save_sources
except ImportError:
    from distutils.core import setup, Extension

import numpy as np

to_del = []

for i, a in enumerate(sys.argv):
    if a == '--disable-cython':
        to_del.append(i)
        have_cython = False

for i in reversed(to_del):
    del sys.argv[i]

del to_del

if have_cython:
    srcs = ['pygpu/gpuarray.pyx']
    blas_src = ['pygpu/blas.pyx']
else:
    srcs = ['pygpu/gpuarray.c']
    blas_src = ['pygpu/blas.c']

exts = [Extension('pygpu.gpuarray',
                  sources = srcs,
                  include_dirs = [np.get_include()],
                  libraries = ['gpuarray'],
                  define_macros = [('GPUARRAY_SHARED', None)],
                  ),
        Extension('pygpu.blas',
                  sources = blas_src,
                  include_dirs = [np.get_include()],
                  libraries = ['gpuarray'],
                  define_macros = [('GPUARRAY_SHARED', None)],
                  )]

setup(name='pygpu',
      version='0.2.1',
      description='numpy-like wrapper on libgpuarray for GPU computations',
      packages = ['pygpu', 'pygpu/tests'],
      data_files = [('pygpu', ['pygpu/gpuarray.h', 'pygpu/gpuarray_api.h',
                               'pygpu/blas_api.h', 'pygpu/numpy_compat.h'])],
      ext_modules=cythonize(exts),
      install_requires=['mako>=0.7'],
      )