lib/Target/R600/AMDGPUMCInstLower.cpp

   1 //===- AMDGPUMCInstLower.cpp - Lower AMDGPU MachineInstr to an MCInst -----===//
   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 /// \file
  11 /// \brief Code to lower AMDGPU MachineInstrs to their corresponding MCInst.
  12 //
  13 //===----------------------------------------------------------------------===//
  14 //
  15
  16 #include "AMDGPUMCInstLower.h"
  17 #include "AMDGPUAsmPrinter.h"
  18 #include "AMDGPUTargetMachine.h"
  19 #include "InstPrinter/AMDGPUInstPrinter.h"
  20 #include "R600InstrInfo.h"
  21 #include "SIInstrInfo.h"
  22 #include "llvm/CodeGen/MachineBasicBlock.h"
  23 #include "llvm/CodeGen/MachineInstr.h"
  24 #include "llvm/IR/Constants.h"
  25 #include "llvm/IR/GlobalVariable.h"
  26 #include "llvm/MC/MCCodeEmitter.h"
  27 #include "llvm/MC/MCContext.h"
  28 #include "llvm/MC/MCExpr.h"
  29 #include "llvm/MC/MCInst.h"
  30 #include "llvm/MC/MCObjectStreamer.h"
  31 #include "llvm/MC/MCStreamer.h"
  32 #include "llvm/Support/ErrorHandling.h"
  33 #include "llvm/Support/Format.h"
  34 #include <algorithm>
  35
  36 using namespace llvm;
  37
  38 AMDGPUMCInstLower::AMDGPUMCInstLower(MCContext &ctx, const AMDGPUSubtarget &st):
  39   Ctx(ctx), ST(st)
  40 { }
  41
  42 enum AMDGPUMCInstLower::SISubtarget
  43 AMDGPUMCInstLower::AMDGPUSubtargetToSISubtarget(unsigned Gen) const {
  44   switch (Gen) {
  45   default:
  46     return AMDGPUMCInstLower::SI;
  47   case AMDGPUSubtarget::VOLCANIC_ISLANDS:
  48     return AMDGPUMCInstLower::VI;
  49   }
  50 }
  51
  52 unsigned AMDGPUMCInstLower::getMCOpcode(unsigned MIOpcode) const {
  53
  54   int MCOpcode = AMDGPU::getMCOpcode(MIOpcode,
  55                               AMDGPUSubtargetToSISubtarget(ST.getGeneration()));
  56   if (MCOpcode == -1)
  57     MCOpcode = MIOpcode;
  58
  59   return MCOpcode;
  60 }
  61
  62 void AMDGPUMCInstLower::lower(const MachineInstr *MI, MCInst &OutMI) const {
  63
  64   OutMI.setOpcode(getMCOpcode(MI->getOpcode()));
  65
  66   for (const MachineOperand &MO : MI->explicit_operands()) {
  67     MCOperand MCOp;
  68     switch (MO.getType()) {
  69     default:
  70       llvm_unreachable("unknown operand type");
  71     case MachineOperand::MO_FPImmediate: {
  72       const APFloat &FloatValue = MO.getFPImm()->getValueAPF();
  73
  74       if (&FloatValue.getSemantics() == &APFloat::IEEEsingle)
  75         MCOp = MCOperand::CreateFPImm(FloatValue.convertToFloat());
  76       else if (&FloatValue.getSemantics() == &APFloat::IEEEdouble)
  77         MCOp = MCOperand::CreateFPImm(FloatValue.convertToDouble());
  78       else
  79         llvm_unreachable("Unhandled floating point type");
  80
  81       break;
  82     }
  83     case MachineOperand::MO_Immediate:
  84       MCOp = MCOperand::CreateImm(MO.getImm());
  85       break;
  86     case MachineOperand::MO_Register:
  87       MCOp = MCOperand::CreateReg(MO.getReg());
  88       break;
  89     case MachineOperand::MO_MachineBasicBlock:
  90       MCOp = MCOperand::CreateExpr(MCSymbolRefExpr::Create(
  91                                    MO.getMBB()->getSymbol(), Ctx));
  92       break;
  93     case MachineOperand::MO_GlobalAddress: {
  94       const GlobalValue *GV = MO.getGlobal();
  95       MCSymbol *Sym = Ctx.GetOrCreateSymbol(StringRef(GV->getName()));
  96       MCOp = MCOperand::CreateExpr(MCSymbolRefExpr::Create(Sym, Ctx));
  97       break;
  98     }
  99     case MachineOperand::MO_TargetIndex: {
 100       assert(MO.getIndex() == AMDGPU::TI_CONSTDATA_START);
 101       MCSymbol *Sym = Ctx.GetOrCreateSymbol(StringRef(END_OF_TEXT_LABEL_NAME));
 102       const MCSymbolRefExpr *Expr = MCSymbolRefExpr::Create(Sym, Ctx);
 103       MCOp = MCOperand::CreateExpr(Expr);
 104       break;
 105     }
 106     }
 107     OutMI.addOperand(MCOp);
 108   }
 109 }
 110
 111 void AMDGPUAsmPrinter::EmitInstruction(const MachineInstr *MI) {
 112   AMDGPUMCInstLower MCInstLowering(OutContext,
 113                                MF->getTarget().getSubtarget<AMDGPUSubtarget>());
 114
 115 #ifdef _DEBUG
 116   StringRef Err;
 117   if (!TM.getSubtargetImpl()->getInstrInfo()->verifyInstruction(MI, Err)) {
 118     errs() << "Warning: Illegal instruction detected: " << Err << "\n";
 119     MI->dump();
 120   }
 121 #endif
 122   if (MI->isBundle()) {
 123     const MachineBasicBlock *MBB = MI->getParent();
 124     MachineBasicBlock::const_instr_iterator I = MI;
 125     ++I;
 126     while (I != MBB->end() && I->isInsideBundle()) {
 127       EmitInstruction(I);
 128       ++I;
 129     }
 130   } else {
 131     MCInst TmpInst;
 132     MCInstLowering.lower(MI, TmpInst);
 133     EmitToStreamer(OutStreamer, TmpInst);
 134
 135     if (DisasmEnabled) {
 136       // Disassemble instruction/operands to text.
 137       DisasmLines.resize(DisasmLines.size() + 1);
 138       std::string &DisasmLine = DisasmLines.back();
 139       raw_string_ostream DisasmStream(DisasmLine);
 140
 141       AMDGPUInstPrinter InstPrinter(*TM.getMCAsmInfo(),
 142                                     *TM.getSubtargetImpl()->getInstrInfo(),
 143                                     *TM.getSubtargetImpl()->getRegisterInfo());
 144       InstPrinter.printInst(&TmpInst, DisasmStream, StringRef());
 145
 146       // Disassemble instruction/operands to hex representation.
 147       SmallVector<MCFixup, 4> Fixups;
 148       SmallVector<char, 16> CodeBytes;
 149       raw_svector_ostream CodeStream(CodeBytes);
 150
 151       MCObjectStreamer &ObjStreamer = (MCObjectStreamer &)OutStreamer;
 152       MCCodeEmitter &InstEmitter = ObjStreamer.getAssembler().getEmitter();
 153       InstEmitter.EncodeInstruction(TmpInst, CodeStream, Fixups,
 154                                     TM.getSubtarget<MCSubtargetInfo>());
 155       CodeStream.flush();
 156
 157       HexLines.resize(HexLines.size() + 1);
 158       std::string &HexLine = HexLines.back();
 159       raw_string_ostream HexStream(HexLine);
 160
 161       for (size_t i = 0; i < CodeBytes.size(); i += 4) {
 162         unsigned int CodeDWord = *(unsigned int *)&CodeBytes[i];
 163         HexStream << format("%s%08X", (i > 0 ? " " : ""), CodeDWord);
 164       }
 165
 166       DisasmStream.flush();
 167       DisasmLineMaxLen = std::max(DisasmLineMaxLen, DisasmLine.size());
 168     }
 169   }
 170 }