|
|
|
import sys |
|
|
|
from numpy.distutils.ccompiler import simple_version_match |
|
from numpy.distutils.fcompiler import FCompiler, dummy_fortran_file |
|
|
|
compilers = ['IntelFCompiler', 'IntelVisualFCompiler', |
|
'IntelItaniumFCompiler', 'IntelItaniumVisualFCompiler', |
|
'IntelEM64VisualFCompiler', 'IntelEM64TFCompiler'] |
|
|
|
|
|
def intel_version_match(type): |
|
|
|
return simple_version_match(start=r'Intel.*?Fortran.*?(?:%s).*?Version' % (type,)) |
|
|
|
|
|
class BaseIntelFCompiler(FCompiler): |
|
def update_executables(self): |
|
f = dummy_fortran_file() |
|
self.executables['version_cmd'] = ['<F77>', '-FI', '-V', '-c', |
|
f + '.f', '-o', f + '.o'] |
|
|
|
def runtime_library_dir_option(self, dir): |
|
|
|
assert "," not in dir |
|
|
|
return '-Wl,-rpath=%s' % dir |
|
|
|
|
|
class IntelFCompiler(BaseIntelFCompiler): |
|
|
|
compiler_type = 'intel' |
|
compiler_aliases = ('ifort',) |
|
description = 'Intel Fortran Compiler for 32-bit apps' |
|
version_match = intel_version_match('32-bit|IA-32') |
|
|
|
possible_executables = ['ifort', 'ifc'] |
|
|
|
executables = { |
|
'version_cmd' : None, |
|
'compiler_f77' : [None, "-72", "-w90", "-w95"], |
|
'compiler_f90' : [None], |
|
'compiler_fix' : [None, "-FI"], |
|
'linker_so' : ["<F90>", "-shared"], |
|
'archiver' : ["ar", "-cr"], |
|
'ranlib' : ["ranlib"] |
|
} |
|
|
|
pic_flags = ['-fPIC'] |
|
module_dir_switch = '-module ' |
|
module_include_switch = '-I' |
|
|
|
def get_flags_free(self): |
|
return ['-FR'] |
|
|
|
def get_flags(self): |
|
return ['-fPIC'] |
|
|
|
def get_flags_opt(self): |
|
v = self.get_version() |
|
mpopt = 'openmp' if v and v < '15' else 'qopenmp' |
|
return ['-fp-model', 'strict', '-O1', |
|
'-assume', 'minus0', '-{}'.format(mpopt)] |
|
|
|
def get_flags_arch(self): |
|
return [] |
|
|
|
def get_flags_linker_so(self): |
|
opt = FCompiler.get_flags_linker_so(self) |
|
v = self.get_version() |
|
if v and v >= '8.0': |
|
opt.append('-nofor_main') |
|
if sys.platform == 'darwin': |
|
|
|
try: |
|
idx = opt.index('-shared') |
|
opt.remove('-shared') |
|
except ValueError: |
|
idx = 0 |
|
opt[idx:idx] = ['-dynamiclib', '-Wl,-undefined,dynamic_lookup'] |
|
return opt |
|
|
|
|
|
class IntelItaniumFCompiler(IntelFCompiler): |
|
compiler_type = 'intele' |
|
compiler_aliases = () |
|
description = 'Intel Fortran Compiler for Itanium apps' |
|
|
|
version_match = intel_version_match('Itanium|IA-64') |
|
|
|
possible_executables = ['ifort', 'efort', 'efc'] |
|
|
|
executables = { |
|
'version_cmd' : None, |
|
'compiler_f77' : [None, "-FI", "-w90", "-w95"], |
|
'compiler_fix' : [None, "-FI"], |
|
'compiler_f90' : [None], |
|
'linker_so' : ['<F90>', "-shared"], |
|
'archiver' : ["ar", "-cr"], |
|
'ranlib' : ["ranlib"] |
|
} |
|
|
|
|
|
class IntelEM64TFCompiler(IntelFCompiler): |
|
compiler_type = 'intelem' |
|
compiler_aliases = () |
|
description = 'Intel Fortran Compiler for 64-bit apps' |
|
|
|
version_match = intel_version_match('EM64T-based|Intel\\(R\\) 64|64|IA-64|64-bit') |
|
|
|
possible_executables = ['ifort', 'efort', 'efc'] |
|
|
|
executables = { |
|
'version_cmd' : None, |
|
'compiler_f77' : [None, "-FI"], |
|
'compiler_fix' : [None, "-FI"], |
|
'compiler_f90' : [None], |
|
'linker_so' : ['<F90>', "-shared"], |
|
'archiver' : ["ar", "-cr"], |
|
'ranlib' : ["ranlib"] |
|
} |
|
|
|
|
|
|
|
|
|
|
|
class IntelVisualFCompiler(BaseIntelFCompiler): |
|
compiler_type = 'intelv' |
|
description = 'Intel Visual Fortran Compiler for 32-bit apps' |
|
version_match = intel_version_match('32-bit|IA-32') |
|
|
|
def update_executables(self): |
|
f = dummy_fortran_file() |
|
self.executables['version_cmd'] = ['<F77>', '/FI', '/c', |
|
f + '.f', '/o', f + '.o'] |
|
|
|
ar_exe = 'lib.exe' |
|
possible_executables = ['ifort', 'ifl'] |
|
|
|
executables = { |
|
'version_cmd' : None, |
|
'compiler_f77' : [None], |
|
'compiler_fix' : [None], |
|
'compiler_f90' : [None], |
|
'linker_so' : [None], |
|
'archiver' : [ar_exe, "/verbose", "/OUT:"], |
|
'ranlib' : None |
|
} |
|
|
|
compile_switch = '/c ' |
|
object_switch = '/Fo' |
|
library_switch = '/OUT:' |
|
module_dir_switch = '/module:' |
|
module_include_switch = '/I' |
|
|
|
def get_flags(self): |
|
opt = ['/nologo', '/MD', '/nbs', '/names:lowercase', |
|
'/assume:underscore', '/fpp'] |
|
return opt |
|
|
|
def get_flags_free(self): |
|
return [] |
|
|
|
def get_flags_debug(self): |
|
return ['/4Yb', '/d2'] |
|
|
|
def get_flags_opt(self): |
|
return ['/O1', '/assume:minus0'] |
|
|
|
def get_flags_arch(self): |
|
return ["/arch:IA32", "/QaxSSE3"] |
|
|
|
def runtime_library_dir_option(self, dir): |
|
raise NotImplementedError |
|
|
|
|
|
class IntelItaniumVisualFCompiler(IntelVisualFCompiler): |
|
compiler_type = 'intelev' |
|
description = 'Intel Visual Fortran Compiler for Itanium apps' |
|
|
|
version_match = intel_version_match('Itanium') |
|
|
|
possible_executables = ['efl'] |
|
ar_exe = IntelVisualFCompiler.ar_exe |
|
|
|
executables = { |
|
'version_cmd' : None, |
|
'compiler_f77' : [None, "-FI", "-w90", "-w95"], |
|
'compiler_fix' : [None, "-FI", "-4L72", "-w"], |
|
'compiler_f90' : [None], |
|
'linker_so' : ['<F90>', "-shared"], |
|
'archiver' : [ar_exe, "/verbose", "/OUT:"], |
|
'ranlib' : None |
|
} |
|
|
|
|
|
class IntelEM64VisualFCompiler(IntelVisualFCompiler): |
|
compiler_type = 'intelvem' |
|
description = 'Intel Visual Fortran Compiler for 64-bit apps' |
|
|
|
version_match = simple_version_match(start=r'Intel\(R\).*?64,') |
|
|
|
def get_flags_arch(self): |
|
return [] |
|
|
|
|
|
if __name__ == '__main__': |
|
from distutils import log |
|
log.set_verbosity(2) |
|
from numpy.distutils import customized_fcompiler |
|
print(customized_fcompiler(compiler='intel').get_version()) |
|
|