blob: 77dead8d24137a7bdd0b05b3ea5c0ebc609278b3 [file] [log] [blame]
Igor Bregerf7359d82017-02-22 12:25:09 +00001//===- X86InstructionSelector.cpp ----------------------------*- 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 targeting of the InstructionSelector class for
11/// X86.
12/// \todo This should be generated by TableGen.
13//===----------------------------------------------------------------------===//
14
Igor Bregera8ba5722017-03-23 15:25:57 +000015#include "X86InstrBuilder.h"
Igor Bregerf7359d82017-02-22 12:25:09 +000016#include "X86InstrInfo.h"
17#include "X86RegisterBankInfo.h"
18#include "X86RegisterInfo.h"
19#include "X86Subtarget.h"
20#include "X86TargetMachine.h"
Igor Breger3b97ea32017-04-12 12:54:54 +000021#include "llvm/CodeGen/GlobalISel/InstructionSelector.h"
Igor Breger28f290f2017-05-17 12:48:08 +000022#include "llvm/CodeGen/GlobalISel/Utils.h"
Igor Bregerf7359d82017-02-22 12:25:09 +000023#include "llvm/CodeGen/MachineBasicBlock.h"
24#include "llvm/CodeGen/MachineFunction.h"
25#include "llvm/CodeGen/MachineInstr.h"
26#include "llvm/CodeGen/MachineInstrBuilder.h"
Daniel Sanders0b5293f2017-04-06 09:49:34 +000027#include "llvm/CodeGen/MachineOperand.h"
Igor Bregerf7359d82017-02-22 12:25:09 +000028#include "llvm/CodeGen/MachineRegisterInfo.h"
29#include "llvm/IR/Type.h"
30#include "llvm/Support/Debug.h"
31#include "llvm/Support/raw_ostream.h"
32
33#define DEBUG_TYPE "X86-isel"
34
35using namespace llvm;
36
37#ifndef LLVM_BUILD_GLOBAL_ISEL
38#error "You shouldn't build this"
39#endif
40
Daniel Sanders0b5293f2017-04-06 09:49:34 +000041namespace {
42
Daniel Sanderse7b0d662017-04-21 15:59:56 +000043#define GET_GLOBALISEL_PREDICATE_BITSET
44#include "X86GenGlobalISel.inc"
45#undef GET_GLOBALISEL_PREDICATE_BITSET
46
Daniel Sanders0b5293f2017-04-06 09:49:34 +000047class X86InstructionSelector : public InstructionSelector {
48public:
Daniel Sanderse7b0d662017-04-21 15:59:56 +000049 X86InstructionSelector(const X86TargetMachine &TM, const X86Subtarget &STI,
Daniel Sanders0b5293f2017-04-06 09:49:34 +000050 const X86RegisterBankInfo &RBI);
51
52 bool select(MachineInstr &I) const override;
53
54private:
55 /// tblgen-erated 'select' implementation, used as the initial selector for
56 /// the patterns that don't require complex C++.
57 bool selectImpl(MachineInstr &I) const;
58
Igor Breger2452ef02017-05-01 07:06:08 +000059 // TODO: remove after suported by Tablegen-erated instruction selection.
Daniel Sanders0b5293f2017-04-06 09:49:34 +000060 unsigned getLoadStoreOp(LLT &Ty, const RegisterBank &RB, unsigned Opc,
61 uint64_t Alignment) const;
62
Daniel Sanders0b5293f2017-04-06 09:49:34 +000063 bool selectLoadStoreOp(MachineInstr &I, MachineRegisterInfo &MRI,
64 MachineFunction &MF) const;
Igor Breger810c6252017-05-08 09:40:43 +000065 bool selectFrameIndexOrGep(MachineInstr &I, MachineRegisterInfo &MRI,
66 MachineFunction &MF) const;
Igor Breger3b97ea32017-04-12 12:54:54 +000067 bool selectConstant(MachineInstr &I, MachineRegisterInfo &MRI,
68 MachineFunction &MF) const;
Igor Breger4fdf1e42017-04-19 11:34:59 +000069 bool selectTrunc(MachineInstr &I, MachineRegisterInfo &MRI,
70 MachineFunction &MF) const;
Igor Bregerfda31e62017-05-10 06:52:58 +000071 bool selectZext(MachineInstr &I, MachineRegisterInfo &MRI,
72 MachineFunction &MF) const;
Igor Bregerc7b59772017-05-11 07:17:40 +000073 bool selectCmp(MachineInstr &I, MachineRegisterInfo &MRI,
74 MachineFunction &MF) const;
Daniel Sanders0b5293f2017-04-06 09:49:34 +000075
Igor Breger28f290f2017-05-17 12:48:08 +000076 bool selectUadde(MachineInstr &I, MachineRegisterInfo &MRI,
77 MachineFunction &MF) const;
78
Daniel Sanderse7b0d662017-04-21 15:59:56 +000079 const X86TargetMachine &TM;
Daniel Sanders0b5293f2017-04-06 09:49:34 +000080 const X86Subtarget &STI;
81 const X86InstrInfo &TII;
82 const X86RegisterInfo &TRI;
83 const X86RegisterBankInfo &RBI;
Daniel Sanderse7b0d662017-04-21 15:59:56 +000084
Daniel Sanderse9fdba32017-04-29 17:30:09 +000085#define GET_GLOBALISEL_PREDICATES_DECL
86#include "X86GenGlobalISel.inc"
87#undef GET_GLOBALISEL_PREDICATES_DECL
Daniel Sanders0b5293f2017-04-06 09:49:34 +000088
89#define GET_GLOBALISEL_TEMPORARIES_DECL
90#include "X86GenGlobalISel.inc"
91#undef GET_GLOBALISEL_TEMPORARIES_DECL
92};
93
94} // end anonymous namespace
95
Daniel Sanders8a4bae92017-03-14 21:32:08 +000096#define GET_GLOBALISEL_IMPL
Igor Bregerf7359d82017-02-22 12:25:09 +000097#include "X86GenGlobalISel.inc"
Daniel Sanders8a4bae92017-03-14 21:32:08 +000098#undef GET_GLOBALISEL_IMPL
Igor Bregerf7359d82017-02-22 12:25:09 +000099
Daniel Sanderse7b0d662017-04-21 15:59:56 +0000100X86InstructionSelector::X86InstructionSelector(const X86TargetMachine &TM,
101 const X86Subtarget &STI,
Igor Bregerf7359d82017-02-22 12:25:09 +0000102 const X86RegisterBankInfo &RBI)
Daniel Sanderse7b0d662017-04-21 15:59:56 +0000103 : InstructionSelector(), TM(TM), STI(STI), TII(*STI.getInstrInfo()),
Daniel Sanderse9fdba32017-04-29 17:30:09 +0000104 TRI(*STI.getRegisterInfo()), RBI(RBI),
105#define GET_GLOBALISEL_PREDICATES_INIT
106#include "X86GenGlobalISel.inc"
107#undef GET_GLOBALISEL_PREDICATES_INIT
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000108#define GET_GLOBALISEL_TEMPORARIES_INIT
109#include "X86GenGlobalISel.inc"
110#undef GET_GLOBALISEL_TEMPORARIES_INIT
111{
112}
Igor Bregerf7359d82017-02-22 12:25:09 +0000113
114// FIXME: This should be target-independent, inferred from the types declared
115// for each class in the bank.
116static const TargetRegisterClass *
117getRegClassForTypeOnBank(LLT Ty, const RegisterBank &RB) {
118 if (RB.getID() == X86::GPRRegBankID) {
Igor Breger4fdf1e42017-04-19 11:34:59 +0000119 if (Ty.getSizeInBits() <= 8)
120 return &X86::GR8RegClass;
121 if (Ty.getSizeInBits() == 16)
122 return &X86::GR16RegClass;
Igor Breger321cf3c2017-03-03 08:06:46 +0000123 if (Ty.getSizeInBits() == 32)
Igor Bregerf7359d82017-02-22 12:25:09 +0000124 return &X86::GR32RegClass;
125 if (Ty.getSizeInBits() == 64)
126 return &X86::GR64RegClass;
127 }
Igor Breger321cf3c2017-03-03 08:06:46 +0000128 if (RB.getID() == X86::VECRRegBankID) {
129 if (Ty.getSizeInBits() == 32)
130 return &X86::FR32XRegClass;
131 if (Ty.getSizeInBits() == 64)
132 return &X86::FR64XRegClass;
133 if (Ty.getSizeInBits() == 128)
134 return &X86::VR128XRegClass;
135 if (Ty.getSizeInBits() == 256)
136 return &X86::VR256XRegClass;
137 if (Ty.getSizeInBits() == 512)
138 return &X86::VR512RegClass;
139 }
Igor Bregerf7359d82017-02-22 12:25:09 +0000140
141 llvm_unreachable("Unknown RegBank!");
142}
143
144// Set X86 Opcode and constrain DestReg.
145static bool selectCopy(MachineInstr &I, const TargetInstrInfo &TII,
146 MachineRegisterInfo &MRI, const TargetRegisterInfo &TRI,
147 const RegisterBankInfo &RBI) {
148
149 unsigned DstReg = I.getOperand(0).getReg();
150 if (TargetRegisterInfo::isPhysicalRegister(DstReg)) {
151 assert(I.isCopy() && "Generic operators do not allow physical registers");
152 return true;
153 }
154
155 const RegisterBank &RegBank = *RBI.getRegBank(DstReg, MRI, TRI);
156 const unsigned DstSize = MRI.getType(DstReg).getSizeInBits();
157 unsigned SrcReg = I.getOperand(1).getReg();
158 const unsigned SrcSize = RBI.getSizeInBits(SrcReg, MRI, TRI);
Igor Breger360d0f22017-04-27 08:02:03 +0000159
Igor Bregerf7359d82017-02-22 12:25:09 +0000160 assert((!TargetRegisterInfo::isPhysicalRegister(SrcReg) || I.isCopy()) &&
161 "No phys reg on generic operators");
162 assert((DstSize == SrcSize ||
163 // Copies are a mean to setup initial types, the number of
164 // bits may not exactly match.
165 (TargetRegisterInfo::isPhysicalRegister(SrcReg) &&
166 DstSize <= RBI.getSizeInBits(SrcReg, MRI, TRI))) &&
167 "Copy with different width?!");
168
169 const TargetRegisterClass *RC = nullptr;
170
171 switch (RegBank.getID()) {
172 case X86::GPRRegBankID:
173 assert((DstSize <= 64) && "GPRs cannot get more than 64-bit width values.");
174 RC = getRegClassForTypeOnBank(MRI.getType(DstReg), RegBank);
Igor Breger360d0f22017-04-27 08:02:03 +0000175
176 // Change the physical register
177 if (SrcSize > DstSize && TargetRegisterInfo::isPhysicalRegister(SrcReg)) {
178 if (RC == &X86::GR32RegClass)
179 I.getOperand(1).setSubReg(X86::sub_32bit);
180 else if (RC == &X86::GR16RegClass)
181 I.getOperand(1).setSubReg(X86::sub_16bit);
182 else if (RC == &X86::GR8RegClass)
183 I.getOperand(1).setSubReg(X86::sub_8bit);
184
185 I.getOperand(1).substPhysReg(SrcReg, TRI);
186 }
Igor Bregerf7359d82017-02-22 12:25:09 +0000187 break;
Igor Breger321cf3c2017-03-03 08:06:46 +0000188 case X86::VECRRegBankID:
189 RC = getRegClassForTypeOnBank(MRI.getType(DstReg), RegBank);
190 break;
Igor Bregerf7359d82017-02-22 12:25:09 +0000191 default:
192 llvm_unreachable("Unknown RegBank!");
193 }
194
195 // No need to constrain SrcReg. It will get constrained when
196 // we hit another of its use or its defs.
197 // Copies do not have constraints.
Igor Breger8a924be2017-03-23 12:13:29 +0000198 const TargetRegisterClass *OldRC = MRI.getRegClassOrNull(DstReg);
Igor Breger321cf3c2017-03-03 08:06:46 +0000199 if (!OldRC || !RC->hasSubClassEq(OldRC)) {
200 if (!RBI.constrainGenericRegister(DstReg, *RC, MRI)) {
Igor Breger8a924be2017-03-23 12:13:29 +0000201 DEBUG(dbgs() << "Failed to constrain " << TII.getName(I.getOpcode())
202 << " operand\n");
203 return false;
204 }
Igor Bregerf7359d82017-02-22 12:25:09 +0000205 }
206 I.setDesc(TII.get(X86::COPY));
207 return true;
208}
209
210bool X86InstructionSelector::select(MachineInstr &I) const {
211 assert(I.getParent() && "Instruction should be in a basic block!");
212 assert(I.getParent()->getParent() && "Instruction should be in a function!");
213
214 MachineBasicBlock &MBB = *I.getParent();
215 MachineFunction &MF = *MBB.getParent();
216 MachineRegisterInfo &MRI = MF.getRegInfo();
217
218 unsigned Opcode = I.getOpcode();
219 if (!isPreISelGenericOpcode(Opcode)) {
220 // Certain non-generic instructions also need some special handling.
221
222 if (I.isCopy())
223 return selectCopy(I, TII, MRI, TRI, RBI);
224
225 // TODO: handle more cases - LOAD_STACK_GUARD, PHI
226 return true;
227 }
228
Benjamin Kramer5a7e0f82017-02-22 12:59:47 +0000229 assert(I.getNumOperands() == I.getNumExplicitOperands() &&
230 "Generic instruction has unexpected implicit operands\n");
Igor Bregerf7359d82017-02-22 12:25:09 +0000231
Igor Breger2452ef02017-05-01 07:06:08 +0000232 if (selectImpl(I))
Igor Bregerfda31e62017-05-10 06:52:58 +0000233 return true;
Igor Breger2452ef02017-05-01 07:06:08 +0000234
235 DEBUG(dbgs() << " C++ instruction selection: "; I.print(dbgs()));
236
237 // TODO: This should be implemented by tblgen.
Igor Bregera8ba5722017-03-23 15:25:57 +0000238 if (selectLoadStoreOp(I, MRI, MF))
Igor Breger321cf3c2017-03-03 08:06:46 +0000239 return true;
Igor Breger810c6252017-05-08 09:40:43 +0000240 if (selectFrameIndexOrGep(I, MRI, MF))
Igor Breger531a2032017-03-26 08:11:12 +0000241 return true;
Igor Breger3b97ea32017-04-12 12:54:54 +0000242 if (selectConstant(I, MRI, MF))
243 return true;
Igor Breger4fdf1e42017-04-19 11:34:59 +0000244 if (selectTrunc(I, MRI, MF))
245 return true;
Igor Bregerfda31e62017-05-10 06:52:58 +0000246 if (selectZext(I, MRI, MF))
247 return true;
Igor Bregerc7b59772017-05-11 07:17:40 +0000248 if (selectCmp(I, MRI, MF))
249 return true;
Igor Breger28f290f2017-05-17 12:48:08 +0000250 if (selectUadde(I, MRI, MF))
251 return true;
Igor Breger321cf3c2017-03-03 08:06:46 +0000252
Igor Breger2452ef02017-05-01 07:06:08 +0000253 return false;
Igor Bregerf7359d82017-02-22 12:25:09 +0000254}
Igor Breger321cf3c2017-03-03 08:06:46 +0000255
Igor Bregera8ba5722017-03-23 15:25:57 +0000256unsigned X86InstructionSelector::getLoadStoreOp(LLT &Ty, const RegisterBank &RB,
257 unsigned Opc,
258 uint64_t Alignment) const {
259 bool Isload = (Opc == TargetOpcode::G_LOAD);
260 bool HasAVX = STI.hasAVX();
261 bool HasAVX512 = STI.hasAVX512();
262 bool HasVLX = STI.hasVLX();
263
264 if (Ty == LLT::scalar(8)) {
265 if (X86::GPRRegBankID == RB.getID())
266 return Isload ? X86::MOV8rm : X86::MOV8mr;
267 } else if (Ty == LLT::scalar(16)) {
268 if (X86::GPRRegBankID == RB.getID())
269 return Isload ? X86::MOV16rm : X86::MOV16mr;
Igor Bregera9edb882017-05-01 06:08:32 +0000270 } else if (Ty == LLT::scalar(32) || Ty == LLT::pointer(0, 32)) {
Igor Bregera8ba5722017-03-23 15:25:57 +0000271 if (X86::GPRRegBankID == RB.getID())
272 return Isload ? X86::MOV32rm : X86::MOV32mr;
273 if (X86::VECRRegBankID == RB.getID())
274 return Isload ? (HasAVX512 ? X86::VMOVSSZrm
275 : HasAVX ? X86::VMOVSSrm : X86::MOVSSrm)
276 : (HasAVX512 ? X86::VMOVSSZmr
277 : HasAVX ? X86::VMOVSSmr : X86::MOVSSmr);
Igor Bregera9edb882017-05-01 06:08:32 +0000278 } else if (Ty == LLT::scalar(64) || Ty == LLT::pointer(0, 64)) {
Igor Bregera8ba5722017-03-23 15:25:57 +0000279 if (X86::GPRRegBankID == RB.getID())
280 return Isload ? X86::MOV64rm : X86::MOV64mr;
281 if (X86::VECRRegBankID == RB.getID())
282 return Isload ? (HasAVX512 ? X86::VMOVSDZrm
283 : HasAVX ? X86::VMOVSDrm : X86::MOVSDrm)
284 : (HasAVX512 ? X86::VMOVSDZmr
285 : HasAVX ? X86::VMOVSDmr : X86::MOVSDmr);
286 } else if (Ty.isVector() && Ty.getSizeInBits() == 128) {
287 if (Alignment >= 16)
288 return Isload ? (HasVLX ? X86::VMOVAPSZ128rm
289 : HasAVX512
290 ? X86::VMOVAPSZ128rm_NOVLX
291 : HasAVX ? X86::VMOVAPSrm : X86::MOVAPSrm)
292 : (HasVLX ? X86::VMOVAPSZ128mr
293 : HasAVX512
294 ? X86::VMOVAPSZ128mr_NOVLX
295 : HasAVX ? X86::VMOVAPSmr : X86::MOVAPSmr);
296 else
297 return Isload ? (HasVLX ? X86::VMOVUPSZ128rm
298 : HasAVX512
299 ? X86::VMOVUPSZ128rm_NOVLX
300 : HasAVX ? X86::VMOVUPSrm : X86::MOVUPSrm)
301 : (HasVLX ? X86::VMOVUPSZ128mr
302 : HasAVX512
303 ? X86::VMOVUPSZ128mr_NOVLX
304 : HasAVX ? X86::VMOVUPSmr : X86::MOVUPSmr);
Igor Breger617be6e2017-05-23 08:23:51 +0000305 } else if (Ty.isVector() && Ty.getSizeInBits() == 256) {
306 if (Alignment >= 32)
307 return Isload ? (HasVLX ? X86::VMOVAPSZ256rm
308 : HasAVX512 ? X86::VMOVAPSZ256rm_NOVLX
309 : X86::VMOVAPSYrm)
310 : (HasVLX ? X86::VMOVAPSZ256mr
311 : HasAVX512 ? X86::VMOVAPSZ256mr_NOVLX
312 : X86::VMOVAPSYmr);
313 else
314 return Isload ? (HasVLX ? X86::VMOVUPSZ256rm
315 : HasAVX512 ? X86::VMOVUPSZ256rm_NOVLX
316 : X86::VMOVUPSYrm)
317 : (HasVLX ? X86::VMOVUPSZ256mr
318 : HasAVX512 ? X86::VMOVUPSZ256mr_NOVLX
319 : X86::VMOVUPSYmr);
320 } else if (Ty.isVector() && Ty.getSizeInBits() == 512) {
321 if (Alignment >= 64)
322 return Isload ? X86::VMOVAPSZrm : X86::VMOVAPSZmr;
323 else
324 return Isload ? X86::VMOVUPSZrm : X86::VMOVUPSZmr;
Igor Bregera8ba5722017-03-23 15:25:57 +0000325 }
326 return Opc;
327}
328
329bool X86InstructionSelector::selectLoadStoreOp(MachineInstr &I,
330 MachineRegisterInfo &MRI,
331 MachineFunction &MF) const {
332
333 unsigned Opc = I.getOpcode();
334
335 if (Opc != TargetOpcode::G_STORE && Opc != TargetOpcode::G_LOAD)
336 return false;
337
338 const unsigned DefReg = I.getOperand(0).getReg();
339 LLT Ty = MRI.getType(DefReg);
340 const RegisterBank &RB = *RBI.getRegBank(DefReg, MRI, TRI);
341
342 auto &MemOp = **I.memoperands_begin();
343 unsigned NewOpc = getLoadStoreOp(Ty, RB, Opc, MemOp.getAlignment());
344 if (NewOpc == Opc)
345 return false;
346
347 I.setDesc(TII.get(NewOpc));
348 MachineInstrBuilder MIB(MF, I);
349 if (Opc == TargetOpcode::G_LOAD)
350 addOffset(MIB, 0);
351 else {
352 // G_STORE (VAL, Addr), X86Store instruction (Addr, VAL)
353 I.RemoveOperand(0);
354 addOffset(MIB, 0).addUse(DefReg);
355 }
356 return constrainSelectedInstRegOperands(I, TII, TRI, RBI);
357}
358
Igor Breger810c6252017-05-08 09:40:43 +0000359bool X86InstructionSelector::selectFrameIndexOrGep(MachineInstr &I,
360 MachineRegisterInfo &MRI,
361 MachineFunction &MF) const {
362 unsigned Opc = I.getOpcode();
363
364 if (Opc != TargetOpcode::G_FRAME_INDEX && Opc != TargetOpcode::G_GEP)
Igor Breger531a2032017-03-26 08:11:12 +0000365 return false;
366
367 const unsigned DefReg = I.getOperand(0).getReg();
368 LLT Ty = MRI.getType(DefReg);
369
Igor Breger810c6252017-05-08 09:40:43 +0000370 // Use LEA to calculate frame index and GEP
Igor Breger531a2032017-03-26 08:11:12 +0000371 unsigned NewOpc;
372 if (Ty == LLT::pointer(0, 64))
373 NewOpc = X86::LEA64r;
374 else if (Ty == LLT::pointer(0, 32))
375 NewOpc = STI.isTarget64BitILP32() ? X86::LEA64_32r : X86::LEA32r;
376 else
Igor Breger810c6252017-05-08 09:40:43 +0000377 llvm_unreachable("Can't select G_FRAME_INDEX/G_GEP, unsupported type.");
Igor Breger531a2032017-03-26 08:11:12 +0000378
379 I.setDesc(TII.get(NewOpc));
380 MachineInstrBuilder MIB(MF, I);
Igor Breger810c6252017-05-08 09:40:43 +0000381
382 if (Opc == TargetOpcode::G_FRAME_INDEX) {
383 addOffset(MIB, 0);
384 } else {
385 MachineOperand &InxOp = I.getOperand(2);
386 I.addOperand(InxOp); // set IndexReg
387 InxOp.ChangeToImmediate(1); // set Scale
388 MIB.addImm(0).addReg(0);
389 }
Igor Breger531a2032017-03-26 08:11:12 +0000390
391 return constrainSelectedInstRegOperands(I, TII, TRI, RBI);
392}
Daniel Sanders0b5293f2017-04-06 09:49:34 +0000393
Igor Breger3b97ea32017-04-12 12:54:54 +0000394bool X86InstructionSelector::selectConstant(MachineInstr &I,
395 MachineRegisterInfo &MRI,
396 MachineFunction &MF) const {
397 if (I.getOpcode() != TargetOpcode::G_CONSTANT)
398 return false;
399
400 const unsigned DefReg = I.getOperand(0).getReg();
401 LLT Ty = MRI.getType(DefReg);
402
403 assert(Ty.isScalar() && "invalid element type.");
404
405 uint64_t Val = 0;
406 if (I.getOperand(1).isCImm()) {
407 Val = I.getOperand(1).getCImm()->getZExtValue();
408 I.getOperand(1).ChangeToImmediate(Val);
409 } else if (I.getOperand(1).isImm()) {
410 Val = I.getOperand(1).getImm();
411 } else
412 llvm_unreachable("Unsupported operand type.");
413
414 unsigned NewOpc;
415 switch (Ty.getSizeInBits()) {
416 case 8:
417 NewOpc = X86::MOV8ri;
418 break;
419 case 16:
420 NewOpc = X86::MOV16ri;
421 break;
422 case 32:
423 NewOpc = X86::MOV32ri;
424 break;
425 case 64: {
426 // TODO: in case isUInt<32>(Val), X86::MOV32ri can be used
427 if (isInt<32>(Val))
428 NewOpc = X86::MOV64ri32;
429 else
430 NewOpc = X86::MOV64ri;
431 break;
432 }
433 default:
434 llvm_unreachable("Can't select G_CONSTANT, unsupported type.");
435 }
436
437 I.setDesc(TII.get(NewOpc));
438 return constrainSelectedInstRegOperands(I, TII, TRI, RBI);
439}
440
Igor Breger4fdf1e42017-04-19 11:34:59 +0000441bool X86InstructionSelector::selectTrunc(MachineInstr &I,
442 MachineRegisterInfo &MRI,
443 MachineFunction &MF) const {
444 if (I.getOpcode() != TargetOpcode::G_TRUNC)
445 return false;
446
447 const unsigned DstReg = I.getOperand(0).getReg();
448 const unsigned SrcReg = I.getOperand(1).getReg();
449
450 const LLT DstTy = MRI.getType(DstReg);
451 const LLT SrcTy = MRI.getType(SrcReg);
452
453 const RegisterBank &DstRB = *RBI.getRegBank(DstReg, MRI, TRI);
454 const RegisterBank &SrcRB = *RBI.getRegBank(SrcReg, MRI, TRI);
455
456 if (DstRB.getID() != SrcRB.getID()) {
457 DEBUG(dbgs() << "G_TRUNC input/output on different banks\n");
458 return false;
459 }
460
461 if (DstRB.getID() != X86::GPRRegBankID)
462 return false;
463
464 const TargetRegisterClass *DstRC = getRegClassForTypeOnBank(DstTy, DstRB);
465 if (!DstRC)
466 return false;
467
468 const TargetRegisterClass *SrcRC = getRegClassForTypeOnBank(SrcTy, SrcRB);
469 if (!SrcRC)
470 return false;
471
Igor Breger014fc562017-05-21 11:13:56 +0000472 unsigned SubIdx;
473 if (DstRC == SrcRC) {
474 // Nothing to be done
475 SubIdx = X86::NoSubRegister;
476 } else if (DstRC == &X86::GR32RegClass) {
477 SubIdx = X86::sub_32bit;
478 } else if (DstRC == &X86::GR16RegClass) {
479 SubIdx = X86::sub_16bit;
480 } else if (DstRC == &X86::GR8RegClass) {
481 SubIdx = X86::sub_8bit;
482 } else {
483 return false;
484 }
485
486 SrcRC = TRI.getSubClassWithSubReg(SrcRC, SubIdx);
487
Igor Breger4fdf1e42017-04-19 11:34:59 +0000488 if (!RBI.constrainGenericRegister(SrcReg, *SrcRC, MRI) ||
489 !RBI.constrainGenericRegister(DstReg, *DstRC, MRI)) {
490 DEBUG(dbgs() << "Failed to constrain G_TRUNC\n");
491 return false;
492 }
493
Igor Breger014fc562017-05-21 11:13:56 +0000494 I.getOperand(1).setSubReg(SubIdx);
Igor Breger4fdf1e42017-04-19 11:34:59 +0000495
496 I.setDesc(TII.get(X86::COPY));
497 return true;
498}
499
Igor Bregerfda31e62017-05-10 06:52:58 +0000500bool X86InstructionSelector::selectZext(MachineInstr &I,
501 MachineRegisterInfo &MRI,
502 MachineFunction &MF) const {
503 if (I.getOpcode() != TargetOpcode::G_ZEXT)
504 return false;
505
506 const unsigned DstReg = I.getOperand(0).getReg();
507 const unsigned SrcReg = I.getOperand(1).getReg();
508
509 const LLT DstTy = MRI.getType(DstReg);
510 const LLT SrcTy = MRI.getType(SrcReg);
511
512 if (SrcTy == LLT::scalar(1)) {
513
514 unsigned AndOpc;
515 if (DstTy == LLT::scalar(32))
516 AndOpc = X86::AND32ri8;
517 else if (DstTy == LLT::scalar(64))
518 AndOpc = X86::AND64ri8;
519 else
520 return false;
521
522 const RegisterBank &RegBank = *RBI.getRegBank(DstReg, MRI, TRI);
523 unsigned DefReg =
524 MRI.createVirtualRegister(getRegClassForTypeOnBank(DstTy, RegBank));
525
526 BuildMI(*I.getParent(), I, I.getDebugLoc(),
527 TII.get(TargetOpcode::SUBREG_TO_REG), DefReg)
528 .addImm(0)
529 .addReg(SrcReg)
530 .addImm(X86::sub_8bit);
531
532 MachineInstr &AndInst =
533 *BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(AndOpc), DstReg)
534 .addReg(DefReg)
535 .addImm(1);
536
537 constrainSelectedInstRegOperands(AndInst, TII, TRI, RBI);
538
539 I.eraseFromParent();
540 return true;
541 }
542
543 return false;
544}
545
Igor Bregerc7b59772017-05-11 07:17:40 +0000546bool X86InstructionSelector::selectCmp(MachineInstr &I,
547 MachineRegisterInfo &MRI,
548 MachineFunction &MF) const {
549 if (I.getOpcode() != TargetOpcode::G_ICMP)
550 return false;
551
552 X86::CondCode CC;
553 bool SwapArgs;
554 std::tie(CC, SwapArgs) = X86::getX86ConditionCode(
555 (CmpInst::Predicate)I.getOperand(1).getPredicate());
556 unsigned OpSet = X86::getSETFromCond(CC);
557
558 unsigned LHS = I.getOperand(2).getReg();
559 unsigned RHS = I.getOperand(3).getReg();
560
561 if (SwapArgs)
562 std::swap(LHS, RHS);
563
564 unsigned OpCmp;
565 LLT Ty = MRI.getType(LHS);
566
567 switch (Ty.getSizeInBits()) {
568 default:
569 return false;
570 case 8:
571 OpCmp = X86::CMP8rr;
572 break;
573 case 16:
574 OpCmp = X86::CMP16rr;
575 break;
576 case 32:
577 OpCmp = X86::CMP32rr;
578 break;
579 case 64:
580 OpCmp = X86::CMP64rr;
581 break;
582 }
583
584 MachineInstr &CmpInst =
585 *BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(OpCmp))
586 .addReg(LHS)
587 .addReg(RHS);
588
589 MachineInstr &SetInst = *BuildMI(*I.getParent(), I, I.getDebugLoc(),
590 TII.get(OpSet), I.getOperand(0).getReg());
591
592 constrainSelectedInstRegOperands(CmpInst, TII, TRI, RBI);
593 constrainSelectedInstRegOperands(SetInst, TII, TRI, RBI);
594
595 I.eraseFromParent();
596 return true;
597}
598
Igor Breger28f290f2017-05-17 12:48:08 +0000599bool X86InstructionSelector::selectUadde(MachineInstr &I,
600 MachineRegisterInfo &MRI,
601 MachineFunction &MF) const {
602 if (I.getOpcode() != TargetOpcode::G_UADDE)
603 return false;
604
605 const unsigned DstReg = I.getOperand(0).getReg();
606 const unsigned CarryOutReg = I.getOperand(1).getReg();
607 const unsigned Op0Reg = I.getOperand(2).getReg();
608 const unsigned Op1Reg = I.getOperand(3).getReg();
609 unsigned CarryInReg = I.getOperand(4).getReg();
610
611 const LLT DstTy = MRI.getType(DstReg);
612
613 if (DstTy != LLT::scalar(32))
614 return false;
615
616 // find CarryIn def instruction.
617 MachineInstr *Def = MRI.getVRegDef(CarryInReg);
618 while (Def->getOpcode() == TargetOpcode::G_TRUNC) {
619 CarryInReg = Def->getOperand(1).getReg();
620 Def = MRI.getVRegDef(CarryInReg);
621 }
622
623 unsigned Opcode;
624 if (Def->getOpcode() == TargetOpcode::G_UADDE) {
625 // carry set by prev ADD.
626
627 BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(X86::COPY), X86::EFLAGS)
628 .addReg(CarryInReg);
629
630 if (!RBI.constrainGenericRegister(CarryInReg, X86::GR32RegClass, MRI))
631 return false;
632
633 Opcode = X86::ADC32rr;
634 } else if (auto val = getConstantVRegVal(CarryInReg, MRI)) {
635 // carry is constant, support only 0.
636 if (*val != 0)
637 return false;
638
639 Opcode = X86::ADD32rr;
640 } else
641 return false;
642
643 MachineInstr &AddInst =
644 *BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(Opcode), DstReg)
645 .addReg(Op0Reg)
646 .addReg(Op1Reg);
647
648 BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(X86::COPY), CarryOutReg)
649 .addReg(X86::EFLAGS);
650
651 if (!constrainSelectedInstRegOperands(AddInst, TII, TRI, RBI) ||
652 !RBI.constrainGenericRegister(CarryOutReg, X86::GR32RegClass, MRI))
653 return false;
654
655 I.eraseFromParent();
656 return true;
657}
658
Daniel Sanders0b5293f2017-04-06 09:49:34 +0000659InstructionSelector *
Daniel Sanderse7b0d662017-04-21 15:59:56 +0000660llvm::createX86InstructionSelector(const X86TargetMachine &TM,
661 X86Subtarget &Subtarget,
Daniel Sanders0b5293f2017-04-06 09:49:34 +0000662 X86RegisterBankInfo &RBI) {
Daniel Sanderse7b0d662017-04-21 15:59:56 +0000663 return new X86InstructionSelector(TM, Subtarget, RBI);
Daniel Sanders0b5293f2017-04-06 09:49:34 +0000664}