Refactor generic op printing: extract a public printFunctionalType() on OpAsmPrinter (NFC)

PiperOrigin-RevId: 253674584

mlir/include/mlir/IR/OpImplementation.h

@@ -98,6 +98,23 @@ public:
       os << ')';
   }
 
+  /// Print the complete type of an operation in functional form.
+  void printFunctionalType(Operation *op) {
+    auto &os = getStream();
+    os << "(";
+    interleaveComma(op->getNonSuccessorOperands(), os,
+                    [&](Value *operand) { printType(operand->getType()); });
+    os << ") -> ";
+    if (op->getNumResults() == 1 &&
+        !op->getResult(0)->getType().isa<FunctionType>()) {
+      printType(op->getResult(0)->getType());
+    } else {
+      os << '(';
+      interleaveComma(op->getResultTypes(), os);
+      os << ')';
+    }
+  }
+
 private:
   OpAsmPrinter(const OpAsmPrinter &) = delete;
   void operator=(const OpAsmPrinter &) = delete;

mlir/include/mlir/IR/Operation.h

@@ -320,6 +320,7 @@ public:
     return getOperand(getSuccessorOperandIndex(succIndex) + opIndex);
   }
 
+  bool hasSuccessors() { return numSuccs != 0; }
   unsigned getNumSuccessors() { return numSuccs; }
   unsigned getNumSuccessorOperands(unsigned index) {
     assert(!isKnownNonTerminator() && "only terminators may have successors");

mlir/lib/IR/AsmPrinter.cpp

@@ -1575,20 +1575,8 @@ void FunctionPrinter::printGenericOp(Operation *op) {
   printOptionalAttrDict(attrs);
 
   // Print the type signature of the operation.
-  os << " : (";
+  os << " : ";
-  interleaveComma(properOperands,
+  printFunctionalType(op);
-                  [&](Value *value) { printType(value->getType()); });
-  os << ") -> ";
-
-  if (op->getNumResults() == 1 &&
-      !op->getResult(0)->getType().isa<FunctionType>()) {
-    printType(op->getResult(0)->getType());
-  } else {
-    os << '(';
-    interleaveComma(op->getResults(),
-                    [&](Value *result) { printType(result->getType()); });
-    os << ')';
-  }
 }
 
 void FunctionPrinter::printSuccessorAndUseList(Operation *term,

mlir/lib/IR/Operation.cpp

@@ -495,7 +495,8 @@ void Operation::setSuccessor(Block *block, unsigned index) {
 
 auto Operation::getNonSuccessorOperands() -> operand_range {
   return {operand_iterator(this, 0),
-          operand_iterator(this, getSuccessorOperandIndex(0))};
+          operand_iterator(this, hasSuccessors() ? getSuccessorOperandIndex(0)
+                                                 : getNumOperands())};
 }
 
 /// Get the index of the first operand of the successor at the provided