commit 384da8c3e8b9b6a92cd52329b0ea2f22f3bee800
author Chris Lattner <clattner@google.com> Thu Aug 02 17:16:58 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:53:44 2019 -0700
tree 44211ee122022e919d6b504f0487c1a44f01085c
parent 85cf26d214e5e8ce2c33d38f7ee04e342924bd5f

Have the asmprinter give true/false constants nice names, add a dump/print
method to SSAValue.

PiperOrigin-RevId: 207193088

lib/IR/AsmPrinter.cpp

diff --git a/lib/IR/AsmPrinter.cpp b/lib/IR/AsmPrinter.cpp
index 221f453..6177e19 100644
--- a/lib/IR/AsmPrinter.cpp
+++ b/lib/IR/AsmPrinter.cpp
@@ -594,9 +594,14 @@
     // Give constant integers special names.
     if (auto *op = value->getDefiningOperation()) {
       if (auto intOp = op->getAs<ConstantIntOp>()) {
-        specialName << 'c' << intOp->getValue();
-        if (!intOp->getType()->isAffineInt())
-          specialName << '_' << *intOp->getType();
+        // i1 constants get special names.
+        if (intOp->getType()->isInteger(1)) {
+          specialName << (intOp->getValue() ? "true" : "false");
+        } else {
+          specialName << 'c' << intOp->getValue();
+          if (!intOp->getType()->isAffineInt())
+            specialName << '_' << *intOp->getType();
+        }
       }
     }
 
@@ -1151,6 +1156,27 @@
   ModulePrinter(os, state).printAffineMap(this);
 }
 
+void SSAValue::print(raw_ostream &os) const {
+  switch (getKind()) {
+  case SSAValueKind::BBArgument:
+    // TODO: Improve this.
+    os << "<bb argument>\n";
+    return;
+  case SSAValueKind::InstResult:
+    return getDefiningInst()->print(os);
+  case SSAValueKind::FnArgument:
+    // TODO: Improve this.
+    os << "<function argument>\n";
+    return;
+  case SSAValueKind::StmtResult:
+    return getDefiningStmt()->print(os);
+  case SSAValueKind::ForStmt:
+    return cast<ForStmt>(this)->print(os);
+  }
+}
+
+void SSAValue::dump() const { print(llvm::errs()); }
+
 void Instruction::print(raw_ostream &os) const {
   ModuleState state(getFunction()->getContext());
   ModulePrinter modulePrinter(os, state);