提交 dd3608d7 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix the problem with partial set_subtensor coverage in gpuarray.

上级 ab1d2d48
...@@ -308,7 +308,7 @@ class GpuIncSubtensor(IncSubtensor): ...@@ -308,7 +308,7 @@ class GpuIncSubtensor(IncSubtensor):
returns a C code expression to copy source into view, and returns a C code expression to copy source into view, and
return 0 on success return 0 on success
""" """
return """GpuArray_move(&%(view)s->ga, &%(source)s->ga)""" % locals() return """GpuArray_setarray(&%(view)s->ga, &%(source)s->ga)""" % locals()
def c_support_code_apply(self, node, nodename): def c_support_code_apply(self, node, nodename):
gop = self.iadd_node.op gop = self.iadd_node.op
...@@ -351,7 +351,7 @@ class GpuIncSubtensor(IncSubtensor): ...@@ -351,7 +351,7 @@ class GpuIncSubtensor(IncSubtensor):
elemwise_version = self.iadd_node.c_code_cache_version() elemwise_version = self.iadd_node.c_code_cache_version()
if not parent_version or not elemwise_version: if not parent_version or not elemwise_version:
return return
return parent_version + elemwise_version + (0,) return parent_version + elemwise_version + (1,)
class GpuAdvancedIncSubtensor1(HideC, tensor.AdvancedIncSubtensor1): class GpuAdvancedIncSubtensor1(HideC, tensor.AdvancedIncSubtensor1):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论