[PIR] Reify InferSymbolicShapeInterface (#60438)

* Reify InferSymbolicShapeInterface
PaddlePaddle · Jan 9, 2024 · b80ece9 · b80ece9
1 parent 0dfba1a
commit b80ece9
Show file tree

Hide file tree

Showing 16 changed files with 411 additions and 234 deletions.
diff --git a/paddle/cinn/hlir/framework/pir/utils.cc b/paddle/cinn/hlir/framework/pir/utils.cc
@@ -25,6 +25,7 @@
 #include "paddle/phi/common/data_type.h"
 #include "paddle/pir/core/builtin_op.h"
 #include "paddle/pir/core/builtin_type.h"
+#include "paddle/pir/dialect/shape/ir/shape_attribute.h"
 
 namespace cinn {
 namespace hlir {
@@ -146,6 +147,8 @@ utils::Attribute CompatibleInfo::ConvertAttribute(
   } else if (src_attr.isa<paddle::dialect::DataTypeAttribute>()) {
     auto dtype = src_attr.dyn_cast<paddle::dialect::DataTypeAttribute>().data();
     dst_attr = phi::DataTypeToString(dtype);
+  } else if (src_attr.isa<::pir::shape::SymbolAttribute>()) {
+    auto dst_attr = src_attr.dyn_cast<::pir::shape::SymbolAttribute>().data();
   } else if (src_attr.isa<::pir::ArrayAttribute>()) {
     auto attr_vec = src_attr.dyn_cast<::pir::ArrayAttribute>().AsVector();
     if (attr_vec.size() > 0) {

diff --git a/paddle/fluid/inference/paddle_inference.map b/paddle/fluid/inference/paddle_inference.map
@@ -82,7 +82,6 @@
 		*Pass*;
 		*profile*;
 		*phi*;
-		*pir*;
 		PD_*;
 		*cinn*;
 	local:

diff --git a/paddle/fluid/inference/utils/CMakeLists.txt b/paddle/fluid/inference/utils/CMakeLists.txt
@@ -13,7 +13,7 @@ cc_library(
   DEPS proto_desc enforce common)
 
 cc_library(table_printer SRCS table_printer.cc)
-paddle_test(test_table_printer SRCS table_printer_tester.cc)
+paddle_test(test_table_printer SRCS table_printer_tester.cc DEPS pir)
 
 proto_library(shape_range_info_proto SRCS shape_range_info.proto)