提交 2d79c5f4 authored 作者: Roy Xue's avatar Roy Xue 提交者: Lijun Xue

Check format

上级 ae06fd72
......@@ -887,7 +887,7 @@ class VM_Linker(link.LocalLinker):
fgraph, order, input_storage, output_storage)
compute_map = {}
for k in storage_map:
compute_map[k] = [k.owner is None]
compute_map[k] = [k.owner is None]
thunks = []
......@@ -895,12 +895,12 @@ class VM_Linker(link.LocalLinker):
idx = order.index(node)
for ins in node.inputs:
if ins.allow_gc and ins.ndim == 0:
# check if input variable can be gc and ndim = 0
for i in range(idx+1, len(order)):
# check if input variable can be gc and ndim = 0
for i in range(idx + 1, len(order)):
for outs in order[i].outputs:
if outs.ndim == 0 and outs == ins:
storage_map[outs] == storage_map[ins]
for node in order:
try:
thunks.append(node.op.make_thunk(node,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论