stable-diffusion-implementation
/
main
/myenv
/lib
/python3.10
/site-packages
/numpy
/distutils
/fcompiler
/arm.py
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 | |
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()) | |