|
import sys |
|
|
|
from numpy.distutils.fcompiler import FCompiler, dummy_fortran_file |
|
from sys import platform |
|
from os.path import join, dirname, normpath |
|
|
|
compilers = ['ArmFlangCompiler'] |
|
|
|
import functools |
|
|
|
class ArmFlangCompiler(FCompiler): |
|
compiler_type = 'arm' |
|
description = 'Arm Compiler' |
|
version_pattern = r'\s*Arm.*version (?P<version>[\d.-]+).*' |
|
|
|
ar_exe = 'lib.exe' |
|
possible_executables = ['armflang'] |
|
|
|
executables = { |
|
'version_cmd': ["", "--version"], |
|
'compiler_f77': ["armflang", "-fPIC"], |
|
'compiler_fix': ["armflang", "-fPIC", "-ffixed-form"], |
|
'compiler_f90': ["armflang", "-fPIC"], |
|
'linker_so': ["armflang", "-fPIC", "-shared"], |
|
'archiver': ["ar", "-cr"], |
|
'ranlib': None |
|
} |
|
|
|
pic_flags = ["-fPIC", "-DPIC"] |
|
c_compiler = 'arm' |
|
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): |
|
return '-Wl,-rpath=%s' % dir |
|
|
|
|
|
if __name__ == '__main__': |
|
from distutils import log |
|
log.set_verbosity(2) |
|
from numpy.distutils import customized_fcompiler |
|
print(customized_fcompiler(compiler='armflang').get_version()) |
|
|
|
|
