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

Check format

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