Skip to content

Commit b6a38d0

Browse files
[security] refine _get_program_cache_key (#61827) (#61896)
* security, refine _get_program_cache_key
1 parent 39010bf commit b6a38d0

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

python/paddle/base/executor.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -682,15 +682,19 @@ def _get_varname_from_block(block):
682682
)
683683

684684

685-
def _get_program_cache_key(feed, fetch_list):
685+
def _get_feed_fetch_var_names(feed, fetch_list):
686686
feed_var_names = []
687687
if isinstance(feed, dict):
688688
feed_var_names = list(feed.keys())
689689
elif isinstance(feed, (list, tuple)):
690690
for i, each in enumerate(feed):
691691
feed_var_names += list(each.keys())
692692
fetch_var_names = list(map(_to_name_str, fetch_list))
693-
return str(feed_var_names + fetch_var_names)
693+
return feed_var_names + fetch_var_names
694+
695+
696+
def _get_program_cache_key(feed, fetch_list):
697+
return str(_get_feed_fetch_var_names(feed, fetch_list))
694698

695699

696700
def _as_lodtensor(data, place, dtype=None):
@@ -1026,7 +1030,7 @@ def _get_program_and_executor(self, cached_data):
10261030

10271031
if enable_inplace or enable_addto:
10281032
# inplace should skip feed and fetch var
1029-
skip_var_names = eval(_get_program_cache_key(feed, fetch_list))
1033+
skip_var_names = _get_feed_fetch_var_names(feed, fetch_list)
10301034
_apply_inplace_addto_pass(
10311035
program, enable_inplace, enable_addto, skip_var_names
10321036
)

0 commit comments

Comments
 (0)