提交 e7c738a9 authored 作者: Frederic's avatar Frederic

Fix test crash in FAST_COMPILE.

This allow to free the temp memory with the old linkers.
上级 813c0076
...@@ -616,6 +616,7 @@ class PerformLinker(LocalLinker): ...@@ -616,6 +616,7 @@ class PerformLinker(LocalLinker):
f.allow_gc = self.allow_gc #HACK: this is a way of passing an arg to Function.__call__ f.allow_gc = self.allow_gc #HACK: this is a way of passing an arg to Function.__call__
add_clear_storage(f, computed, storage_map) add_clear_storage(f, computed, storage_map)
f.storage_map = storage_map
return f, [Container(input, storage) for input, storage in zip(fgraph.inputs, input_storage)], \ return f, [Container(input, storage) for input, storage in zip(fgraph.inputs, input_storage)], \
[Container(output, storage, True) for output, storage in zip(fgraph.outputs, output_storage)], \ [Container(output, storage, True) for output, storage in zip(fgraph.outputs, output_storage)], \
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论