提交 e266f7af authored 作者: Lijun Xue's avatar Lijun Xue

Fix

上级 f50aa790
...@@ -953,7 +953,8 @@ class VM_Linker(link.LocalLinker): ...@@ -953,7 +953,8 @@ class VM_Linker(link.LocalLinker):
if reuse_out: if reuse_out:
break break
for out in order[i].outputs: for out in order[i].outputs:
if getattr(out, 'ndim', None) == 0 and out not in pre_allocated: if (getattr(out, 'ndim', None) == 0 and out not in pre_allocated
and ins.type == out.type):
reuse_out = out reuse_out = out
pre_allocated.add(out) pre_allocated.add(out)
elif ins in view_of: elif ins in view_of:
...@@ -968,7 +969,8 @@ class VM_Linker(link.LocalLinker): ...@@ -968,7 +969,8 @@ class VM_Linker(link.LocalLinker):
if reuse_out: if reuse_out:
break break
for out in order[i].outputs: for out in order[i].outputs:
if getattr(out, 'ndim', None) == 0 and out not in pre_allocated: if (getattr(out, 'ndim', None) == 0 and out not in pre_allocated
and ins.type == out.type):
reuse_out = out reuse_out = out
pre_allocated.add(out) pre_allocated.add(out)
...@@ -976,7 +978,7 @@ class VM_Linker(link.LocalLinker): ...@@ -976,7 +978,7 @@ class VM_Linker(link.LocalLinker):
reallocated_info[ins] = [ins, reuse_out] reallocated_info[ins] = [ins, reuse_out]
for pair in reallocated_info.values(): for pair in reallocated_info.values():
storage_map[pair[1]] = storage_map[pair[0]] storage_map[pair[1]][0] = storage_map[pair[0]][0]
for node in order: for node in order:
try: try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论