commit dd0c2ca567c1690a5ae42bb547764993ac0c0208
author Chris Lattner <clattner@google.com> Tue Jul 24 16:07:22 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:42:08 2019 -0700
tree 845442cb419fbbf364e1c8938d0b29e1c6534ff2
parent 4f78837f9a05f17a4a2077d2b00d298e762b0fad

Expose custom asmprinter support to core operations and have them adopt it,
fixing the printing syntax for dim, constant, fadd, etc.

PiperOrigin-RevId: 205908627

lib/IR/StandardOps.cpp

diff --git a/lib/IR/StandardOps.cpp b/lib/IR/StandardOps.cpp
index 1307b2a..7114d49 100644
--- a/lib/IR/StandardOps.cpp
+++ b/lib/IR/StandardOps.cpp
@@ -17,14 +17,16 @@
 
 #include "mlir/IR/StandardOps.h"
 #include "mlir/IR/AffineMap.h"
+#include "mlir/IR/OpImplementation.h"
 #include "mlir/IR/OperationSet.h"
 #include "mlir/IR/SSAValue.h"
 #include "mlir/IR/Types.h"
 #include "llvm/Support/raw_ostream.h"
 using namespace mlir;
 
-void AddFOp::print(raw_ostream &os) const {
-  os << "addf xx, yy : sometype";
+void AddFOp::print(OpAsmPrinter *p) const {
+  *p << "addf " << *getOperand(0) << ", " << *getOperand(1) << " : "
+     << *getType();
 }
 
 // Return an error message on failure.
@@ -62,8 +64,9 @@
          isa<IntegerType>(op->getResult(0)->getType());
 }
 
-void DimOp::print(raw_ostream &os) const {
-  os << "dim xxx, " << getIndex() << " : sometype";
+void DimOp::print(OpAsmPrinter *p) const {
+  *p << "dim " << *getOperand() << ", " << getIndex() << " : "
+     << *getOperand()->getType();
 }
 
 const char *DimOp::verify() const {
@@ -90,9 +93,9 @@
   return nullptr;
 }
 
-void AffineApplyOp::print(raw_ostream &os) const {
-  os << "affine_apply map: ";
-  getAffineMap()->print(os);
+void AffineApplyOp::print(OpAsmPrinter *p) const {
+  // TODO: Print operands etc.
+  *p << "affine_apply map: " << *getAffineMap();
 }
 
 const char *AffineApplyOp::verify() const {