提交 3aa7a24b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Always silently add the default library dirs, but don't append them to flags.

上级 034c9c37
...@@ -346,21 +346,15 @@ def try_blas_flag(flags): ...@@ -346,21 +346,15 @@ def try_blas_flag(flags):
return 0; return 0;
} }
""") """)
cflags = flags + ['-L' + d for d in theano.gof.cmodule.std_lib_dirs()]
res = GCC_compiler.try_compile_tmp( res = GCC_compiler.try_compile_tmp(
test_code, tmp_prefix='try_blas_', test_code, tmp_prefix='try_blas_',
flags=flags, try_run=True) flags=cflags, try_run=True)
# res[0]: shows successful compilation # res[0]: shows successful compilation
# res[1]: shows successful execution # res[1]: shows successful execution
if res and res[0] and res[1]: if res and res[0] and res[1]:
return ' '.join(flags) return ' '.join(flags)
else: else:
# Retry adding '-L' flags which may be required
flags = flags + ['-L' + d for d in theano.gof.cmodule.std_lib_dirs()]
res = GCC_compiler.try_compile_tmp(
test_code, tmp_prefix='try_blas_',
flags=flags, try_run=True)
if res and res[0] and res[1]:
return ' '.join(flags)
return "" return ""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论