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

Fix version for GpuCAReduceCPY

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