提交 12ff9ab5 authored 作者: ruslanagit's avatar ruslanagit

fix merge conflict

上级 27146e2e
...@@ -281,28 +281,6 @@ def dnn_available(): ...@@ -281,28 +281,6 @@ def dnn_available():
dnn_available.msg = "Device not supported" dnn_available.msg = "Device not supported"
dnn_available.avail = False dnn_available.avail = False
else: else:
<<<<<<< HEAD
preambule = """
#include <stdio.h>
#include <cuda.h>
#include <cudnn.h>
#include <cudnn_helper.h>
"""
body = """
cudnnHandle_t _handle = NULL;
cudnnStatus_t err;
if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
fprintf(stderr, "could not create cuDNN handle: %s",
cudnnGetErrorString(err));
return 1;
}
"""
# to support path that includes spaces, we need to wrap it with double quotes on Windows
path_wrapper = "\"" if os.name =='nt' else ""
params = ["-l", "cudnn"]
params.extend(['-I%s%s%s' % (path_wrapper,os.path.dirname(__file__),path_wrapper)])
=======
preambule = textwrap.dedent( preambule = textwrap.dedent(
""" """
#include <stdio.h> #include <stdio.h>
...@@ -321,10 +299,12 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) { ...@@ -321,10 +299,12 @@ if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
return 1; return 1;
} }
""") """)
params = ["-l", "cudnn", "-I" + os.path.dirname(__file__)] # to support path that includes spaces, we need to wrap it with double quotes on Windows
>>>>>>> refs/remotes/Theano/master path_wrapper = "\"" if os.name =='nt' else ""
params = ["-l", "cudnn"]
params.extend(['-I%s%s%s' % (path_wrapper, os.path.dirname(__file__), path_wrapper)])
if config.dnn.include_path: if config.dnn.include_path:
params.extend(['-I%s%s%s' % (path_wrapper, config.dnn.include_path, path_wrapper)]) params.extend(['-I%s%s%s' % (path_wrapper, config.dnn.include_path, path_wrapper)])
if config.dnn.library_path: if config.dnn.library_path:
params.extend(['-L%s%s%s' % (path_wrapper, config.dnn.library_path, path_wrapper)]) params.extend(['-L%s%s%s' % (path_wrapper, config.dnn.library_path, path_wrapper)])
if config.nvcc.compiler_bindir: if config.nvcc.compiler_bindir:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论