提交 1cc3e5d6 authored 作者: Frederic Bastien's avatar Frederic Bastien

Add cuDNN version in the c code version in the old back-end as done in the new back-end.

上级 cc377cb5
...@@ -101,6 +101,9 @@ class DnnBase(GpuOp, COp): ...@@ -101,6 +101,9 @@ class DnnBase(GpuOp, COp):
def c_compile_args(self): def c_compile_args(self):
return ['-Wl,-rpath,' + config.dnn.library_path] return ['-Wl,-rpath,' + config.dnn.library_path]
def c_code_cache_version(self):
return (super(DnnBase, self).c_code_cache_version(), version())
class GpuDnnConvDesc(GpuOp): class GpuDnnConvDesc(GpuOp):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论