-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
34 lines (30 loc) · 882 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
28
29
30
31
32
33
34
from setuptools import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
clBLASdir = '/opt/clBLAS-2.10.0-Hawaii-Linux-x64-CL2.0/'
ext_modules = []
ext_modules.append(Extension(
'pyopencl_blas.blas',
['pyopencl_blas/blas.pyx'],
include_dirs=[clBLASdir + 'include'],
library_dirs=[clBLASdir + 'lib64'],
libraries=['clBLAS'],
language='c++',
extra_compile_args=['-w', '-O3']))
setup(
name='pyopencl_blas',
version='0.0.1',
author="Eric Hunsberger",
author_email="erichuns@gmail.com",
url="https://github.com/hunse/pyopencl_blas",
license="MIT",
description="PyOpenCL wrapper for AMD clMathLibraries.clBLAS",
requires=[
'pyopencl',
'cython',
],
packages=['pyopencl_blas'],
scripts=[],
ext_modules=ext_modules,
cmdclass = {'build_ext': build_ext},
)