提交 2a9acfeb authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add -rpath for cudnn library.

上级 fac25eaf
......@@ -362,6 +362,9 @@ class DnnVersion(GpuOp):
def c_lib_dirs(self):
return [config.dnn.library_path]
def c_compile_args(self):
return ['-rpath', config.dnn.library_path]
def c_support_code(self):
return """
#if PY_MAJOR_VERSION >= 3
......
......@@ -94,6 +94,9 @@ class DnnBase(GpuOp, COp):
def c_lib_dirs(self):
return [config.dnn.library_path]
def c_compile_args(self):
return ['-rpath', config.dnn.library_path]
class GpuDnnConvDesc(GpuOp):
"""
......
......@@ -171,6 +171,9 @@ class DnnBase(COp):
def c_lib_dirs(self):
return [config.dnn.library_path]
def c_compile_args(self):
return ['-rpath', config.dnn.library_path]
def c_code_cache_version(self):
return (super(DnnBase, self).c_code_cache_version(), version())
......@@ -190,6 +193,9 @@ class DnnVersion(Op):
def c_lib_dirs(self):
return [config.dnn.library_path]
def c_compile_args(self):
return ['-rpath', config.dnn.library_path]
def c_support_code(self):
return """
#if PY_MAJOR_VERSION >= 3
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论