support selectedrows print (#56621)

paddle/fluid/ir/dialect/paddle_kernel_dialect/ir/kernel_dialect.cc

@@ -39,17 +39,31 @@ void PaddleKernelDialect::initialize() {
 }
 
 void PaddleKernelDialect::PrintType(ir::Type type, std::ostream &os) const {
-  AllocatedDenseTensorType tensor_type =
-      type.dyn_cast<AllocatedDenseTensorType>();
+  if (type.isa<AllocatedDenseTensorType>()) {
+    AllocatedDenseTensorType tensor_type =
+        type.dyn_cast<AllocatedDenseTensorType>();
 
-  os << phi::AllocationTypeStr(tensor_type.place().GetType()) << "_";
-  os << "tensor<";
-  for (auto d : phi::vectorize(tensor_type.dims())) {
-    os << d;
-    os << "x";
+    os << phi::AllocationTypeStr(tensor_type.place().GetType()) << "_";
+    os << "tensor<";
+    for (auto d : phi::vectorize(tensor_type.dims())) {
+      os << d;
+      os << "x";
+    }
+    tensor_type.dtype().Print(os);
+    os << ">";
+  } else if (type.isa<AllocatedSelectedRowsType>()) {
+    AllocatedSelectedRowsType tensor_type =
+        type.dyn_cast<AllocatedSelectedRowsType>();
+
+    os << phi::AllocationTypeStr(tensor_type.place().GetType()) << "_";
+    os << "tensor<";
+    for (auto d : phi::vectorize(tensor_type.dims())) {
+      os << d;
+      os << "x";
+    }
+    tensor_type.dtype().Print(os);
+    os << ">";
   }
-  tensor_type.dtype().Print(os);
-  os << ">";
 }
 
 void PaddleKernelDialect::PrintAttribute(ir::Attribute attr,