提交 29abda4b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

On windows, check for intel OpenMP if gnu failed for mkl

上级 12afd65f
...@@ -1404,13 +1404,23 @@ def default_blas_ldflags(): ...@@ -1404,13 +1404,23 @@ def default_blas_ldflags():
thr = 'mkl_gnu_thread' thr = 'mkl_gnu_thread'
else: else:
thr = 'mkl_intel_thread' thr = 'mkl_intel_thread'
base_flags = list(flags)
flags += ['-l%s' % l for l in ["mkl_core", flags += ['-l%s' % l for l in ["mkl_core",
thr, thr,
"mkl_rt"]] "mkl_rt"]]
res = try_blas_flag(flags) res = try_blas_flag(flags)
if not res and sys.platform == "win32" and thr == "mkl_gnu_thread":
# Check if it would work for intel OpenMP on windows
flags = base_flags + ['-l%s' % l for l in ["mkl_core",
'mkl_intel_thread',
"mkl_rt"]]
res = try_blas_flag(flags)
if res: if res:
check_mkl_openmp() check_mkl_openmp()
return res return res
flags.extend(['-Wl,-rpath,' + l for l in flags.extend(['-Wl,-rpath,' + l for l in
blas_info.get('library_dirs', [])]) blas_info.get('library_dirs', [])])
res = try_blas_flag(flags) res = try_blas_flag(flags)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论