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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
|
## Automatically adapted for scipy Oct 18, 2005 by
#!/usr/bin/env python
import os
import sys
import re
from distutils.dep_util import newer_group, newer
from glob import glob
from os.path import join
#-------------------
# To skip wrapping single precision atlas/lapack/blas routines, set
# the following flag to True:
skip_single_routines = 0
# Some OS distributions (e.g. Redhat, Suse) provide a blas library that
# is built using incomplete blas sources that come with lapack tar-ball.
# In order to use such a library in scipy.linalg, the following flag
# must be set to True:
using_lapack_blas = 0
#--------------------
def needs_cblas_wrapper(info):
"""Returns true if needs c wrapper around cblas for calling from
fortran."""
import re
r_accel = re.compile("Accelerate")
r_vec = re.compile("vecLib")
res = False
try:
tmpstr = info['extra_link_args']
for i in tmpstr:
if r_accel.search(i) or r_vec.search(i):
res = True
except KeyError:
pass
return res
def configuration(parent_package='',top_path=None):
from numpy.distutils.system_info import get_info, NotFoundError
from numpy.distutils.misc_util import Configuration
from interface_gen import generate_interface
config = Configuration('linalg',parent_package,top_path)
lapack_opt = get_info('lapack_opt')
if not lapack_opt:
raise NotFoundError,'no lapack/blas resources found'
atlas_version = ([v[3:-3] for k,v in lapack_opt.get('define_macros',[]) \
if k=='ATLAS_INFO']+[None])[0]
if atlas_version:
print 'ATLAS version',atlas_version
target_dir = ''
skip_names = {'clapack':[],'flapack':[],'cblas':[],'fblas':[]}
if skip_single_routines:
target_dir = 'dbl'
skip_names['clapack'].extend(\
'sgesv cgesv sgetrf cgetrf sgetrs cgetrs sgetri cgetri'\
' sposv cposv spotrf cpotrf spotrs cpotrs spotri cpotri'\
' slauum clauum strtri ctrtri'.split())
skip_names['flapack'].extend(skip_names['clapack'])
skip_names['flapack'].extend(\
'sgesdd cgesdd sgelss cgelss sgeqrf cgeqrf sgeev cgeev'\
' sgegv cgegv ssyev cheev slaswp claswp sgees cgees'
' sggev cggev'.split())
skip_names['cblas'].extend('saxpy caxpy'.split())
skip_names['fblas'].extend(skip_names['cblas'])
skip_names['fblas'].extend(\
'srotg crotg srotmg srot csrot srotm sswap cswap sscal cscal'\
' csscal scopy ccopy sdot cdotu cdotc snrm2 scnrm2 sasum scasum'\
' isamax icamax sgemv cgemv chemv ssymv strmv ctrmv'\
' sgemm cgemm'.split())
if using_lapack_blas:
target_dir = join(target_dir,'blas')
skip_names['fblas'].extend(\
'drotmg srotmg drotm srotm'.split())
if atlas_version=='3.2.1_pre3.3.6':
target_dir = join(target_dir,'atlas321')
skip_names['clapack'].extend(\
'sgetri dgetri cgetri zgetri spotri dpotri cpotri zpotri'\
' slauum dlauum clauum zlauum strtri dtrtri ctrtri ztrtri'.split())
elif atlas_version>'3.4.0' and atlas_version<='3.5.12':
skip_names['clapack'].extend('cpotrf zpotrf'.split())
def generate_pyf(extension, build_dir):
name = extension.name.split('.')[-1]
target = join(build_dir,target_dir,name+'.pyf')
if name[0]=='c' and atlas_version is None and newer(__file__,target):
f = open(target,'w')
f.write('python module '+name+'\n')
f.write('usercode void empty_module(void) {}\n')
f.write('interface\n')
f.write('subroutine empty_module()\n')
f.write('intent(c) empty_module\n')
f.write('end subroutine empty_module\n')
f.write('end interface\nend python module'+name+'\n')
f.close()
return target
if newer_group(extension.depends,target):
generate_interface(name,
extension.depends[0],
target,
skip_names[name])
return target
depends = ['generic_fblas.pyf',
'generic_fblas1.pyf',
'generic_fblas2.pyf',
'generic_fblas3.pyf',
'interface_gen.py',
join('src','fblaswrap_veclib_c.c'),
join('src','fblaswrap.f'),
]
# fblas:
if needs_cblas_wrapper(lapack_opt):
config.add_extension('fblas',
sources = [generate_pyf,
join('src','fblaswrap_veclib_c.c')],
depends = depends,
extra_info = lapack_opt
)
else:
config.add_extension('fblas',
sources = [generate_pyf,
join('src','fblaswrap.f')],
depends = depends,
extra_info = lapack_opt
)
# cblas:
config.add_extension('cblas',
sources = [generate_pyf],
depends = ['generic_cblas.pyf',
'generic_cblas1.pyf',
'interface_gen.py'],
extra_info = lapack_opt
)
# flapack:
config.add_extension('flapack',
sources = [generate_pyf],
depends = ['generic_flapack.pyf',
'flapack_user_routines.pyf',
'interface_gen.py'],
extra_info = lapack_opt
)
# clapack:
config.add_extension('clapack',
sources = [generate_pyf],
depends = ['generic_clapack.pyf',
'interface_gen.py'],
extra_info = lapack_opt
)
# _flinalg:
config.add_extension('_flinalg',
sources = [join('src','det.f'),join('src','lu.f')],
extra_info = lapack_opt
)
# calc_lwork:
config.add_extension('calc_lwork',
[join('src','calc_lwork.f')],
extra_info = lapack_opt
)
# atlas_version:
config.add_extension('atlas_version',
['atlas_version.c'],
extra_info = lapack_opt
)
config.add_data_dir('tests')
config.add_data_dir('benchmarks')
return config
if __name__ == '__main__':
from numpy.distutils.core import setup
from linalg_version import linalg_version
setup(version=linalg_version,
**configuration(top_path='').todict())
|
