[IR] Support custom op printer

paddle/fluid/ir_adaptor/translator/op_translator.cc

@@ -23,6 +23,7 @@
 #include <vector>
 
 #include "paddle/fluid/framework/op_desc.h"
+#include "paddle/fluid/ir/dialect/pd_attribute.h"
 #include "paddle/fluid/ir/dialect/pd_type.h"
 #include "paddle/fluid/ir/interface/op_yaml_info.h"
 #include "paddle/fluid/ir_adaptor/translator/attribute_translator.h"
@@ -198,20 +199,18 @@ inline ir::Operation* InsertFullOperationForAttributeInput(ir::IrContext* ctx,
 
 inline ir::Operation* InsertFullArrayOperationForAttributeInput(
     ir::IrContext* ctx, ir::Program* program, ir::Attribute attr) {
-  std::string constant_op_name(ir::ConstantOp::name());
-  ir::OpInfo op_info = ctx->GetRegisteredOpInfo(constant_op_name);
-
-  ir::Type null_type = paddle::dialect::DenseTensorType::get(
-      ctx,
-      ir::Type(nullptr),
-      phi::DDim{},
-      paddle::dialect::DenseTensorTypeStorage::DataLayout::UNDEFINED,
-      phi::LoD{},
-      0);  // TODO(lyk): to be done
-  ir::Operation* operation =
-      ir::Operation::Create({}, {{"value", attr}}, {null_type}, op_info);
-  program->block()->push_back(operation);
-  return operation;
+  IR_ENFORCE(attr.isa<paddle::dialect::IntArrayAttribute>(),
+             "Encounter non IntArray type when trying to insert IntArray "
+             "mutable attribute");
+
+  phi::IntArray int_array =
+      attr.dyn_cast<paddle::dialect::IntArrayAttribute>().data();
+
+  ir::Builder builder = ir::Builder::AtBlockEnd(ctx, program->block());
+  paddle::dialect::FullIntArrayOp full_int_array_op =
+      builder.Build<paddle::dialect::FullIntArrayOp>(
+          int_array.GetData(), phi::DataType::INT64, phi::CPUPlace());
+  return full_int_array_op.operation();
 }
 
 inline ir::OpResult GetAttributeAsInput(ir::IrContext* ctx,

paddle/ir/core/dialect.h

@@ -14,6 +14,7 @@
 
 #pragma once
 
+#include <functional>
 #include <ostream>
 
 #include "paddle/ir/core/attribute.h"
@@ -25,6 +26,10 @@
 #include "paddle/ir/core/type_base.h"
 
 namespace ir {
+
+class Operation;
+class IrPrinter;
+
 class DialectInterface;
 ///
 /// \brief Dialect can basically be understood as a namespace. In Dialect, we
@@ -136,10 +141,13 @@ class Dialect {
     IR_THROW("dialect has no registered type printing hook");
   }
 
-  virtual void PrintAttribute(Attribute type, std::ostream &os) const {
+  virtual void PrintAttribute(Attribute attr, std::ostream &os) const {
     IR_THROW("dialect has no registered attribute printing hook");
   }
 
+  virtual void PrintOperation(Operation *op,
+                              IrPrinter &printer) const;  // NOLINT
+
  private:
   Dialect(const Dialect &) = delete;