提交 848fae56 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix version for GpuCAReduceCPY

上级 31c951ee
......@@ -2885,8 +2885,8 @@ class GpuCAReduceCPY(GpuKernelBase, HideC, CAReduceDtype):
return code
def c_code_cache_version(self):
return (2, self.GpuKernelBase_version)
def c_code_cache_version_apply(self, node):
return (2, self.kernel_version(node))
def generate_kernel(self, node, odtype, redux):
if isinstance(self.scalar_op, scalar.basic.Add):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论