提交 9711cadd authored 作者: notoraptor's avatar notoraptor

Avoid rpath on Windows for cuDNN.

上级 71644ebe
...@@ -227,7 +227,7 @@ dnn_available.msg = None ...@@ -227,7 +227,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 and sys.platform != 'win32':
cargs.append('-Wl,-rpath,' + config.dnn.bin_path) cargs.append('-Wl,-rpath,' + config.dnn.bin_path)
return CDataType(name, freefunc, return CDataType(name, freefunc,
...@@ -256,7 +256,7 @@ class DnnVersion(Op): ...@@ -256,7 +256,7 @@ class DnnVersion(Op):
return [config.dnn.library_path] return [config.dnn.library_path]
def c_compile_args(self): def c_compile_args(self):
if config.dnn.bin_path: if config.dnn.bin_path and sys.platform != 'win32':
return ['-Wl,-rpath,' + config.dnn.bin_path] return ['-Wl,-rpath,' + config.dnn.bin_path]
return [] return []
...@@ -385,7 +385,7 @@ class DnnBase(COp): ...@@ -385,7 +385,7 @@ class DnnBase(COp):
return [config.dnn.library_path] return [config.dnn.library_path]
def c_compile_args(self): def c_compile_args(self):
if config.dnn.bin_path: if config.dnn.bin_path and sys.platform != 'win32':
return ['-Wl,-rpath,' + config.dnn.bin_path] return ['-Wl,-rpath,' + config.dnn.bin_path]
return [] return []
...@@ -430,7 +430,7 @@ class GpuDnnConvDesc(COp): ...@@ -430,7 +430,7 @@ class GpuDnnConvDesc(COp):
return [config.dnn.library_path] return [config.dnn.library_path]
def c_compile_args(self): def c_compile_args(self):
if config.dnn.bin_path: if config.dnn.bin_path and sys.platform != 'win32':
return ['-Wl,-rpath,' + config.dnn.bin_path] return ['-Wl,-rpath,' + config.dnn.bin_path]
return [] return []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论