提交 130e6464 authored 作者: Lijun Xue's avatar Lijun Xue

Fix list remove value error

上级 f479a4f8
......@@ -948,6 +948,7 @@ class ProfileStats(object):
mem_freed += var_mem[ins]
elif ins in view_of:
origin = view_of[ins]
if ins in viewed_by[origin]:
viewed_by[origin].remove(ins)
viewedby_remove[origin].append(ins)
if (not viewed_by[origin] and
......
......@@ -982,6 +982,7 @@ class VM_Linker(link.LocalLinker):
pre_allocated.add(out)
elif ins in view_of:
origin = view_of[ins]
if ins in viewed_by[origin]:
viewed_by[origin].remove(ins)
if (not viewed_by[origin] and
origin not in fgraph.inputs and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论