Fix I4TZAN

If key not exist in map(while deep copy a cell), not throw exception
while pop key.
This commit is contained in:
zhangzhaoju 2022-02-25 15:31:57 +08:00
parent d8feb92a86
commit c59e31f373
1 changed files with 3 additions and 1 deletions

View File

@ -322,7 +322,9 @@ class Cell(Cell_):
if context.get_context is not None and context._get_mode() == context.PYNATIVE_MODE:
_pynative_executor.del_cell(str(id(self)))
cells_compile_cache.pop(id(self))
# while deepcopy a cell instance, the copied cell instance can't be added to cells_compile_cache
# here using pop(id(self), None) to avoid KeyError exception
cells_compile_cache.pop(id(self), None)
if self.compile_cache:
_cell_graph_executor.del_net_res(self.compile_cache)