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
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
|
import re
import os
import sys
import warnings
from numpy.distutils.cpuinfo import cpu
from numpy.distutils.fcompiler import FCompiler
from numpy.distutils.exec_command import exec_command, find_executable
from numpy.distutils.misc_util import mingw32
class GnuFCompiler(FCompiler):
compiler_type = 'gnu'
version_pattern = r'GNU Fortran ((\(GCC[^\)]*(\)\)|\)))|)\s*'\
'(?P<version>[^\s*\)]+)'
# 'g77 --version' results
# SunOS: GNU Fortran (GCC 3.2) 3.2 20020814 (release)
# Debian: GNU Fortran (GCC) 3.3.3 20040110 (prerelease) (Debian)
# GNU Fortran (GCC) 3.3.3 (Debian 20040401)
# GNU Fortran 0.5.25 20010319 (prerelease)
# Redhat: GNU Fortran (GCC 3.2.2 20030222 (Red Hat Linux 3.2.2-5)) 3.2.2 20030222 (Red Hat Linux 3.2.2-5)
for fc_exe in map(find_executable,['g77','f77']):
if os.path.isfile(fc_exe):
break
executables = {
'version_cmd' : [fc_exe,"--version"],
'compiler_f77' : [fc_exe,"-Wall","-fno-second-underscore"],
'compiler_f90' : None,
'compiler_fix' : None,
'linker_so' : [fc_exe,"-Wall"],
'archiver' : ["ar", "-cr"],
'ranlib' : ["ranlib"],
'linker_exe' : [fc_exe,"-Wall"]
}
module_dir_switch = None
module_include_switch = None
# Cygwin: f771: warning: -fPIC ignored for target (all code is position independent)
if os.name != 'nt' and sys.platform!='cygwin':
pic_flags = ['-fPIC']
g2c = 'g2c'
#def get_linker_so(self):
# # win32 linking should be handled by standard linker
# # Darwin g77 cannot be used as a linker.
# #if re.match(r'(darwin)', sys.platform):
# # return
# return FCompiler.get_linker_so(self)
def get_flags_linker_so(self):
opt = []
if sys.platform=='darwin':
target = os.environ.get('MACOSX_DEPLOYMENT_TARGET', None)
if target is None:
target = '10.3'
major, minor = target.split('.')
if int(minor) < 3:
minor = '3'
warnings.warn('Environment variable '
'MACOSX_DEPLOYMENT_TARGET reset to 10.3')
os.environ['MACOSX_DEPLOYMENT_TARGET'] = '%s.%s' % (major,
minor)
opt.extend(['-undefined', 'dynamic_lookup', '-bundle'])
else:
opt.append("-shared")
if sys.platform[:5]=='sunos':
# SunOS often has dynamically loaded symbols defined in the
# static library libg2c.a The linker doesn't like this. To
# ignore the problem, use the -mimpure-text flag. It isn't
# the safest thing, but seems to work. 'man gcc' says:
# ".. Instead of using -mimpure-text, you should compile all
# source code with -fpic or -fPIC."
opt.append('-mimpure-text')
return opt
def get_libgcc_dir(self):
status, output = exec_command('%s -print-libgcc-file-name' \
% (self.compiler_f77[0]),use_tee=0)
if not status:
return os.path.dirname(output)
return
def get_library_dirs(self):
opt = []
if sys.platform[:5] != 'linux':
d = self.get_libgcc_dir()
if d:
opt.append(d)
return opt
def get_libraries(self):
opt = []
d = self.get_libgcc_dir()
if d is not None:
g2c = self.g2c + '-pic'
f = self.static_lib_format % (g2c, self.static_lib_extension)
if not os.path.isfile(os.path.join(d,f)):
g2c = self.g2c
else:
g2c = self.g2c
if sys.platform=='win32':
# To avoid undefined reference __EH_FRAME_BEGIN__ linker error,
# don't use -lgcc option for mingw32 g77 linker.
if not mingw32():
opt.append('gcc')
if g2c is not None:
opt.append(g2c)
if sys.platform == 'darwin':
opt.append('cc_dynamic')
return opt
def get_flags_debug(self):
return ['-g']
def get_flags_opt(self):
if self.get_version()<='3.3.3':
# With this compiler version building Fortran BLAS/LAPACK
# with -O3 caused failures in lib.lapack heevr,syevr tests.
opt = ['-O2']
else:
opt = ['-O3']
opt.append('-funroll-loops')
return opt
def get_flags_arch(self):
opt = []
if sys.platform=='darwin':
if os.name != 'posix':
# this should presumably correspond to Apple
if cpu.is_ppc():
opt.append('-arch ppc')
elif cpu.is_i386():
opt.append('-arch i386')
for a in '601 602 603 603e 604 604e 620 630 740 7400 7450 750'\
'403 505 801 821 823 860'.split():
if getattr(cpu,'is_ppc%s'%a)():
opt.append('-mcpu='+a)
opt.append('-mtune='+a)
break
return opt
# default march options in case we find nothing better
if cpu.is_i686():
march_opt = '-march=i686'
elif cpu.is_i586():
march_opt = '-march=i586'
elif cpu.is_i486():
march_opt = '-march=i486'
elif cpu.is_i386():
march_opt = '-march=i386'
else:
march_opt = ''
gnu_ver = self.get_version()
if gnu_ver >= '0.5.26': # gcc 3.0
if cpu.is_AthlonK6():
march_opt = '-march=k6'
elif cpu.is_AthlonK7():
march_opt = '-march=athlon'
if gnu_ver >= '3.1.1':
if cpu.is_AthlonK6_2():
march_opt = '-march=k6-2'
elif cpu.is_AthlonK6_3():
march_opt = '-march=k6-3'
elif cpu.is_AthlonMP():
march_opt = '-march=athlon-mp'
# there's also: athlon-tbird, athlon-4, athlon-xp
elif cpu.is_Nocona():
march_opt = '-march=nocona'
elif cpu.is_Prescott():
march_opt = '-march=prescott'
elif cpu.is_PentiumIV():
march_opt = '-march=pentium4'
elif cpu.is_PentiumIII():
march_opt = '-march=pentium3'
elif cpu.is_PentiumM():
march_opt = '-march=pentium3'
elif cpu.is_PentiumII():
march_opt = '-march=pentium2'
if gnu_ver >= '3.4':
if cpu.is_Opteron():
march_opt = '-march=opteron'
elif cpu.is_Athlon64():
march_opt = '-march=athlon64'
if gnu_ver >= '3.4.4':
if cpu.is_PentiumM():
march_opt = '-march=pentium-m'
# Note: gcc 3.2 on win32 has breakage with -march specified
if '3.1.1' <= gnu_ver <= '3.4' and sys.platform=='win32':
march_opt = ''
if march_opt:
opt.append(march_opt)
# other CPU flags
if gnu_ver >= '3.1.1':
if cpu.has_mmx(): opt.append('-mmmx')
if cpu.has_3dnow(): opt.append('-m3dnow')
if gnu_ver > '3.2.2':
if cpu.has_sse2(): opt.append('-msse2')
if cpu.has_sse(): opt.append('-msse')
if gnu_ver >= '3.4':
if cpu.has_sse3(): opt.append('-msse3')
if cpu.is_Intel():
opt.append('-fomit-frame-pointer')
if cpu.is_32bit():
opt.append('-malign-double')
return opt
class Gnu95FCompiler(GnuFCompiler):
compiler_type = 'gnu95'
version_pattern = r'GNU Fortran 95 \(GCC (?P<version>[^\s*\)]+)'
# 'gfortran --version' results:
# Debian: GNU Fortran 95 (GCC 4.0.3 20051023 (prerelease) (Debian 4.0.2-3))
for fc_exe in map(find_executable,['gfortran','f95']):
if os.path.isfile(fc_exe):
break
executables = {
'version_cmd' : [fc_exe,"--version"],
'compiler_f77' : [fc_exe,"-Wall","-ffixed-form","-fno-second-underscore"],
'compiler_f90' : [fc_exe,"-Wall","-fno-second-underscore"],
'compiler_fix' : [fc_exe,"-Wall","-ffixed-form","-fno-second-underscore"],
'linker_so' : [fc_exe,"-Wall"],
'archiver' : ["ar", "-cr"],
'ranlib' : ["ranlib"],
'linker_exe' : [fc_exe,"-Wall"]
}
module_dir_switch = '-J'
module_include_switch = '-I'
g2c = 'gfortran'
if __name__ == '__main__':
from distutils import log
log.set_verbosity(2)
from numpy.distutils.fcompiler import new_fcompiler
#compiler = new_fcompiler(compiler='gnu')
compiler = GnuFCompiler()
compiler.customize()
print compiler.get_version()
|
