utils/TableGen/PseudoLoweringEmitter.cpp

   1 //===- PseudoLoweringEmitter.cpp - PseudoLowering Generator -----*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #define DEBUG_TYPE "pseudo-lowering"
  11 #include "CodeGenInstruction.h"
  12 #include "PseudoLoweringEmitter.h"
  13 #include "llvm/TableGen/Error.h"
  14 #include "llvm/TableGen/Record.h"
  15 #include "llvm/ADT/IndexedMap.h"
  16 #include "llvm/ADT/StringMap.h"
  17 #include "llvm/Support/ErrorHandling.h"
  18 #include "llvm/Support/Debug.h"
  19 #include <vector>
  20 using namespace llvm;
  21
  22 // FIXME: This pass currently can only expand a pseudo to a single instruction.
  23 //        The pseudo expansion really should take a list of dags, not just
  24 //        a single dag, so we can do fancier things.
  25
  26 unsigned PseudoLoweringEmitter::
  27 addDagOperandMapping(Record *Rec, DagInit *Dag, CodeGenInstruction &Insn,
  28                      IndexedMap<OpData> &OperandMap, unsigned BaseIdx) {
  29   unsigned OpsAdded = 0;
  30   for (unsigned i = 0, e = Dag->getNumArgs(); i != e; ++i) {
  31     if (DefInit *DI = dynamic_cast<DefInit*>(Dag->getArg(i))) {
  32       // Physical register reference. Explicit check for the special case
  33       // "zero_reg" definition.
  34       if (DI->getDef()->isSubClassOf("Register") ||
  35           DI->getDef()->getName() == "zero_reg") {
  36         OperandMap[BaseIdx + i].Kind = OpData::Reg;
  37         OperandMap[BaseIdx + i].Data.Reg = DI->getDef();
  38         ++OpsAdded;
  39         continue;
  40       }
  41
  42       // Normal operands should always have the same type, or we have a
  43       // problem.
  44       // FIXME: We probably shouldn't ever get a non-zero BaseIdx here.
  45       assert(BaseIdx == 0 && "Named subargument in pseudo expansion?!");
  46       if (DI->getDef() != Insn.Operands[BaseIdx + i].Rec)
  47         throw TGError(Rec->getLoc(),
  48                       "Pseudo operand type '" + DI->getDef()->getName() +
  49                       "' does not match expansion operand type '" +
  50                       Insn.Operands[BaseIdx + i].Rec->getName() + "'");
  51       // Source operand maps to destination operand. The Data element
  52       // will be filled in later, just set the Kind for now. Do it
  53       // for each corresponding MachineInstr operand, not just the first.
  54       for (unsigned I = 0, E = Insn.Operands[i].MINumOperands; I != E; ++I)
  55         OperandMap[BaseIdx + i + I].Kind = OpData::Operand;
  56       OpsAdded += Insn.Operands[i].MINumOperands;
  57     } else if (IntInit *II = dynamic_cast<IntInit*>(Dag->getArg(i))) {
  58       OperandMap[BaseIdx + i].Kind = OpData::Imm;
  59       OperandMap[BaseIdx + i].Data.Imm = II->getValue();
  60       ++OpsAdded;
  61     } else if (DagInit *SubDag = dynamic_cast<DagInit*>(Dag->getArg(i))) {
  62       // Just add the operands recursively. This is almost certainly
  63       // a constant value for a complex operand (> 1 MI operand).
  64       unsigned NewOps =
  65         addDagOperandMapping(Rec, SubDag, Insn, OperandMap, BaseIdx + i);
  66       OpsAdded += NewOps;
  67       // Since we added more than one, we also need to adjust the base.
  68       BaseIdx += NewOps - 1;
  69     } else
  70       llvm_unreachable("Unhandled pseudo-expansion argument type!");
  71   }
  72   return OpsAdded;
  73 }
  74
  75 void PseudoLoweringEmitter::evaluateExpansion(Record *Rec) {
  76   DEBUG(dbgs() << "Pseudo definition: " << Rec->getName() << "\n");
  77
  78   // Validate that the result pattern has the corrent number and types
  79   // of arguments for the instruction it references.
  80   DagInit *Dag = Rec->getValueAsDag("ResultInst");
  81   assert(Dag && "Missing result instruction in pseudo expansion!");
  82   DEBUG(dbgs() << "  Result: " << *Dag << "\n");
  83
  84   DefInit *OpDef = dynamic_cast<DefInit*>(Dag->getOperator());
  85   if (!OpDef)
  86     throw TGError(Rec->getLoc(), Rec->getName() +
  87                   " has unexpected operator type!");
  88   Record *Operator = OpDef->getDef();
  89   if (!Operator->isSubClassOf("Instruction"))
  90     throw TGError(Rec->getLoc(), "Pseudo result '" + Operator->getName() +
  91                                  "' is not an instruction!");
  92
  93   CodeGenInstruction Insn(Operator);
  94
  95   if (Insn.isCodeGenOnly || Insn.isPseudo)
  96     throw TGError(Rec->getLoc(), "Pseudo result '" + Operator->getName() +
  97                                  "' cannot be another pseudo instruction!");
  98
  99   if (Insn.Operands.size() != Dag->getNumArgs())
 100     throw TGError(Rec->getLoc(), "Pseudo result '" + Operator->getName() +
 101                                  "' operand count mismatch");
 102
 103   unsigned NumMIOperands = 0;
 104   for (unsigned i = 0, e = Insn.Operands.size(); i != e; ++i)
 105     NumMIOperands += Insn.Operands[i].MINumOperands;
 106   IndexedMap<OpData> OperandMap;
 107   OperandMap.grow(NumMIOperands);
 108
 109   addDagOperandMapping(Rec, Dag, Insn, OperandMap, 0);
 110
 111   // If there are more operands that weren't in the DAG, they have to
 112   // be operands that have default values, or we have an error. Currently,
 113   // PredicateOperand and OptionalDefOperand both have default values.
 114
 115
 116   // Validate that each result pattern argument has a matching (by name)
 117   // argument in the source instruction, in either the (outs) or (ins) list.
 118   // Also check that the type of the arguments match.
 119   //
 120   // Record the mapping of the source to result arguments for use by
 121   // the lowering emitter.
 122   CodeGenInstruction SourceInsn(Rec);
 123   StringMap<unsigned> SourceOperands;
 124   for (unsigned i = 0, e = SourceInsn.Operands.size(); i != e; ++i)
 125     SourceOperands[SourceInsn.Operands[i].Name] = i;
 126
 127   DEBUG(dbgs() << "  Operand mapping:\n");
 128   for (unsigned i = 0, e = Insn.Operands.size(); i != e; ++i) {
 129     // We've already handled constant values. Just map instruction operands
 130     // here.
 131     if (OperandMap[Insn.Operands[i].MIOperandNo].Kind != OpData::Operand)
 132       continue;
 133     StringMap<unsigned>::iterator SourceOp =
 134       SourceOperands.find(Dag->getArgName(i));
 135     if (SourceOp == SourceOperands.end())
 136       throw TGError(Rec->getLoc(),
 137                     "Pseudo output operand '" + Dag->getArgName(i) +
 138                     "' has no matching source operand.");
 139     // Map the source operand to the destination operand index for each
 140     // MachineInstr operand.
 141     for (unsigned I = 0, E = Insn.Operands[i].MINumOperands; I != E; ++I)
 142       OperandMap[Insn.Operands[i].MIOperandNo + I].Data.Operand =
 143         SourceOp->getValue();
 144
 145     DEBUG(dbgs() << "    " << SourceOp->getValue() << " ==> " << i << "\n");
 146   }
 147
 148   Expansions.push_back(PseudoExpansion(SourceInsn, Insn, OperandMap));
 149 }
 150
 151 void PseudoLoweringEmitter::emitLoweringEmitter(raw_ostream &o) {
 152   // Emit file header.
 153   EmitSourceFileHeader("Pseudo-instruction MC lowering Source Fragment", o);
 154
 155   o << "bool " << Target.getName() + "AsmPrinter" << "::\n"
 156     << "emitPseudoExpansionLowering(MCStreamer &OutStreamer,\n"
 157     << "                            const MachineInstr *MI) {\n"
 158     << "  switch (MI->getOpcode()) {\n"
 159     << "    default: return false;\n";
 160   for (unsigned i = 0, e = Expansions.size(); i != e; ++i) {
 161     PseudoExpansion &Expansion = Expansions[i];
 162     CodeGenInstruction &Source = Expansion.Source;
 163     CodeGenInstruction &Dest = Expansion.Dest;
 164     o << "    case " << Source.Namespace << "::"
 165       << Source.TheDef->getName() << ": {\n"
 166       << "      MCInst TmpInst;\n"
 167       << "      MCOperand MCOp;\n"
 168       << "      TmpInst.setOpcode(" << Dest.Namespace << "::"
 169       << Dest.TheDef->getName() << ");\n";
 170
 171     // Copy the operands from the source instruction.
 172     // FIXME: Instruction operands with defaults values (predicates and cc_out
 173     //        in ARM, for example shouldn't need explicit values in the
 174     //        expansion DAG.
 175     unsigned MIOpNo = 0;
 176     for (unsigned OpNo = 0, E = Dest.Operands.size(); OpNo != E;
 177          ++OpNo) {
 178       o << "      // Operand: " << Dest.Operands[OpNo].Name << "\n";
 179       for (unsigned i = 0, e = Dest.Operands[OpNo].MINumOperands;
 180            i != e; ++i) {
 181         switch (Expansion.OperandMap[MIOpNo + i].Kind) {
 182         case OpData::Operand:
 183           o << "      lowerOperand(MI->getOperand("
 184             << Source.Operands[Expansion.OperandMap[MIOpNo].Data
 185                 .Operand].MIOperandNo + i
 186             << "), MCOp);\n"
 187             << "      TmpInst.addOperand(MCOp);\n";
 188           break;
 189         case OpData::Imm:
 190           o << "      TmpInst.addOperand(MCOperand::CreateImm("
 191             << Expansion.OperandMap[MIOpNo + i].Data.Imm << "));\n";
 192           break;
 193         case OpData::Reg: {
 194           Record *Reg = Expansion.OperandMap[MIOpNo + i].Data.Reg;
 195           o << "      TmpInst.addOperand(MCOperand::CreateReg(";
 196           // "zero_reg" is special.
 197           if (Reg->getName() == "zero_reg")
 198             o << "0";
 199           else
 200             o << Reg->getValueAsString("Namespace") << "::" << Reg->getName();
 201           o << "));\n";
 202           break;
 203         }
 204         }
 205       }
 206       MIOpNo += Dest.Operands[OpNo].MINumOperands;
 207     }
 208     if (Dest.Operands.isVariadic) {
 209       o << "      // variable_ops\n";
 210       o << "      for (unsigned i = " << MIOpNo
 211         << ", e = MI->getNumOperands(); i != e; ++i)\n"
 212         << "        if (lowerOperand(MI->getOperand(i), MCOp))\n"
 213         << "          TmpInst.addOperand(MCOp);\n";
 214     }
 215     o << "      OutStreamer.EmitInstruction(TmpInst);\n"
 216       << "      break;\n"
 217       << "    }\n";
 218   }
 219   o << "  }\n  return true;\n}\n\n";
 220 }
 221
 222 void PseudoLoweringEmitter::run(raw_ostream &o) {
 223   Record *ExpansionClass = Records.getClass("PseudoInstExpansion");
 224   Record *InstructionClass = Records.getClass("PseudoInstExpansion");
 225   assert(ExpansionClass && "PseudoInstExpansion class definition missing!");
 226   assert(InstructionClass && "Instruction class definition missing!");
 227
 228   std::vector<Record*> Insts;
 229   for (std::map<std::string, Record*>::const_iterator I =
 230          Records.getDefs().begin(), E = Records.getDefs().end(); I != E; ++I) {
 231     if (I->second->isSubClassOf(ExpansionClass) &&
 232         I->second->isSubClassOf(InstructionClass))
 233       Insts.push_back(I->second);
 234   }
 235
 236   // Process the pseudo expansion definitions, validating them as we do so.
 237   for (unsigned i = 0, e = Insts.size(); i != e; ++i)
 238     evaluateExpansion(Insts[i]);
 239
 240   // Generate expansion code to lower the pseudo to an MCInst of the real
 241   // instruction.
 242   emitLoweringEmitter(o);
 243 }
 244