提交 f5802f01 authored 作者: Roy Xue's avatar Roy Xue

Fix idx problems

上级 4abd8af7
...@@ -698,17 +698,17 @@ class ProfileStats(object): ...@@ -698,17 +698,17 @@ class ProfileStats(object):
# Update the Python emulating dicts and add the memory allocated by the node # Update the Python emulating dicts and add the memory allocated by the node
idx2 = 0 idx2 = 0
for out in node.outputs: for out in node.outputs:
if (dmap and idx2 in dmap) or (vmap and idx2 in vmap): if (dmap and idx2 in dmap) or (vmap and idx2 in vmap):
# This is needed for destroy_map in case it return a partial view that is destroyed. # This is needed for destroy_map in case it return a partial view that is destroyed.
# So the output could be different then the input. # So the output could be different then the input.
for ins in node.inputs: for ins in node.inputs:
assert len[ins] == 1 assert len[ins] == 1
view_of[out] = view_of.get(ins, ins)# This get make that we keep trac of view only again the original view_of[out] = view_of.get(ins, ins)# This get make that we keep trac of view only again the original
viewed_by[ins].append(out) viewed_by[ins].append(out)
else: else:
running_memory_size += var_mem[out] running_memory_size += var_mem[out]
node_memory_size += var_mem[out] node_memory_size += var_mem[out]
idx2 += 1 idx2 += 1
running_max_memory_size = max(running_max_memory_size, running_memory_size) running_max_memory_size = max(running_max_memory_size, running_memory_size)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论