From e03b27b1da7719246ed1f556d95436ce739f81a7 Mon Sep 17 00:00:00 2001 From: xiongkun Date: Wed, 1 Jun 2022 03:57:19 +0000 Subject: [PATCH] fix memory leakage --- .../fluid/dygraph/dygraph_to_static/program_translator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/dygraph/dygraph_to_static/program_translator.py b/python/paddle/fluid/dygraph/dygraph_to_static/program_translator.py index 2efb6965085de..eac2941c09778 100644 --- a/python/paddle/fluid/dygraph/dygraph_to_static/program_translator.py +++ b/python/paddle/fluid/dygraph/dygraph_to_static/program_translator.py @@ -61,7 +61,7 @@ class FunctionCache(object): def __init__(self): # Caches the converted static functions. {dygraph_func: static_func} - self._converted_static_func_caches = dict() + self._converted_static_func_caches = weakref.WeakKeyDictionary() # Caches the converted ast node for same source code. {source_code: ast_root} self._code_to_ast_caches = dict() self._dygraph_to_static = DygraphToStaticAst()