提交 d88c1a84 authored 作者: Frederic Bastien's avatar Frederic Bastien

GpuKernelBase if the subclass don't implement the version, don't give one.

上级 1b530f72
...@@ -293,6 +293,9 @@ class GpuKernelBase(object): ...@@ -293,6 +293,9 @@ class GpuKernelBase(object):
# This is a shorthand for if your op only has a fixed version # This is a shorthand for if your op only has a fixed version
# You can reimplement it, but make sure to call kernel_version() # You can reimplement it, but make sure to call kernel_version()
def c_code_cache_version_apply(self, node): def c_code_cache_version_apply(self, node):
v = self.c_code_cache_version()
if not v:
return ()
return (self.c_code_cache_version(), self.kernel_version(node)) return (self.c_code_cache_version(), self.kernel_version(node))
def kernel_version(self, node): def kernel_version(self, node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论