72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
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 ' # Don't remove ending space!
|
|
|
|
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())
|
|
|