提交 17f74589 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Some minor fixes from review.

上级 de4be8be
......@@ -317,7 +317,7 @@ class ConfigParam(object):
val_str = self.default()
else:
val_str = self.default
self.__set__(None, val_str)
self.__set__(cls, val_str)
# print "RVAL", self.val
return self.val
......
......@@ -55,15 +55,17 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
return 1;
}
"""
params = ["-l", "cudnn", "-I" + os.path.dirname(__file__)]
if config.dnn.include_path:
params.append("-I" + config.dnn.include_path)
if config.dnn.library_path:
params.append("-L" + config.dnn.library_path)
# Do not run here the test program. It would run on the
# default gpu, not the one selected by the user. If mixed
# GPU are installed or if the GPUs are configured in
# exclusive mode, this cause bad detection.
comp, out, err = NVCC_compiler.try_flags(
["-l", "cudnn", "-I" + os.path.dirname(__file__),
"-I" + config.dnn.include_path,
"-L" + config.dnn.library_path],
preambule=preambule, body=body,
params=params, preambule=preambule, body=body,
try_run=False, output=True)
dnn_available.avail = comp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论