|
|
|
import sys |
|
|
|
from numpy.distutils.fcompiler import FCompiler |
|
from sys import platform |
|
from os.path import join, dirname, normpath |
|
|
|
compilers = ['PGroupFCompiler', 'PGroupFlangCompiler'] |
|
|
|
|
|
class PGroupFCompiler(FCompiler): |
|
|
|
compiler_type = 'pg' |
|
description = 'Portland Group Fortran Compiler' |
|
version_pattern = r'\s*pg(f77|f90|hpf|fortran) (?P<version>[\d.-]+).*' |
|
|
|
if platform == 'darwin': |
|
executables = { |
|
'version_cmd': ["<F77>", "-V"], |
|
'compiler_f77': ["pgfortran", "-dynamiclib"], |
|
'compiler_fix': ["pgfortran", "-Mfixed", "-dynamiclib"], |
|
'compiler_f90': ["pgfortran", "-dynamiclib"], |
|
'linker_so': ["libtool"], |
|
'archiver': ["ar", "-cr"], |
|
'ranlib': ["ranlib"] |
|
} |
|
pic_flags = [''] |
|
else: |
|
executables = { |
|
'version_cmd': ["<F77>", "-V"], |
|
'compiler_f77': ["pgfortran"], |
|
'compiler_fix': ["pgfortran", "-Mfixed"], |
|
'compiler_f90': ["pgfortran"], |
|
'linker_so': ["<F90>"], |
|
'archiver': ["ar", "-cr"], |
|
'ranlib': ["ranlib"] |
|
} |
|
pic_flags = ['-fpic'] |
|
|
|
module_dir_switch = '-module ' |
|
module_include_switch = '-I' |
|
|
|
def get_flags(self): |
|
opt = ['-Minform=inform', '-Mnosecond_underscore'] |
|
return self.pic_flags + opt |
|
|
|
def get_flags_opt(self): |
|
return ['-fast'] |
|
|
|
def get_flags_debug(self): |
|
return ['-g'] |
|
|
|
if platform == 'darwin': |
|
def get_flags_linker_so(self): |
|
return ["-dynamic", '-undefined', 'dynamic_lookup'] |
|
|
|
else: |
|
def get_flags_linker_so(self): |
|
return ["-shared", '-fpic'] |
|
|
|
def runtime_library_dir_option(self, dir): |
|
return '-R%s' % dir |
|
|
|
|
|
import functools |
|
|
|
class PGroupFlangCompiler(FCompiler): |
|
compiler_type = 'flang' |
|
description = 'Portland Group Fortran LLVM Compiler' |
|
version_pattern = r'\s*(flang|clang) version (?P<version>[\d.-]+).*' |
|
|
|
ar_exe = 'lib.exe' |
|
possible_executables = ['flang'] |
|
|
|
executables = { |
|
'version_cmd': ["<F77>", "--version"], |
|
'compiler_f77': ["flang"], |
|
'compiler_fix': ["flang"], |
|
'compiler_f90': ["flang"], |
|
'linker_so': [None], |
|
'archiver': [ar_exe, "/verbose", "/OUT:"], |
|
'ranlib': None |
|
} |
|
|
|
library_switch = '/OUT:' |
|
module_dir_switch = '-module ' |
|
|
|
def get_libraries(self): |
|
opt = FCompiler.get_libraries(self) |
|
opt.extend(['flang', 'flangrti', 'ompstub']) |
|
return opt |
|
|
|
@functools.lru_cache(maxsize=128) |
|
def get_library_dirs(self): |
|
"""List of compiler library directories.""" |
|
opt = FCompiler.get_library_dirs(self) |
|
flang_dir = dirname(self.executables['compiler_f77'][0]) |
|
opt.append(normpath(join(flang_dir, '..', 'lib'))) |
|
|
|
return opt |
|
|
|
def get_flags(self): |
|
return [] |
|
|
|
def get_flags_free(self): |
|
return [] |
|
|
|
def get_flags_debug(self): |
|
return ['-g'] |
|
|
|
def get_flags_opt(self): |
|
return ['-O3'] |
|
|
|
def get_flags_arch(self): |
|
return [] |
|
|
|
def runtime_library_dir_option(self, dir): |
|
raise NotImplementedError |
|
|
|
|
|
if __name__ == '__main__': |
|
from distutils import log |
|
log.set_verbosity(2) |
|
from numpy.distutils import customized_fcompiler |
|
if 'flang' in sys.argv: |
|
print(customized_fcompiler(compiler='flang').get_version()) |
|
else: |
|
print(customized_fcompiler(compiler='pg').get_version()) |
|
|
