-
Notifications
You must be signed in to change notification settings - Fork 5
/
setup.py
28 lines (22 loc) · 825 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
from numpy.distutils.core import setup, Extension
import os
source = ['dgbmv.pyf']
# Set up the libraries for building against BLAS. Either use an internal copy
# of dgbmv.f, or try and build against MKL.
if 'MKLROOT' in os.environ: # Check to see if we have MKL available
blas_lib = ['mkl_rt', 'iomp5', 'pthread', 'm'] # MKL libraries we need to link against
blas_libdir = [os.environ['MKLROOT']+'/lib/intel64']
else:
source += ['dgbmv.f']
blas_lib = []
blas_libdir = []
use_omp = True
omp_args = ['-fopenmp'] if use_omp else []
dgbmv_ext = Extension('dgbmv', source,
library_dirs=blas_libdir, libraries=blas_lib,
extra_compile_args=omp_args,
extra_link_args=omp_args)
setup(
name='pfb-inverse',
ext_modules=[dgbmv_ext]
)