diff --git a/paddle/fluid/framework/operator.cc b/paddle/fluid/framework/operator.cc index b91ee3c2d633dff148e9118ea5c3bea50768ca89..ffdc3e6d3c2bc7349fc6c83a5aa8f62d95406c6a 100644 --- a/paddle/fluid/framework/operator.cc +++ b/paddle/fluid/framework/operator.cc @@ -1210,6 +1210,9 @@ void OperatorWithKernel::RunImpl(const Scope& scope, VLOG(6) << "Static mode ChoosePhiKernel - kernel `" << pt_kernel_name << "` not found."; } + } else { + pt_kernel_name = pt_kernel_signature_->name; + pt_kernel_key = TransOpKernelTypeToPhiKernelKey(*kernel_type_.get()); } #ifdef PADDLE_WITH_XPU bool is_xpu_unsupport =