1 //===- llvm/CodeGen/GlobalISel/InstructionSelector.cpp -----------*- C++ -*-==//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 /// \file
10 /// This file implements the InstructionSelector class.
11 //===----------------------------------------------------------------------===//
12 
13 #include "llvm/CodeGen/GlobalISel/InstructionSelector.h"
14 #include "llvm/CodeGen/GlobalISel/RegisterBankInfo.h"
15 #include "llvm/CodeGen/GlobalISel/Utils.h"
16 #include "llvm/CodeGen/MachineInstr.h"
17 #include "llvm/CodeGen/MachineRegisterInfo.h"
18 #include "llvm/IR/Constants.h"
19 #include "llvm/Target/TargetInstrInfo.h"
20 #include "llvm/Target/TargetRegisterInfo.h"
21 
22 #define DEBUG_TYPE "instructionselector"
23 
24 using namespace llvm;
25 
26 InstructionSelector::InstructionSelector() {}
27 
28 bool InstructionSelector::constrainSelectedInstRegOperands(
29     MachineInstr &I, const TargetInstrInfo &TII, const TargetRegisterInfo &TRI,
30     const RegisterBankInfo &RBI) const {
31   MachineBasicBlock &MBB = *I.getParent();
32   MachineFunction &MF = *MBB.getParent();
33   MachineRegisterInfo &MRI = MF.getRegInfo();
34 
35   for (unsigned OpI = 0, OpE = I.getNumExplicitOperands(); OpI != OpE; ++OpI) {
36     MachineOperand &MO = I.getOperand(OpI);
37 
38     // There's nothing to be done on non-register operands.
39     if (!MO.isReg())
40       continue;
41 
42     DEBUG(dbgs() << "Converting operand: " << MO << '\n');
43     assert(MO.isReg() && "Unsupported non-reg operand");
44 
45     unsigned Reg = MO.getReg();
46     // Physical registers don't need to be constrained.
47     if (TRI.isPhysicalRegister(Reg))
48       continue;
49 
50     // Register operands with a value of 0 (e.g. predicate operands) don't need
51     // to be constrained.
52     if (Reg == 0)
53       continue;
54 
55     // If the operand is a vreg, we should constrain its regclass, and only
56     // insert COPYs if that's impossible.
57     // constrainOperandRegClass does that for us.
58     MO.setReg(constrainOperandRegClass(MF, TRI, MRI, TII, RBI, I, I.getDesc(),
59                                        Reg, OpI));
60 
61     // Tie uses to defs as indicated in MCInstrDesc.
62     if (MO.isUse()) {
63       int DefIdx = I.getDesc().getOperandConstraint(OpI, MCOI::TIED_TO);
64       if (DefIdx != -1)
65         I.tieOperands(DefIdx, OpI);
66     }
67   }
68   return true;
69 }
70 
71 Optional<int64_t>
72 InstructionSelector::getConstantVRegVal(unsigned VReg,
73                                         const MachineRegisterInfo &MRI) const {
74   MachineInstr *MI = MRI.getVRegDef(VReg);
75   if (MI->getOpcode() != TargetOpcode::G_CONSTANT)
76     return None;
77 
78   if (MI->getOperand(1).isImm())
79     return MI->getOperand(1).getImm();
80 
81   if (MI->getOperand(1).isCImm() &&
82       MI->getOperand(1).getCImm()->getBitWidth() <= 64)
83     return MI->getOperand(1).getCImm()->getSExtValue();
84 
85   return None;
86 }
87 
88 bool InstructionSelector::isOperandImmEqual(
89     const MachineOperand &MO, int64_t Value,
90     const MachineRegisterInfo &MRI) const {
91 
92   if (MO.getReg())
93     if (auto VRegVal = getConstantVRegVal(MO.getReg(), MRI))
94       return *VRegVal == Value;
95   return false;
96 }
97 
98 bool InstructionSelector::isObviouslySafeToFold(MachineInstr &MI) const {
99   return !MI.mayLoadOrStore() && !MI.hasUnmodeledSideEffects() &&
100          MI.implicit_operands().begin() == MI.implicit_operands().end();
101 }
102