提交 6ce6e2e5 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #5544 from notoraptor/wl-rpath

Fix `-Wl,rpath` parameter for Windows (fix #5463).
...@@ -233,17 +233,23 @@ class DnnBase(COp): ...@@ -233,17 +233,23 @@ class DnnBase(COp):
'gpuarray_helper.h'] 'gpuarray_helper.h']
def c_header_dirs(self): def c_header_dirs(self):
return [os.path.dirname(__file__), pygpu.get_include(), dirs = [os.path.dirname(__file__), pygpu.get_include()]
config.dnn.include_path] if config.dnn.include_path:
dirs.append(config.dnn.include_path)
return dirs
def c_libraries(self): def c_libraries(self):
return ['cudnn', 'gpuarray'] return ['cudnn', 'gpuarray']
def c_lib_dirs(self): def c_lib_dirs(self):
if config.dnn.library_path:
return [config.dnn.library_path] return [config.dnn.library_path]
return []
def c_compile_args(self): def c_compile_args(self):
return ['-Wl,-rpath,' + config.dnn.library_path] if config.dnn.library_path:
return ['-Wl,-rpath,"' + config.dnn.library_path + '"']
return []
def c_code_cache_version(self): def c_code_cache_version(self):
return (super(DnnBase, self).c_code_cache_version(), version(), 1) return (super(DnnBase, self).c_code_cache_version(), version(), 1)
...@@ -256,16 +262,20 @@ class DnnVersion(Op): ...@@ -256,16 +262,20 @@ class DnnVersion(Op):
return ['cudnn.h'] return ['cudnn.h']
def c_header_dirs(self): def c_header_dirs(self):
return [config.dnn.include_path] return [config.dnn.include_path] if config.dnn.include_path else []
def c_libraries(self): def c_libraries(self):
return ['cudnn'] return ['cudnn']
def c_lib_dirs(self): def c_lib_dirs(self):
if config.dnn.library_path:
return [config.dnn.library_path] return [config.dnn.library_path]
return []
def c_compile_args(self): def c_compile_args(self):
return ['-Wl,-rpath,' + config.dnn.library_path] if config.dnn.library_path:
return ['-Wl,-rpath,"' + config.dnn.library_path + '"']
return []
def c_support_code(self): def c_support_code(self):
return """ return """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论