提交 b4ec4265 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

Fixed a linker bug when an output is a Constant

上级 22cebe6c
...@@ -184,6 +184,9 @@ def map_storage(env, order, input_storage, output_storage): ...@@ -184,6 +184,9 @@ def map_storage(env, order, input_storage, output_storage):
storage_map[r] = [r.data] storage_map[r] = [r.data]
for r in node.outputs: for r in node.outputs:
storage_map.setdefault(r, [None]) storage_map.setdefault(r, [None])
for r in env.outputs:
if isinstance(r, graph.Constant):
storage_map.setdefault(r, [r.data])
if output_storage is None: if output_storage is None:
output_storage = [storage_map[r] for r in env.outputs] output_storage = [storage_map[r] for r in env.outputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论