summaryrefslogtreecommitdiff
path: root/numpy/distutils
diff options
context:
space:
mode:
authorDmitry Zagorny <dmitry.zagorny@intel.com>2015-09-14 13:03:38 -0500
committerDmitry Zagorny <dmitry.zagorny@intel.com>2015-09-15 14:34:43 +0300
commit72753bbdf8736a13f1cb60c25cf8683608f46e29 (patch)
tree5a0f17e10c88a02e4b9d6438ad88a447172f5174 /numpy/distutils
parent44c9b2aba51f73e50c17b7f990a054d0d4804269 (diff)
downloadnumpy-72753bbdf8736a13f1cb60c25cf8683608f46e29.tar.gz
MSVCCompiler overwrite 'lib' and 'include' environment variables. This
behavior affect at least python 3.5 and SciPy build and build failed. During initialization <python>.distutils.MSVCCompiler replace Intel environment('include' and 'lib' paths). This fix decorate 'initialize' function in MSVCCompiler and extend 'lib' and 'include' environment variables. Changed compilation keys: generate optimized code specialized for Intel processors with SSE4.2 support.
Diffstat (limited to 'numpy/distutils')
-rw-r--r--numpy/distutils/fcompiler/compaq.py2
-rw-r--r--numpy/distutils/fcompiler/intel.py29
-rw-r--r--numpy/distutils/intelccompiler.py33
-rw-r--r--numpy/distutils/msvc9compiler.py21
-rw-r--r--numpy/distutils/msvccompiler.py17
-rw-r--r--numpy/distutils/system_info.py4
6 files changed, 71 insertions, 35 deletions
diff --git a/numpy/distutils/fcompiler/compaq.py b/numpy/distutils/fcompiler/compaq.py
index 5162b168c..2dd6c01e6 100644
--- a/numpy/distutils/fcompiler/compaq.py
+++ b/numpy/distutils/fcompiler/compaq.py
@@ -74,7 +74,7 @@ class CompaqVisualFCompiler(FCompiler):
fc_exe = 'DF'
if sys.platform=='win32':
- from distutils.msvccompiler import MSVCCompiler
+ from numpy.distutils.msvccompiler import MSVCCompiler
try:
m = MSVCCompiler()
diff --git a/numpy/distutils/fcompiler/intel.py b/numpy/distutils/fcompiler/intel.py
index ef0bcc30b..28624918d 100644
--- a/numpy/distutils/fcompiler/intel.py
+++ b/numpy/distutils/fcompiler/intel.py
@@ -10,6 +10,7 @@ compilers = ['IntelFCompiler', 'IntelVisualFCompiler',
'IntelItaniumFCompiler', 'IntelItaniumVisualFCompiler',
'IntelEM64VisualFCompiler', 'IntelEM64TFCompiler']
+
def intel_version_match(type):
# Match against the important stuff in the version string
return simple_version_match(start=r'Intel.*?Fortran.*?(?:%s).*?Version' % (type,))
@@ -45,17 +46,16 @@ class IntelFCompiler(BaseIntelFCompiler):
}
pic_flags = ['-fPIC']
- module_dir_switch = '-module ' # Don't remove ending space!
+ module_dir_switch = '-module ' # Don't remove ending space!
module_include_switch = '-I'
def get_flags_free(self):
- return ["-FR"]
+ return ['-FR']
def get_flags(self):
return ['-fPIC']
def get_flags_opt(self):
- #return ['-i8 -xhost -openmp -fp-model strict']
return ['-xhost -openmp -fp-model strict']
def get_flags_arch(self):
@@ -120,11 +120,10 @@ class IntelEM64TFCompiler(IntelFCompiler):
return ['-fPIC']
def get_flags_opt(self):
- #return ['-i8 -xhost -openmp -fp-model strict']
- return ['-xhost -openmp -fp-model strict']
+ return ['-openmp -fp-model strict']
def get_flags_arch(self):
- return []
+ return ['-xSSE4.2']
# Is there no difference in the version string between the above compilers
# and the Visual compilers?
@@ -145,18 +144,18 @@ class IntelVisualFCompiler(BaseIntelFCompiler):
executables = {
'version_cmd' : None,
- 'compiler_f77' : [None, "-FI", "-w90", "-w95"],
- 'compiler_fix' : [None, "-FI", "-4L72", "-w"],
+ 'compiler_f77' : [None],
+ 'compiler_fix' : [None],
'compiler_f90' : [None],
- 'linker_so' : ['<F90>', "-shared"],
+ 'linker_so' : [None],
'archiver' : [ar_exe, "/verbose", "/OUT:"],
'ranlib' : None
}
compile_switch = '/c '
- object_switch = '/Fo' #No space after /Fo!
- library_switch = '/OUT:' #No space after /OUT:!
- module_dir_switch = '/module:' #No space after /module:
+ object_switch = '/Fo' # No space after /Fo!
+ library_switch = '/OUT:' # No space after /OUT:!
+ module_dir_switch = '/module:' # No space after /module:
module_include_switch = '/I'
def get_flags(self):
@@ -164,7 +163,7 @@ class IntelVisualFCompiler(BaseIntelFCompiler):
return opt
def get_flags_free(self):
- return ["-FR"]
+ return []
def get_flags_debug(self):
return ['/4Yb', '/d2']
@@ -185,7 +184,7 @@ class IntelItaniumVisualFCompiler(IntelVisualFCompiler):
version_match = intel_version_match('Itanium')
- possible_executables = ['efl'] # XXX this is a wild guess
+ possible_executables = ['efl'] # XXX this is a wild guess
ar_exe = IntelVisualFCompiler.ar_exe
executables = {
@@ -206,7 +205,7 @@ class IntelEM64VisualFCompiler(IntelVisualFCompiler):
version_match = simple_version_match(start='Intel\(R\).*?64,')
def get_flags_arch(self):
- return ["/arch:SSE2"]
+ return ['/QxSSE4.2']
if __name__ == '__main__':
diff --git a/numpy/distutils/intelccompiler.py b/numpy/distutils/intelccompiler.py
index db6ef80bd..2635424e3 100644
--- a/numpy/distutils/intelccompiler.py
+++ b/numpy/distutils/intelccompiler.py
@@ -1,8 +1,10 @@
from __future__ import division, absolute_import, print_function
-import sys
+import platform
from distutils.unixccompiler import UnixCCompiler
+if platform.system() == 'Windows':
+ from numpy.distutils.msvc9compiler import MSVCCompiler
from numpy.distutils.exec_command import find_executable
from numpy.distutils.ccompiler import simple_version_match
@@ -15,14 +17,14 @@ class IntelCCompiler(UnixCCompiler):
def __init__(self, verbose=0, dry_run=0, force=0):
UnixCCompiler.__init__(self, verbose, dry_run, force)
- self.cc_exe = 'icc -fPIC'
+ self.cc_exe = 'icc -fPIC -fp-model strict -O3 -fomit-frame-pointer -openmp'
compiler = self.cc_exe
self.set_executables(compiler=compiler,
compiler_so=compiler,
compiler_cxx=compiler,
archiver='xiar' + ' cru',
- linker_exe=compiler,
- linker_so=compiler + ' -shared')
+ linker_exe=compiler + ' -shared-intel',
+ linker_so=compiler + ' -shared -shared-intel')
class IntelItaniumCCompiler(IntelCCompiler):
@@ -40,24 +42,22 @@ class IntelEM64TCCompiler(UnixCCompiler):
A modified Intel x86_64 compiler compatible with a 64bit GCC-built Python.
"""
compiler_type = 'intelem'
- cc_exe = 'icc -m64 -fPIC'
- cc_args = "-fPIC"
+ cc_exe = 'icc -m64'
+ cc_args = '-fPIC'
def __init__(self, verbose=0, dry_run=0, force=0):
UnixCCompiler.__init__(self, verbose, dry_run, force)
- self.cc_exe = 'icc -m64 -fPIC'
+ self.cc_exe = 'icc -m64 -fPIC -fp-model strict -O3 -fomit-frame-pointer -openmp -xSSE4.2'
compiler = self.cc_exe
self.set_executables(compiler=compiler,
compiler_so=compiler,
compiler_cxx=compiler,
archiver='xiar' + ' cru',
- linker_exe=compiler,
- linker_so=compiler + ' -shared')
+ linker_exe=compiler + ' -shared-intel',
+ linker_so=compiler + ' -shared -shared-intel')
-if sys.platform == 'win32':
- from distutils.msvc9compiler import MSVCCompiler
-
+if platform.system() == 'Windows':
class IntelCCompilerW(MSVCCompiler):
"""
A modified Intel compiler compatible with an MSVC-built Python.
@@ -72,11 +72,11 @@ if sys.platform == 'win32':
def initialize(self, plat_name=None):
MSVCCompiler.initialize(self, plat_name)
- self.cc = self.find_exe("icl.exe")
- self.lib = self.find_exe("xilib")
- self.linker = self.find_exe("xilink")
+ self.cc = self.find_exe('icl.exe')
+ self.lib = self.find_exe('xilib')
+ self.linker = self.find_exe('xilink')
self.compile_options = ['/nologo', '/O3', '/MD', '/W3',
- '/Qstd=c99']
+ '/Qstd=c99', '/QxSSE4.2']
self.compile_options_debug = ['/nologo', '/Od', '/MDd', '/W3',
'/Qstd=c99', '/Z7', '/D_DEBUG']
@@ -91,4 +91,3 @@ if sys.platform == 'win32':
MSVCCompiler.__init__(self, verbose, dry_run, force)
version_match = simple_version_match(start='Intel\(R\).*?64,')
self.__version = version_match
-
diff --git a/numpy/distutils/msvc9compiler.py b/numpy/distutils/msvc9compiler.py
new file mode 100644
index 000000000..c60826de9
--- /dev/null
+++ b/numpy/distutils/msvc9compiler.py
@@ -0,0 +1,21 @@
+import os
+from distutils.msvccompiler import *
+from distutils.msvc9compiler import MSVCCompiler as distutils_MSVCCompiler
+
+
+class MSVCCompiler(distutils_MSVCCompiler):
+ def __init__(self, verbose=0, dry_run=0, force=0):
+ distutils_MSVCCompiler.__init__(self, verbose, dry_run, force)
+
+ def initialize(self, plat_name=None):
+ environ_lib = os.getenv('lib')
+ environ_include = os.getenv('include')
+ distutils_MSVCCompiler.initialize(self, plat_name)
+ if environ_lib is not None:
+ os.environ['lib'] = environ_lib + os.environ['lib']
+ if environ_include is not None:
+ os.environ['include'] = environ_include + os.environ['include']
+
+ def manifest_setup_ldargs(self, output_filename, build_temp, ld_args):
+ ld_args.append('/MANIFEST')
+ distutils_MSVCCompiler.manifest_setup_ldargs(self, output_filename, build_temp, ld_args)
diff --git a/numpy/distutils/msvccompiler.py b/numpy/distutils/msvccompiler.py
new file mode 100644
index 000000000..cb9b81a71
--- /dev/null
+++ b/numpy/distutils/msvccompiler.py
@@ -0,0 +1,17 @@
+import os
+from distutils.msvccompiler import *
+from distutils.msvccompiler import MSVCCompiler as distutils_MSVCCompiler
+
+
+class MSVCCompiler(distutils_MSVCCompiler):
+ def __init__(self, verbose=0, dry_run=0, force=0):
+ distutils_MSVCCompiler.__init__(self, verbose, dry_run, force)
+
+ def initialize(self, plat_name=None):
+ environ_lib = os.getenv('lib')
+ environ_include = os.getenv('include')
+ distutils_MSVCCompiler.initialize(self, plat_name)
+ if environ_lib is not None:
+ os.environ['lib'] = environ_lib + os.environ['lib']
+ if environ_include is not None:
+ os.environ['include'] = environ_include + os.environ['include']
diff --git a/numpy/distutils/system_info.py b/numpy/distutils/system_info.py
index 90c053298..b09b5d40b 100644
--- a/numpy/distutils/system_info.py
+++ b/numpy/distutils/system_info.py
@@ -999,8 +999,8 @@ class mkl_info(system_info):
plt = '64'
#l = 'mkl_ipf'
elif cpu.is_Xeon():
- plt = 'em64t'
- #l = 'mkl_em64t'
+ plt = 'intel64'
+ #l = 'mkl_intel64'
else:
plt = '32'
#l = 'mkl_ia32'