提交 ec1fc9de authored 作者: Zotov Yuriy's avatar Zotov Yuriy 提交者: Arnaud Bergeron

Function.__call__(): correct indexing variable for group of args with shared memory

上级 e89aee9d
...@@ -851,9 +851,9 @@ class Function(object): ...@@ -851,9 +851,9 @@ class Function(object):
for group in args_share_memory: for group in args_share_memory:
if len(group) > 1: if len(group) > 1:
# copy all but the first # copy all but the first
for idx in group[1:]: for j in group[1:]:
self.input_storage[i].storage[0] = copy.copy( self.input_storage[j].storage[0] = copy.copy(
self.input_storage[i].storage[0]) self.input_storage[j].storage[0])
# Check if inputs are missing, or if inputs were set more than once, or # Check if inputs are missing, or if inputs were set more than once, or
# if we tried to provide inputs that are supposed to be implicit. # if we tried to provide inputs that are supposed to be implicit.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论