From 2e0c86780bf0909a621a220ed8afeb9f70463681 Mon Sep 17 00:00:00 2001 From: zhangbo9674 <82555433+zhangbo9674@users.noreply.github.com> Date: Mon, 5 Jun 2023 20:11:00 +0800 Subject: [PATCH] fix bug (#54363) --- test/cpp/ir/core/ir_exe_test.cc | 2 +- test/cpp/ir/core/phi_kernel_adaptor.h | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/test/cpp/ir/core/ir_exe_test.cc b/test/cpp/ir/core/ir_exe_test.cc index 844dfee82f3..898e46bbfb4 100644 --- a/test/cpp/ir/core/ir_exe_test.cc +++ b/test/cpp/ir/core/ir_exe_test.cc @@ -15,9 +15,9 @@ #include #include "paddle/fluid/ir/dialect/pd_dialect.h" -#include "paddle/fluid/ir/dialect/pd_interface.h" #include "paddle/fluid/ir/dialect/pd_type.h" #include "paddle/fluid/ir/dialect/utils.h" +#include "paddle/fluid/ir/interface/op_yaml_info.h" #include "paddle/ir/core/builtin_attribute.h" #include "paddle/ir/core/builtin_dialect.h" #include "paddle/ir/core/builtin_op.h" diff --git a/test/cpp/ir/core/phi_kernel_adaptor.h b/test/cpp/ir/core/phi_kernel_adaptor.h index 7b6a2253596..601e37649c5 100644 --- a/test/cpp/ir/core/phi_kernel_adaptor.h +++ b/test/cpp/ir/core/phi_kernel_adaptor.h @@ -15,10 +15,10 @@ #pragma once #include "paddle/fluid/ir/dialect/pd_dialect.h" -#include "paddle/fluid/ir/dialect/pd_interface.h" #include "paddle/fluid/ir/dialect/pd_op.h" #include "paddle/fluid/ir/dialect/pd_type.h" #include "paddle/fluid/ir/dialect/utils.h" +#include "paddle/fluid/ir/interface/op_yaml_info.h" #include "paddle/ir/core/builtin_attribute.h" #include "paddle/ir/core/builtin_dialect.h" #include "paddle/ir/core/builtin_op.h" @@ -94,8 +94,8 @@ void build_context(ir::Operation* op, paddle::framework::Scope* scope, T* ctx, bool is_infer_meta = true) { - paddle::dialect::GetOpInfoInterface op_info_interface = - op->dyn_cast(); + paddle::dialect::OpYamlInfoInterface op_info_interface = + op->dyn_cast(); auto op_info_res = op_info_interface.GetOpInfo(); auto input_info = std::get<0>(op_info_res); @@ -188,8 +188,8 @@ class PhiKernelAdaptor { interface.InferShape(&ctx); - paddle::dialect::GetOpInfoInterface op_info_interface = - (*it)->dyn_cast(); + paddle::dialect::OpYamlInfoInterface op_info_interface = + (*it)->dyn_cast(); auto op_info_res = op_info_interface.GetOpInfo(); auto runtime_info = std::get<3>(op_info_res); -- GitLab