提交 332d2fd7 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Use "mkl" package (mkl-service in conda) to check for MKL availability

上级 9d0f6f8b
...@@ -291,20 +291,27 @@ SOMEPATH/Canopy_64bit/User/lib/python2.7/site-packages/numpy/distutils/system_in ...@@ -291,20 +291,27 @@ SOMEPATH/Canopy_64bit/User/lib/python2.7/site-packages/numpy/distutils/system_in
"mk2_rt"]]) "mk2_rt"]])
# Anaconda # Anaconda
if "Anaconda" in sys.version and sys.platform == "win32": if "Anaconda" in sys.version and sys.platform == "win32":
lib_path = os.path.join(sys.prefix, 'pkgs') # If the "mkl-service" conda package (available through Python
for dir in os.listdir(lib_path): # package "mkl") is installed and importable, then the libraries
if dir.startswith("mkl-rt-"): # (installed by conda package "mkl-rt") are actually available.
lib_path = os.path.join(lib_path, dir, "DLLs") # Using "conda install mkl" will install both, as well as
break # optimized versions of numpy and scipy.
if os.path.exists(lib_path): try:
#-LC:\\Users\\*\\Anaconda\\libs import mkl
except ImportError as e:
_logger.info('Conda mkl is not available: %s', e)
else:
# This branch is executed if no exception was raised
lib_path = os.path.join(sys.prefix, 'DLLs')
flags = ['-L%s' % lib_path] flags = ['-L%s' % lib_path]
flags += ['-l%s' % l for l in ["mkl_core", flags += ['-l%s' % l for l in ["mkl_core",
"mkl_intel_thread", "mkl_intel_thread",
"mkl_rt"]] "mkl_rt"]]
if GCC_compiler.try_flags(flags):
return ' '.join(flags) return ' '.join(flags)
# if numpy was linked with library that are not installed, we # If numpy was linked with library that are not installed or
# the dev version of the package is not currently available, we
# can't reuse them. # can't reuse them.
if any(os.path.exists(dir) for dir in blas_info['library_dirs']): if any(os.path.exists(dir) for dir in blas_info['library_dirs']):
ret = ( ret = (
...@@ -316,10 +323,6 @@ SOMEPATH/Canopy_64bit/User/lib/python2.7/site-packages/numpy/distutils/system_in ...@@ -316,10 +323,6 @@ SOMEPATH/Canopy_64bit/User/lib/python2.7/site-packages/numpy/distutils/system_in
['-L%s' % l for l in blas_info['library_dirs']] + ['-L%s' % l for l in blas_info['library_dirs']] +
['-l%s' % l for l in blas_info['libraries']] + ['-l%s' % l for l in blas_info['libraries']] +
[]) [])
# ['-I%s' % l for l in blas_info['include_dirs']])
# if numpy was linked with library that are not installed or
# the dev version of the package is not currently available, we
# can't reuse them.
if GCC_compiler.try_flags(ret): if GCC_compiler.try_flags(ret):
return ' '.join(ret) return ' '.join(ret)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论