blob: 0395c48272a5b2d43fdf5253e8b5d263f0e43122 [file] [log] [blame]
Tim Northover3b0846e2014-05-24 12:50:23 +00001//===- AArch64InstrInfo.cpp - AArch64 Instruction Information -------------===//
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// This file contains the AArch64 implementation of the TargetInstrInfo class.
11//
12//===----------------------------------------------------------------------===//
13
14#include "AArch64InstrInfo.h"
15#include "AArch64Subtarget.h"
16#include "MCTargetDesc/AArch64AddressingModes.h"
17#include "llvm/CodeGen/MachineFrameInfo.h"
18#include "llvm/CodeGen/MachineInstrBuilder.h"
19#include "llvm/CodeGen/MachineMemOperand.h"
20#include "llvm/CodeGen/MachineRegisterInfo.h"
21#include "llvm/CodeGen/PseudoSourceValue.h"
22#include "llvm/MC/MCInst.h"
23#include "llvm/Support/ErrorHandling.h"
24#include "llvm/Support/TargetRegistry.h"
25
26using namespace llvm;
27
28#define GET_INSTRINFO_CTOR_DTOR
29#include "AArch64GenInstrInfo.inc"
30
31AArch64InstrInfo::AArch64InstrInfo(const AArch64Subtarget &STI)
32 : AArch64GenInstrInfo(AArch64::ADJCALLSTACKDOWN, AArch64::ADJCALLSTACKUP),
Eric Christophera0de2532015-03-18 20:37:30 +000033 RI(STI.getTargetTriple()), Subtarget(STI) {}
Tim Northover3b0846e2014-05-24 12:50:23 +000034
35/// GetInstSize - Return the number of bytes of code the specified
36/// instruction may be. This returns the maximum number of bytes.
37unsigned AArch64InstrInfo::GetInstSizeInBytes(const MachineInstr *MI) const {
Tim Northoverd5531f72014-06-17 11:31:42 +000038 const MachineBasicBlock &MBB = *MI->getParent();
39 const MachineFunction *MF = MBB.getParent();
40 const MCAsmInfo *MAI = MF->getTarget().getMCAsmInfo();
Tim Northover3b0846e2014-05-24 12:50:23 +000041
Tim Northoverd5531f72014-06-17 11:31:42 +000042 if (MI->getOpcode() == AArch64::INLINEASM)
43 return getInlineAsmLength(MI->getOperand(0).getSymbolName(), *MAI);
44
45 const MCInstrDesc &Desc = MI->getDesc();
Tim Northover3b0846e2014-05-24 12:50:23 +000046 switch (Desc.getOpcode()) {
47 default:
48 // Anything not explicitly designated otherwise is a nomal 4-byte insn.
49 return 4;
50 case TargetOpcode::DBG_VALUE:
51 case TargetOpcode::EH_LABEL:
52 case TargetOpcode::IMPLICIT_DEF:
53 case TargetOpcode::KILL:
54 return 0;
55 }
56
57 llvm_unreachable("GetInstSizeInBytes()- Unable to determin insn size");
58}
59
60static void parseCondBranch(MachineInstr *LastInst, MachineBasicBlock *&Target,
61 SmallVectorImpl<MachineOperand> &Cond) {
62 // Block ends with fall-through condbranch.
63 switch (LastInst->getOpcode()) {
64 default:
65 llvm_unreachable("Unknown branch instruction?");
66 case AArch64::Bcc:
67 Target = LastInst->getOperand(1).getMBB();
68 Cond.push_back(LastInst->getOperand(0));
69 break;
70 case AArch64::CBZW:
71 case AArch64::CBZX:
72 case AArch64::CBNZW:
73 case AArch64::CBNZX:
74 Target = LastInst->getOperand(1).getMBB();
75 Cond.push_back(MachineOperand::CreateImm(-1));
76 Cond.push_back(MachineOperand::CreateImm(LastInst->getOpcode()));
77 Cond.push_back(LastInst->getOperand(0));
78 break;
79 case AArch64::TBZW:
80 case AArch64::TBZX:
81 case AArch64::TBNZW:
82 case AArch64::TBNZX:
83 Target = LastInst->getOperand(2).getMBB();
84 Cond.push_back(MachineOperand::CreateImm(-1));
85 Cond.push_back(MachineOperand::CreateImm(LastInst->getOpcode()));
86 Cond.push_back(LastInst->getOperand(0));
87 Cond.push_back(LastInst->getOperand(1));
88 }
89}
90
91// Branch analysis.
92bool AArch64InstrInfo::AnalyzeBranch(MachineBasicBlock &MBB,
93 MachineBasicBlock *&TBB,
94 MachineBasicBlock *&FBB,
95 SmallVectorImpl<MachineOperand> &Cond,
96 bool AllowModify) const {
97 // If the block has no terminators, it just falls into the block after it.
Benjamin Kramere61cbd12015-06-25 13:28:24 +000098 MachineBasicBlock::iterator I = MBB.getLastNonDebugInstr();
99 if (I == MBB.end())
Tim Northover3b0846e2014-05-24 12:50:23 +0000100 return false;
Benjamin Kramere61cbd12015-06-25 13:28:24 +0000101
Tim Northover3b0846e2014-05-24 12:50:23 +0000102 if (!isUnpredicatedTerminator(I))
103 return false;
104
105 // Get the last instruction in the block.
106 MachineInstr *LastInst = I;
107
108 // If there is only one terminator instruction, process it.
109 unsigned LastOpc = LastInst->getOpcode();
110 if (I == MBB.begin() || !isUnpredicatedTerminator(--I)) {
111 if (isUncondBranchOpcode(LastOpc)) {
112 TBB = LastInst->getOperand(0).getMBB();
113 return false;
114 }
115 if (isCondBranchOpcode(LastOpc)) {
116 // Block ends with fall-through condbranch.
117 parseCondBranch(LastInst, TBB, Cond);
118 return false;
119 }
120 return true; // Can't handle indirect branch.
121 }
122
123 // Get the instruction before it if it is a terminator.
124 MachineInstr *SecondLastInst = I;
125 unsigned SecondLastOpc = SecondLastInst->getOpcode();
126
127 // If AllowModify is true and the block ends with two or more unconditional
128 // branches, delete all but the first unconditional branch.
129 if (AllowModify && isUncondBranchOpcode(LastOpc)) {
130 while (isUncondBranchOpcode(SecondLastOpc)) {
131 LastInst->eraseFromParent();
132 LastInst = SecondLastInst;
133 LastOpc = LastInst->getOpcode();
134 if (I == MBB.begin() || !isUnpredicatedTerminator(--I)) {
135 // Return now the only terminator is an unconditional branch.
136 TBB = LastInst->getOperand(0).getMBB();
137 return false;
138 } else {
139 SecondLastInst = I;
140 SecondLastOpc = SecondLastInst->getOpcode();
141 }
142 }
143 }
144
145 // If there are three terminators, we don't know what sort of block this is.
146 if (SecondLastInst && I != MBB.begin() && isUnpredicatedTerminator(--I))
147 return true;
148
149 // If the block ends with a B and a Bcc, handle it.
150 if (isCondBranchOpcode(SecondLastOpc) && isUncondBranchOpcode(LastOpc)) {
151 parseCondBranch(SecondLastInst, TBB, Cond);
152 FBB = LastInst->getOperand(0).getMBB();
153 return false;
154 }
155
156 // If the block ends with two unconditional branches, handle it. The second
157 // one is not executed, so remove it.
158 if (isUncondBranchOpcode(SecondLastOpc) && isUncondBranchOpcode(LastOpc)) {
159 TBB = SecondLastInst->getOperand(0).getMBB();
160 I = LastInst;
161 if (AllowModify)
162 I->eraseFromParent();
163 return false;
164 }
165
166 // ...likewise if it ends with an indirect branch followed by an unconditional
167 // branch.
168 if (isIndirectBranchOpcode(SecondLastOpc) && isUncondBranchOpcode(LastOpc)) {
169 I = LastInst;
170 if (AllowModify)
171 I->eraseFromParent();
172 return true;
173 }
174
175 // Otherwise, can't handle this.
176 return true;
177}
178
179bool AArch64InstrInfo::ReverseBranchCondition(
180 SmallVectorImpl<MachineOperand> &Cond) const {
181 if (Cond[0].getImm() != -1) {
182 // Regular Bcc
183 AArch64CC::CondCode CC = (AArch64CC::CondCode)(int)Cond[0].getImm();
184 Cond[0].setImm(AArch64CC::getInvertedCondCode(CC));
185 } else {
186 // Folded compare-and-branch
187 switch (Cond[1].getImm()) {
188 default:
189 llvm_unreachable("Unknown conditional branch!");
190 case AArch64::CBZW:
191 Cond[1].setImm(AArch64::CBNZW);
192 break;
193 case AArch64::CBNZW:
194 Cond[1].setImm(AArch64::CBZW);
195 break;
196 case AArch64::CBZX:
197 Cond[1].setImm(AArch64::CBNZX);
198 break;
199 case AArch64::CBNZX:
200 Cond[1].setImm(AArch64::CBZX);
201 break;
202 case AArch64::TBZW:
203 Cond[1].setImm(AArch64::TBNZW);
204 break;
205 case AArch64::TBNZW:
206 Cond[1].setImm(AArch64::TBZW);
207 break;
208 case AArch64::TBZX:
209 Cond[1].setImm(AArch64::TBNZX);
210 break;
211 case AArch64::TBNZX:
212 Cond[1].setImm(AArch64::TBZX);
213 break;
214 }
215 }
216
217 return false;
218}
219
220unsigned AArch64InstrInfo::RemoveBranch(MachineBasicBlock &MBB) const {
Benjamin Kramere61cbd12015-06-25 13:28:24 +0000221 MachineBasicBlock::iterator I = MBB.getLastNonDebugInstr();
222 if (I == MBB.end())
Tim Northover3b0846e2014-05-24 12:50:23 +0000223 return 0;
Benjamin Kramere61cbd12015-06-25 13:28:24 +0000224
Tim Northover3b0846e2014-05-24 12:50:23 +0000225 if (!isUncondBranchOpcode(I->getOpcode()) &&
226 !isCondBranchOpcode(I->getOpcode()))
227 return 0;
228
229 // Remove the branch.
230 I->eraseFromParent();
231
232 I = MBB.end();
233
234 if (I == MBB.begin())
235 return 1;
236 --I;
237 if (!isCondBranchOpcode(I->getOpcode()))
238 return 1;
239
240 // Remove the branch.
241 I->eraseFromParent();
242 return 2;
243}
244
245void AArch64InstrInfo::instantiateCondBranch(
246 MachineBasicBlock &MBB, DebugLoc DL, MachineBasicBlock *TBB,
Ahmed Bougachac88bf542015-06-11 19:30:37 +0000247 ArrayRef<MachineOperand> Cond) const {
Tim Northover3b0846e2014-05-24 12:50:23 +0000248 if (Cond[0].getImm() != -1) {
249 // Regular Bcc
250 BuildMI(&MBB, DL, get(AArch64::Bcc)).addImm(Cond[0].getImm()).addMBB(TBB);
251 } else {
252 // Folded compare-and-branch
Ahmed Bougacha72001cf2014-11-07 02:50:00 +0000253 // Note that we use addOperand instead of addReg to keep the flags.
Tim Northover3b0846e2014-05-24 12:50:23 +0000254 const MachineInstrBuilder MIB =
Ahmed Bougacha72001cf2014-11-07 02:50:00 +0000255 BuildMI(&MBB, DL, get(Cond[1].getImm())).addOperand(Cond[2]);
Tim Northover3b0846e2014-05-24 12:50:23 +0000256 if (Cond.size() > 3)
257 MIB.addImm(Cond[3].getImm());
258 MIB.addMBB(TBB);
259 }
260}
261
262unsigned AArch64InstrInfo::InsertBranch(
263 MachineBasicBlock &MBB, MachineBasicBlock *TBB, MachineBasicBlock *FBB,
Ahmed Bougachac88bf542015-06-11 19:30:37 +0000264 ArrayRef<MachineOperand> Cond, DebugLoc DL) const {
Tim Northover3b0846e2014-05-24 12:50:23 +0000265 // Shouldn't be a fall through.
266 assert(TBB && "InsertBranch must not be told to insert a fallthrough");
267
268 if (!FBB) {
269 if (Cond.empty()) // Unconditional branch?
270 BuildMI(&MBB, DL, get(AArch64::B)).addMBB(TBB);
271 else
272 instantiateCondBranch(MBB, DL, TBB, Cond);
273 return 1;
274 }
275
276 // Two-way conditional branch.
277 instantiateCondBranch(MBB, DL, TBB, Cond);
278 BuildMI(&MBB, DL, get(AArch64::B)).addMBB(FBB);
279 return 2;
280}
281
282// Find the original register that VReg is copied from.
283static unsigned removeCopies(const MachineRegisterInfo &MRI, unsigned VReg) {
284 while (TargetRegisterInfo::isVirtualRegister(VReg)) {
285 const MachineInstr *DefMI = MRI.getVRegDef(VReg);
286 if (!DefMI->isFullCopy())
287 return VReg;
288 VReg = DefMI->getOperand(1).getReg();
289 }
290 return VReg;
291}
292
293// Determine if VReg is defined by an instruction that can be folded into a
294// csel instruction. If so, return the folded opcode, and the replacement
295// register.
296static unsigned canFoldIntoCSel(const MachineRegisterInfo &MRI, unsigned VReg,
297 unsigned *NewVReg = nullptr) {
298 VReg = removeCopies(MRI, VReg);
299 if (!TargetRegisterInfo::isVirtualRegister(VReg))
300 return 0;
301
302 bool Is64Bit = AArch64::GPR64allRegClass.hasSubClassEq(MRI.getRegClass(VReg));
303 const MachineInstr *DefMI = MRI.getVRegDef(VReg);
304 unsigned Opc = 0;
305 unsigned SrcOpNum = 0;
306 switch (DefMI->getOpcode()) {
307 case AArch64::ADDSXri:
308 case AArch64::ADDSWri:
309 // if NZCV is used, do not fold.
310 if (DefMI->findRegisterDefOperandIdx(AArch64::NZCV, true) == -1)
311 return 0;
312 // fall-through to ADDXri and ADDWri.
313 case AArch64::ADDXri:
314 case AArch64::ADDWri:
315 // add x, 1 -> csinc.
316 if (!DefMI->getOperand(2).isImm() || DefMI->getOperand(2).getImm() != 1 ||
317 DefMI->getOperand(3).getImm() != 0)
318 return 0;
319 SrcOpNum = 1;
320 Opc = Is64Bit ? AArch64::CSINCXr : AArch64::CSINCWr;
321 break;
322
323 case AArch64::ORNXrr:
324 case AArch64::ORNWrr: {
325 // not x -> csinv, represented as orn dst, xzr, src.
326 unsigned ZReg = removeCopies(MRI, DefMI->getOperand(1).getReg());
327 if (ZReg != AArch64::XZR && ZReg != AArch64::WZR)
328 return 0;
329 SrcOpNum = 2;
330 Opc = Is64Bit ? AArch64::CSINVXr : AArch64::CSINVWr;
331 break;
332 }
333
334 case AArch64::SUBSXrr:
335 case AArch64::SUBSWrr:
336 // if NZCV is used, do not fold.
337 if (DefMI->findRegisterDefOperandIdx(AArch64::NZCV, true) == -1)
338 return 0;
339 // fall-through to SUBXrr and SUBWrr.
340 case AArch64::SUBXrr:
341 case AArch64::SUBWrr: {
342 // neg x -> csneg, represented as sub dst, xzr, src.
343 unsigned ZReg = removeCopies(MRI, DefMI->getOperand(1).getReg());
344 if (ZReg != AArch64::XZR && ZReg != AArch64::WZR)
345 return 0;
346 SrcOpNum = 2;
347 Opc = Is64Bit ? AArch64::CSNEGXr : AArch64::CSNEGWr;
348 break;
349 }
350 default:
351 return 0;
352 }
353 assert(Opc && SrcOpNum && "Missing parameters");
354
355 if (NewVReg)
356 *NewVReg = DefMI->getOperand(SrcOpNum).getReg();
357 return Opc;
358}
359
360bool AArch64InstrInfo::canInsertSelect(
Ahmed Bougachac88bf542015-06-11 19:30:37 +0000361 const MachineBasicBlock &MBB, ArrayRef<MachineOperand> Cond,
Tim Northover3b0846e2014-05-24 12:50:23 +0000362 unsigned TrueReg, unsigned FalseReg, int &CondCycles, int &TrueCycles,
363 int &FalseCycles) const {
364 // Check register classes.
365 const MachineRegisterInfo &MRI = MBB.getParent()->getRegInfo();
366 const TargetRegisterClass *RC =
Eric Christophera0de2532015-03-18 20:37:30 +0000367 RI.getCommonSubClass(MRI.getRegClass(TrueReg), MRI.getRegClass(FalseReg));
Tim Northover3b0846e2014-05-24 12:50:23 +0000368 if (!RC)
369 return false;
370
371 // Expanding cbz/tbz requires an extra cycle of latency on the condition.
372 unsigned ExtraCondLat = Cond.size() != 1;
373
374 // GPRs are handled by csel.
375 // FIXME: Fold in x+1, -x, and ~x when applicable.
376 if (AArch64::GPR64allRegClass.hasSubClassEq(RC) ||
377 AArch64::GPR32allRegClass.hasSubClassEq(RC)) {
378 // Single-cycle csel, csinc, csinv, and csneg.
379 CondCycles = 1 + ExtraCondLat;
380 TrueCycles = FalseCycles = 1;
381 if (canFoldIntoCSel(MRI, TrueReg))
382 TrueCycles = 0;
383 else if (canFoldIntoCSel(MRI, FalseReg))
384 FalseCycles = 0;
385 return true;
386 }
387
388 // Scalar floating point is handled by fcsel.
389 // FIXME: Form fabs, fmin, and fmax when applicable.
390 if (AArch64::FPR64RegClass.hasSubClassEq(RC) ||
391 AArch64::FPR32RegClass.hasSubClassEq(RC)) {
392 CondCycles = 5 + ExtraCondLat;
393 TrueCycles = FalseCycles = 2;
394 return true;
395 }
396
397 // Can't do vectors.
398 return false;
399}
400
401void AArch64InstrInfo::insertSelect(MachineBasicBlock &MBB,
402 MachineBasicBlock::iterator I, DebugLoc DL,
403 unsigned DstReg,
Ahmed Bougachac88bf542015-06-11 19:30:37 +0000404 ArrayRef<MachineOperand> Cond,
Tim Northover3b0846e2014-05-24 12:50:23 +0000405 unsigned TrueReg, unsigned FalseReg) const {
406 MachineRegisterInfo &MRI = MBB.getParent()->getRegInfo();
407
408 // Parse the condition code, see parseCondBranch() above.
409 AArch64CC::CondCode CC;
410 switch (Cond.size()) {
411 default:
412 llvm_unreachable("Unknown condition opcode in Cond");
413 case 1: // b.cc
414 CC = AArch64CC::CondCode(Cond[0].getImm());
415 break;
416 case 3: { // cbz/cbnz
417 // We must insert a compare against 0.
418 bool Is64Bit;
419 switch (Cond[1].getImm()) {
420 default:
421 llvm_unreachable("Unknown branch opcode in Cond");
422 case AArch64::CBZW:
423 Is64Bit = 0;
424 CC = AArch64CC::EQ;
425 break;
426 case AArch64::CBZX:
427 Is64Bit = 1;
428 CC = AArch64CC::EQ;
429 break;
430 case AArch64::CBNZW:
431 Is64Bit = 0;
432 CC = AArch64CC::NE;
433 break;
434 case AArch64::CBNZX:
435 Is64Bit = 1;
436 CC = AArch64CC::NE;
437 break;
438 }
439 unsigned SrcReg = Cond[2].getReg();
440 if (Is64Bit) {
441 // cmp reg, #0 is actually subs xzr, reg, #0.
442 MRI.constrainRegClass(SrcReg, &AArch64::GPR64spRegClass);
443 BuildMI(MBB, I, DL, get(AArch64::SUBSXri), AArch64::XZR)
444 .addReg(SrcReg)
445 .addImm(0)
446 .addImm(0);
447 } else {
448 MRI.constrainRegClass(SrcReg, &AArch64::GPR32spRegClass);
449 BuildMI(MBB, I, DL, get(AArch64::SUBSWri), AArch64::WZR)
450 .addReg(SrcReg)
451 .addImm(0)
452 .addImm(0);
453 }
454 break;
455 }
456 case 4: { // tbz/tbnz
457 // We must insert a tst instruction.
458 switch (Cond[1].getImm()) {
459 default:
460 llvm_unreachable("Unknown branch opcode in Cond");
461 case AArch64::TBZW:
462 case AArch64::TBZX:
463 CC = AArch64CC::EQ;
464 break;
465 case AArch64::TBNZW:
466 case AArch64::TBNZX:
467 CC = AArch64CC::NE;
468 break;
469 }
470 // cmp reg, #foo is actually ands xzr, reg, #1<<foo.
471 if (Cond[1].getImm() == AArch64::TBZW || Cond[1].getImm() == AArch64::TBNZW)
472 BuildMI(MBB, I, DL, get(AArch64::ANDSWri), AArch64::WZR)
473 .addReg(Cond[2].getReg())
474 .addImm(
475 AArch64_AM::encodeLogicalImmediate(1ull << Cond[3].getImm(), 32));
476 else
477 BuildMI(MBB, I, DL, get(AArch64::ANDSXri), AArch64::XZR)
478 .addReg(Cond[2].getReg())
479 .addImm(
480 AArch64_AM::encodeLogicalImmediate(1ull << Cond[3].getImm(), 64));
481 break;
482 }
483 }
484
485 unsigned Opc = 0;
486 const TargetRegisterClass *RC = nullptr;
487 bool TryFold = false;
488 if (MRI.constrainRegClass(DstReg, &AArch64::GPR64RegClass)) {
489 RC = &AArch64::GPR64RegClass;
490 Opc = AArch64::CSELXr;
491 TryFold = true;
492 } else if (MRI.constrainRegClass(DstReg, &AArch64::GPR32RegClass)) {
493 RC = &AArch64::GPR32RegClass;
494 Opc = AArch64::CSELWr;
495 TryFold = true;
496 } else if (MRI.constrainRegClass(DstReg, &AArch64::FPR64RegClass)) {
497 RC = &AArch64::FPR64RegClass;
498 Opc = AArch64::FCSELDrrr;
499 } else if (MRI.constrainRegClass(DstReg, &AArch64::FPR32RegClass)) {
500 RC = &AArch64::FPR32RegClass;
501 Opc = AArch64::FCSELSrrr;
502 }
503 assert(RC && "Unsupported regclass");
504
505 // Try folding simple instructions into the csel.
506 if (TryFold) {
507 unsigned NewVReg = 0;
508 unsigned FoldedOpc = canFoldIntoCSel(MRI, TrueReg, &NewVReg);
509 if (FoldedOpc) {
510 // The folded opcodes csinc, csinc and csneg apply the operation to
511 // FalseReg, so we need to invert the condition.
512 CC = AArch64CC::getInvertedCondCode(CC);
513 TrueReg = FalseReg;
514 } else
515 FoldedOpc = canFoldIntoCSel(MRI, FalseReg, &NewVReg);
516
517 // Fold the operation. Leave any dead instructions for DCE to clean up.
518 if (FoldedOpc) {
519 FalseReg = NewVReg;
520 Opc = FoldedOpc;
521 // The extends the live range of NewVReg.
522 MRI.clearKillFlags(NewVReg);
523 }
524 }
525
526 // Pull all virtual register into the appropriate class.
527 MRI.constrainRegClass(TrueReg, RC);
528 MRI.constrainRegClass(FalseReg, RC);
529
530 // Insert the csel.
531 BuildMI(MBB, I, DL, get(Opc), DstReg).addReg(TrueReg).addReg(FalseReg).addImm(
532 CC);
533}
534
Lawrence Hu687097a2015-07-23 23:55:28 +0000535/// Returns true if a MOVi32imm or MOVi64imm can be expanded to an ORRxx.
Weiming Zhaob33a5552015-07-23 19:24:53 +0000536static bool canBeExpandedToORR(const MachineInstr *MI, unsigned BitSize) {
537 uint64_t Imm = MI->getOperand(1).getImm();
538 uint64_t UImm = Imm << (64 - BitSize) >> (64 - BitSize);
539 uint64_t Encoding;
540 return AArch64_AM::processLogicalImmediate(UImm, BitSize, Encoding);
541}
542
Jiangning Liucd296372014-07-29 02:09:26 +0000543// FIXME: this implementation should be micro-architecture dependent, so a
544// micro-architecture target hook should be introduced here in future.
545bool AArch64InstrInfo::isAsCheapAsAMove(const MachineInstr *MI) const {
Chad Rosiercd2be7f2016-02-12 15:51:51 +0000546 if (!Subtarget.isCortexA57() && !Subtarget.isCortexA53() &&
547 !Subtarget.isKryo())
Jiangning Liucd296372014-07-29 02:09:26 +0000548 return MI->isAsCheapAsAMove();
549
550 switch (MI->getOpcode()) {
551 default:
552 return false;
553
554 // add/sub on register without shift
555 case AArch64::ADDWri:
556 case AArch64::ADDXri:
557 case AArch64::SUBWri:
558 case AArch64::SUBXri:
559 return (MI->getOperand(3).getImm() == 0);
560
561 // logical ops on immediate
562 case AArch64::ANDWri:
563 case AArch64::ANDXri:
564 case AArch64::EORWri:
565 case AArch64::EORXri:
566 case AArch64::ORRWri:
567 case AArch64::ORRXri:
568 return true;
569
570 // logical ops on register without shift
571 case AArch64::ANDWrr:
572 case AArch64::ANDXrr:
573 case AArch64::BICWrr:
574 case AArch64::BICXrr:
575 case AArch64::EONWrr:
576 case AArch64::EONXrr:
577 case AArch64::EORWrr:
578 case AArch64::EORXrr:
579 case AArch64::ORNWrr:
580 case AArch64::ORNXrr:
581 case AArch64::ORRWrr:
582 case AArch64::ORRXrr:
583 return true;
Weiming Zhaob33a5552015-07-23 19:24:53 +0000584 // If MOVi32imm or MOVi64imm can be expanded into ORRWri or
585 // ORRXri, it is as cheap as MOV
586 case AArch64::MOVi32imm:
587 return canBeExpandedToORR(MI, 32);
588 case AArch64::MOVi64imm:
589 return canBeExpandedToORR(MI, 64);
Jiangning Liucd296372014-07-29 02:09:26 +0000590 }
591
592 llvm_unreachable("Unknown opcode to check as cheap as a move!");
593}
594
Tim Northover3b0846e2014-05-24 12:50:23 +0000595bool AArch64InstrInfo::isCoalescableExtInstr(const MachineInstr &MI,
596 unsigned &SrcReg, unsigned &DstReg,
597 unsigned &SubIdx) const {
598 switch (MI.getOpcode()) {
599 default:
600 return false;
601 case AArch64::SBFMXri: // aka sxtw
602 case AArch64::UBFMXri: // aka uxtw
603 // Check for the 32 -> 64 bit extension case, these instructions can do
604 // much more.
605 if (MI.getOperand(2).getImm() != 0 || MI.getOperand(3).getImm() != 31)
606 return false;
607 // This is a signed or unsigned 32 -> 64 bit extension.
608 SrcReg = MI.getOperand(1).getReg();
609 DstReg = MI.getOperand(0).getReg();
610 SubIdx = AArch64::sub_32;
611 return true;
612 }
613}
614
Chad Rosier3528c1e2014-09-08 14:43:48 +0000615bool
616AArch64InstrInfo::areMemAccessesTriviallyDisjoint(MachineInstr *MIa,
617 MachineInstr *MIb,
618 AliasAnalysis *AA) const {
Eric Christophera0de2532015-03-18 20:37:30 +0000619 const TargetRegisterInfo *TRI = &getRegisterInfo();
Chad Rosier3528c1e2014-09-08 14:43:48 +0000620 unsigned BaseRegA = 0, BaseRegB = 0;
621 int OffsetA = 0, OffsetB = 0;
622 int WidthA = 0, WidthB = 0;
623
Chad Rosiera73b3592015-05-21 21:59:57 +0000624 assert(MIa && MIa->mayLoadOrStore() && "MIa must be a load or store.");
625 assert(MIb && MIb->mayLoadOrStore() && "MIb must be a load or store.");
Chad Rosier3528c1e2014-09-08 14:43:48 +0000626
627 if (MIa->hasUnmodeledSideEffects() || MIb->hasUnmodeledSideEffects() ||
628 MIa->hasOrderedMemoryRef() || MIb->hasOrderedMemoryRef())
629 return false;
630
631 // Retrieve the base register, offset from the base register and width. Width
632 // is the size of memory that is being loaded/stored (e.g. 1, 2, 4, 8). If
633 // base registers are identical, and the offset of a lower memory access +
634 // the width doesn't overlap the offset of a higher memory access,
635 // then the memory accesses are different.
Sanjoy Dasb666ea32015-06-15 18:44:14 +0000636 if (getMemOpBaseRegImmOfsWidth(MIa, BaseRegA, OffsetA, WidthA, TRI) &&
637 getMemOpBaseRegImmOfsWidth(MIb, BaseRegB, OffsetB, WidthB, TRI)) {
Chad Rosier3528c1e2014-09-08 14:43:48 +0000638 if (BaseRegA == BaseRegB) {
639 int LowOffset = OffsetA < OffsetB ? OffsetA : OffsetB;
640 int HighOffset = OffsetA < OffsetB ? OffsetB : OffsetA;
641 int LowWidth = (LowOffset == OffsetA) ? WidthA : WidthB;
642 if (LowOffset + LowWidth <= HighOffset)
643 return true;
644 }
645 }
646 return false;
647}
648
Tim Northover3b0846e2014-05-24 12:50:23 +0000649/// analyzeCompare - For a comparison instruction, return the source registers
650/// in SrcReg and SrcReg2, and the value it compares against in CmpValue.
651/// Return true if the comparison instruction can be analyzed.
652bool AArch64InstrInfo::analyzeCompare(const MachineInstr *MI, unsigned &SrcReg,
653 unsigned &SrcReg2, int &CmpMask,
654 int &CmpValue) const {
655 switch (MI->getOpcode()) {
656 default:
657 break;
658 case AArch64::SUBSWrr:
659 case AArch64::SUBSWrs:
660 case AArch64::SUBSWrx:
661 case AArch64::SUBSXrr:
662 case AArch64::SUBSXrs:
663 case AArch64::SUBSXrx:
664 case AArch64::ADDSWrr:
665 case AArch64::ADDSWrs:
666 case AArch64::ADDSWrx:
667 case AArch64::ADDSXrr:
668 case AArch64::ADDSXrs:
669 case AArch64::ADDSXrx:
670 // Replace SUBSWrr with SUBWrr if NZCV is not used.
671 SrcReg = MI->getOperand(1).getReg();
672 SrcReg2 = MI->getOperand(2).getReg();
673 CmpMask = ~0;
674 CmpValue = 0;
675 return true;
676 case AArch64::SUBSWri:
677 case AArch64::ADDSWri:
678 case AArch64::SUBSXri:
679 case AArch64::ADDSXri:
680 SrcReg = MI->getOperand(1).getReg();
681 SrcReg2 = 0;
682 CmpMask = ~0;
Jiangning Liudcc651f2014-08-08 14:19:29 +0000683 // FIXME: In order to convert CmpValue to 0 or 1
684 CmpValue = (MI->getOperand(2).getImm() != 0);
Tim Northover3b0846e2014-05-24 12:50:23 +0000685 return true;
686 case AArch64::ANDSWri:
687 case AArch64::ANDSXri:
688 // ANDS does not use the same encoding scheme as the others xxxS
689 // instructions.
690 SrcReg = MI->getOperand(1).getReg();
691 SrcReg2 = 0;
692 CmpMask = ~0;
Jiangning Liudcc651f2014-08-08 14:19:29 +0000693 // FIXME:The return val type of decodeLogicalImmediate is uint64_t,
694 // while the type of CmpValue is int. When converting uint64_t to int,
695 // the high 32 bits of uint64_t will be lost.
696 // In fact it causes a bug in spec2006-483.xalancbmk
697 // CmpValue is only used to compare with zero in OptimizeCompareInstr
698 CmpValue = (AArch64_AM::decodeLogicalImmediate(
699 MI->getOperand(2).getImm(),
700 MI->getOpcode() == AArch64::ANDSWri ? 32 : 64) != 0);
Tim Northover3b0846e2014-05-24 12:50:23 +0000701 return true;
702 }
703
704 return false;
705}
706
707static bool UpdateOperandRegClass(MachineInstr *Instr) {
708 MachineBasicBlock *MBB = Instr->getParent();
709 assert(MBB && "Can't get MachineBasicBlock here");
710 MachineFunction *MF = MBB->getParent();
711 assert(MF && "Can't get MachineFunction here");
Eric Christopher6c901622015-01-28 03:51:33 +0000712 const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
713 const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
Tim Northover3b0846e2014-05-24 12:50:23 +0000714 MachineRegisterInfo *MRI = &MF->getRegInfo();
715
716 for (unsigned OpIdx = 0, EndIdx = Instr->getNumOperands(); OpIdx < EndIdx;
717 ++OpIdx) {
718 MachineOperand &MO = Instr->getOperand(OpIdx);
719 const TargetRegisterClass *OpRegCstraints =
720 Instr->getRegClassConstraint(OpIdx, TII, TRI);
721
722 // If there's no constraint, there's nothing to do.
723 if (!OpRegCstraints)
724 continue;
725 // If the operand is a frame index, there's nothing to do here.
726 // A frame index operand will resolve correctly during PEI.
727 if (MO.isFI())
728 continue;
729
730 assert(MO.isReg() &&
731 "Operand has register constraints without being a register!");
732
733 unsigned Reg = MO.getReg();
734 if (TargetRegisterInfo::isPhysicalRegister(Reg)) {
735 if (!OpRegCstraints->contains(Reg))
736 return false;
737 } else if (!OpRegCstraints->hasSubClassEq(MRI->getRegClass(Reg)) &&
738 !MRI->constrainRegClass(Reg, OpRegCstraints))
739 return false;
740 }
741
742 return true;
743}
744
Juergen Ributzka7a7c4682014-11-18 21:02:40 +0000745/// \brief Return the opcode that does not set flags when possible - otherwise
746/// return the original opcode. The caller is responsible to do the actual
747/// substitution and legality checking.
748static unsigned convertFlagSettingOpcode(const MachineInstr *MI) {
749 // Don't convert all compare instructions, because for some the zero register
750 // encoding becomes the sp register.
751 bool MIDefinesZeroReg = false;
752 if (MI->definesRegister(AArch64::WZR) || MI->definesRegister(AArch64::XZR))
753 MIDefinesZeroReg = true;
754
755 switch (MI->getOpcode()) {
756 default:
757 return MI->getOpcode();
758 case AArch64::ADDSWrr:
759 return AArch64::ADDWrr;
760 case AArch64::ADDSWri:
761 return MIDefinesZeroReg ? AArch64::ADDSWri : AArch64::ADDWri;
762 case AArch64::ADDSWrs:
763 return MIDefinesZeroReg ? AArch64::ADDSWrs : AArch64::ADDWrs;
764 case AArch64::ADDSWrx:
765 return AArch64::ADDWrx;
766 case AArch64::ADDSXrr:
767 return AArch64::ADDXrr;
768 case AArch64::ADDSXri:
769 return MIDefinesZeroReg ? AArch64::ADDSXri : AArch64::ADDXri;
770 case AArch64::ADDSXrs:
771 return MIDefinesZeroReg ? AArch64::ADDSXrs : AArch64::ADDXrs;
772 case AArch64::ADDSXrx:
773 return AArch64::ADDXrx;
774 case AArch64::SUBSWrr:
775 return AArch64::SUBWrr;
776 case AArch64::SUBSWri:
777 return MIDefinesZeroReg ? AArch64::SUBSWri : AArch64::SUBWri;
778 case AArch64::SUBSWrs:
779 return MIDefinesZeroReg ? AArch64::SUBSWrs : AArch64::SUBWrs;
780 case AArch64::SUBSWrx:
781 return AArch64::SUBWrx;
782 case AArch64::SUBSXrr:
783 return AArch64::SUBXrr;
784 case AArch64::SUBSXri:
785 return MIDefinesZeroReg ? AArch64::SUBSXri : AArch64::SUBXri;
786 case AArch64::SUBSXrs:
787 return MIDefinesZeroReg ? AArch64::SUBSXrs : AArch64::SUBXrs;
788 case AArch64::SUBSXrx:
789 return AArch64::SUBXrx;
790 }
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +0000791}
Tim Northover3b0846e2014-05-24 12:50:23 +0000792
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +0000793/// True when condition code could be modified on the instruction
794/// trace starting at from and ending at to.
795static bool modifiesConditionCode(MachineInstr *From, MachineInstr *To,
796 const bool CheckOnlyCCWrites,
797 const TargetRegisterInfo *TRI) {
798 // We iterate backward starting \p To until we hit \p From
799 MachineBasicBlock::iterator I = To, E = From, B = To->getParent()->begin();
800
801 // Early exit if To is at the beginning of the BB.
802 if (I == B)
803 return true;
804
805 // Check whether the definition of SrcReg is in the same basic block as
806 // Compare. If not, assume the condition code gets modified on some path.
807 if (To->getParent() != From->getParent())
808 return true;
809
810 // Check that NZCV isn't set on the trace.
811 for (--I; I != E; --I) {
812 const MachineInstr &Instr = *I;
813
814 if (Instr.modifiesRegister(AArch64::NZCV, TRI) ||
815 (!CheckOnlyCCWrites && Instr.readsRegister(AArch64::NZCV, TRI)))
816 // This instruction modifies or uses NZCV after the one we want to
817 // change.
818 return true;
819 if (I == B)
820 // We currently don't allow the instruction trace to cross basic
821 // block boundaries
822 return true;
823 }
824 return false;
825}
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +0000826/// optimizeCompareInstr - Convert the instruction supplying the argument to the
827/// comparison into one that sets the zero bit in the flags register.
828bool AArch64InstrInfo::optimizeCompareInstr(
829 MachineInstr *CmpInstr, unsigned SrcReg, unsigned SrcReg2, int CmpMask,
830 int CmpValue, const MachineRegisterInfo *MRI) const {
831
832 // Replace SUBSWrr with SUBWrr if NZCV is not used.
833 int Cmp_NZCV = CmpInstr->findRegisterDefOperandIdx(AArch64::NZCV, true);
834 if (Cmp_NZCV != -1) {
Juergen Ributzka7a7c4682014-11-18 21:02:40 +0000835 if (CmpInstr->definesRegister(AArch64::WZR) ||
836 CmpInstr->definesRegister(AArch64::XZR)) {
837 CmpInstr->eraseFromParent();
838 return true;
839 }
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +0000840 unsigned Opc = CmpInstr->getOpcode();
841 unsigned NewOpc = convertFlagSettingOpcode(CmpInstr);
842 if (NewOpc == Opc)
843 return false;
Tim Northover3b0846e2014-05-24 12:50:23 +0000844 const MCInstrDesc &MCID = get(NewOpc);
845 CmpInstr->setDesc(MCID);
846 CmpInstr->RemoveOperand(Cmp_NZCV);
847 bool succeeded = UpdateOperandRegClass(CmpInstr);
848 (void)succeeded;
849 assert(succeeded && "Some operands reg class are incompatible!");
850 return true;
851 }
852
853 // Continue only if we have a "ri" where immediate is zero.
Jiangning Liudcc651f2014-08-08 14:19:29 +0000854 // FIXME:CmpValue has already been converted to 0 or 1 in analyzeCompare
855 // function.
856 assert((CmpValue == 0 || CmpValue == 1) && "CmpValue must be 0 or 1!");
Tim Northover3b0846e2014-05-24 12:50:23 +0000857 if (CmpValue != 0 || SrcReg2 != 0)
858 return false;
859
860 // CmpInstr is a Compare instruction if destination register is not used.
861 if (!MRI->use_nodbg_empty(CmpInstr->getOperand(0).getReg()))
862 return false;
863
864 // Get the unique definition of SrcReg.
865 MachineInstr *MI = MRI->getUniqueVRegDef(SrcReg);
866 if (!MI)
867 return false;
868
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +0000869 bool CheckOnlyCCWrites = false;
Eric Christophera0de2532015-03-18 20:37:30 +0000870 const TargetRegisterInfo *TRI = &getRegisterInfo();
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +0000871 if (modifiesConditionCode(MI, CmpInstr, CheckOnlyCCWrites, TRI))
872 return false;
Tim Northover3b0846e2014-05-24 12:50:23 +0000873
874 unsigned NewOpc = MI->getOpcode();
875 switch (MI->getOpcode()) {
876 default:
877 return false;
878 case AArch64::ADDSWrr:
879 case AArch64::ADDSWri:
880 case AArch64::ADDSXrr:
881 case AArch64::ADDSXri:
882 case AArch64::SUBSWrr:
883 case AArch64::SUBSWri:
884 case AArch64::SUBSXrr:
885 case AArch64::SUBSXri:
886 break;
887 case AArch64::ADDWrr: NewOpc = AArch64::ADDSWrr; break;
888 case AArch64::ADDWri: NewOpc = AArch64::ADDSWri; break;
889 case AArch64::ADDXrr: NewOpc = AArch64::ADDSXrr; break;
890 case AArch64::ADDXri: NewOpc = AArch64::ADDSXri; break;
891 case AArch64::ADCWr: NewOpc = AArch64::ADCSWr; break;
892 case AArch64::ADCXr: NewOpc = AArch64::ADCSXr; break;
893 case AArch64::SUBWrr: NewOpc = AArch64::SUBSWrr; break;
894 case AArch64::SUBWri: NewOpc = AArch64::SUBSWri; break;
895 case AArch64::SUBXrr: NewOpc = AArch64::SUBSXrr; break;
896 case AArch64::SUBXri: NewOpc = AArch64::SUBSXri; break;
897 case AArch64::SBCWr: NewOpc = AArch64::SBCSWr; break;
898 case AArch64::SBCXr: NewOpc = AArch64::SBCSXr; break;
899 case AArch64::ANDWri: NewOpc = AArch64::ANDSWri; break;
900 case AArch64::ANDXri: NewOpc = AArch64::ANDSXri; break;
901 }
902
903 // Scan forward for the use of NZCV.
904 // When checking against MI: if it's a conditional code requires
905 // checking of V bit, then this is not safe to do.
906 // It is safe to remove CmpInstr if NZCV is redefined or killed.
907 // If we are done with the basic block, we need to check whether NZCV is
908 // live-out.
909 bool IsSafe = false;
910 for (MachineBasicBlock::iterator I = CmpInstr,
911 E = CmpInstr->getParent()->end();
912 !IsSafe && ++I != E;) {
913 const MachineInstr &Instr = *I;
914 for (unsigned IO = 0, EO = Instr.getNumOperands(); !IsSafe && IO != EO;
915 ++IO) {
916 const MachineOperand &MO = Instr.getOperand(IO);
917 if (MO.isRegMask() && MO.clobbersPhysReg(AArch64::NZCV)) {
918 IsSafe = true;
919 break;
920 }
921 if (!MO.isReg() || MO.getReg() != AArch64::NZCV)
922 continue;
923 if (MO.isDef()) {
924 IsSafe = true;
925 break;
926 }
927
928 // Decode the condition code.
929 unsigned Opc = Instr.getOpcode();
930 AArch64CC::CondCode CC;
931 switch (Opc) {
932 default:
933 return false;
934 case AArch64::Bcc:
935 CC = (AArch64CC::CondCode)Instr.getOperand(IO - 2).getImm();
936 break;
937 case AArch64::CSINVWr:
938 case AArch64::CSINVXr:
939 case AArch64::CSINCWr:
940 case AArch64::CSINCXr:
941 case AArch64::CSELWr:
942 case AArch64::CSELXr:
943 case AArch64::CSNEGWr:
944 case AArch64::CSNEGXr:
945 case AArch64::FCSELSrrr:
946 case AArch64::FCSELDrrr:
947 CC = (AArch64CC::CondCode)Instr.getOperand(IO - 1).getImm();
948 break;
949 }
950
951 // It is not safe to remove Compare instruction if Overflow(V) is used.
952 switch (CC) {
953 default:
954 // NZCV can be used multiple times, we should continue.
955 break;
956 case AArch64CC::VS:
957 case AArch64CC::VC:
958 case AArch64CC::GE:
959 case AArch64CC::LT:
960 case AArch64CC::GT:
961 case AArch64CC::LE:
962 return false;
963 }
964 }
965 }
966
967 // If NZCV is not killed nor re-defined, we should check whether it is
968 // live-out. If it is live-out, do not optimize.
969 if (!IsSafe) {
970 MachineBasicBlock *ParentBlock = CmpInstr->getParent();
971 for (auto *MBB : ParentBlock->successors())
972 if (MBB->isLiveIn(AArch64::NZCV))
973 return false;
974 }
975
976 // Update the instruction to set NZCV.
977 MI->setDesc(get(NewOpc));
978 CmpInstr->eraseFromParent();
979 bool succeeded = UpdateOperandRegClass(MI);
980 (void)succeeded;
981 assert(succeeded && "Some operands reg class are incompatible!");
982 MI->addRegisterDefined(AArch64::NZCV, TRI);
983 return true;
984}
985
Akira Hatanakae5b6e0d2014-07-25 19:31:34 +0000986bool
987AArch64InstrInfo::expandPostRAPseudo(MachineBasicBlock::iterator MI) const {
988 if (MI->getOpcode() != TargetOpcode::LOAD_STACK_GUARD)
989 return false;
990
991 MachineBasicBlock &MBB = *MI->getParent();
992 DebugLoc DL = MI->getDebugLoc();
993 unsigned Reg = MI->getOperand(0).getReg();
994 const GlobalValue *GV =
995 cast<GlobalValue>((*MI->memoperands_begin())->getValue());
996 const TargetMachine &TM = MBB.getParent()->getTarget();
997 unsigned char OpFlags = Subtarget.ClassifyGlobalReference(GV, TM);
998 const unsigned char MO_NC = AArch64II::MO_NC;
999
1000 if ((OpFlags & AArch64II::MO_GOT) != 0) {
1001 BuildMI(MBB, MI, DL, get(AArch64::LOADgot), Reg)
1002 .addGlobalAddress(GV, 0, AArch64II::MO_GOT);
1003 BuildMI(MBB, MI, DL, get(AArch64::LDRXui), Reg)
1004 .addReg(Reg, RegState::Kill).addImm(0)
1005 .addMemOperand(*MI->memoperands_begin());
1006 } else if (TM.getCodeModel() == CodeModel::Large) {
1007 BuildMI(MBB, MI, DL, get(AArch64::MOVZXi), Reg)
1008 .addGlobalAddress(GV, 0, AArch64II::MO_G3).addImm(48);
1009 BuildMI(MBB, MI, DL, get(AArch64::MOVKXi), Reg)
1010 .addReg(Reg, RegState::Kill)
1011 .addGlobalAddress(GV, 0, AArch64II::MO_G2 | MO_NC).addImm(32);
1012 BuildMI(MBB, MI, DL, get(AArch64::MOVKXi), Reg)
1013 .addReg(Reg, RegState::Kill)
1014 .addGlobalAddress(GV, 0, AArch64II::MO_G1 | MO_NC).addImm(16);
1015 BuildMI(MBB, MI, DL, get(AArch64::MOVKXi), Reg)
1016 .addReg(Reg, RegState::Kill)
1017 .addGlobalAddress(GV, 0, AArch64II::MO_G0 | MO_NC).addImm(0);
1018 BuildMI(MBB, MI, DL, get(AArch64::LDRXui), Reg)
1019 .addReg(Reg, RegState::Kill).addImm(0)
1020 .addMemOperand(*MI->memoperands_begin());
1021 } else {
1022 BuildMI(MBB, MI, DL, get(AArch64::ADRP), Reg)
1023 .addGlobalAddress(GV, 0, OpFlags | AArch64II::MO_PAGE);
1024 unsigned char LoFlags = OpFlags | AArch64II::MO_PAGEOFF | MO_NC;
1025 BuildMI(MBB, MI, DL, get(AArch64::LDRXui), Reg)
1026 .addReg(Reg, RegState::Kill)
1027 .addGlobalAddress(GV, 0, LoFlags)
1028 .addMemOperand(*MI->memoperands_begin());
1029 }
1030
1031 MBB.erase(MI);
1032
1033 return true;
1034}
1035
Tim Northover3b0846e2014-05-24 12:50:23 +00001036/// Return true if this is this instruction has a non-zero immediate
1037bool AArch64InstrInfo::hasShiftedReg(const MachineInstr *MI) const {
1038 switch (MI->getOpcode()) {
1039 default:
1040 break;
1041 case AArch64::ADDSWrs:
1042 case AArch64::ADDSXrs:
1043 case AArch64::ADDWrs:
1044 case AArch64::ADDXrs:
1045 case AArch64::ANDSWrs:
1046 case AArch64::ANDSXrs:
1047 case AArch64::ANDWrs:
1048 case AArch64::ANDXrs:
1049 case AArch64::BICSWrs:
1050 case AArch64::BICSXrs:
1051 case AArch64::BICWrs:
1052 case AArch64::BICXrs:
1053 case AArch64::CRC32Brr:
1054 case AArch64::CRC32CBrr:
1055 case AArch64::CRC32CHrr:
1056 case AArch64::CRC32CWrr:
1057 case AArch64::CRC32CXrr:
1058 case AArch64::CRC32Hrr:
1059 case AArch64::CRC32Wrr:
1060 case AArch64::CRC32Xrr:
1061 case AArch64::EONWrs:
1062 case AArch64::EONXrs:
1063 case AArch64::EORWrs:
1064 case AArch64::EORXrs:
1065 case AArch64::ORNWrs:
1066 case AArch64::ORNXrs:
1067 case AArch64::ORRWrs:
1068 case AArch64::ORRXrs:
1069 case AArch64::SUBSWrs:
1070 case AArch64::SUBSXrs:
1071 case AArch64::SUBWrs:
1072 case AArch64::SUBXrs:
1073 if (MI->getOperand(3).isImm()) {
1074 unsigned val = MI->getOperand(3).getImm();
1075 return (val != 0);
1076 }
1077 break;
1078 }
1079 return false;
1080}
1081
1082/// Return true if this is this instruction has a non-zero immediate
1083bool AArch64InstrInfo::hasExtendedReg(const MachineInstr *MI) const {
1084 switch (MI->getOpcode()) {
1085 default:
1086 break;
1087 case AArch64::ADDSWrx:
1088 case AArch64::ADDSXrx:
1089 case AArch64::ADDSXrx64:
1090 case AArch64::ADDWrx:
1091 case AArch64::ADDXrx:
1092 case AArch64::ADDXrx64:
1093 case AArch64::SUBSWrx:
1094 case AArch64::SUBSXrx:
1095 case AArch64::SUBSXrx64:
1096 case AArch64::SUBWrx:
1097 case AArch64::SUBXrx:
1098 case AArch64::SUBXrx64:
1099 if (MI->getOperand(3).isImm()) {
1100 unsigned val = MI->getOperand(3).getImm();
1101 return (val != 0);
1102 }
1103 break;
1104 }
1105
1106 return false;
1107}
1108
1109// Return true if this instruction simply sets its single destination register
1110// to zero. This is equivalent to a register rename of the zero-register.
1111bool AArch64InstrInfo::isGPRZero(const MachineInstr *MI) const {
1112 switch (MI->getOpcode()) {
1113 default:
1114 break;
1115 case AArch64::MOVZWi:
1116 case AArch64::MOVZXi: // movz Rd, #0 (LSL #0)
1117 if (MI->getOperand(1).isImm() && MI->getOperand(1).getImm() == 0) {
1118 assert(MI->getDesc().getNumOperands() == 3 &&
1119 MI->getOperand(2).getImm() == 0 && "invalid MOVZi operands");
1120 return true;
1121 }
1122 break;
1123 case AArch64::ANDWri: // and Rd, Rzr, #imm
1124 return MI->getOperand(1).getReg() == AArch64::WZR;
1125 case AArch64::ANDXri:
1126 return MI->getOperand(1).getReg() == AArch64::XZR;
1127 case TargetOpcode::COPY:
1128 return MI->getOperand(1).getReg() == AArch64::WZR;
1129 }
1130 return false;
1131}
1132
1133// Return true if this instruction simply renames a general register without
1134// modifying bits.
1135bool AArch64InstrInfo::isGPRCopy(const MachineInstr *MI) const {
1136 switch (MI->getOpcode()) {
1137 default:
1138 break;
1139 case TargetOpcode::COPY: {
1140 // GPR32 copies will by lowered to ORRXrs
1141 unsigned DstReg = MI->getOperand(0).getReg();
1142 return (AArch64::GPR32RegClass.contains(DstReg) ||
1143 AArch64::GPR64RegClass.contains(DstReg));
1144 }
1145 case AArch64::ORRXrs: // orr Xd, Xzr, Xm (LSL #0)
1146 if (MI->getOperand(1).getReg() == AArch64::XZR) {
1147 assert(MI->getDesc().getNumOperands() == 4 &&
1148 MI->getOperand(3).getImm() == 0 && "invalid ORRrs operands");
1149 return true;
1150 }
Renato Golin541d7e72014-08-01 17:27:31 +00001151 break;
Tim Northover3b0846e2014-05-24 12:50:23 +00001152 case AArch64::ADDXri: // add Xd, Xn, #0 (LSL #0)
1153 if (MI->getOperand(2).getImm() == 0) {
1154 assert(MI->getDesc().getNumOperands() == 4 &&
1155 MI->getOperand(3).getImm() == 0 && "invalid ADDXri operands");
1156 return true;
1157 }
Renato Golin541d7e72014-08-01 17:27:31 +00001158 break;
Tim Northover3b0846e2014-05-24 12:50:23 +00001159 }
1160 return false;
1161}
1162
1163// Return true if this instruction simply renames a general register without
1164// modifying bits.
1165bool AArch64InstrInfo::isFPRCopy(const MachineInstr *MI) const {
1166 switch (MI->getOpcode()) {
1167 default:
1168 break;
1169 case TargetOpcode::COPY: {
1170 // FPR64 copies will by lowered to ORR.16b
1171 unsigned DstReg = MI->getOperand(0).getReg();
1172 return (AArch64::FPR64RegClass.contains(DstReg) ||
1173 AArch64::FPR128RegClass.contains(DstReg));
1174 }
1175 case AArch64::ORRv16i8:
1176 if (MI->getOperand(1).getReg() == MI->getOperand(2).getReg()) {
1177 assert(MI->getDesc().getNumOperands() == 3 && MI->getOperand(0).isReg() &&
1178 "invalid ORRv16i8 operands");
1179 return true;
1180 }
Renato Golin541d7e72014-08-01 17:27:31 +00001181 break;
Tim Northover3b0846e2014-05-24 12:50:23 +00001182 }
1183 return false;
1184}
1185
1186unsigned AArch64InstrInfo::isLoadFromStackSlot(const MachineInstr *MI,
1187 int &FrameIndex) const {
1188 switch (MI->getOpcode()) {
1189 default:
1190 break;
1191 case AArch64::LDRWui:
1192 case AArch64::LDRXui:
1193 case AArch64::LDRBui:
1194 case AArch64::LDRHui:
1195 case AArch64::LDRSui:
1196 case AArch64::LDRDui:
1197 case AArch64::LDRQui:
1198 if (MI->getOperand(0).getSubReg() == 0 && MI->getOperand(1).isFI() &&
1199 MI->getOperand(2).isImm() && MI->getOperand(2).getImm() == 0) {
1200 FrameIndex = MI->getOperand(1).getIndex();
1201 return MI->getOperand(0).getReg();
1202 }
1203 break;
1204 }
1205
1206 return 0;
1207}
1208
1209unsigned AArch64InstrInfo::isStoreToStackSlot(const MachineInstr *MI,
1210 int &FrameIndex) const {
1211 switch (MI->getOpcode()) {
1212 default:
1213 break;
1214 case AArch64::STRWui:
1215 case AArch64::STRXui:
1216 case AArch64::STRBui:
1217 case AArch64::STRHui:
1218 case AArch64::STRSui:
1219 case AArch64::STRDui:
1220 case AArch64::STRQui:
1221 if (MI->getOperand(0).getSubReg() == 0 && MI->getOperand(1).isFI() &&
1222 MI->getOperand(2).isImm() && MI->getOperand(2).getImm() == 0) {
1223 FrameIndex = MI->getOperand(1).getIndex();
1224 return MI->getOperand(0).getReg();
1225 }
1226 break;
1227 }
1228 return 0;
1229}
1230
1231/// Return true if this is load/store scales or extends its register offset.
1232/// This refers to scaling a dynamic index as opposed to scaled immediates.
1233/// MI should be a memory op that allows scaled addressing.
1234bool AArch64InstrInfo::isScaledAddr(const MachineInstr *MI) const {
1235 switch (MI->getOpcode()) {
1236 default:
1237 break;
1238 case AArch64::LDRBBroW:
1239 case AArch64::LDRBroW:
1240 case AArch64::LDRDroW:
1241 case AArch64::LDRHHroW:
1242 case AArch64::LDRHroW:
1243 case AArch64::LDRQroW:
1244 case AArch64::LDRSBWroW:
1245 case AArch64::LDRSBXroW:
1246 case AArch64::LDRSHWroW:
1247 case AArch64::LDRSHXroW:
1248 case AArch64::LDRSWroW:
1249 case AArch64::LDRSroW:
1250 case AArch64::LDRWroW:
1251 case AArch64::LDRXroW:
1252 case AArch64::STRBBroW:
1253 case AArch64::STRBroW:
1254 case AArch64::STRDroW:
1255 case AArch64::STRHHroW:
1256 case AArch64::STRHroW:
1257 case AArch64::STRQroW:
1258 case AArch64::STRSroW:
1259 case AArch64::STRWroW:
1260 case AArch64::STRXroW:
1261 case AArch64::LDRBBroX:
1262 case AArch64::LDRBroX:
1263 case AArch64::LDRDroX:
1264 case AArch64::LDRHHroX:
1265 case AArch64::LDRHroX:
1266 case AArch64::LDRQroX:
1267 case AArch64::LDRSBWroX:
1268 case AArch64::LDRSBXroX:
1269 case AArch64::LDRSHWroX:
1270 case AArch64::LDRSHXroX:
1271 case AArch64::LDRSWroX:
1272 case AArch64::LDRSroX:
1273 case AArch64::LDRWroX:
1274 case AArch64::LDRXroX:
1275 case AArch64::STRBBroX:
1276 case AArch64::STRBroX:
1277 case AArch64::STRDroX:
1278 case AArch64::STRHHroX:
1279 case AArch64::STRHroX:
1280 case AArch64::STRQroX:
1281 case AArch64::STRSroX:
1282 case AArch64::STRWroX:
1283 case AArch64::STRXroX:
1284
1285 unsigned Val = MI->getOperand(3).getImm();
1286 AArch64_AM::ShiftExtendType ExtType = AArch64_AM::getMemExtendType(Val);
1287 return (ExtType != AArch64_AM::UXTX) || AArch64_AM::getMemDoShift(Val);
1288 }
1289 return false;
1290}
1291
1292/// Check all MachineMemOperands for a hint to suppress pairing.
1293bool AArch64InstrInfo::isLdStPairSuppressed(const MachineInstr *MI) const {
1294 assert(MOSuppressPair < (1 << MachineMemOperand::MOTargetNumBits) &&
1295 "Too many target MO flags");
1296 for (auto *MM : MI->memoperands()) {
1297 if (MM->getFlags() &
1298 (MOSuppressPair << MachineMemOperand::MOTargetStartBit)) {
1299 return true;
1300 }
1301 }
1302 return false;
1303}
1304
1305/// Set a flag on the first MachineMemOperand to suppress pairing.
1306void AArch64InstrInfo::suppressLdStPair(MachineInstr *MI) const {
1307 if (MI->memoperands_empty())
1308 return;
1309
1310 assert(MOSuppressPair < (1 << MachineMemOperand::MOTargetNumBits) &&
1311 "Too many target MO flags");
1312 (*MI->memoperands_begin())
1313 ->setFlags(MOSuppressPair << MachineMemOperand::MOTargetStartBit);
1314}
1315
1316bool
Sanjoy Dasb666ea32015-06-15 18:44:14 +00001317AArch64InstrInfo::getMemOpBaseRegImmOfs(MachineInstr *LdSt, unsigned &BaseReg,
1318 unsigned &Offset,
1319 const TargetRegisterInfo *TRI) const {
Tim Northover3b0846e2014-05-24 12:50:23 +00001320 switch (LdSt->getOpcode()) {
1321 default:
1322 return false;
1323 case AArch64::STRSui:
1324 case AArch64::STRDui:
1325 case AArch64::STRQui:
1326 case AArch64::STRXui:
1327 case AArch64::STRWui:
1328 case AArch64::LDRSui:
1329 case AArch64::LDRDui:
1330 case AArch64::LDRQui:
1331 case AArch64::LDRXui:
1332 case AArch64::LDRWui:
1333 if (!LdSt->getOperand(1).isReg() || !LdSt->getOperand(2).isImm())
1334 return false;
1335 BaseReg = LdSt->getOperand(1).getReg();
1336 MachineFunction &MF = *LdSt->getParent()->getParent();
1337 unsigned Width = getRegClass(LdSt->getDesc(), 0, TRI, MF)->getSize();
1338 Offset = LdSt->getOperand(2).getImm() * Width;
1339 return true;
1340 };
1341}
1342
Sanjoy Dasb666ea32015-06-15 18:44:14 +00001343bool AArch64InstrInfo::getMemOpBaseRegImmOfsWidth(
Chad Rosier3528c1e2014-09-08 14:43:48 +00001344 MachineInstr *LdSt, unsigned &BaseReg, int &Offset, int &Width,
1345 const TargetRegisterInfo *TRI) const {
1346 // Handle only loads/stores with base register followed by immediate offset.
1347 if (LdSt->getNumOperands() != 3)
1348 return false;
1349 if (!LdSt->getOperand(1).isReg() || !LdSt->getOperand(2).isImm())
1350 return false;
1351
1352 // Offset is calculated as the immediate operand multiplied by the scaling factor.
1353 // Unscaled instructions have scaling factor set to 1.
1354 int Scale = 0;
1355 switch (LdSt->getOpcode()) {
1356 default:
1357 return false;
1358 case AArch64::LDURQi:
1359 case AArch64::STURQi:
1360 Width = 16;
1361 Scale = 1;
1362 break;
1363 case AArch64::LDURXi:
1364 case AArch64::LDURDi:
1365 case AArch64::STURXi:
1366 case AArch64::STURDi:
1367 Width = 8;
1368 Scale = 1;
1369 break;
1370 case AArch64::LDURWi:
1371 case AArch64::LDURSi:
1372 case AArch64::LDURSWi:
1373 case AArch64::STURWi:
1374 case AArch64::STURSi:
1375 Width = 4;
1376 Scale = 1;
1377 break;
1378 case AArch64::LDURHi:
1379 case AArch64::LDURHHi:
1380 case AArch64::LDURSHXi:
1381 case AArch64::LDURSHWi:
1382 case AArch64::STURHi:
1383 case AArch64::STURHHi:
1384 Width = 2;
1385 Scale = 1;
1386 break;
1387 case AArch64::LDURBi:
1388 case AArch64::LDURBBi:
1389 case AArch64::LDURSBXi:
1390 case AArch64::LDURSBWi:
1391 case AArch64::STURBi:
1392 case AArch64::STURBBi:
1393 Width = 1;
1394 Scale = 1;
1395 break;
Chad Rosierd90e2eb2015-09-18 14:15:19 +00001396 case AArch64::LDRQui:
1397 case AArch64::STRQui:
1398 Scale = Width = 16;
1399 break;
Chad Rosier3528c1e2014-09-08 14:43:48 +00001400 case AArch64::LDRXui:
Chad Rosier84a0afd2015-09-18 14:13:18 +00001401 case AArch64::LDRDui:
Chad Rosier3528c1e2014-09-08 14:43:48 +00001402 case AArch64::STRXui:
Chad Rosier84a0afd2015-09-18 14:13:18 +00001403 case AArch64::STRDui:
Chad Rosier3528c1e2014-09-08 14:43:48 +00001404 Scale = Width = 8;
1405 break;
1406 case AArch64::LDRWui:
Chad Rosier3528c1e2014-09-08 14:43:48 +00001407 case AArch64::LDRSui:
Chad Rosier84a0afd2015-09-18 14:13:18 +00001408 case AArch64::STRWui:
Chad Rosier3528c1e2014-09-08 14:43:48 +00001409 case AArch64::STRSui:
1410 Scale = Width = 4;
1411 break;
Chad Rosier84a0afd2015-09-18 14:13:18 +00001412 case AArch64::LDRHui:
1413 case AArch64::LDRHHui:
1414 case AArch64::STRHui:
1415 case AArch64::STRHHui:
1416 Scale = Width = 2;
Chad Rosier3528c1e2014-09-08 14:43:48 +00001417 break;
Chad Rosierd90e2eb2015-09-18 14:15:19 +00001418 case AArch64::LDRBui:
1419 case AArch64::LDRBBui:
1420 case AArch64::STRBui:
1421 case AArch64::STRBBui:
1422 Scale = Width = 1;
Chad Rosier3528c1e2014-09-08 14:43:48 +00001423 break;
Chad Rosier064261d2016-02-01 20:54:36 +00001424 }
Chad Rosier3528c1e2014-09-08 14:43:48 +00001425
1426 BaseReg = LdSt->getOperand(1).getReg();
1427 Offset = LdSt->getOperand(2).getImm() * Scale;
1428 return true;
1429}
1430
Tim Northover3b0846e2014-05-24 12:50:23 +00001431/// Detect opportunities for ldp/stp formation.
1432///
Sanjoy Dasb666ea32015-06-15 18:44:14 +00001433/// Only called for LdSt for which getMemOpBaseRegImmOfs returns true.
Tim Northover3b0846e2014-05-24 12:50:23 +00001434bool AArch64InstrInfo::shouldClusterLoads(MachineInstr *FirstLdSt,
1435 MachineInstr *SecondLdSt,
1436 unsigned NumLoads) const {
1437 // Only cluster up to a single pair.
1438 if (NumLoads > 1)
1439 return false;
1440 if (FirstLdSt->getOpcode() != SecondLdSt->getOpcode())
1441 return false;
Sanjoy Dasb666ea32015-06-15 18:44:14 +00001442 // getMemOpBaseRegImmOfs guarantees that oper 2 isImm.
Tim Northover3b0846e2014-05-24 12:50:23 +00001443 unsigned Ofs1 = FirstLdSt->getOperand(2).getImm();
1444 // Allow 6 bits of positive range.
1445 if (Ofs1 > 64)
1446 return false;
1447 // The caller should already have ordered First/SecondLdSt by offset.
1448 unsigned Ofs2 = SecondLdSt->getOperand(2).getImm();
1449 return Ofs1 + 1 == Ofs2;
1450}
1451
1452bool AArch64InstrInfo::shouldScheduleAdjacent(MachineInstr *First,
1453 MachineInstr *Second) const {
Matthias Braunc8b67e62015-07-20 23:11:42 +00001454 if (Subtarget.isCyclone()) {
1455 // Cyclone can fuse CMN, CMP, TST followed by Bcc.
1456 unsigned SecondOpcode = Second->getOpcode();
1457 if (SecondOpcode == AArch64::Bcc) {
1458 switch (First->getOpcode()) {
1459 default:
1460 return false;
1461 case AArch64::SUBSWri:
1462 case AArch64::ADDSWri:
1463 case AArch64::ANDSWri:
1464 case AArch64::SUBSXri:
1465 case AArch64::ADDSXri:
1466 case AArch64::ANDSXri:
1467 return true;
1468 }
Matthias Braune536f4f2015-07-20 22:34:47 +00001469 }
Matthias Braunc8b67e62015-07-20 23:11:42 +00001470 // Cyclone B0 also supports ALU operations followed by CBZ/CBNZ.
1471 if (SecondOpcode == AArch64::CBNZW || SecondOpcode == AArch64::CBNZX ||
1472 SecondOpcode == AArch64::CBZW || SecondOpcode == AArch64::CBZX) {
1473 switch (First->getOpcode()) {
1474 default:
1475 return false;
1476 case AArch64::ADDWri:
1477 case AArch64::ADDXri:
1478 case AArch64::ANDWri:
1479 case AArch64::ANDXri:
1480 case AArch64::EORWri:
1481 case AArch64::EORXri:
1482 case AArch64::ORRWri:
1483 case AArch64::ORRXri:
1484 case AArch64::SUBWri:
1485 case AArch64::SUBXri:
1486 return true;
1487 }
Matthias Braune536f4f2015-07-20 22:34:47 +00001488 }
1489 }
1490 return false;
Tim Northover3b0846e2014-05-24 12:50:23 +00001491}
1492
Adrian Prantl87b7eb92014-10-01 18:55:02 +00001493MachineInstr *AArch64InstrInfo::emitFrameIndexDebugValue(
1494 MachineFunction &MF, int FrameIx, uint64_t Offset, const MDNode *Var,
1495 const MDNode *Expr, DebugLoc DL) const {
Tim Northover3b0846e2014-05-24 12:50:23 +00001496 MachineInstrBuilder MIB = BuildMI(MF, DL, get(AArch64::DBG_VALUE))
1497 .addFrameIndex(FrameIx)
1498 .addImm(0)
1499 .addImm(Offset)
Adrian Prantl87b7eb92014-10-01 18:55:02 +00001500 .addMetadata(Var)
1501 .addMetadata(Expr);
Tim Northover3b0846e2014-05-24 12:50:23 +00001502 return &*MIB;
1503}
1504
1505static const MachineInstrBuilder &AddSubReg(const MachineInstrBuilder &MIB,
1506 unsigned Reg, unsigned SubIdx,
1507 unsigned State,
1508 const TargetRegisterInfo *TRI) {
1509 if (!SubIdx)
1510 return MIB.addReg(Reg, State);
1511
1512 if (TargetRegisterInfo::isPhysicalRegister(Reg))
1513 return MIB.addReg(TRI->getSubReg(Reg, SubIdx), State);
1514 return MIB.addReg(Reg, State, SubIdx);
1515}
1516
1517static bool forwardCopyWillClobberTuple(unsigned DestReg, unsigned SrcReg,
1518 unsigned NumRegs) {
1519 // We really want the positive remainder mod 32 here, that happens to be
1520 // easily obtainable with a mask.
1521 return ((DestReg - SrcReg) & 0x1f) < NumRegs;
1522}
1523
1524void AArch64InstrInfo::copyPhysRegTuple(
1525 MachineBasicBlock &MBB, MachineBasicBlock::iterator I, DebugLoc DL,
1526 unsigned DestReg, unsigned SrcReg, bool KillSrc, unsigned Opcode,
1527 llvm::ArrayRef<unsigned> Indices) const {
Eric Christopher58f32662014-06-10 22:57:21 +00001528 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001529 "Unexpected register copy without NEON");
Eric Christophera0de2532015-03-18 20:37:30 +00001530 const TargetRegisterInfo *TRI = &getRegisterInfo();
Tim Northover3b0846e2014-05-24 12:50:23 +00001531 uint16_t DestEncoding = TRI->getEncodingValue(DestReg);
1532 uint16_t SrcEncoding = TRI->getEncodingValue(SrcReg);
1533 unsigned NumRegs = Indices.size();
1534
1535 int SubReg = 0, End = NumRegs, Incr = 1;
1536 if (forwardCopyWillClobberTuple(DestEncoding, SrcEncoding, NumRegs)) {
1537 SubReg = NumRegs - 1;
1538 End = -1;
1539 Incr = -1;
1540 }
1541
1542 for (; SubReg != End; SubReg += Incr) {
James Molloyf8aa57a2015-04-16 11:37:40 +00001543 const MachineInstrBuilder MIB = BuildMI(MBB, I, DL, get(Opcode));
Tim Northover3b0846e2014-05-24 12:50:23 +00001544 AddSubReg(MIB, DestReg, Indices[SubReg], RegState::Define, TRI);
1545 AddSubReg(MIB, SrcReg, Indices[SubReg], 0, TRI);
1546 AddSubReg(MIB, SrcReg, Indices[SubReg], getKillRegState(KillSrc), TRI);
1547 }
1548}
1549
1550void AArch64InstrInfo::copyPhysReg(MachineBasicBlock &MBB,
1551 MachineBasicBlock::iterator I, DebugLoc DL,
1552 unsigned DestReg, unsigned SrcReg,
1553 bool KillSrc) const {
1554 if (AArch64::GPR32spRegClass.contains(DestReg) &&
1555 (AArch64::GPR32spRegClass.contains(SrcReg) || SrcReg == AArch64::WZR)) {
Eric Christophera0de2532015-03-18 20:37:30 +00001556 const TargetRegisterInfo *TRI = &getRegisterInfo();
1557
Tim Northover3b0846e2014-05-24 12:50:23 +00001558 if (DestReg == AArch64::WSP || SrcReg == AArch64::WSP) {
1559 // If either operand is WSP, expand to ADD #0.
1560 if (Subtarget.hasZeroCycleRegMove()) {
1561 // Cyclone recognizes "ADD Xd, Xn, #0" as a zero-cycle register move.
1562 unsigned DestRegX = TRI->getMatchingSuperReg(DestReg, AArch64::sub_32,
1563 &AArch64::GPR64spRegClass);
1564 unsigned SrcRegX = TRI->getMatchingSuperReg(SrcReg, AArch64::sub_32,
1565 &AArch64::GPR64spRegClass);
1566 // This instruction is reading and writing X registers. This may upset
1567 // the register scavenger and machine verifier, so we need to indicate
1568 // that we are reading an undefined value from SrcRegX, but a proper
1569 // value from SrcReg.
1570 BuildMI(MBB, I, DL, get(AArch64::ADDXri), DestRegX)
1571 .addReg(SrcRegX, RegState::Undef)
1572 .addImm(0)
1573 .addImm(AArch64_AM::getShifterImm(AArch64_AM::LSL, 0))
1574 .addReg(SrcReg, RegState::Implicit | getKillRegState(KillSrc));
1575 } else {
1576 BuildMI(MBB, I, DL, get(AArch64::ADDWri), DestReg)
1577 .addReg(SrcReg, getKillRegState(KillSrc))
1578 .addImm(0)
1579 .addImm(AArch64_AM::getShifterImm(AArch64_AM::LSL, 0));
1580 }
1581 } else if (SrcReg == AArch64::WZR && Subtarget.hasZeroCycleZeroing()) {
1582 BuildMI(MBB, I, DL, get(AArch64::MOVZWi), DestReg).addImm(0).addImm(
1583 AArch64_AM::getShifterImm(AArch64_AM::LSL, 0));
1584 } else {
1585 if (Subtarget.hasZeroCycleRegMove()) {
1586 // Cyclone recognizes "ORR Xd, XZR, Xm" as a zero-cycle register move.
1587 unsigned DestRegX = TRI->getMatchingSuperReg(DestReg, AArch64::sub_32,
1588 &AArch64::GPR64spRegClass);
1589 unsigned SrcRegX = TRI->getMatchingSuperReg(SrcReg, AArch64::sub_32,
1590 &AArch64::GPR64spRegClass);
1591 // This instruction is reading and writing X registers. This may upset
1592 // the register scavenger and machine verifier, so we need to indicate
1593 // that we are reading an undefined value from SrcRegX, but a proper
1594 // value from SrcReg.
1595 BuildMI(MBB, I, DL, get(AArch64::ORRXrr), DestRegX)
1596 .addReg(AArch64::XZR)
1597 .addReg(SrcRegX, RegState::Undef)
1598 .addReg(SrcReg, RegState::Implicit | getKillRegState(KillSrc));
1599 } else {
1600 // Otherwise, expand to ORR WZR.
1601 BuildMI(MBB, I, DL, get(AArch64::ORRWrr), DestReg)
1602 .addReg(AArch64::WZR)
1603 .addReg(SrcReg, getKillRegState(KillSrc));
1604 }
1605 }
1606 return;
1607 }
1608
1609 if (AArch64::GPR64spRegClass.contains(DestReg) &&
1610 (AArch64::GPR64spRegClass.contains(SrcReg) || SrcReg == AArch64::XZR)) {
1611 if (DestReg == AArch64::SP || SrcReg == AArch64::SP) {
1612 // If either operand is SP, expand to ADD #0.
1613 BuildMI(MBB, I, DL, get(AArch64::ADDXri), DestReg)
1614 .addReg(SrcReg, getKillRegState(KillSrc))
1615 .addImm(0)
1616 .addImm(AArch64_AM::getShifterImm(AArch64_AM::LSL, 0));
1617 } else if (SrcReg == AArch64::XZR && Subtarget.hasZeroCycleZeroing()) {
1618 BuildMI(MBB, I, DL, get(AArch64::MOVZXi), DestReg).addImm(0).addImm(
1619 AArch64_AM::getShifterImm(AArch64_AM::LSL, 0));
1620 } else {
1621 // Otherwise, expand to ORR XZR.
1622 BuildMI(MBB, I, DL, get(AArch64::ORRXrr), DestReg)
1623 .addReg(AArch64::XZR)
1624 .addReg(SrcReg, getKillRegState(KillSrc));
1625 }
1626 return;
1627 }
1628
1629 // Copy a DDDD register quad by copying the individual sub-registers.
1630 if (AArch64::DDDDRegClass.contains(DestReg) &&
1631 AArch64::DDDDRegClass.contains(SrcReg)) {
1632 static const unsigned Indices[] = { AArch64::dsub0, AArch64::dsub1,
1633 AArch64::dsub2, AArch64::dsub3 };
1634 copyPhysRegTuple(MBB, I, DL, DestReg, SrcReg, KillSrc, AArch64::ORRv8i8,
1635 Indices);
1636 return;
1637 }
1638
1639 // Copy a DDD register triple by copying the individual sub-registers.
1640 if (AArch64::DDDRegClass.contains(DestReg) &&
1641 AArch64::DDDRegClass.contains(SrcReg)) {
1642 static const unsigned Indices[] = { AArch64::dsub0, AArch64::dsub1,
1643 AArch64::dsub2 };
1644 copyPhysRegTuple(MBB, I, DL, DestReg, SrcReg, KillSrc, AArch64::ORRv8i8,
1645 Indices);
1646 return;
1647 }
1648
1649 // Copy a DD register pair by copying the individual sub-registers.
1650 if (AArch64::DDRegClass.contains(DestReg) &&
1651 AArch64::DDRegClass.contains(SrcReg)) {
1652 static const unsigned Indices[] = { AArch64::dsub0, AArch64::dsub1 };
1653 copyPhysRegTuple(MBB, I, DL, DestReg, SrcReg, KillSrc, AArch64::ORRv8i8,
1654 Indices);
1655 return;
1656 }
1657
1658 // Copy a QQQQ register quad by copying the individual sub-registers.
1659 if (AArch64::QQQQRegClass.contains(DestReg) &&
1660 AArch64::QQQQRegClass.contains(SrcReg)) {
1661 static const unsigned Indices[] = { AArch64::qsub0, AArch64::qsub1,
1662 AArch64::qsub2, AArch64::qsub3 };
1663 copyPhysRegTuple(MBB, I, DL, DestReg, SrcReg, KillSrc, AArch64::ORRv16i8,
1664 Indices);
1665 return;
1666 }
1667
1668 // Copy a QQQ register triple by copying the individual sub-registers.
1669 if (AArch64::QQQRegClass.contains(DestReg) &&
1670 AArch64::QQQRegClass.contains(SrcReg)) {
1671 static const unsigned Indices[] = { AArch64::qsub0, AArch64::qsub1,
1672 AArch64::qsub2 };
1673 copyPhysRegTuple(MBB, I, DL, DestReg, SrcReg, KillSrc, AArch64::ORRv16i8,
1674 Indices);
1675 return;
1676 }
1677
1678 // Copy a QQ register pair by copying the individual sub-registers.
1679 if (AArch64::QQRegClass.contains(DestReg) &&
1680 AArch64::QQRegClass.contains(SrcReg)) {
1681 static const unsigned Indices[] = { AArch64::qsub0, AArch64::qsub1 };
1682 copyPhysRegTuple(MBB, I, DL, DestReg, SrcReg, KillSrc, AArch64::ORRv16i8,
1683 Indices);
1684 return;
1685 }
1686
1687 if (AArch64::FPR128RegClass.contains(DestReg) &&
1688 AArch64::FPR128RegClass.contains(SrcReg)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001689 if(Subtarget.hasNEON()) {
Tim Northover3b0846e2014-05-24 12:50:23 +00001690 BuildMI(MBB, I, DL, get(AArch64::ORRv16i8), DestReg)
1691 .addReg(SrcReg)
1692 .addReg(SrcReg, getKillRegState(KillSrc));
1693 } else {
1694 BuildMI(MBB, I, DL, get(AArch64::STRQpre))
1695 .addReg(AArch64::SP, RegState::Define)
1696 .addReg(SrcReg, getKillRegState(KillSrc))
1697 .addReg(AArch64::SP)
1698 .addImm(-16);
1699 BuildMI(MBB, I, DL, get(AArch64::LDRQpre))
1700 .addReg(AArch64::SP, RegState::Define)
1701 .addReg(DestReg, RegState::Define)
1702 .addReg(AArch64::SP)
1703 .addImm(16);
1704 }
1705 return;
1706 }
1707
1708 if (AArch64::FPR64RegClass.contains(DestReg) &&
1709 AArch64::FPR64RegClass.contains(SrcReg)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001710 if(Subtarget.hasNEON()) {
Eric Christophera0de2532015-03-18 20:37:30 +00001711 DestReg = RI.getMatchingSuperReg(DestReg, AArch64::dsub,
1712 &AArch64::FPR128RegClass);
1713 SrcReg = RI.getMatchingSuperReg(SrcReg, AArch64::dsub,
1714 &AArch64::FPR128RegClass);
Tim Northover3b0846e2014-05-24 12:50:23 +00001715 BuildMI(MBB, I, DL, get(AArch64::ORRv16i8), DestReg)
1716 .addReg(SrcReg)
1717 .addReg(SrcReg, getKillRegState(KillSrc));
1718 } else {
1719 BuildMI(MBB, I, DL, get(AArch64::FMOVDr), DestReg)
1720 .addReg(SrcReg, getKillRegState(KillSrc));
1721 }
1722 return;
1723 }
1724
1725 if (AArch64::FPR32RegClass.contains(DestReg) &&
1726 AArch64::FPR32RegClass.contains(SrcReg)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001727 if(Subtarget.hasNEON()) {
Eric Christophera0de2532015-03-18 20:37:30 +00001728 DestReg = RI.getMatchingSuperReg(DestReg, AArch64::ssub,
1729 &AArch64::FPR128RegClass);
1730 SrcReg = RI.getMatchingSuperReg(SrcReg, AArch64::ssub,
1731 &AArch64::FPR128RegClass);
Tim Northover3b0846e2014-05-24 12:50:23 +00001732 BuildMI(MBB, I, DL, get(AArch64::ORRv16i8), DestReg)
1733 .addReg(SrcReg)
1734 .addReg(SrcReg, getKillRegState(KillSrc));
1735 } else {
1736 BuildMI(MBB, I, DL, get(AArch64::FMOVSr), DestReg)
1737 .addReg(SrcReg, getKillRegState(KillSrc));
1738 }
1739 return;
1740 }
1741
1742 if (AArch64::FPR16RegClass.contains(DestReg) &&
1743 AArch64::FPR16RegClass.contains(SrcReg)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001744 if(Subtarget.hasNEON()) {
Eric Christophera0de2532015-03-18 20:37:30 +00001745 DestReg = RI.getMatchingSuperReg(DestReg, AArch64::hsub,
1746 &AArch64::FPR128RegClass);
1747 SrcReg = RI.getMatchingSuperReg(SrcReg, AArch64::hsub,
1748 &AArch64::FPR128RegClass);
Tim Northover3b0846e2014-05-24 12:50:23 +00001749 BuildMI(MBB, I, DL, get(AArch64::ORRv16i8), DestReg)
1750 .addReg(SrcReg)
1751 .addReg(SrcReg, getKillRegState(KillSrc));
1752 } else {
Eric Christophera0de2532015-03-18 20:37:30 +00001753 DestReg = RI.getMatchingSuperReg(DestReg, AArch64::hsub,
1754 &AArch64::FPR32RegClass);
1755 SrcReg = RI.getMatchingSuperReg(SrcReg, AArch64::hsub,
1756 &AArch64::FPR32RegClass);
Tim Northover3b0846e2014-05-24 12:50:23 +00001757 BuildMI(MBB, I, DL, get(AArch64::FMOVSr), DestReg)
1758 .addReg(SrcReg, getKillRegState(KillSrc));
1759 }
1760 return;
1761 }
1762
1763 if (AArch64::FPR8RegClass.contains(DestReg) &&
1764 AArch64::FPR8RegClass.contains(SrcReg)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001765 if(Subtarget.hasNEON()) {
Eric Christophera0de2532015-03-18 20:37:30 +00001766 DestReg = RI.getMatchingSuperReg(DestReg, AArch64::bsub,
Tim Northover3b0846e2014-05-24 12:50:23 +00001767 &AArch64::FPR128RegClass);
Eric Christophera0de2532015-03-18 20:37:30 +00001768 SrcReg = RI.getMatchingSuperReg(SrcReg, AArch64::bsub,
1769 &AArch64::FPR128RegClass);
Tim Northover3b0846e2014-05-24 12:50:23 +00001770 BuildMI(MBB, I, DL, get(AArch64::ORRv16i8), DestReg)
1771 .addReg(SrcReg)
1772 .addReg(SrcReg, getKillRegState(KillSrc));
1773 } else {
Eric Christophera0de2532015-03-18 20:37:30 +00001774 DestReg = RI.getMatchingSuperReg(DestReg, AArch64::bsub,
1775 &AArch64::FPR32RegClass);
1776 SrcReg = RI.getMatchingSuperReg(SrcReg, AArch64::bsub,
1777 &AArch64::FPR32RegClass);
Tim Northover3b0846e2014-05-24 12:50:23 +00001778 BuildMI(MBB, I, DL, get(AArch64::FMOVSr), DestReg)
1779 .addReg(SrcReg, getKillRegState(KillSrc));
1780 }
1781 return;
1782 }
1783
1784 // Copies between GPR64 and FPR64.
1785 if (AArch64::FPR64RegClass.contains(DestReg) &&
1786 AArch64::GPR64RegClass.contains(SrcReg)) {
1787 BuildMI(MBB, I, DL, get(AArch64::FMOVXDr), DestReg)
1788 .addReg(SrcReg, getKillRegState(KillSrc));
1789 return;
1790 }
1791 if (AArch64::GPR64RegClass.contains(DestReg) &&
1792 AArch64::FPR64RegClass.contains(SrcReg)) {
1793 BuildMI(MBB, I, DL, get(AArch64::FMOVDXr), DestReg)
1794 .addReg(SrcReg, getKillRegState(KillSrc));
1795 return;
1796 }
1797 // Copies between GPR32 and FPR32.
1798 if (AArch64::FPR32RegClass.contains(DestReg) &&
1799 AArch64::GPR32RegClass.contains(SrcReg)) {
1800 BuildMI(MBB, I, DL, get(AArch64::FMOVWSr), DestReg)
1801 .addReg(SrcReg, getKillRegState(KillSrc));
1802 return;
1803 }
1804 if (AArch64::GPR32RegClass.contains(DestReg) &&
1805 AArch64::FPR32RegClass.contains(SrcReg)) {
1806 BuildMI(MBB, I, DL, get(AArch64::FMOVSWr), DestReg)
1807 .addReg(SrcReg, getKillRegState(KillSrc));
1808 return;
1809 }
1810
Tim Northover1bed9af2014-05-27 12:16:02 +00001811 if (DestReg == AArch64::NZCV) {
1812 assert(AArch64::GPR64RegClass.contains(SrcReg) && "Invalid NZCV copy");
1813 BuildMI(MBB, I, DL, get(AArch64::MSR))
1814 .addImm(AArch64SysReg::NZCV)
1815 .addReg(SrcReg, getKillRegState(KillSrc))
1816 .addReg(AArch64::NZCV, RegState::Implicit | RegState::Define);
1817 return;
1818 }
1819
1820 if (SrcReg == AArch64::NZCV) {
1821 assert(AArch64::GPR64RegClass.contains(DestReg) && "Invalid NZCV copy");
1822 BuildMI(MBB, I, DL, get(AArch64::MRS))
1823 .addReg(DestReg)
1824 .addImm(AArch64SysReg::NZCV)
1825 .addReg(AArch64::NZCV, RegState::Implicit | getKillRegState(KillSrc));
1826 return;
1827 }
1828
1829 llvm_unreachable("unimplemented reg-to-reg copy");
Tim Northover3b0846e2014-05-24 12:50:23 +00001830}
1831
1832void AArch64InstrInfo::storeRegToStackSlot(
1833 MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, unsigned SrcReg,
1834 bool isKill, int FI, const TargetRegisterClass *RC,
1835 const TargetRegisterInfo *TRI) const {
1836 DebugLoc DL;
1837 if (MBBI != MBB.end())
1838 DL = MBBI->getDebugLoc();
1839 MachineFunction &MF = *MBB.getParent();
1840 MachineFrameInfo &MFI = *MF.getFrameInfo();
1841 unsigned Align = MFI.getObjectAlignment(FI);
1842
Alex Lorenze40c8a22015-08-11 23:09:45 +00001843 MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(MF, FI);
Tim Northover3b0846e2014-05-24 12:50:23 +00001844 MachineMemOperand *MMO = MF.getMachineMemOperand(
1845 PtrInfo, MachineMemOperand::MOStore, MFI.getObjectSize(FI), Align);
1846 unsigned Opc = 0;
1847 bool Offset = true;
1848 switch (RC->getSize()) {
1849 case 1:
1850 if (AArch64::FPR8RegClass.hasSubClassEq(RC))
1851 Opc = AArch64::STRBui;
1852 break;
1853 case 2:
1854 if (AArch64::FPR16RegClass.hasSubClassEq(RC))
1855 Opc = AArch64::STRHui;
1856 break;
1857 case 4:
1858 if (AArch64::GPR32allRegClass.hasSubClassEq(RC)) {
1859 Opc = AArch64::STRWui;
1860 if (TargetRegisterInfo::isVirtualRegister(SrcReg))
1861 MF.getRegInfo().constrainRegClass(SrcReg, &AArch64::GPR32RegClass);
1862 else
1863 assert(SrcReg != AArch64::WSP);
1864 } else if (AArch64::FPR32RegClass.hasSubClassEq(RC))
1865 Opc = AArch64::STRSui;
1866 break;
1867 case 8:
1868 if (AArch64::GPR64allRegClass.hasSubClassEq(RC)) {
1869 Opc = AArch64::STRXui;
1870 if (TargetRegisterInfo::isVirtualRegister(SrcReg))
1871 MF.getRegInfo().constrainRegClass(SrcReg, &AArch64::GPR64RegClass);
1872 else
1873 assert(SrcReg != AArch64::SP);
1874 } else if (AArch64::FPR64RegClass.hasSubClassEq(RC))
1875 Opc = AArch64::STRDui;
1876 break;
1877 case 16:
1878 if (AArch64::FPR128RegClass.hasSubClassEq(RC))
1879 Opc = AArch64::STRQui;
1880 else if (AArch64::DDRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001881 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001882 "Unexpected register store without NEON");
1883 Opc = AArch64::ST1Twov1d, Offset = false;
1884 }
1885 break;
1886 case 24:
1887 if (AArch64::DDDRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001888 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001889 "Unexpected register store without NEON");
1890 Opc = AArch64::ST1Threev1d, Offset = false;
1891 }
1892 break;
1893 case 32:
1894 if (AArch64::DDDDRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001895 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001896 "Unexpected register store without NEON");
1897 Opc = AArch64::ST1Fourv1d, Offset = false;
1898 } else if (AArch64::QQRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001899 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001900 "Unexpected register store without NEON");
1901 Opc = AArch64::ST1Twov2d, Offset = false;
1902 }
1903 break;
1904 case 48:
1905 if (AArch64::QQQRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001906 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001907 "Unexpected register store without NEON");
1908 Opc = AArch64::ST1Threev2d, Offset = false;
1909 }
1910 break;
1911 case 64:
1912 if (AArch64::QQQQRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001913 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001914 "Unexpected register store without NEON");
1915 Opc = AArch64::ST1Fourv2d, Offset = false;
1916 }
1917 break;
1918 }
1919 assert(Opc && "Unknown register class");
1920
James Molloyf8aa57a2015-04-16 11:37:40 +00001921 const MachineInstrBuilder MI = BuildMI(MBB, MBBI, DL, get(Opc))
Tim Northover3b0846e2014-05-24 12:50:23 +00001922 .addReg(SrcReg, getKillRegState(isKill))
1923 .addFrameIndex(FI);
1924
1925 if (Offset)
1926 MI.addImm(0);
1927 MI.addMemOperand(MMO);
1928}
1929
1930void AArch64InstrInfo::loadRegFromStackSlot(
1931 MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, unsigned DestReg,
1932 int FI, const TargetRegisterClass *RC,
1933 const TargetRegisterInfo *TRI) const {
1934 DebugLoc DL;
1935 if (MBBI != MBB.end())
1936 DL = MBBI->getDebugLoc();
1937 MachineFunction &MF = *MBB.getParent();
1938 MachineFrameInfo &MFI = *MF.getFrameInfo();
1939 unsigned Align = MFI.getObjectAlignment(FI);
Alex Lorenze40c8a22015-08-11 23:09:45 +00001940 MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(MF, FI);
Tim Northover3b0846e2014-05-24 12:50:23 +00001941 MachineMemOperand *MMO = MF.getMachineMemOperand(
1942 PtrInfo, MachineMemOperand::MOLoad, MFI.getObjectSize(FI), Align);
1943
1944 unsigned Opc = 0;
1945 bool Offset = true;
1946 switch (RC->getSize()) {
1947 case 1:
1948 if (AArch64::FPR8RegClass.hasSubClassEq(RC))
1949 Opc = AArch64::LDRBui;
1950 break;
1951 case 2:
1952 if (AArch64::FPR16RegClass.hasSubClassEq(RC))
1953 Opc = AArch64::LDRHui;
1954 break;
1955 case 4:
1956 if (AArch64::GPR32allRegClass.hasSubClassEq(RC)) {
1957 Opc = AArch64::LDRWui;
1958 if (TargetRegisterInfo::isVirtualRegister(DestReg))
1959 MF.getRegInfo().constrainRegClass(DestReg, &AArch64::GPR32RegClass);
1960 else
1961 assert(DestReg != AArch64::WSP);
1962 } else if (AArch64::FPR32RegClass.hasSubClassEq(RC))
1963 Opc = AArch64::LDRSui;
1964 break;
1965 case 8:
1966 if (AArch64::GPR64allRegClass.hasSubClassEq(RC)) {
1967 Opc = AArch64::LDRXui;
1968 if (TargetRegisterInfo::isVirtualRegister(DestReg))
1969 MF.getRegInfo().constrainRegClass(DestReg, &AArch64::GPR64RegClass);
1970 else
1971 assert(DestReg != AArch64::SP);
1972 } else if (AArch64::FPR64RegClass.hasSubClassEq(RC))
1973 Opc = AArch64::LDRDui;
1974 break;
1975 case 16:
1976 if (AArch64::FPR128RegClass.hasSubClassEq(RC))
1977 Opc = AArch64::LDRQui;
1978 else if (AArch64::DDRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001979 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001980 "Unexpected register load without NEON");
1981 Opc = AArch64::LD1Twov1d, Offset = false;
1982 }
1983 break;
1984 case 24:
1985 if (AArch64::DDDRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001986 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001987 "Unexpected register load without NEON");
1988 Opc = AArch64::LD1Threev1d, Offset = false;
1989 }
1990 break;
1991 case 32:
1992 if (AArch64::DDDDRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001993 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001994 "Unexpected register load without NEON");
1995 Opc = AArch64::LD1Fourv1d, Offset = false;
1996 } else if (AArch64::QQRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00001997 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00001998 "Unexpected register load without NEON");
1999 Opc = AArch64::LD1Twov2d, Offset = false;
2000 }
2001 break;
2002 case 48:
2003 if (AArch64::QQQRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00002004 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00002005 "Unexpected register load without NEON");
2006 Opc = AArch64::LD1Threev2d, Offset = false;
2007 }
2008 break;
2009 case 64:
2010 if (AArch64::QQQQRegClass.hasSubClassEq(RC)) {
Eric Christopher58f32662014-06-10 22:57:21 +00002011 assert(Subtarget.hasNEON() &&
Tim Northover3b0846e2014-05-24 12:50:23 +00002012 "Unexpected register load without NEON");
2013 Opc = AArch64::LD1Fourv2d, Offset = false;
2014 }
2015 break;
2016 }
2017 assert(Opc && "Unknown register class");
2018
James Molloyf8aa57a2015-04-16 11:37:40 +00002019 const MachineInstrBuilder MI = BuildMI(MBB, MBBI, DL, get(Opc))
Tim Northover3b0846e2014-05-24 12:50:23 +00002020 .addReg(DestReg, getDefRegState(true))
2021 .addFrameIndex(FI);
2022 if (Offset)
2023 MI.addImm(0);
2024 MI.addMemOperand(MMO);
2025}
2026
2027void llvm::emitFrameOffset(MachineBasicBlock &MBB,
2028 MachineBasicBlock::iterator MBBI, DebugLoc DL,
2029 unsigned DestReg, unsigned SrcReg, int Offset,
Eric Christopherbc76b972014-06-10 17:33:39 +00002030 const TargetInstrInfo *TII,
Tim Northover3b0846e2014-05-24 12:50:23 +00002031 MachineInstr::MIFlag Flag, bool SetNZCV) {
2032 if (DestReg == SrcReg && Offset == 0)
2033 return;
2034
2035 bool isSub = Offset < 0;
2036 if (isSub)
2037 Offset = -Offset;
2038
2039 // FIXME: If the offset won't fit in 24-bits, compute the offset into a
2040 // scratch register. If DestReg is a virtual register, use it as the
2041 // scratch register; otherwise, create a new virtual register (to be
2042 // replaced by the scavenger at the end of PEI). That case can be optimized
2043 // slightly if DestReg is SP which is always 16-byte aligned, so the scratch
2044 // register can be loaded with offset%8 and the add/sub can use an extending
2045 // instruction with LSL#3.
2046 // Currently the function handles any offsets but generates a poor sequence
2047 // of code.
2048 // assert(Offset < (1 << 24) && "unimplemented reg plus immediate");
2049
2050 unsigned Opc;
2051 if (SetNZCV)
2052 Opc = isSub ? AArch64::SUBSXri : AArch64::ADDSXri;
2053 else
2054 Opc = isSub ? AArch64::SUBXri : AArch64::ADDXri;
2055 const unsigned MaxEncoding = 0xfff;
2056 const unsigned ShiftSize = 12;
2057 const unsigned MaxEncodableValue = MaxEncoding << ShiftSize;
2058 while (((unsigned)Offset) >= (1 << ShiftSize)) {
2059 unsigned ThisVal;
2060 if (((unsigned)Offset) > MaxEncodableValue) {
2061 ThisVal = MaxEncodableValue;
2062 } else {
2063 ThisVal = Offset & MaxEncodableValue;
2064 }
2065 assert((ThisVal >> ShiftSize) <= MaxEncoding &&
2066 "Encoding cannot handle value that big");
2067 BuildMI(MBB, MBBI, DL, TII->get(Opc), DestReg)
2068 .addReg(SrcReg)
2069 .addImm(ThisVal >> ShiftSize)
2070 .addImm(AArch64_AM::getShifterImm(AArch64_AM::LSL, ShiftSize))
2071 .setMIFlag(Flag);
2072
2073 SrcReg = DestReg;
2074 Offset -= ThisVal;
2075 if (Offset == 0)
2076 return;
2077 }
2078 BuildMI(MBB, MBBI, DL, TII->get(Opc), DestReg)
2079 .addReg(SrcReg)
2080 .addImm(Offset)
2081 .addImm(AArch64_AM::getShifterImm(AArch64_AM::LSL, 0))
2082 .setMIFlag(Flag);
2083}
2084
Keno Fischere70b31f2015-06-08 20:09:58 +00002085MachineInstr *AArch64InstrInfo::foldMemoryOperandImpl(
2086 MachineFunction &MF, MachineInstr *MI, ArrayRef<unsigned> Ops,
2087 MachineBasicBlock::iterator InsertPt, int FrameIndex) const {
Tim Northover3b0846e2014-05-24 12:50:23 +00002088 // This is a bit of a hack. Consider this instruction:
2089 //
2090 // %vreg0<def> = COPY %SP; GPR64all:%vreg0
2091 //
2092 // We explicitly chose GPR64all for the virtual register so such a copy might
2093 // be eliminated by RegisterCoalescer. However, that may not be possible, and
2094 // %vreg0 may even spill. We can't spill %SP, and since it is in the GPR64all
2095 // register class, TargetInstrInfo::foldMemoryOperand() is going to try.
2096 //
2097 // To prevent that, we are going to constrain the %vreg0 register class here.
2098 //
2099 // <rdar://problem/11522048>
2100 //
2101 if (MI->isCopy()) {
2102 unsigned DstReg = MI->getOperand(0).getReg();
2103 unsigned SrcReg = MI->getOperand(1).getReg();
2104 if (SrcReg == AArch64::SP &&
2105 TargetRegisterInfo::isVirtualRegister(DstReg)) {
2106 MF.getRegInfo().constrainRegClass(DstReg, &AArch64::GPR64RegClass);
2107 return nullptr;
2108 }
2109 if (DstReg == AArch64::SP &&
2110 TargetRegisterInfo::isVirtualRegister(SrcReg)) {
2111 MF.getRegInfo().constrainRegClass(SrcReg, &AArch64::GPR64RegClass);
2112 return nullptr;
2113 }
2114 }
2115
2116 // Cannot fold.
2117 return nullptr;
2118}
2119
2120int llvm::isAArch64FrameOffsetLegal(const MachineInstr &MI, int &Offset,
2121 bool *OutUseUnscaledOp,
2122 unsigned *OutUnscaledOp,
2123 int *EmittableOffset) {
2124 int Scale = 1;
2125 bool IsSigned = false;
2126 // The ImmIdx should be changed case by case if it is not 2.
2127 unsigned ImmIdx = 2;
2128 unsigned UnscaledOp = 0;
2129 // Set output values in case of early exit.
2130 if (EmittableOffset)
2131 *EmittableOffset = 0;
2132 if (OutUseUnscaledOp)
2133 *OutUseUnscaledOp = false;
2134 if (OutUnscaledOp)
2135 *OutUnscaledOp = 0;
2136 switch (MI.getOpcode()) {
2137 default:
Craig Topper2a30d782014-06-18 05:05:13 +00002138 llvm_unreachable("unhandled opcode in rewriteAArch64FrameIndex");
Tim Northover3b0846e2014-05-24 12:50:23 +00002139 // Vector spills/fills can't take an immediate offset.
2140 case AArch64::LD1Twov2d:
2141 case AArch64::LD1Threev2d:
2142 case AArch64::LD1Fourv2d:
2143 case AArch64::LD1Twov1d:
2144 case AArch64::LD1Threev1d:
2145 case AArch64::LD1Fourv1d:
2146 case AArch64::ST1Twov2d:
2147 case AArch64::ST1Threev2d:
2148 case AArch64::ST1Fourv2d:
2149 case AArch64::ST1Twov1d:
2150 case AArch64::ST1Threev1d:
2151 case AArch64::ST1Fourv1d:
2152 return AArch64FrameOffsetCannotUpdate;
2153 case AArch64::PRFMui:
2154 Scale = 8;
2155 UnscaledOp = AArch64::PRFUMi;
2156 break;
2157 case AArch64::LDRXui:
2158 Scale = 8;
2159 UnscaledOp = AArch64::LDURXi;
2160 break;
2161 case AArch64::LDRWui:
2162 Scale = 4;
2163 UnscaledOp = AArch64::LDURWi;
2164 break;
2165 case AArch64::LDRBui:
2166 Scale = 1;
2167 UnscaledOp = AArch64::LDURBi;
2168 break;
2169 case AArch64::LDRHui:
2170 Scale = 2;
2171 UnscaledOp = AArch64::LDURHi;
2172 break;
2173 case AArch64::LDRSui:
2174 Scale = 4;
2175 UnscaledOp = AArch64::LDURSi;
2176 break;
2177 case AArch64::LDRDui:
2178 Scale = 8;
2179 UnscaledOp = AArch64::LDURDi;
2180 break;
2181 case AArch64::LDRQui:
2182 Scale = 16;
2183 UnscaledOp = AArch64::LDURQi;
2184 break;
2185 case AArch64::LDRBBui:
2186 Scale = 1;
2187 UnscaledOp = AArch64::LDURBBi;
2188 break;
2189 case AArch64::LDRHHui:
2190 Scale = 2;
2191 UnscaledOp = AArch64::LDURHHi;
2192 break;
2193 case AArch64::LDRSBXui:
2194 Scale = 1;
2195 UnscaledOp = AArch64::LDURSBXi;
2196 break;
2197 case AArch64::LDRSBWui:
2198 Scale = 1;
2199 UnscaledOp = AArch64::LDURSBWi;
2200 break;
2201 case AArch64::LDRSHXui:
2202 Scale = 2;
2203 UnscaledOp = AArch64::LDURSHXi;
2204 break;
2205 case AArch64::LDRSHWui:
2206 Scale = 2;
2207 UnscaledOp = AArch64::LDURSHWi;
2208 break;
2209 case AArch64::LDRSWui:
2210 Scale = 4;
2211 UnscaledOp = AArch64::LDURSWi;
2212 break;
2213
2214 case AArch64::STRXui:
2215 Scale = 8;
2216 UnscaledOp = AArch64::STURXi;
2217 break;
2218 case AArch64::STRWui:
2219 Scale = 4;
2220 UnscaledOp = AArch64::STURWi;
2221 break;
2222 case AArch64::STRBui:
2223 Scale = 1;
2224 UnscaledOp = AArch64::STURBi;
2225 break;
2226 case AArch64::STRHui:
2227 Scale = 2;
2228 UnscaledOp = AArch64::STURHi;
2229 break;
2230 case AArch64::STRSui:
2231 Scale = 4;
2232 UnscaledOp = AArch64::STURSi;
2233 break;
2234 case AArch64::STRDui:
2235 Scale = 8;
2236 UnscaledOp = AArch64::STURDi;
2237 break;
2238 case AArch64::STRQui:
2239 Scale = 16;
2240 UnscaledOp = AArch64::STURQi;
2241 break;
2242 case AArch64::STRBBui:
2243 Scale = 1;
2244 UnscaledOp = AArch64::STURBBi;
2245 break;
2246 case AArch64::STRHHui:
2247 Scale = 2;
2248 UnscaledOp = AArch64::STURHHi;
2249 break;
2250
2251 case AArch64::LDPXi:
2252 case AArch64::LDPDi:
2253 case AArch64::STPXi:
2254 case AArch64::STPDi:
Ahmed Bougacha05541452015-09-10 01:54:43 +00002255 case AArch64::LDNPXi:
2256 case AArch64::LDNPDi:
2257 case AArch64::STNPXi:
2258 case AArch64::STNPDi:
2259 ImmIdx = 3;
Tim Northover3b0846e2014-05-24 12:50:23 +00002260 IsSigned = true;
2261 Scale = 8;
2262 break;
2263 case AArch64::LDPQi:
2264 case AArch64::STPQi:
Ahmed Bougacha05541452015-09-10 01:54:43 +00002265 case AArch64::LDNPQi:
2266 case AArch64::STNPQi:
2267 ImmIdx = 3;
Tim Northover3b0846e2014-05-24 12:50:23 +00002268 IsSigned = true;
2269 Scale = 16;
2270 break;
2271 case AArch64::LDPWi:
2272 case AArch64::LDPSi:
2273 case AArch64::STPWi:
2274 case AArch64::STPSi:
Ahmed Bougacha05541452015-09-10 01:54:43 +00002275 case AArch64::LDNPWi:
2276 case AArch64::LDNPSi:
2277 case AArch64::STNPWi:
2278 case AArch64::STNPSi:
2279 ImmIdx = 3;
Tim Northover3b0846e2014-05-24 12:50:23 +00002280 IsSigned = true;
2281 Scale = 4;
2282 break;
2283
2284 case AArch64::LDURXi:
2285 case AArch64::LDURWi:
2286 case AArch64::LDURBi:
2287 case AArch64::LDURHi:
2288 case AArch64::LDURSi:
2289 case AArch64::LDURDi:
2290 case AArch64::LDURQi:
2291 case AArch64::LDURHHi:
2292 case AArch64::LDURBBi:
2293 case AArch64::LDURSBXi:
2294 case AArch64::LDURSBWi:
2295 case AArch64::LDURSHXi:
2296 case AArch64::LDURSHWi:
2297 case AArch64::LDURSWi:
2298 case AArch64::STURXi:
2299 case AArch64::STURWi:
2300 case AArch64::STURBi:
2301 case AArch64::STURHi:
2302 case AArch64::STURSi:
2303 case AArch64::STURDi:
2304 case AArch64::STURQi:
2305 case AArch64::STURBBi:
2306 case AArch64::STURHHi:
2307 Scale = 1;
2308 break;
2309 }
2310
2311 Offset += MI.getOperand(ImmIdx).getImm() * Scale;
2312
2313 bool useUnscaledOp = false;
2314 // If the offset doesn't match the scale, we rewrite the instruction to
2315 // use the unscaled instruction instead. Likewise, if we have a negative
2316 // offset (and have an unscaled op to use).
2317 if ((Offset & (Scale - 1)) != 0 || (Offset < 0 && UnscaledOp != 0))
2318 useUnscaledOp = true;
2319
2320 // Use an unscaled addressing mode if the instruction has a negative offset
2321 // (or if the instruction is already using an unscaled addressing mode).
2322 unsigned MaskBits;
2323 if (IsSigned) {
2324 // ldp/stp instructions.
2325 MaskBits = 7;
2326 Offset /= Scale;
2327 } else if (UnscaledOp == 0 || useUnscaledOp) {
2328 MaskBits = 9;
2329 IsSigned = true;
2330 Scale = 1;
2331 } else {
2332 MaskBits = 12;
2333 IsSigned = false;
2334 Offset /= Scale;
2335 }
2336
2337 // Attempt to fold address computation.
2338 int MaxOff = (1 << (MaskBits - IsSigned)) - 1;
2339 int MinOff = (IsSigned ? (-MaxOff - 1) : 0);
2340 if (Offset >= MinOff && Offset <= MaxOff) {
2341 if (EmittableOffset)
2342 *EmittableOffset = Offset;
2343 Offset = 0;
2344 } else {
2345 int NewOff = Offset < 0 ? MinOff : MaxOff;
2346 if (EmittableOffset)
2347 *EmittableOffset = NewOff;
2348 Offset = (Offset - NewOff) * Scale;
2349 }
2350 if (OutUseUnscaledOp)
2351 *OutUseUnscaledOp = useUnscaledOp;
2352 if (OutUnscaledOp)
2353 *OutUnscaledOp = UnscaledOp;
2354 return AArch64FrameOffsetCanUpdate |
2355 (Offset == 0 ? AArch64FrameOffsetIsLegal : 0);
2356}
2357
2358bool llvm::rewriteAArch64FrameIndex(MachineInstr &MI, unsigned FrameRegIdx,
2359 unsigned FrameReg, int &Offset,
2360 const AArch64InstrInfo *TII) {
2361 unsigned Opcode = MI.getOpcode();
2362 unsigned ImmIdx = FrameRegIdx + 1;
2363
2364 if (Opcode == AArch64::ADDSXri || Opcode == AArch64::ADDXri) {
2365 Offset += MI.getOperand(ImmIdx).getImm();
2366 emitFrameOffset(*MI.getParent(), MI, MI.getDebugLoc(),
2367 MI.getOperand(0).getReg(), FrameReg, Offset, TII,
2368 MachineInstr::NoFlags, (Opcode == AArch64::ADDSXri));
2369 MI.eraseFromParent();
2370 Offset = 0;
2371 return true;
2372 }
2373
2374 int NewOffset;
2375 unsigned UnscaledOp;
2376 bool UseUnscaledOp;
2377 int Status = isAArch64FrameOffsetLegal(MI, Offset, &UseUnscaledOp,
2378 &UnscaledOp, &NewOffset);
2379 if (Status & AArch64FrameOffsetCanUpdate) {
2380 if (Status & AArch64FrameOffsetIsLegal)
2381 // Replace the FrameIndex with FrameReg.
2382 MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
2383 if (UseUnscaledOp)
2384 MI.setDesc(TII->get(UnscaledOp));
2385
2386 MI.getOperand(ImmIdx).ChangeToImmediate(NewOffset);
2387 return Offset == 0;
2388 }
2389
2390 return false;
2391}
2392
2393void AArch64InstrInfo::getNoopForMachoTarget(MCInst &NopInst) const {
2394 NopInst.setOpcode(AArch64::HINT);
Jim Grosbache9119e42015-05-13 18:37:00 +00002395 NopInst.addOperand(MCOperand::createImm(0));
Tim Northover3b0846e2014-05-24 12:50:23 +00002396}
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002397/// useMachineCombiner - return true when a target supports MachineCombiner
Benjamin Kramer8c90fd72014-09-03 11:41:21 +00002398bool AArch64InstrInfo::useMachineCombiner() const {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002399 // AArch64 supports the combiner
2400 return true;
2401}
2402//
2403// True when Opc sets flag
2404static bool isCombineInstrSettingFlag(unsigned Opc) {
2405 switch (Opc) {
2406 case AArch64::ADDSWrr:
2407 case AArch64::ADDSWri:
2408 case AArch64::ADDSXrr:
2409 case AArch64::ADDSXri:
2410 case AArch64::SUBSWrr:
2411 case AArch64::SUBSXrr:
2412 // Note: MSUB Wd,Wn,Wm,Wi -> Wd = Wi - WnxWm, not Wd=WnxWm - Wi.
2413 case AArch64::SUBSWri:
2414 case AArch64::SUBSXri:
2415 return true;
2416 default:
2417 break;
2418 }
2419 return false;
2420}
2421//
2422// 32b Opcodes that can be combined with a MUL
2423static bool isCombineInstrCandidate32(unsigned Opc) {
2424 switch (Opc) {
2425 case AArch64::ADDWrr:
2426 case AArch64::ADDWri:
2427 case AArch64::SUBWrr:
2428 case AArch64::ADDSWrr:
2429 case AArch64::ADDSWri:
2430 case AArch64::SUBSWrr:
2431 // Note: MSUB Wd,Wn,Wm,Wi -> Wd = Wi - WnxWm, not Wd=WnxWm - Wi.
2432 case AArch64::SUBWri:
2433 case AArch64::SUBSWri:
2434 return true;
2435 default:
2436 break;
2437 }
2438 return false;
2439}
2440//
2441// 64b Opcodes that can be combined with a MUL
2442static bool isCombineInstrCandidate64(unsigned Opc) {
2443 switch (Opc) {
2444 case AArch64::ADDXrr:
2445 case AArch64::ADDXri:
2446 case AArch64::SUBXrr:
2447 case AArch64::ADDSXrr:
2448 case AArch64::ADDSXri:
2449 case AArch64::SUBSXrr:
2450 // Note: MSUB Wd,Wn,Wm,Wi -> Wd = Wi - WnxWm, not Wd=WnxWm - Wi.
2451 case AArch64::SUBXri:
2452 case AArch64::SUBSXri:
2453 return true;
2454 default:
2455 break;
2456 }
2457 return false;
2458}
2459//
2460// Opcodes that can be combined with a MUL
2461static bool isCombineInstrCandidate(unsigned Opc) {
2462 return (isCombineInstrCandidate32(Opc) || isCombineInstrCandidate64(Opc));
2463}
2464
2465static bool canCombineWithMUL(MachineBasicBlock &MBB, MachineOperand &MO,
2466 unsigned MulOpc, unsigned ZeroReg) {
2467 MachineRegisterInfo &MRI = MBB.getParent()->getRegInfo();
2468 MachineInstr *MI = nullptr;
2469 // We need a virtual register definition.
2470 if (MO.isReg() && TargetRegisterInfo::isVirtualRegister(MO.getReg()))
2471 MI = MRI.getUniqueVRegDef(MO.getReg());
2472 // And it needs to be in the trace (otherwise, it won't have a depth).
2473 if (!MI || MI->getParent() != &MBB || (unsigned)MI->getOpcode() != MulOpc)
2474 return false;
2475
2476 assert(MI->getNumOperands() >= 4 && MI->getOperand(0).isReg() &&
2477 MI->getOperand(1).isReg() && MI->getOperand(2).isReg() &&
2478 MI->getOperand(3).isReg() && "MAdd/MSub must have a least 4 regs");
2479
2480 // The third input reg must be zero.
2481 if (MI->getOperand(3).getReg() != ZeroReg)
2482 return false;
2483
2484 // Must only used by the user we combine with.
Gerolf Hoflehnerfe2c11f2014-08-13 22:07:36 +00002485 if (!MRI.hasOneNonDBGUse(MI->getOperand(0).getReg()))
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002486 return false;
2487
2488 return true;
2489}
2490
Haicheng Wu08b94622016-01-07 04:01:02 +00002491// TODO: There are many more machine instruction opcodes to match:
2492// 1. Other data types (integer, vectors)
2493// 2. Other math / logic operations (xor, or)
2494// 3. Other forms of the same operation (intrinsics and other variants)
2495bool AArch64InstrInfo::isAssociativeAndCommutative(const MachineInstr &Inst) const {
2496 switch (Inst.getOpcode()) {
2497 case AArch64::FADDDrr:
2498 case AArch64::FADDSrr:
2499 case AArch64::FADDv2f32:
2500 case AArch64::FADDv2f64:
2501 case AArch64::FADDv4f32:
2502 case AArch64::FMULDrr:
2503 case AArch64::FMULSrr:
2504 case AArch64::FMULX32:
2505 case AArch64::FMULX64:
2506 case AArch64::FMULXv2f32:
2507 case AArch64::FMULXv2f64:
2508 case AArch64::FMULXv4f32:
2509 case AArch64::FMULv2f32:
2510 case AArch64::FMULv2f64:
2511 case AArch64::FMULv4f32:
2512 return Inst.getParent()->getParent()->getTarget().Options.UnsafeFPMath;
2513 default:
2514 return false;
2515 }
2516}
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002517
Haicheng Wu08b94622016-01-07 04:01:02 +00002518/// Find instructions that can be turned into madd.
2519static bool getMaddPatterns(MachineInstr &Root,
2520 SmallVectorImpl<MachineCombinerPattern> &Patterns) {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002521 unsigned Opc = Root.getOpcode();
2522 MachineBasicBlock &MBB = *Root.getParent();
2523 bool Found = false;
2524
2525 if (!isCombineInstrCandidate(Opc))
2526 return 0;
2527 if (isCombineInstrSettingFlag(Opc)) {
2528 int Cmp_NZCV = Root.findRegisterDefOperandIdx(AArch64::NZCV, true);
2529 // When NZCV is live bail out.
2530 if (Cmp_NZCV == -1)
2531 return 0;
2532 unsigned NewOpc = convertFlagSettingOpcode(&Root);
2533 // When opcode can't change bail out.
2534 // CHECKME: do we miss any cases for opcode conversion?
2535 if (NewOpc == Opc)
2536 return 0;
2537 Opc = NewOpc;
2538 }
2539
2540 switch (Opc) {
2541 default:
2542 break;
2543 case AArch64::ADDWrr:
2544 assert(Root.getOperand(1).isReg() && Root.getOperand(2).isReg() &&
2545 "ADDWrr does not have register operands");
2546 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDWrrr,
2547 AArch64::WZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002548 Patterns.push_back(MachineCombinerPattern::MULADDW_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002549 Found = true;
2550 }
2551 if (canCombineWithMUL(MBB, Root.getOperand(2), AArch64::MADDWrrr,
2552 AArch64::WZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002553 Patterns.push_back(MachineCombinerPattern::MULADDW_OP2);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002554 Found = true;
2555 }
2556 break;
2557 case AArch64::ADDXrr:
2558 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDXrrr,
2559 AArch64::XZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002560 Patterns.push_back(MachineCombinerPattern::MULADDX_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002561 Found = true;
2562 }
2563 if (canCombineWithMUL(MBB, Root.getOperand(2), AArch64::MADDXrrr,
2564 AArch64::XZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002565 Patterns.push_back(MachineCombinerPattern::MULADDX_OP2);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002566 Found = true;
2567 }
2568 break;
2569 case AArch64::SUBWrr:
2570 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDWrrr,
2571 AArch64::WZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002572 Patterns.push_back(MachineCombinerPattern::MULSUBW_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002573 Found = true;
2574 }
2575 if (canCombineWithMUL(MBB, Root.getOperand(2), AArch64::MADDWrrr,
2576 AArch64::WZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002577 Patterns.push_back(MachineCombinerPattern::MULSUBW_OP2);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002578 Found = true;
2579 }
2580 break;
2581 case AArch64::SUBXrr:
2582 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDXrrr,
2583 AArch64::XZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002584 Patterns.push_back(MachineCombinerPattern::MULSUBX_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002585 Found = true;
2586 }
2587 if (canCombineWithMUL(MBB, Root.getOperand(2), AArch64::MADDXrrr,
2588 AArch64::XZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002589 Patterns.push_back(MachineCombinerPattern::MULSUBX_OP2);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002590 Found = true;
2591 }
2592 break;
2593 case AArch64::ADDWri:
2594 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDWrrr,
2595 AArch64::WZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002596 Patterns.push_back(MachineCombinerPattern::MULADDWI_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002597 Found = true;
2598 }
2599 break;
2600 case AArch64::ADDXri:
2601 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDXrrr,
2602 AArch64::XZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002603 Patterns.push_back(MachineCombinerPattern::MULADDXI_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002604 Found = true;
2605 }
2606 break;
2607 case AArch64::SUBWri:
2608 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDWrrr,
2609 AArch64::WZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002610 Patterns.push_back(MachineCombinerPattern::MULSUBWI_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002611 Found = true;
2612 }
2613 break;
2614 case AArch64::SUBXri:
2615 if (canCombineWithMUL(MBB, Root.getOperand(1), AArch64::MADDXrrr,
2616 AArch64::XZR)) {
Sanjay Patel387e66e2015-11-05 19:34:57 +00002617 Patterns.push_back(MachineCombinerPattern::MULSUBXI_OP1);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002618 Found = true;
2619 }
2620 break;
2621 }
2622 return Found;
2623}
2624
Haicheng Wu08b94622016-01-07 04:01:02 +00002625/// Return true when there is potentially a faster code sequence for an
2626/// instruction chain ending in \p Root. All potential patterns are listed in
2627/// the \p Pattern vector. Pattern should be sorted in priority order since the
2628/// pattern evaluator stops checking as soon as it finds a faster sequence.
2629
2630bool AArch64InstrInfo::getMachineCombinerPatterns(
2631 MachineInstr &Root,
2632 SmallVectorImpl<MachineCombinerPattern> &Patterns) const {
2633 if (getMaddPatterns(Root, Patterns))
2634 return true;
2635
2636 return TargetInstrInfo::getMachineCombinerPatterns(Root, Patterns);
2637}
2638
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002639/// genMadd - Generate madd instruction and combine mul and add.
2640/// Example:
2641/// MUL I=A,B,0
2642/// ADD R,I,C
2643/// ==> MADD R,A,B,C
2644/// \param Root is the ADD instruction
NAKAMURA Takumi40da2672014-08-08 02:04:18 +00002645/// \param [out] InsInstrs is a vector of machine instructions and will
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002646/// contain the generated madd instruction
2647/// \param IdxMulOpd is index of operand in Root that is the result of
2648/// the MUL. In the example above IdxMulOpd is 1.
2649/// \param MaddOpc the opcode fo the madd instruction
2650static MachineInstr *genMadd(MachineFunction &MF, MachineRegisterInfo &MRI,
2651 const TargetInstrInfo *TII, MachineInstr &Root,
2652 SmallVectorImpl<MachineInstr *> &InsInstrs,
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002653 unsigned IdxMulOpd, unsigned MaddOpc,
2654 const TargetRegisterClass *RC) {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002655 assert(IdxMulOpd == 1 || IdxMulOpd == 2);
2656
2657 unsigned IdxOtherOpd = IdxMulOpd == 1 ? 2 : 1;
2658 MachineInstr *MUL = MRI.getUniqueVRegDef(Root.getOperand(IdxMulOpd).getReg());
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002659 unsigned ResultReg = Root.getOperand(0).getReg();
2660 unsigned SrcReg0 = MUL->getOperand(1).getReg();
2661 bool Src0IsKill = MUL->getOperand(1).isKill();
2662 unsigned SrcReg1 = MUL->getOperand(2).getReg();
2663 bool Src1IsKill = MUL->getOperand(2).isKill();
2664 unsigned SrcReg2 = Root.getOperand(IdxOtherOpd).getReg();
2665 bool Src2IsKill = Root.getOperand(IdxOtherOpd).isKill();
2666
2667 if (TargetRegisterInfo::isVirtualRegister(ResultReg))
2668 MRI.constrainRegClass(ResultReg, RC);
2669 if (TargetRegisterInfo::isVirtualRegister(SrcReg0))
2670 MRI.constrainRegClass(SrcReg0, RC);
2671 if (TargetRegisterInfo::isVirtualRegister(SrcReg1))
2672 MRI.constrainRegClass(SrcReg1, RC);
2673 if (TargetRegisterInfo::isVirtualRegister(SrcReg2))
2674 MRI.constrainRegClass(SrcReg2, RC);
2675
2676 MachineInstrBuilder MIB = BuildMI(MF, Root.getDebugLoc(), TII->get(MaddOpc),
2677 ResultReg)
2678 .addReg(SrcReg0, getKillRegState(Src0IsKill))
2679 .addReg(SrcReg1, getKillRegState(Src1IsKill))
2680 .addReg(SrcReg2, getKillRegState(Src2IsKill));
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002681 // Insert the MADD
2682 InsInstrs.push_back(MIB);
2683 return MUL;
2684}
2685
2686/// genMaddR - Generate madd instruction and combine mul and add using
2687/// an extra virtual register
2688/// Example - an ADD intermediate needs to be stored in a register:
2689/// MUL I=A,B,0
2690/// ADD R,I,Imm
2691/// ==> ORR V, ZR, Imm
2692/// ==> MADD R,A,B,V
2693/// \param Root is the ADD instruction
NAKAMURA Takumi40da2672014-08-08 02:04:18 +00002694/// \param [out] InsInstrs is a vector of machine instructions and will
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002695/// contain the generated madd instruction
2696/// \param IdxMulOpd is index of operand in Root that is the result of
2697/// the MUL. In the example above IdxMulOpd is 1.
2698/// \param MaddOpc the opcode fo the madd instruction
2699/// \param VR is a virtual register that holds the value of an ADD operand
2700/// (V in the example above).
2701static MachineInstr *genMaddR(MachineFunction &MF, MachineRegisterInfo &MRI,
2702 const TargetInstrInfo *TII, MachineInstr &Root,
2703 SmallVectorImpl<MachineInstr *> &InsInstrs,
2704 unsigned IdxMulOpd, unsigned MaddOpc,
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002705 unsigned VR, const TargetRegisterClass *RC) {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002706 assert(IdxMulOpd == 1 || IdxMulOpd == 2);
2707
2708 MachineInstr *MUL = MRI.getUniqueVRegDef(Root.getOperand(IdxMulOpd).getReg());
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002709 unsigned ResultReg = Root.getOperand(0).getReg();
2710 unsigned SrcReg0 = MUL->getOperand(1).getReg();
2711 bool Src0IsKill = MUL->getOperand(1).isKill();
2712 unsigned SrcReg1 = MUL->getOperand(2).getReg();
2713 bool Src1IsKill = MUL->getOperand(2).isKill();
2714
2715 if (TargetRegisterInfo::isVirtualRegister(ResultReg))
2716 MRI.constrainRegClass(ResultReg, RC);
2717 if (TargetRegisterInfo::isVirtualRegister(SrcReg0))
2718 MRI.constrainRegClass(SrcReg0, RC);
2719 if (TargetRegisterInfo::isVirtualRegister(SrcReg1))
2720 MRI.constrainRegClass(SrcReg1, RC);
2721 if (TargetRegisterInfo::isVirtualRegister(VR))
2722 MRI.constrainRegClass(VR, RC);
2723
2724 MachineInstrBuilder MIB = BuildMI(MF, Root.getDebugLoc(), TII->get(MaddOpc),
2725 ResultReg)
2726 .addReg(SrcReg0, getKillRegState(Src0IsKill))
2727 .addReg(SrcReg1, getKillRegState(Src1IsKill))
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002728 .addReg(VR);
2729 // Insert the MADD
2730 InsInstrs.push_back(MIB);
2731 return MUL;
2732}
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002733
Sanjay Patelcfe03932015-06-19 23:21:42 +00002734/// When getMachineCombinerPatterns() finds potential patterns,
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002735/// this function generates the instructions that could replace the
2736/// original code sequence
2737void AArch64InstrInfo::genAlternativeCodeSequence(
Sanjay Patel387e66e2015-11-05 19:34:57 +00002738 MachineInstr &Root, MachineCombinerPattern Pattern,
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002739 SmallVectorImpl<MachineInstr *> &InsInstrs,
2740 SmallVectorImpl<MachineInstr *> &DelInstrs,
2741 DenseMap<unsigned, unsigned> &InstrIdxForVirtReg) const {
2742 MachineBasicBlock &MBB = *Root.getParent();
2743 MachineRegisterInfo &MRI = MBB.getParent()->getRegInfo();
2744 MachineFunction &MF = *MBB.getParent();
Eric Christophere0818912014-09-03 20:36:26 +00002745 const TargetInstrInfo *TII = MF.getSubtarget().getInstrInfo();
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002746
2747 MachineInstr *MUL;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002748 const TargetRegisterClass *RC;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002749 unsigned Opc;
2750 switch (Pattern) {
2751 default:
Haicheng Wu08b94622016-01-07 04:01:02 +00002752 // Reassociate instructions.
2753 TargetInstrInfo::genAlternativeCodeSequence(Root, Pattern, InsInstrs,
2754 DelInstrs, InstrIdxForVirtReg);
2755 return;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002756 case MachineCombinerPattern::MULADDW_OP1:
2757 case MachineCombinerPattern::MULADDX_OP1:
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002758 // MUL I=A,B,0
2759 // ADD R,I,C
2760 // ==> MADD R,A,B,C
2761 // --- Create(MADD);
Sanjay Patel387e66e2015-11-05 19:34:57 +00002762 if (Pattern == MachineCombinerPattern::MULADDW_OP1) {
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002763 Opc = AArch64::MADDWrrr;
2764 RC = &AArch64::GPR32RegClass;
2765 } else {
2766 Opc = AArch64::MADDXrrr;
2767 RC = &AArch64::GPR64RegClass;
2768 }
2769 MUL = genMadd(MF, MRI, TII, Root, InsInstrs, 1, Opc, RC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002770 break;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002771 case MachineCombinerPattern::MULADDW_OP2:
2772 case MachineCombinerPattern::MULADDX_OP2:
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002773 // MUL I=A,B,0
2774 // ADD R,C,I
2775 // ==> MADD R,A,B,C
2776 // --- Create(MADD);
Sanjay Patel387e66e2015-11-05 19:34:57 +00002777 if (Pattern == MachineCombinerPattern::MULADDW_OP2) {
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002778 Opc = AArch64::MADDWrrr;
2779 RC = &AArch64::GPR32RegClass;
2780 } else {
2781 Opc = AArch64::MADDXrrr;
2782 RC = &AArch64::GPR64RegClass;
2783 }
2784 MUL = genMadd(MF, MRI, TII, Root, InsInstrs, 2, Opc, RC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002785 break;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002786 case MachineCombinerPattern::MULADDWI_OP1:
2787 case MachineCombinerPattern::MULADDXI_OP1: {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002788 // MUL I=A,B,0
2789 // ADD R,I,Imm
2790 // ==> ORR V, ZR, Imm
2791 // ==> MADD R,A,B,V
2792 // --- Create(MADD);
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002793 const TargetRegisterClass *OrrRC;
2794 unsigned BitSize, OrrOpc, ZeroReg;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002795 if (Pattern == MachineCombinerPattern::MULADDWI_OP1) {
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002796 OrrOpc = AArch64::ORRWri;
2797 OrrRC = &AArch64::GPR32spRegClass;
2798 BitSize = 32;
2799 ZeroReg = AArch64::WZR;
2800 Opc = AArch64::MADDWrrr;
2801 RC = &AArch64::GPR32RegClass;
2802 } else {
2803 OrrOpc = AArch64::ORRXri;
2804 OrrRC = &AArch64::GPR64spRegClass;
2805 BitSize = 64;
2806 ZeroReg = AArch64::XZR;
2807 Opc = AArch64::MADDXrrr;
2808 RC = &AArch64::GPR64RegClass;
2809 }
2810 unsigned NewVR = MRI.createVirtualRegister(OrrRC);
2811 uint64_t Imm = Root.getOperand(2).getImm();
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002812
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002813 if (Root.getOperand(3).isImm()) {
2814 unsigned Val = Root.getOperand(3).getImm();
2815 Imm = Imm << Val;
2816 }
2817 uint64_t UImm = Imm << (64 - BitSize) >> (64 - BitSize);
2818 uint64_t Encoding;
2819 if (AArch64_AM::processLogicalImmediate(UImm, BitSize, Encoding)) {
2820 MachineInstrBuilder MIB1 =
2821 BuildMI(MF, Root.getDebugLoc(), TII->get(OrrOpc), NewVR)
2822 .addReg(ZeroReg)
2823 .addImm(Encoding);
2824 InsInstrs.push_back(MIB1);
2825 InstrIdxForVirtReg.insert(std::make_pair(NewVR, 0));
2826 MUL = genMaddR(MF, MRI, TII, Root, InsInstrs, 1, Opc, NewVR, RC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002827 }
2828 break;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002829 }
Sanjay Patel387e66e2015-11-05 19:34:57 +00002830 case MachineCombinerPattern::MULSUBW_OP1:
2831 case MachineCombinerPattern::MULSUBX_OP1: {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002832 // MUL I=A,B,0
2833 // SUB R,I, C
2834 // ==> SUB V, 0, C
2835 // ==> MADD R,A,B,V // = -C + A*B
2836 // --- Create(MADD);
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002837 const TargetRegisterClass *SubRC;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002838 unsigned SubOpc, ZeroReg;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002839 if (Pattern == MachineCombinerPattern::MULSUBW_OP1) {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002840 SubOpc = AArch64::SUBWrr;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002841 SubRC = &AArch64::GPR32spRegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002842 ZeroReg = AArch64::WZR;
2843 Opc = AArch64::MADDWrrr;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002844 RC = &AArch64::GPR32RegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002845 } else {
2846 SubOpc = AArch64::SUBXrr;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002847 SubRC = &AArch64::GPR64spRegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002848 ZeroReg = AArch64::XZR;
2849 Opc = AArch64::MADDXrrr;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002850 RC = &AArch64::GPR64RegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002851 }
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002852 unsigned NewVR = MRI.createVirtualRegister(SubRC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002853 // SUB NewVR, 0, C
2854 MachineInstrBuilder MIB1 =
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002855 BuildMI(MF, Root.getDebugLoc(), TII->get(SubOpc), NewVR)
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002856 .addReg(ZeroReg)
2857 .addOperand(Root.getOperand(2));
2858 InsInstrs.push_back(MIB1);
2859 InstrIdxForVirtReg.insert(std::make_pair(NewVR, 0));
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002860 MUL = genMaddR(MF, MRI, TII, Root, InsInstrs, 1, Opc, NewVR, RC);
2861 break;
2862 }
Sanjay Patel387e66e2015-11-05 19:34:57 +00002863 case MachineCombinerPattern::MULSUBW_OP2:
2864 case MachineCombinerPattern::MULSUBX_OP2:
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002865 // MUL I=A,B,0
2866 // SUB R,C,I
2867 // ==> MSUB R,A,B,C (computes C - A*B)
2868 // --- Create(MSUB);
Sanjay Patel387e66e2015-11-05 19:34:57 +00002869 if (Pattern == MachineCombinerPattern::MULSUBW_OP2) {
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002870 Opc = AArch64::MSUBWrrr;
2871 RC = &AArch64::GPR32RegClass;
2872 } else {
2873 Opc = AArch64::MSUBXrrr;
2874 RC = &AArch64::GPR64RegClass;
2875 }
2876 MUL = genMadd(MF, MRI, TII, Root, InsInstrs, 2, Opc, RC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002877 break;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002878 case MachineCombinerPattern::MULSUBWI_OP1:
2879 case MachineCombinerPattern::MULSUBXI_OP1: {
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002880 // MUL I=A,B,0
2881 // SUB R,I, Imm
2882 // ==> ORR V, ZR, -Imm
2883 // ==> MADD R,A,B,V // = -Imm + A*B
2884 // --- Create(MADD);
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002885 const TargetRegisterClass *OrrRC;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002886 unsigned BitSize, OrrOpc, ZeroReg;
Sanjay Patel387e66e2015-11-05 19:34:57 +00002887 if (Pattern == MachineCombinerPattern::MULSUBWI_OP1) {
Juergen Ributzka25816b02014-08-30 06:16:26 +00002888 OrrOpc = AArch64::ORRWri;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002889 OrrRC = &AArch64::GPR32spRegClass;
2890 BitSize = 32;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002891 ZeroReg = AArch64::WZR;
2892 Opc = AArch64::MADDWrrr;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002893 RC = &AArch64::GPR32RegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002894 } else {
2895 OrrOpc = AArch64::ORRXri;
Juergen Ributzkaf9660f02014-11-04 22:20:07 +00002896 OrrRC = &AArch64::GPR64spRegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002897 BitSize = 64;
2898 ZeroReg = AArch64::XZR;
2899 Opc = AArch64::MADDXrrr;
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002900 RC = &AArch64::GPR64RegClass;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002901 }
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002902 unsigned NewVR = MRI.createVirtualRegister(OrrRC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002903 int Imm = Root.getOperand(2).getImm();
2904 if (Root.getOperand(3).isImm()) {
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002905 unsigned Val = Root.getOperand(3).getImm();
2906 Imm = Imm << Val;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002907 }
2908 uint64_t UImm = -Imm << (64 - BitSize) >> (64 - BitSize);
2909 uint64_t Encoding;
2910 if (AArch64_AM::processLogicalImmediate(UImm, BitSize, Encoding)) {
2911 MachineInstrBuilder MIB1 =
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002912 BuildMI(MF, Root.getDebugLoc(), TII->get(OrrOpc), NewVR)
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002913 .addReg(ZeroReg)
2914 .addImm(Encoding);
2915 InsInstrs.push_back(MIB1);
2916 InstrIdxForVirtReg.insert(std::make_pair(NewVR, 0));
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002917 MUL = genMaddR(MF, MRI, TII, Root, InsInstrs, 1, Opc, NewVR, RC);
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002918 }
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002919 break;
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002920 }
Juergen Ributzka31e5b7f2014-09-03 07:07:10 +00002921 } // end switch (Pattern)
Gerolf Hoflehner97c383b2014-08-07 21:40:58 +00002922 // Record MUL and ADD/SUB for deletion
2923 DelInstrs.push_back(MUL);
2924 DelInstrs.push_back(&Root);
2925
2926 return;
2927}
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00002928
2929/// \brief Replace csincr-branch sequence by simple conditional branch
2930///
2931/// Examples:
2932/// 1.
2933/// csinc w9, wzr, wzr, <condition code>
2934/// tbnz w9, #0, 0x44
2935/// to
2936/// b.<inverted condition code>
2937///
2938/// 2.
2939/// csinc w9, wzr, wzr, <condition code>
2940/// tbz w9, #0, 0x44
2941/// to
2942/// b.<condition code>
2943///
2944/// \param MI Conditional Branch
2945/// \return True when the simple conditional branch is generated
2946///
2947bool AArch64InstrInfo::optimizeCondBranch(MachineInstr *MI) const {
2948 bool IsNegativeBranch = false;
2949 bool IsTestAndBranch = false;
2950 unsigned TargetBBInMI = 0;
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00002951 switch (MI->getOpcode()) {
2952 default:
2953 llvm_unreachable("Unknown branch instruction?");
2954 case AArch64::Bcc:
2955 return false;
2956 case AArch64::CBZW:
2957 case AArch64::CBZX:
2958 TargetBBInMI = 1;
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00002959 break;
2960 case AArch64::CBNZW:
2961 case AArch64::CBNZX:
2962 TargetBBInMI = 1;
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00002963 IsNegativeBranch = true;
2964 break;
2965 case AArch64::TBZW:
2966 case AArch64::TBZX:
2967 TargetBBInMI = 2;
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00002968 IsTestAndBranch = true;
2969 break;
2970 case AArch64::TBNZW:
2971 case AArch64::TBNZX:
2972 TargetBBInMI = 2;
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00002973 IsNegativeBranch = true;
2974 IsTestAndBranch = true;
2975 break;
2976 }
2977 // So we increment a zero register and test for bits other
2978 // than bit 0? Conservatively bail out in case the verifier
2979 // missed this case.
2980 if (IsTestAndBranch && MI->getOperand(1).getImm())
2981 return false;
2982
2983 // Find Definition.
2984 assert(MI->getParent() && "Incomplete machine instruciton\n");
2985 MachineBasicBlock *MBB = MI->getParent();
2986 MachineFunction *MF = MBB->getParent();
2987 MachineRegisterInfo *MRI = &MF->getRegInfo();
2988 unsigned VReg = MI->getOperand(0).getReg();
2989 if (!TargetRegisterInfo::isVirtualRegister(VReg))
2990 return false;
2991
2992 MachineInstr *DefMI = MRI->getVRegDef(VReg);
2993
2994 // Look for CSINC
2995 if (!(DefMI->getOpcode() == AArch64::CSINCWr &&
2996 DefMI->getOperand(1).getReg() == AArch64::WZR &&
2997 DefMI->getOperand(2).getReg() == AArch64::WZR) &&
2998 !(DefMI->getOpcode() == AArch64::CSINCXr &&
2999 DefMI->getOperand(1).getReg() == AArch64::XZR &&
3000 DefMI->getOperand(2).getReg() == AArch64::XZR))
3001 return false;
3002
3003 if (DefMI->findRegisterDefOperandIdx(AArch64::NZCV, true) != -1)
3004 return false;
3005
3006 AArch64CC::CondCode CC =
Gerolf Hoflehner5d26d402014-10-14 23:55:00 +00003007 (AArch64CC::CondCode)DefMI->getOperand(3).getImm();
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00003008 bool CheckOnlyCCWrites = true;
3009 // Convert only when the condition code is not modified between
3010 // the CSINC and the branch. The CC may be used by other
3011 // instructions in between.
Eric Christophera0de2532015-03-18 20:37:30 +00003012 if (modifiesConditionCode(DefMI, MI, CheckOnlyCCWrites, &getRegisterInfo()))
Gerolf Hoflehnera4c96d02014-10-14 23:07:53 +00003013 return false;
3014 MachineBasicBlock &RefToMBB = *MBB;
3015 MachineBasicBlock *TBB = MI->getOperand(TargetBBInMI).getMBB();
3016 DebugLoc DL = MI->getDebugLoc();
3017 if (IsNegativeBranch)
3018 CC = AArch64CC::getInvertedCondCode(CC);
3019 BuildMI(RefToMBB, MI, DL, get(AArch64::Bcc)).addImm(CC).addMBB(TBB);
3020 MI->eraseFromParent();
3021 return true;
3022}
Alex Lorenzf3630112015-08-18 22:52:15 +00003023
3024std::pair<unsigned, unsigned>
3025AArch64InstrInfo::decomposeMachineOperandsTargetFlags(unsigned TF) const {
3026 const unsigned Mask = AArch64II::MO_FRAGMENT;
3027 return std::make_pair(TF & Mask, TF & ~Mask);
3028}
3029
3030ArrayRef<std::pair<unsigned, const char *>>
3031AArch64InstrInfo::getSerializableDirectMachineOperandTargetFlags() const {
3032 using namespace AArch64II;
Hal Finkel982e8d42015-08-30 08:07:29 +00003033 static const std::pair<unsigned, const char *> TargetFlags[] = {
Alex Lorenzf3630112015-08-18 22:52:15 +00003034 {MO_PAGE, "aarch64-page"},
3035 {MO_PAGEOFF, "aarch64-pageoff"},
3036 {MO_G3, "aarch64-g3"},
3037 {MO_G2, "aarch64-g2"},
3038 {MO_G1, "aarch64-g1"},
3039 {MO_G0, "aarch64-g0"},
3040 {MO_HI12, "aarch64-hi12"}};
3041 return makeArrayRef(TargetFlags);
3042}
3043
3044ArrayRef<std::pair<unsigned, const char *>>
3045AArch64InstrInfo::getSerializableBitmaskMachineOperandTargetFlags() const {
3046 using namespace AArch64II;
Hal Finkel982e8d42015-08-30 08:07:29 +00003047 static const std::pair<unsigned, const char *> TargetFlags[] = {
Alex Lorenzf3630112015-08-18 22:52:15 +00003048 {MO_GOT, "aarch64-got"},
3049 {MO_NC, "aarch64-nc"},
3050 {MO_TLS, "aarch64-tls"},
3051 {MO_CONSTPOOL, "aarch64-constant-pool"}};
3052 return makeArrayRef(TargetFlags);
3053}