12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- import os
- import re
- import sys
- import subprocess
- from numpy.distutils.fcompiler import FCompiler
- from numpy.distutils.exec_command import find_executable
- from numpy.distutils.misc_util import make_temp_file
- from distutils import log
- compilers = ['IBMFCompiler']
- class IBMFCompiler(FCompiler):
- compiler_type = 'ibm'
- description = 'IBM XL Fortran Compiler'
- version_pattern = r'(xlf\(1\)\s*|)IBM XL Fortran ((Advanced Edition |)Version |Enterprise Edition V|for AIX, V)(?P<version>[^\s*]*)'
- #IBM XL Fortran Enterprise Edition V10.1 for AIX \nVersion: 10.01.0000.0004
- executables = {
- 'version_cmd' : ["<F77>", "-qversion"],
- 'compiler_f77' : ["xlf"],
- 'compiler_fix' : ["xlf90", "-qfixed"],
- 'compiler_f90' : ["xlf90"],
- 'linker_so' : ["xlf95"],
- 'archiver' : ["ar", "-cr"],
- 'ranlib' : ["ranlib"]
- }
- def get_version(self,*args,**kwds):
- version = FCompiler.get_version(self,*args,**kwds)
- if version is None and sys.platform.startswith('aix'):
- # use lslpp to find out xlf version
- lslpp = find_executable('lslpp')
- xlf = find_executable('xlf')
- if os.path.exists(xlf) and os.path.exists(lslpp):
- try:
- o = subprocess.check_output([lslpp, '-Lc', 'xlfcmp'])
- except (OSError, subprocess.CalledProcessError):
- pass
- else:
- m = re.search(r'xlfcmp:(?P<version>\d+([.]\d+)+)', o)
- if m: version = m.group('version')
- xlf_dir = '/etc/opt/ibmcmp/xlf'
- if version is None and os.path.isdir(xlf_dir):
- # linux:
- # If the output of xlf does not contain version info
- # (that's the case with xlf 8.1, for instance) then
- # let's try another method:
- l = sorted(os.listdir(xlf_dir))
- l.reverse()
- l = [d for d in l if os.path.isfile(os.path.join(xlf_dir, d, 'xlf.cfg'))]
- if l:
- from distutils.version import LooseVersion
- self.version = version = LooseVersion(l[0])
- return version
- def get_flags(self):
- return ['-qextname']
- def get_flags_debug(self):
- return ['-g']
- def get_flags_linker_so(self):
- opt = []
- if sys.platform=='darwin':
- opt.append('-Wl,-bundle,-flat_namespace,-undefined,suppress')
- else:
- opt.append('-bshared')
- version = self.get_version(ok_status=[0, 40])
- if version is not None:
- if sys.platform.startswith('aix'):
- xlf_cfg = '/etc/xlf.cfg'
- else:
- xlf_cfg = '/etc/opt/ibmcmp/xlf/%s/xlf.cfg' % version
- fo, new_cfg = make_temp_file(suffix='_xlf.cfg')
- log.info('Creating '+new_cfg)
- with open(xlf_cfg, 'r') as fi:
- crt1_match = re.compile(r'\s*crt\s*=\s*(?P<path>.*)/crt1.o').match
- for line in fi:
- m = crt1_match(line)
- if m:
- fo.write('crt = %s/bundle1.o\n' % (m.group('path')))
- else:
- fo.write(line)
- fo.close()
- opt.append('-F'+new_cfg)
- return opt
- def get_flags_opt(self):
- return ['-O3']
- if __name__ == '__main__':
- from numpy.distutils import customized_fcompiler
- log.set_verbosity(2)
- print(customized_fcompiler(compiler='ibm').get_version())
|