提交 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):
def c_compile_args(self):
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):
"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论