diff --git a/paddle/fluid/framework/operator.cc b/paddle/fluid/framework/operator.cc index b294779837fc2c63d95e8646477da3d5f1e46025..8e9753e1d7fbdbfb56434f05abe80869748afb80 100644 --- a/paddle/fluid/framework/operator.cc +++ b/paddle/fluid/framework/operator.cc @@ -3260,6 +3260,7 @@ void OperatorWithKernel::BuildPhiKernelContext( #if defined(PADDLE_WITH_MKLDNN) || defined(PADDLE_WITH_CUDA) auto& runtime_attrs = RuntimeAttrs(); for (const auto& attr_iter : runtime_attrs) { + need_prepare_phi_data_ = true; auto& attr_name = attr_iter.first; auto& attr = attr_iter.second; auto attr_propertys = paddle::operators::GetExtraAttrProperties(attr_name);