提交 71644ebe authored 作者: notoraptor's avatar notoraptor

Fix rpath for cuDNN module.

上级 c32b0db8
...@@ -228,10 +228,7 @@ dnn_available.msg = None ...@@ -228,10 +228,7 @@ dnn_available.msg = None
def CUDNNDataType(name, freefunc=None): def CUDNNDataType(name, freefunc=None):
cargs = [] cargs = []
if config.dnn.bin_path: if config.dnn.bin_path:
if sys.platform == 'darwin':
cargs.append('-Wl,-rpath,' + config.dnn.bin_path) cargs.append('-Wl,-rpath,' + config.dnn.bin_path)
else:
cargs.append('-Wl,-rpath,"' + config.dnn.bin_path + '"')
return CDataType(name, freefunc, return CDataType(name, freefunc,
headers=['cudnn.h'], headers=['cudnn.h'],
...@@ -260,10 +257,7 @@ class DnnVersion(Op): ...@@ -260,10 +257,7 @@ class DnnVersion(Op):
def c_compile_args(self): def c_compile_args(self):
if config.dnn.bin_path: if config.dnn.bin_path:
if sys.platform == 'darwin':
return ['-Wl,-rpath,' + config.dnn.bin_path] return ['-Wl,-rpath,' + config.dnn.bin_path]
else:
return ['-Wl,-rpath,"' + config.dnn.bin_path + '"']
return [] return []
def c_support_code(self): def c_support_code(self):
...@@ -392,10 +386,7 @@ class DnnBase(COp): ...@@ -392,10 +386,7 @@ class DnnBase(COp):
def c_compile_args(self): def c_compile_args(self):
if config.dnn.bin_path: if config.dnn.bin_path:
if sys.platform == 'darwin':
return ['-Wl,-rpath,' + config.dnn.bin_path] return ['-Wl,-rpath,' + config.dnn.bin_path]
else:
return ['-Wl,-rpath,"' + config.dnn.bin_path + '"']
return [] return []
def c_code_cache_version(self): def c_code_cache_version(self):
...@@ -440,10 +431,7 @@ class GpuDnnConvDesc(COp): ...@@ -440,10 +431,7 @@ class GpuDnnConvDesc(COp):
def c_compile_args(self): def c_compile_args(self):
if config.dnn.bin_path: if config.dnn.bin_path:
if sys.platform == 'darwin':
return ['-Wl,-rpath,' + config.dnn.bin_path] return ['-Wl,-rpath,' + config.dnn.bin_path]
else:
return ['-Wl,-rpath,"' + config.dnn.bin_path + '"']
return [] return []
def do_constant_folding(self, node): def do_constant_folding(self, node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论