提交 2367655c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

GpuDnnConvDesc doesn't inherit from DnnBase.

上级 dafbe512
......@@ -426,13 +426,28 @@ class GpuDnnConvDesc(COp):
return ['cudnn.h', 'cudnn_helper.h']
def c_header_dirs(self):
return [os.path.dirname(__file__), config.dnn.include_path]
dirs = [os.path.dirname(__file__)]
if config.dnn.include_path:
dirs.append(config.dnn.include_path)
if config.cuda.include_path:
dirs.append(config.cuda.include_path)
return dirs
def c_libraries(self):
return ['cudnn']
def c_lib_dirs(self):
return [config.dnn.library_path]
if config.dnn.library_path:
return [config.dnn.library_path]
return []
def c_compile_args(self):
if config.dnn.bin_path:
if sys.platform == 'darwin':
return ['-Wl,-rpath,' + config.dnn.bin_path]
else:
return ['-Wl,-rpath,"' + config.dnn.bin_path + '"']
return []
def do_constant_folding(self, node):
return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论