Add checks for dnn config variables before adding them headers and library paths

上级 48325e33
...@@ -2847,13 +2847,19 @@ class GpuDnnTransformerDescriptor(COp): ...@@ -2847,13 +2847,19 @@ class GpuDnnTransformerDescriptor(COp):
return ['cudnn.h', 'cudnn_helper.h'] return ['cudnn.h', 'cudnn_helper.h']
def c_header_dirs(self): def c_header_dirs(self):
return [os.path.dirname(__file__), config.dnn.include_path] header_dirs = [os.path.dirname(__file__)]
if config.dnn.include_path:
headers_dirs += [config.dnn.include_path]
return header_dirs
def c_libraries(self): def c_libraries(self):
return ['cudnn'] return ['cudnn']
def c_lib_dirs(self): def c_lib_dirs(self):
return [config.dnn.library_path] lib_dirs = []
if config.dnn.library_path:
lib_dirs += [config.dnn.library_path]
return lib_dirs
def do_constant_folding(self, node): def do_constant_folding(self, node):
return False return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论