提交 1cab9d11 authored 作者: Frederic's avatar Frederic

add c_code_cache_version to GpuAdvancedIncSubtensor1 as it now have c_code!

上级 accccfe7
...@@ -2442,6 +2442,9 @@ class GpuAdvancedIncSubtensor1(tensor.AdvancedIncSubtensor1, GpuOp): ...@@ -2442,6 +2442,9 @@ class GpuAdvancedIncSubtensor1(tensor.AdvancedIncSubtensor1, GpuOp):
x[i] += y x[i] += y
out[0] = x out[0] = x
def c_code_cache_version(self):
return (1,)
def c_code(self, node, name, inputs, outputs, sub): def c_code(self, node, name, inputs, outputs, sub):
if (self.set_instead_of_inc) or \ if (self.set_instead_of_inc) or \
(node.inputs[0].ndim != node.inputs[1].ndim): (node.inputs[0].ndim != node.inputs[1].ndim):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论