diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 46122635404771c5031a92af97adc925dacc0887..6ab31ec9463b8e0ac609e5feb9e71c1b14112d9e 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -489,7 +489,8 @@ class Operator(object): 'rnn_memory_helper_grad', 'conditional_block', 'while', 'send', 'recv', 'listen_and_serv', 'parallel_do', 'save_combine', 'load_combine', 'ncclInit', 'channel_create', 'channel_close', - 'channel_send', 'channel_recv', 'select', 'checkpoint_save' + 'channel_send', 'channel_recv', 'select', 'checkpoint_save', + 'checkpoint_save' } if type not in no_kernel_op_set: self.desc.infer_var_type(self.block.desc)