File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -260,14 +260,17 @@ def run(self,
260
260
scope = global_scope ()
261
261
262
262
program_cache = None
263
- program_cache_key = None
263
+
264
+ feed_var_names = feed .keys ()
265
+ fetch_var_names = [var .desc .name () for var in fetch_list ]
266
+ program_cache_key = str (feed_var_names + fetch_var_names )
267
+
264
268
if use_program_cache :
265
269
# find program cache by cache_key
266
- feed_var_names = feed .keys ()
267
- fetch_var_names = [var .desc .name () for var in fetch_list ]
268
- program_cache_key = str (feed_var_names + fetch_var_names )
269
270
program_cache = self .program_caches .get (program_cache_key , None )
270
271
# TODO(qiao): Should check program_cache and program are exactly the same.
272
+ else :
273
+ self .program_caches .pop (program_cache_key , None )
271
274
272
275
if program_cache is None :
273
276
program_cache = program .clone ()
You can’t perform that action at this time.
0 commit comments