Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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 20c27bbb00
commit 7a707d465b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/python/mindspore/nn/cell.py
View File

@ -324,7 +324,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)