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())