Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 1 | //===-- llvm/CodeGen/GlobalISel/IRTranslator.cpp - IRTranslator --*- 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 | /// \file |
| 10 | /// This file implements the IRTranslator class. |
| 11 | //===----------------------------------------------------------------------===// |
| 12 | |
| 13 | #include "llvm/CodeGen/GlobalISel/IRTranslator.h" |
| 14 | |
Quentin Colombet | fd9d0a0 | 2016-02-11 19:59:41 +0000 | [diff] [blame] | 15 | #include "llvm/ADT/SmallVector.h" |
Quentin Colombet | ba2a016 | 2016-02-16 19:26:02 +0000 | [diff] [blame] | 16 | #include "llvm/CodeGen/GlobalISel/CallLowering.h" |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 17 | #include "llvm/CodeGen/MachineFunction.h" |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 18 | #include "llvm/CodeGen/MachineFrameInfo.h" |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 19 | #include "llvm/CodeGen/MachineRegisterInfo.h" |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 20 | #include "llvm/CodeGen/TargetPassConfig.h" |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 21 | #include "llvm/IR/Constant.h" |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 22 | #include "llvm/IR/Function.h" |
Tim Northover | a7653b3 | 2016-09-12 11:20:22 +0000 | [diff] [blame] | 23 | #include "llvm/IR/GetElementPtrTypeIterator.h" |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 24 | #include "llvm/IR/IntrinsicInst.h" |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 25 | #include "llvm/IR/Type.h" |
| 26 | #include "llvm/IR/Value.h" |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 27 | #include "llvm/Target/TargetIntrinsicInfo.h" |
Quentin Colombet | 74d7d2f | 2016-02-11 18:53:28 +0000 | [diff] [blame] | 28 | #include "llvm/Target/TargetLowering.h" |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 29 | |
| 30 | #define DEBUG_TYPE "irtranslator" |
| 31 | |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 32 | using namespace llvm; |
| 33 | |
| 34 | char IRTranslator::ID = 0; |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 35 | INITIALIZE_PASS_BEGIN(IRTranslator, DEBUG_TYPE, "IRTranslator LLVM IR -> MI", |
| 36 | false, false) |
| 37 | INITIALIZE_PASS_DEPENDENCY(TargetPassConfig) |
| 38 | INITIALIZE_PASS_END(IRTranslator, DEBUG_TYPE, "IRTranslator LLVM IR -> MI", |
Tim Northover | 884b47e | 2016-07-26 03:29:18 +0000 | [diff] [blame] | 39 | false, false) |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 40 | |
Quentin Colombet | a7fae16 | 2016-02-11 17:53:23 +0000 | [diff] [blame] | 41 | IRTranslator::IRTranslator() : MachineFunctionPass(ID), MRI(nullptr) { |
Quentin Colombet | 39293d3 | 2016-03-08 01:38:55 +0000 | [diff] [blame] | 42 | initializeIRTranslatorPass(*PassRegistry::getPassRegistry()); |
Quentin Colombet | a7fae16 | 2016-02-11 17:53:23 +0000 | [diff] [blame] | 43 | } |
| 44 | |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 45 | void IRTranslator::getAnalysisUsage(AnalysisUsage &AU) const { |
| 46 | AU.addRequired<TargetPassConfig>(); |
| 47 | MachineFunctionPass::getAnalysisUsage(AU); |
| 48 | } |
| 49 | |
| 50 | |
Quentin Colombet | e225e25 | 2016-03-11 17:27:54 +0000 | [diff] [blame] | 51 | unsigned IRTranslator::getOrCreateVReg(const Value &Val) { |
| 52 | unsigned &ValReg = ValToVReg[&Val]; |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 53 | // Check if this is the first time we see Val. |
Quentin Colombet | ccd7725 | 2016-02-11 21:48:32 +0000 | [diff] [blame] | 54 | if (!ValReg) { |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 55 | // Fill ValRegsSequence with the sequence of registers |
| 56 | // we need to concat together to produce the value. |
Quentin Colombet | e225e25 | 2016-03-11 17:27:54 +0000 | [diff] [blame] | 57 | assert(Val.getType()->isSized() && |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 58 | "Don't know how to create an empty vreg"); |
Tim Northover | 5ae8350 | 2016-09-15 09:20:34 +0000 | [diff] [blame] | 59 | unsigned VReg = MRI->createGenericVirtualRegister(LLT{*Val.getType(), *DL}); |
Quentin Colombet | ccd7725 | 2016-02-11 21:48:32 +0000 | [diff] [blame] | 60 | ValReg = VReg; |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 61 | |
| 62 | if (auto CV = dyn_cast<Constant>(&Val)) { |
| 63 | bool Success = translate(*CV, VReg); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 64 | if (!Success) { |
| 65 | if (!TPC->isGlobalISelAbortEnabled()) { |
| 66 | MIRBuilder.getMF().getProperties().set( |
| 67 | MachineFunctionProperties::Property::FailedISel); |
| 68 | return 0; |
| 69 | } |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 70 | report_fatal_error("unable to translate constant"); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 71 | } |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 72 | } |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 73 | } |
Quentin Colombet | ccd7725 | 2016-02-11 21:48:32 +0000 | [diff] [blame] | 74 | return ValReg; |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 75 | } |
| 76 | |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 77 | unsigned IRTranslator::getMemOpAlignment(const Instruction &I) { |
| 78 | unsigned Alignment = 0; |
| 79 | Type *ValTy = nullptr; |
| 80 | if (const StoreInst *SI = dyn_cast<StoreInst>(&I)) { |
| 81 | Alignment = SI->getAlignment(); |
| 82 | ValTy = SI->getValueOperand()->getType(); |
| 83 | } else if (const LoadInst *LI = dyn_cast<LoadInst>(&I)) { |
| 84 | Alignment = LI->getAlignment(); |
| 85 | ValTy = LI->getType(); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 86 | } else if (!TPC->isGlobalISelAbortEnabled()) { |
| 87 | MIRBuilder.getMF().getProperties().set( |
| 88 | MachineFunctionProperties::Property::FailedISel); |
| 89 | return 1; |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 90 | } else |
| 91 | llvm_unreachable("unhandled memory instruction"); |
| 92 | |
| 93 | return Alignment ? Alignment : DL->getABITypeAlignment(ValTy); |
| 94 | } |
| 95 | |
Quentin Colombet | 53237a9 | 2016-03-11 17:27:43 +0000 | [diff] [blame] | 96 | MachineBasicBlock &IRTranslator::getOrCreateBB(const BasicBlock &BB) { |
| 97 | MachineBasicBlock *&MBB = BBToMBB[&BB]; |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 98 | if (!MBB) { |
Quentin Colombet | a7fae16 | 2016-02-11 17:53:23 +0000 | [diff] [blame] | 99 | MachineFunction &MF = MIRBuilder.getMF(); |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 100 | MBB = MF.CreateMachineBasicBlock(); |
| 101 | MF.push_back(MBB); |
| 102 | } |
| 103 | return *MBB; |
| 104 | } |
| 105 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 106 | bool IRTranslator::translateBinaryOp(unsigned Opcode, const User &U) { |
Tim Northover | 0d56e05 | 2016-07-29 18:11:21 +0000 | [diff] [blame] | 107 | // FIXME: handle signed/unsigned wrapping flags. |
| 108 | |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 109 | // Get or create a virtual register for each value. |
| 110 | // Unless the value is a Constant => loadimm cst? |
| 111 | // or inline constant each time? |
| 112 | // Creation of a virtual register needs to have a size. |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 113 | unsigned Op0 = getOrCreateVReg(*U.getOperand(0)); |
| 114 | unsigned Op1 = getOrCreateVReg(*U.getOperand(1)); |
| 115 | unsigned Res = getOrCreateVReg(U); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 116 | MIRBuilder.buildInstr(Opcode).addDef(Res).addUse(Op0).addUse(Op1); |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 117 | return true; |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 118 | } |
| 119 | |
Tim Northover | d5c23bc | 2016-08-19 20:48:16 +0000 | [diff] [blame] | 120 | bool IRTranslator::translateCompare(const User &U) { |
| 121 | const CmpInst *CI = dyn_cast<CmpInst>(&U); |
| 122 | unsigned Op0 = getOrCreateVReg(*U.getOperand(0)); |
| 123 | unsigned Op1 = getOrCreateVReg(*U.getOperand(1)); |
| 124 | unsigned Res = getOrCreateVReg(U); |
| 125 | CmpInst::Predicate Pred = |
| 126 | CI ? CI->getPredicate() : static_cast<CmpInst::Predicate>( |
| 127 | cast<ConstantExpr>(U).getPredicate()); |
Tim Northover | de3aea041 | 2016-08-17 20:25:25 +0000 | [diff] [blame] | 128 | |
Tim Northover | d5c23bc | 2016-08-19 20:48:16 +0000 | [diff] [blame] | 129 | if (CmpInst::isIntPredicate(Pred)) |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 130 | MIRBuilder.buildICmp(Pred, Res, Op0, Op1); |
Tim Northover | d5c23bc | 2016-08-19 20:48:16 +0000 | [diff] [blame] | 131 | else |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 132 | MIRBuilder.buildFCmp(Pred, Res, Op0, Op1); |
Tim Northover | d5c23bc | 2016-08-19 20:48:16 +0000 | [diff] [blame] | 133 | |
Tim Northover | de3aea041 | 2016-08-17 20:25:25 +0000 | [diff] [blame] | 134 | return true; |
| 135 | } |
| 136 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 137 | bool IRTranslator::translateRet(const User &U) { |
| 138 | const ReturnInst &RI = cast<ReturnInst>(U); |
Tim Northover | 0d56e05 | 2016-07-29 18:11:21 +0000 | [diff] [blame] | 139 | const Value *Ret = RI.getReturnValue(); |
Quentin Colombet | 74d7d2f | 2016-02-11 18:53:28 +0000 | [diff] [blame] | 140 | // The target may mess up with the insertion point, but |
| 141 | // this is not important as a return is the last instruction |
| 142 | // of the block anyway. |
Tom Stellard | b72a65f | 2016-04-14 17:23:33 +0000 | [diff] [blame] | 143 | return CLI->lowerReturn(MIRBuilder, Ret, !Ret ? 0 : getOrCreateVReg(*Ret)); |
Quentin Colombet | 74d7d2f | 2016-02-11 18:53:28 +0000 | [diff] [blame] | 144 | } |
| 145 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 146 | bool IRTranslator::translateBr(const User &U) { |
| 147 | const BranchInst &BrInst = cast<BranchInst>(U); |
Tim Northover | 69c2ba5 | 2016-07-29 17:58:00 +0000 | [diff] [blame] | 148 | unsigned Succ = 0; |
| 149 | if (!BrInst.isUnconditional()) { |
| 150 | // We want a G_BRCOND to the true BB followed by an unconditional branch. |
| 151 | unsigned Tst = getOrCreateVReg(*BrInst.getCondition()); |
| 152 | const BasicBlock &TrueTgt = *cast<BasicBlock>(BrInst.getSuccessor(Succ++)); |
| 153 | MachineBasicBlock &TrueBB = getOrCreateBB(TrueTgt); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 154 | MIRBuilder.buildBrCond(Tst, TrueBB); |
Quentin Colombet | dd4b137 | 2016-03-11 17:28:03 +0000 | [diff] [blame] | 155 | } |
Tim Northover | 69c2ba5 | 2016-07-29 17:58:00 +0000 | [diff] [blame] | 156 | |
| 157 | const BasicBlock &BrTgt = *cast<BasicBlock>(BrInst.getSuccessor(Succ)); |
| 158 | MachineBasicBlock &TgtBB = getOrCreateBB(BrTgt); |
| 159 | MIRBuilder.buildBr(TgtBB); |
| 160 | |
Quentin Colombet | dd4b137 | 2016-03-11 17:28:03 +0000 | [diff] [blame] | 161 | // Link successors. |
| 162 | MachineBasicBlock &CurBB = MIRBuilder.getMBB(); |
| 163 | for (const BasicBlock *Succ : BrInst.successors()) |
| 164 | CurBB.addSuccessor(&getOrCreateBB(*Succ)); |
| 165 | return true; |
| 166 | } |
| 167 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 168 | bool IRTranslator::translateLoad(const User &U) { |
| 169 | const LoadInst &LI = cast<LoadInst>(U); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 170 | |
| 171 | if (!TPC->isGlobalISelAbortEnabled() && !LI.isSimple()) |
| 172 | return false; |
| 173 | |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 174 | assert(LI.isSimple() && "only simple loads are supported at the moment"); |
| 175 | |
| 176 | MachineFunction &MF = MIRBuilder.getMF(); |
| 177 | unsigned Res = getOrCreateVReg(LI); |
| 178 | unsigned Addr = getOrCreateVReg(*LI.getPointerOperand()); |
Tim Northover | 5ae8350 | 2016-09-15 09:20:34 +0000 | [diff] [blame] | 179 | LLT VTy{*LI.getType(), *DL}, PTy{*LI.getPointerOperand()->getType(), *DL}; |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 180 | |
| 181 | MIRBuilder.buildLoad( |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 182 | Res, Addr, |
Tim Northover | 28fdc42 | 2016-08-15 21:13:17 +0000 | [diff] [blame] | 183 | *MF.getMachineMemOperand( |
| 184 | MachinePointerInfo(LI.getPointerOperand()), MachineMemOperand::MOLoad, |
| 185 | DL->getTypeStoreSize(LI.getType()), getMemOpAlignment(LI))); |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 186 | return true; |
| 187 | } |
| 188 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 189 | bool IRTranslator::translateStore(const User &U) { |
| 190 | const StoreInst &SI = cast<StoreInst>(U); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 191 | |
| 192 | if (!TPC->isGlobalISelAbortEnabled() && !SI.isSimple()) |
| 193 | return false; |
| 194 | |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 195 | assert(SI.isSimple() && "only simple loads are supported at the moment"); |
| 196 | |
| 197 | MachineFunction &MF = MIRBuilder.getMF(); |
| 198 | unsigned Val = getOrCreateVReg(*SI.getValueOperand()); |
| 199 | unsigned Addr = getOrCreateVReg(*SI.getPointerOperand()); |
Tim Northover | 5ae8350 | 2016-09-15 09:20:34 +0000 | [diff] [blame] | 200 | LLT VTy{*SI.getValueOperand()->getType(), *DL}, |
| 201 | PTy{*SI.getPointerOperand()->getType(), *DL}; |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 202 | |
| 203 | MIRBuilder.buildStore( |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 204 | Val, Addr, |
Tim Northover | 28fdc42 | 2016-08-15 21:13:17 +0000 | [diff] [blame] | 205 | *MF.getMachineMemOperand( |
| 206 | MachinePointerInfo(SI.getPointerOperand()), |
| 207 | MachineMemOperand::MOStore, |
| 208 | DL->getTypeStoreSize(SI.getValueOperand()->getType()), |
| 209 | getMemOpAlignment(SI))); |
Tim Northover | ad2b717 | 2016-07-26 20:23:26 +0000 | [diff] [blame] | 210 | return true; |
| 211 | } |
| 212 | |
Tim Northover | 6f80b08 | 2016-08-19 17:47:05 +0000 | [diff] [blame] | 213 | bool IRTranslator::translateExtractValue(const User &U) { |
Tim Northover | b604622 | 2016-08-19 20:09:03 +0000 | [diff] [blame] | 214 | const Value *Src = U.getOperand(0); |
| 215 | Type *Int32Ty = Type::getInt32Ty(U.getContext()); |
Tim Northover | 6f80b08 | 2016-08-19 17:47:05 +0000 | [diff] [blame] | 216 | SmallVector<Value *, 1> Indices; |
| 217 | |
| 218 | // getIndexedOffsetInType is designed for GEPs, so the first index is the |
| 219 | // usual array element rather than looking into the actual aggregate. |
| 220 | Indices.push_back(ConstantInt::get(Int32Ty, 0)); |
Tim Northover | b604622 | 2016-08-19 20:09:03 +0000 | [diff] [blame] | 221 | |
| 222 | if (const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(&U)) { |
| 223 | for (auto Idx : EVI->indices()) |
| 224 | Indices.push_back(ConstantInt::get(Int32Ty, Idx)); |
| 225 | } else { |
| 226 | for (unsigned i = 1; i < U.getNumOperands(); ++i) |
| 227 | Indices.push_back(U.getOperand(i)); |
| 228 | } |
Tim Northover | 6f80b08 | 2016-08-19 17:47:05 +0000 | [diff] [blame] | 229 | |
| 230 | uint64_t Offset = 8 * DL->getIndexedOffsetInType(Src->getType(), Indices); |
| 231 | |
Tim Northover | b604622 | 2016-08-19 20:09:03 +0000 | [diff] [blame] | 232 | unsigned Res = getOrCreateVReg(U); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 233 | MIRBuilder.buildExtract(Res, Offset, getOrCreateVReg(*Src)); |
Tim Northover | 6f80b08 | 2016-08-19 17:47:05 +0000 | [diff] [blame] | 234 | |
| 235 | return true; |
| 236 | } |
| 237 | |
Tim Northover | bbbfb1c | 2016-08-19 20:08:55 +0000 | [diff] [blame] | 238 | bool IRTranslator::translateInsertValue(const User &U) { |
Tim Northover | b604622 | 2016-08-19 20:09:03 +0000 | [diff] [blame] | 239 | const Value *Src = U.getOperand(0); |
| 240 | Type *Int32Ty = Type::getInt32Ty(U.getContext()); |
Tim Northover | bbbfb1c | 2016-08-19 20:08:55 +0000 | [diff] [blame] | 241 | SmallVector<Value *, 1> Indices; |
| 242 | |
| 243 | // getIndexedOffsetInType is designed for GEPs, so the first index is the |
| 244 | // usual array element rather than looking into the actual aggregate. |
| 245 | Indices.push_back(ConstantInt::get(Int32Ty, 0)); |
Tim Northover | b604622 | 2016-08-19 20:09:03 +0000 | [diff] [blame] | 246 | |
| 247 | if (const InsertValueInst *IVI = dyn_cast<InsertValueInst>(&U)) { |
| 248 | for (auto Idx : IVI->indices()) |
| 249 | Indices.push_back(ConstantInt::get(Int32Ty, Idx)); |
| 250 | } else { |
| 251 | for (unsigned i = 2; i < U.getNumOperands(); ++i) |
| 252 | Indices.push_back(U.getOperand(i)); |
| 253 | } |
Tim Northover | bbbfb1c | 2016-08-19 20:08:55 +0000 | [diff] [blame] | 254 | |
| 255 | uint64_t Offset = 8 * DL->getIndexedOffsetInType(Src->getType(), Indices); |
| 256 | |
Tim Northover | b604622 | 2016-08-19 20:09:03 +0000 | [diff] [blame] | 257 | unsigned Res = getOrCreateVReg(U); |
| 258 | const Value &Inserted = *U.getOperand(1); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 259 | MIRBuilder.buildInsert(Res, getOrCreateVReg(*Src), getOrCreateVReg(Inserted), |
| 260 | Offset); |
Tim Northover | bbbfb1c | 2016-08-19 20:08:55 +0000 | [diff] [blame] | 261 | |
| 262 | return true; |
| 263 | } |
| 264 | |
Tim Northover | 5a28c36 | 2016-08-19 20:09:07 +0000 | [diff] [blame] | 265 | bool IRTranslator::translateSelect(const User &U) { |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 266 | MIRBuilder.buildSelect(getOrCreateVReg(U), getOrCreateVReg(*U.getOperand(0)), |
| 267 | getOrCreateVReg(*U.getOperand(1)), |
| 268 | getOrCreateVReg(*U.getOperand(2))); |
Tim Northover | 5a28c36 | 2016-08-19 20:09:07 +0000 | [diff] [blame] | 269 | return true; |
| 270 | } |
| 271 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 272 | bool IRTranslator::translateBitCast(const User &U) { |
Tim Northover | 5ae8350 | 2016-09-15 09:20:34 +0000 | [diff] [blame] | 273 | if (LLT{*U.getOperand(0)->getType(), *DL} == LLT{*U.getType(), *DL}) { |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 274 | unsigned &Reg = ValToVReg[&U]; |
Tim Northover | 7552ef5 | 2016-08-10 16:51:14 +0000 | [diff] [blame] | 275 | if (Reg) |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 276 | MIRBuilder.buildCopy(Reg, getOrCreateVReg(*U.getOperand(0))); |
Tim Northover | 7552ef5 | 2016-08-10 16:51:14 +0000 | [diff] [blame] | 277 | else |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 278 | Reg = getOrCreateVReg(*U.getOperand(0)); |
Tim Northover | 7c9eba9 | 2016-07-25 21:01:29 +0000 | [diff] [blame] | 279 | return true; |
| 280 | } |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 281 | return translateCast(TargetOpcode::G_BITCAST, U); |
Tim Northover | 7c9eba9 | 2016-07-25 21:01:29 +0000 | [diff] [blame] | 282 | } |
| 283 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 284 | bool IRTranslator::translateCast(unsigned Opcode, const User &U) { |
| 285 | unsigned Op = getOrCreateVReg(*U.getOperand(0)); |
| 286 | unsigned Res = getOrCreateVReg(U); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 287 | MIRBuilder.buildInstr(Opcode).addDef(Res).addUse(Op); |
Tim Northover | 7c9eba9 | 2016-07-25 21:01:29 +0000 | [diff] [blame] | 288 | return true; |
| 289 | } |
| 290 | |
Tim Northover | a7653b3 | 2016-09-12 11:20:22 +0000 | [diff] [blame] | 291 | bool IRTranslator::translateGetElementPtr(const User &U) { |
| 292 | // FIXME: support vector GEPs. |
| 293 | if (U.getType()->isVectorTy()) |
| 294 | return false; |
| 295 | |
| 296 | Value &Op0 = *U.getOperand(0); |
| 297 | unsigned BaseReg = getOrCreateVReg(Op0); |
Tim Northover | 5ae8350 | 2016-09-15 09:20:34 +0000 | [diff] [blame] | 298 | LLT PtrTy{*Op0.getType(), *DL}; |
Tim Northover | a7653b3 | 2016-09-12 11:20:22 +0000 | [diff] [blame] | 299 | unsigned PtrSize = DL->getPointerSizeInBits(PtrTy.getAddressSpace()); |
| 300 | LLT OffsetTy = LLT::scalar(PtrSize); |
| 301 | |
| 302 | int64_t Offset = 0; |
| 303 | for (gep_type_iterator GTI = gep_type_begin(&U), E = gep_type_end(&U); |
| 304 | GTI != E; ++GTI) { |
| 305 | const Value *Idx = GTI.getOperand(); |
| 306 | if (StructType *StTy = dyn_cast<StructType>(*GTI)) { |
| 307 | unsigned Field = cast<Constant>(Idx)->getUniqueInteger().getZExtValue(); |
| 308 | Offset += DL->getStructLayout(StTy)->getElementOffset(Field); |
| 309 | continue; |
| 310 | } else { |
| 311 | uint64_t ElementSize = DL->getTypeAllocSize(GTI.getIndexedType()); |
| 312 | |
| 313 | // If this is a scalar constant or a splat vector of constants, |
| 314 | // handle it quickly. |
| 315 | if (const auto *CI = dyn_cast<ConstantInt>(Idx)) { |
| 316 | Offset += ElementSize * CI->getSExtValue(); |
| 317 | continue; |
| 318 | } |
| 319 | |
| 320 | if (Offset != 0) { |
| 321 | unsigned NewBaseReg = MRI->createGenericVirtualRegister(PtrTy); |
| 322 | unsigned OffsetReg = MRI->createGenericVirtualRegister(OffsetTy); |
| 323 | MIRBuilder.buildConstant(OffsetReg, Offset); |
| 324 | MIRBuilder.buildGEP(NewBaseReg, BaseReg, OffsetReg); |
| 325 | |
| 326 | BaseReg = NewBaseReg; |
| 327 | Offset = 0; |
| 328 | } |
| 329 | |
| 330 | // N = N + Idx * ElementSize; |
| 331 | unsigned ElementSizeReg = MRI->createGenericVirtualRegister(OffsetTy); |
| 332 | MIRBuilder.buildConstant(ElementSizeReg, ElementSize); |
| 333 | |
| 334 | unsigned IdxReg = getOrCreateVReg(*Idx); |
| 335 | if (MRI->getType(IdxReg) != OffsetTy) { |
| 336 | unsigned NewIdxReg = MRI->createGenericVirtualRegister(OffsetTy); |
| 337 | MIRBuilder.buildSExtOrTrunc(NewIdxReg, IdxReg); |
| 338 | IdxReg = NewIdxReg; |
| 339 | } |
| 340 | |
| 341 | unsigned OffsetReg = MRI->createGenericVirtualRegister(OffsetTy); |
| 342 | MIRBuilder.buildMul(OffsetReg, ElementSizeReg, IdxReg); |
| 343 | |
| 344 | unsigned NewBaseReg = MRI->createGenericVirtualRegister(PtrTy); |
| 345 | MIRBuilder.buildGEP(NewBaseReg, BaseReg, OffsetReg); |
| 346 | BaseReg = NewBaseReg; |
| 347 | } |
| 348 | } |
| 349 | |
| 350 | if (Offset != 0) { |
| 351 | unsigned OffsetReg = MRI->createGenericVirtualRegister(OffsetTy); |
| 352 | MIRBuilder.buildConstant(OffsetReg, Offset); |
| 353 | MIRBuilder.buildGEP(getOrCreateVReg(U), BaseReg, OffsetReg); |
| 354 | return true; |
| 355 | } |
| 356 | |
| 357 | MIRBuilder.buildCopy(getOrCreateVReg(U), BaseReg); |
| 358 | return true; |
| 359 | } |
| 360 | |
Tim Northover | 3f18603 | 2016-10-18 20:03:45 +0000 | [diff] [blame] | 361 | bool IRTranslator::translateMemcpy(const CallInst &CI) { |
| 362 | LLT SizeTy{*CI.getArgOperand(2)->getType(), *DL}; |
| 363 | if (cast<PointerType>(CI.getArgOperand(0)->getType())->getAddressSpace() != |
| 364 | 0 || |
| 365 | cast<PointerType>(CI.getArgOperand(1)->getType())->getAddressSpace() != |
| 366 | 0 || |
| 367 | SizeTy.getSizeInBits() != DL->getPointerSizeInBits(0)) |
| 368 | return false; |
| 369 | |
| 370 | SmallVector<CallLowering::ArgInfo, 8> Args; |
| 371 | for (int i = 0; i < 3; ++i) { |
| 372 | const auto &Arg = CI.getArgOperand(i); |
| 373 | Args.emplace_back(getOrCreateVReg(*Arg), Arg->getType()); |
| 374 | } |
| 375 | |
| 376 | MachineOperand Callee = MachineOperand::CreateES("memcpy"); |
| 377 | |
| 378 | return CLI->lowerCall(MIRBuilder, Callee, |
| 379 | CallLowering::ArgInfo(0, CI.getType()), Args); |
| 380 | } |
Tim Northover | a7653b3 | 2016-09-12 11:20:22 +0000 | [diff] [blame] | 381 | |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 382 | bool IRTranslator::translateKnownIntrinsic(const CallInst &CI, |
| 383 | Intrinsic::ID ID) { |
| 384 | unsigned Op = 0; |
| 385 | switch (ID) { |
| 386 | default: return false; |
| 387 | case Intrinsic::uadd_with_overflow: Op = TargetOpcode::G_UADDE; break; |
| 388 | case Intrinsic::sadd_with_overflow: Op = TargetOpcode::G_SADDO; break; |
| 389 | case Intrinsic::usub_with_overflow: Op = TargetOpcode::G_USUBE; break; |
| 390 | case Intrinsic::ssub_with_overflow: Op = TargetOpcode::G_SSUBO; break; |
| 391 | case Intrinsic::umul_with_overflow: Op = TargetOpcode::G_UMULO; break; |
| 392 | case Intrinsic::smul_with_overflow: Op = TargetOpcode::G_SMULO; break; |
Tim Northover | 3f18603 | 2016-10-18 20:03:45 +0000 | [diff] [blame] | 393 | case Intrinsic::memcpy: |
| 394 | return translateMemcpy(CI); |
Tim Northover | 6e90430 | 2016-10-18 20:03:51 +0000 | [diff] [blame] | 395 | case Intrinsic::objectsize: { |
| 396 | // If we don't know by now, we're never going to know. |
| 397 | const ConstantInt *Min = cast<ConstantInt>(CI.getArgOperand(1)); |
| 398 | |
| 399 | MIRBuilder.buildConstant(getOrCreateVReg(CI), Min->isZero() ? -1ULL : 0); |
| 400 | return true; |
| 401 | } |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 402 | } |
| 403 | |
Tim Northover | 5ae8350 | 2016-09-15 09:20:34 +0000 | [diff] [blame] | 404 | LLT Ty{*CI.getOperand(0)->getType(), *DL}; |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 405 | LLT s1 = LLT::scalar(1); |
| 406 | unsigned Width = Ty.getSizeInBits(); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 407 | unsigned Res = MRI->createGenericVirtualRegister(Ty); |
| 408 | unsigned Overflow = MRI->createGenericVirtualRegister(s1); |
| 409 | auto MIB = MIRBuilder.buildInstr(Op) |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 410 | .addDef(Res) |
| 411 | .addDef(Overflow) |
| 412 | .addUse(getOrCreateVReg(*CI.getOperand(0))) |
| 413 | .addUse(getOrCreateVReg(*CI.getOperand(1))); |
| 414 | |
| 415 | if (Op == TargetOpcode::G_UADDE || Op == TargetOpcode::G_USUBE) { |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 416 | unsigned Zero = MRI->createGenericVirtualRegister(s1); |
| 417 | EntryBuilder.buildConstant(Zero, 0); |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 418 | MIB.addUse(Zero); |
| 419 | } |
| 420 | |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 421 | MIRBuilder.buildSequence(getOrCreateVReg(CI), Res, 0, Overflow, Width); |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 422 | return true; |
| 423 | } |
| 424 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 425 | bool IRTranslator::translateCall(const User &U) { |
| 426 | const CallInst &CI = cast<CallInst>(U); |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 427 | auto TII = MIRBuilder.getMF().getTarget().getIntrinsicInfo(); |
Tim Northover | 406024a | 2016-08-10 21:44:01 +0000 | [diff] [blame] | 428 | const Function *F = CI.getCalledFunction(); |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 429 | |
Tim Northover | 406024a | 2016-08-10 21:44:01 +0000 | [diff] [blame] | 430 | if (!F || !F->isIntrinsic()) { |
Tim Northover | 406024a | 2016-08-10 21:44:01 +0000 | [diff] [blame] | 431 | unsigned Res = CI.getType()->isVoidTy() ? 0 : getOrCreateVReg(CI); |
| 432 | SmallVector<unsigned, 8> Args; |
| 433 | for (auto &Arg: CI.arg_operands()) |
| 434 | Args.push_back(getOrCreateVReg(*Arg)); |
| 435 | |
Tim Northover | fe5f89b | 2016-08-29 19:07:08 +0000 | [diff] [blame] | 436 | return CLI->lowerCall(MIRBuilder, CI, Res, Args, [&]() { |
| 437 | return getOrCreateVReg(*CI.getCalledValue()); |
| 438 | }); |
Tim Northover | 406024a | 2016-08-10 21:44:01 +0000 | [diff] [blame] | 439 | } |
| 440 | |
| 441 | Intrinsic::ID ID = F->getIntrinsicID(); |
| 442 | if (TII && ID == Intrinsic::not_intrinsic) |
| 443 | ID = static_cast<Intrinsic::ID>(TII->getIntrinsicID(F)); |
| 444 | |
| 445 | assert(ID != Intrinsic::not_intrinsic && "unknown intrinsic"); |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 446 | |
Tim Northover | 91c8173 | 2016-08-19 17:17:06 +0000 | [diff] [blame] | 447 | if (translateKnownIntrinsic(CI, ID)) |
| 448 | return true; |
| 449 | |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 450 | unsigned Res = CI.getType()->isVoidTy() ? 0 : getOrCreateVReg(CI); |
| 451 | MachineInstrBuilder MIB = |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 452 | MIRBuilder.buildIntrinsic(ID, Res, !CI.doesNotAccessMemory()); |
Tim Northover | 5fb414d | 2016-07-29 22:32:36 +0000 | [diff] [blame] | 453 | |
| 454 | for (auto &Arg : CI.arg_operands()) { |
| 455 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Arg)) |
| 456 | MIB.addImm(CI->getSExtValue()); |
| 457 | else |
| 458 | MIB.addUse(getOrCreateVReg(*Arg)); |
| 459 | } |
| 460 | return true; |
| 461 | } |
| 462 | |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 463 | bool IRTranslator::translateStaticAlloca(const AllocaInst &AI) { |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 464 | if (!TPC->isGlobalISelAbortEnabled() && !AI.isStaticAlloca()) |
| 465 | return false; |
| 466 | |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 467 | assert(AI.isStaticAlloca() && "only handle static allocas now"); |
| 468 | MachineFunction &MF = MIRBuilder.getMF(); |
| 469 | unsigned ElementSize = DL->getTypeStoreSize(AI.getAllocatedType()); |
| 470 | unsigned Size = |
| 471 | ElementSize * cast<ConstantInt>(AI.getArraySize())->getZExtValue(); |
| 472 | |
Tim Northover | 8d2f52e | 2016-07-27 17:47:54 +0000 | [diff] [blame] | 473 | // Always allocate at least one byte. |
| 474 | Size = std::max(Size, 1u); |
| 475 | |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 476 | unsigned Alignment = AI.getAlignment(); |
| 477 | if (!Alignment) |
| 478 | Alignment = DL->getABITypeAlignment(AI.getAllocatedType()); |
| 479 | |
| 480 | unsigned Res = getOrCreateVReg(AI); |
Matthias Braun | 9332039 | 2016-07-28 20:13:42 +0000 | [diff] [blame] | 481 | int FI = MF.getFrameInfo().CreateStackObject(Size, Alignment, false, &AI); |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 482 | MIRBuilder.buildFrameIndex(Res, FI); |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 483 | return true; |
| 484 | } |
| 485 | |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 486 | bool IRTranslator::translatePHI(const User &U) { |
| 487 | const PHINode &PI = cast<PHINode>(U); |
Tim Northover | 25d1286 | 2016-09-09 11:47:31 +0000 | [diff] [blame] | 488 | auto MIB = MIRBuilder.buildInstr(TargetOpcode::PHI); |
Tim Northover | 97d0cb3 | 2016-08-05 17:16:40 +0000 | [diff] [blame] | 489 | MIB.addDef(getOrCreateVReg(PI)); |
| 490 | |
| 491 | PendingPHIs.emplace_back(&PI, MIB.getInstr()); |
| 492 | return true; |
| 493 | } |
| 494 | |
| 495 | void IRTranslator::finishPendingPhis() { |
| 496 | for (std::pair<const PHINode *, MachineInstr *> &Phi : PendingPHIs) { |
| 497 | const PHINode *PI = Phi.first; |
| 498 | MachineInstrBuilder MIB(MIRBuilder.getMF(), Phi.second); |
| 499 | |
| 500 | // All MachineBasicBlocks exist, add them to the PHI. We assume IRTranslator |
| 501 | // won't create extra control flow here, otherwise we need to find the |
| 502 | // dominating predecessor here (or perhaps force the weirder IRTranslators |
| 503 | // to provide a simple boundary). |
| 504 | for (unsigned i = 0; i < PI->getNumIncomingValues(); ++i) { |
| 505 | assert(BBToMBB[PI->getIncomingBlock(i)]->isSuccessor(MIB->getParent()) && |
| 506 | "I appear to have misunderstood Machine PHIs"); |
| 507 | MIB.addUse(getOrCreateVReg(*PI->getIncomingValue(i))); |
| 508 | MIB.addMBB(BBToMBB[PI->getIncomingBlock(i)]); |
| 509 | } |
| 510 | } |
Tim Northover | 14e7f73 | 2016-08-05 17:50:36 +0000 | [diff] [blame] | 511 | |
| 512 | PendingPHIs.clear(); |
Tim Northover | 97d0cb3 | 2016-08-05 17:16:40 +0000 | [diff] [blame] | 513 | } |
| 514 | |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 515 | bool IRTranslator::translate(const Instruction &Inst) { |
Quentin Colombet | a7fae16 | 2016-02-11 17:53:23 +0000 | [diff] [blame] | 516 | MIRBuilder.setDebugLoc(Inst.getDebugLoc()); |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 517 | switch(Inst.getOpcode()) { |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 518 | #define HANDLE_INST(NUM, OPCODE, CLASS) \ |
| 519 | case Instruction::OPCODE: return translate##OPCODE(Inst); |
| 520 | #include "llvm/IR/Instruction.def" |
Quentin Colombet | 74d7d2f | 2016-02-11 18:53:28 +0000 | [diff] [blame] | 521 | default: |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 522 | if (!TPC->isGlobalISelAbortEnabled()) |
| 523 | return false; |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 524 | llvm_unreachable("unknown opcode"); |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 525 | } |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 526 | } |
| 527 | |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 528 | bool IRTranslator::translate(const Constant &C, unsigned Reg) { |
Tim Northover | d403a3d | 2016-08-09 23:01:30 +0000 | [diff] [blame] | 529 | if (auto CI = dyn_cast<ConstantInt>(&C)) |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 530 | EntryBuilder.buildConstant(Reg, CI->getZExtValue()); |
Tim Northover | b16734f | 2016-08-19 20:09:15 +0000 | [diff] [blame] | 531 | else if (auto CF = dyn_cast<ConstantFP>(&C)) |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 532 | EntryBuilder.buildFConstant(Reg, *CF); |
Tim Northover | d403a3d | 2016-08-09 23:01:30 +0000 | [diff] [blame] | 533 | else if (isa<UndefValue>(C)) |
| 534 | EntryBuilder.buildInstr(TargetOpcode::IMPLICIT_DEF).addDef(Reg); |
Tim Northover | 8e0c53a | 2016-08-11 21:40:55 +0000 | [diff] [blame] | 535 | else if (isa<ConstantPointerNull>(C)) |
Tim Northover | 0f140c7 | 2016-09-09 11:46:34 +0000 | [diff] [blame] | 536 | EntryBuilder.buildInstr(TargetOpcode::G_CONSTANT) |
Tim Northover | 8e0c53a | 2016-08-11 21:40:55 +0000 | [diff] [blame] | 537 | .addDef(Reg) |
| 538 | .addImm(0); |
Tim Northover | 032548f | 2016-09-12 12:10:41 +0000 | [diff] [blame] | 539 | else if (auto GV = dyn_cast<GlobalValue>(&C)) |
| 540 | EntryBuilder.buildGlobalValue(Reg, GV); |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 541 | else if (auto CE = dyn_cast<ConstantExpr>(&C)) { |
| 542 | switch(CE->getOpcode()) { |
| 543 | #define HANDLE_INST(NUM, OPCODE, CLASS) \ |
| 544 | case Instruction::OPCODE: return translate##OPCODE(*CE); |
| 545 | #include "llvm/IR/Instruction.def" |
| 546 | default: |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 547 | if (!TPC->isGlobalISelAbortEnabled()) |
| 548 | return false; |
Tim Northover | 357f1be | 2016-08-10 23:02:41 +0000 | [diff] [blame] | 549 | llvm_unreachable("unknown opcode"); |
| 550 | } |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 551 | } else if (!TPC->isGlobalISelAbortEnabled()) |
| 552 | return false; |
| 553 | else |
Tim Northover | d403a3d | 2016-08-09 23:01:30 +0000 | [diff] [blame] | 554 | llvm_unreachable("unhandled constant kind"); |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 555 | |
Tim Northover | d403a3d | 2016-08-09 23:01:30 +0000 | [diff] [blame] | 556 | return true; |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 557 | } |
| 558 | |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 559 | |
Tim Northover | 0d51044 | 2016-08-11 16:21:29 +0000 | [diff] [blame] | 560 | void IRTranslator::finalizeFunction() { |
| 561 | finishPendingPhis(); |
| 562 | |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 563 | // Release the memory used by the different maps we |
| 564 | // needed during the translation. |
Quentin Colombet | ccd7725 | 2016-02-11 21:48:32 +0000 | [diff] [blame] | 565 | ValToVReg.clear(); |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 566 | Constants.clear(); |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 567 | } |
| 568 | |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 569 | bool IRTranslator::runOnMachineFunction(MachineFunction &MF) { |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 570 | const Function &F = *MF.getFunction(); |
Quentin Colombet | fd9d0a0 | 2016-02-11 19:59:41 +0000 | [diff] [blame] | 571 | if (F.empty()) |
| 572 | return false; |
Quentin Colombet | ba2a016 | 2016-02-16 19:26:02 +0000 | [diff] [blame] | 573 | CLI = MF.getSubtarget().getCallLowering(); |
Quentin Colombet | 000b580 | 2016-03-11 17:27:51 +0000 | [diff] [blame] | 574 | MIRBuilder.setMF(MF); |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 575 | EntryBuilder.setMF(MF); |
Quentin Colombet | 17c494b | 2016-02-11 17:51:31 +0000 | [diff] [blame] | 576 | MRI = &MF.getRegInfo(); |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 577 | DL = &F.getParent()->getDataLayout(); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 578 | TPC = &getAnalysis<TargetPassConfig>(); |
Tim Northover | bd50546 | 2016-07-22 16:59:52 +0000 | [diff] [blame] | 579 | |
Tim Northover | 14e7f73 | 2016-08-05 17:50:36 +0000 | [diff] [blame] | 580 | assert(PendingPHIs.empty() && "stale PHIs"); |
| 581 | |
Quentin Colombet | fd9d0a0 | 2016-02-11 19:59:41 +0000 | [diff] [blame] | 582 | // Setup the arguments. |
Quentin Colombet | 53237a9 | 2016-03-11 17:27:43 +0000 | [diff] [blame] | 583 | MachineBasicBlock &MBB = getOrCreateBB(F.front()); |
Quentin Colombet | 91ebd71 | 2016-03-11 17:27:47 +0000 | [diff] [blame] | 584 | MIRBuilder.setMBB(MBB); |
Quentin Colombet | fd9d0a0 | 2016-02-11 19:59:41 +0000 | [diff] [blame] | 585 | SmallVector<unsigned, 8> VRegArgs; |
| 586 | for (const Argument &Arg: F.args()) |
Quentin Colombet | e225e25 | 2016-03-11 17:27:54 +0000 | [diff] [blame] | 587 | VRegArgs.push_back(getOrCreateVReg(Arg)); |
Tim Northover | 862758ec | 2016-09-21 12:57:35 +0000 | [diff] [blame] | 588 | bool Succeeded = CLI->lowerFormalArguments(MIRBuilder, F, VRegArgs); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 589 | if (!Succeeded) { |
| 590 | if (!TPC->isGlobalISelAbortEnabled()) { |
| 591 | MIRBuilder.getMF().getProperties().set( |
| 592 | MachineFunctionProperties::Property::FailedISel); |
| 593 | return false; |
| 594 | } |
Quentin Colombet | fd9d0a0 | 2016-02-11 19:59:41 +0000 | [diff] [blame] | 595 | report_fatal_error("Unable to lower arguments"); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 596 | } |
Quentin Colombet | fd9d0a0 | 2016-02-11 19:59:41 +0000 | [diff] [blame] | 597 | |
Tim Northover | 5ed648e | 2016-08-09 21:28:04 +0000 | [diff] [blame] | 598 | // Now that we've got the ABI handling code, it's safe to set a location for |
| 599 | // any Constants we find in the IR. |
| 600 | if (MBB.empty()) |
| 601 | EntryBuilder.setMBB(MBB); |
| 602 | else |
| 603 | EntryBuilder.setInstr(MBB.back(), /* Before */ false); |
| 604 | |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 605 | for (const BasicBlock &BB: F) { |
Quentin Colombet | 53237a9 | 2016-03-11 17:27:43 +0000 | [diff] [blame] | 606 | MachineBasicBlock &MBB = getOrCreateBB(BB); |
Quentin Colombet | 91ebd71 | 2016-03-11 17:27:47 +0000 | [diff] [blame] | 607 | // Set the insertion point of all the following translations to |
| 608 | // the end of this basic block. |
| 609 | MIRBuilder.setMBB(MBB); |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 610 | for (const Instruction &Inst: BB) { |
| 611 | bool Succeeded = translate(Inst); |
| 612 | if (!Succeeded) { |
| 613 | DEBUG(dbgs() << "Cannot translate: " << Inst << '\n'); |
Quentin Colombet | 3bb32cc | 2016-08-26 23:49:05 +0000 | [diff] [blame] | 614 | if (TPC->isGlobalISelAbortEnabled()) |
| 615 | report_fatal_error("Unable to translate instruction"); |
| 616 | MF.getProperties().set(MachineFunctionProperties::Property::FailedISel); |
| 617 | break; |
Quentin Colombet | 2ecff3b | 2016-02-10 22:59:27 +0000 | [diff] [blame] | 618 | } |
| 619 | } |
| 620 | } |
Tim Northover | 72eebfa | 2016-07-12 22:23:42 +0000 | [diff] [blame] | 621 | |
Tim Northover | 0d51044 | 2016-08-11 16:21:29 +0000 | [diff] [blame] | 622 | finalizeFunction(); |
Tim Northover | 97d0cb3 | 2016-08-05 17:16:40 +0000 | [diff] [blame] | 623 | |
Tim Northover | 72eebfa | 2016-07-12 22:23:42 +0000 | [diff] [blame] | 624 | // Now that the MachineFrameInfo has been configured, no further changes to |
| 625 | // the reserved registers are possible. |
| 626 | MRI->freezeReservedRegs(MF); |
| 627 | |
Quentin Colombet | 105cf2b | 2016-01-20 20:58:56 +0000 | [diff] [blame] | 628 | return false; |
| 629 | } |