提交 954f5fce authored 作者: Lijun Xue's avatar Lijun Xue

wrap code in to a function

上级 c41c4307
...@@ -900,13 +900,13 @@ class VM_Linker(link.LocalLinker): ...@@ -900,13 +900,13 @@ class VM_Linker(link.LocalLinker):
for var in fgraph.inputs: for var in fgraph.inputs:
compute_map_re[var][0] = 1 compute_map_re[var][0] = 1
reallocated_info = {}
if getattr(fgraph.profile, 'dependencies', None): if getattr(fgraph.profile, 'dependencies', None):
dependencies = getattr(fgraph.profile, 'dependencies') dependencies = getattr(fgraph.profile, 'dependencies')
else: else:
dependencies = self.compute_gc_dependencies(storage_map) dependencies = self.compute_gc_dependencies(storage_map)
def calculate_reallocate_info(order, fgraph, dependencies):
reallocated_info = {}
viewed_by = {} viewed_by = {}
for var in fgraph.variables: for var in fgraph.variables:
viewed_by[var] = [] viewed_by[var] = []
...@@ -982,6 +982,10 @@ class VM_Linker(link.LocalLinker): ...@@ -982,6 +982,10 @@ class VM_Linker(link.LocalLinker):
if reuse_out: if reuse_out:
reallocated_info[ins] = [ins, reuse_out] reallocated_info[ins] = [ins, reuse_out]
return reallocated_info
reallocated_info = calculate_reallocate_info(order, fgraph, dependencies)
for node in order: for node in order:
try: try:
thunks.append(node.op.make_thunk(node, thunks.append(node.op.make_thunk(node,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论