[AMDGPU] Check for immediate SrcC in mfma in AsmParser
[llvm-core.git] / lib / Target / ARC / ARCRegisterInfo.cpp
bloba7f89b385ffe9a0016fb9b7d8f3f3d0b7c16c530
1 //===- ARCRegisterInfo.cpp - ARC Register Information -----------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains the ARC implementation of the MRegisterInfo class.
11 //===----------------------------------------------------------------------===//
13 #include "ARCRegisterInfo.h"
14 #include "ARC.h"
15 #include "ARCInstrInfo.h"
16 #include "ARCMachineFunctionInfo.h"
17 #include "ARCSubtarget.h"
18 #include "llvm/ADT/BitVector.h"
19 #include "llvm/CodeGen/MachineFrameInfo.h"
20 #include "llvm/CodeGen/MachineFunction.h"
21 #include "llvm/CodeGen/MachineInstrBuilder.h"
22 #include "llvm/CodeGen/MachineModuleInfo.h"
23 #include "llvm/CodeGen/MachineRegisterInfo.h"
24 #include "llvm/CodeGen/RegisterScavenging.h"
25 #include "llvm/IR/Function.h"
26 #include "llvm/Support/Debug.h"
27 #include "llvm/CodeGen/TargetFrameLowering.h"
28 #include "llvm/Target/TargetMachine.h"
29 #include "llvm/Target/TargetOptions.h"
31 using namespace llvm;
33 #define DEBUG_TYPE "arc-reg-info"
35 #define GET_REGINFO_TARGET_DESC
36 #include "ARCGenRegisterInfo.inc"
38 static void ReplaceFrameIndex(MachineBasicBlock::iterator II,
39 const ARCInstrInfo &TII, unsigned Reg,
40 unsigned FrameReg, int Offset, int StackSize,
41 int ObjSize, RegScavenger *RS, int SPAdj) {
42 assert(RS && "Need register scavenger.");
43 MachineInstr &MI = *II;
44 MachineBasicBlock &MBB = *MI.getParent();
45 DebugLoc dl = MI.getDebugLoc();
46 unsigned BaseReg = FrameReg;
47 unsigned KillState = 0;
48 if (MI.getOpcode() == ARC::LD_rs9 && (Offset >= 256 || Offset < -256)) {
49 // Loads can always be reached with LD_rlimm.
50 BuildMI(MBB, II, dl, TII.get(ARC::LD_rlimm), Reg)
51 .addReg(BaseReg)
52 .addImm(Offset)
53 .addMemOperand(*MI.memoperands_begin());
54 MBB.erase(II);
55 return;
58 if (MI.getOpcode() != ARC::GETFI && (Offset >= 256 || Offset < -256)) {
59 // We need to use a scratch register to reach the far-away frame indexes.
60 BaseReg = RS->FindUnusedReg(&ARC::GPR32RegClass);
61 if (!BaseReg) {
62 // We can be sure that the scavenged-register slot is within the range
63 // of the load offset.
64 const TargetRegisterInfo *TRI =
65 MBB.getParent()->getSubtarget().getRegisterInfo();
66 BaseReg = RS->scavengeRegister(&ARC::GPR32RegClass, II, SPAdj);
67 assert(BaseReg && "Register scavenging failed.");
68 LLVM_DEBUG(dbgs() << "Scavenged register " << printReg(BaseReg, TRI)
69 << " for FrameReg=" << printReg(FrameReg, TRI)
70 << "+Offset=" << Offset << "\n");
71 (void)TRI;
72 RS->setRegUsed(BaseReg);
74 unsigned AddOpc = isUInt<6>(Offset) ? ARC::ADD_rru6 : ARC::ADD_rrlimm;
75 BuildMI(MBB, II, dl, TII.get(AddOpc))
76 .addReg(BaseReg, RegState::Define)
77 .addReg(FrameReg)
78 .addImm(Offset);
79 Offset = 0;
80 KillState = RegState::Kill;
82 switch (MI.getOpcode()) {
83 case ARC::LD_rs9:
84 assert((Offset % 4 == 0) && "LD needs 4 byte alignment.");
85 LLVM_FALLTHROUGH;
86 case ARC::LDH_rs9:
87 case ARC::LDH_X_rs9:
88 assert((Offset % 2 == 0) && "LDH needs 2 byte alignment.");
89 LLVM_FALLTHROUGH;
90 case ARC::LDB_rs9:
91 case ARC::LDB_X_rs9:
92 LLVM_DEBUG(dbgs() << "Building LDFI\n");
93 BuildMI(MBB, II, dl, TII.get(MI.getOpcode()), Reg)
94 .addReg(BaseReg, KillState)
95 .addImm(Offset)
96 .addMemOperand(*MI.memoperands_begin());
97 break;
98 case ARC::ST_rs9:
99 assert((Offset % 4 == 0) && "ST needs 4 byte alignment.");
100 LLVM_FALLTHROUGH;
101 case ARC::STH_rs9:
102 assert((Offset % 2 == 0) && "STH needs 2 byte alignment.");
103 LLVM_FALLTHROUGH;
104 case ARC::STB_rs9:
105 LLVM_DEBUG(dbgs() << "Building STFI\n");
106 BuildMI(MBB, II, dl, TII.get(MI.getOpcode()))
107 .addReg(Reg, getKillRegState(MI.getOperand(0).isKill()))
108 .addReg(BaseReg, KillState)
109 .addImm(Offset)
110 .addMemOperand(*MI.memoperands_begin());
111 break;
112 case ARC::GETFI:
113 LLVM_DEBUG(dbgs() << "Building GETFI\n");
114 BuildMI(MBB, II, dl,
115 TII.get(isUInt<6>(Offset) ? ARC::ADD_rru6 : ARC::ADD_rrlimm))
116 .addReg(Reg, RegState::Define)
117 .addReg(FrameReg)
118 .addImm(Offset);
119 break;
120 default:
121 llvm_unreachable("Unhandled opcode.");
124 // Erase old instruction.
125 MBB.erase(II);
128 ARCRegisterInfo::ARCRegisterInfo() : ARCGenRegisterInfo(ARC::BLINK) {}
130 bool ARCRegisterInfo::needsFrameMoves(const MachineFunction &MF) {
131 return MF.getMMI().hasDebugInfo() || MF.getFunction().needsUnwindTableEntry();
134 const MCPhysReg *
135 ARCRegisterInfo::getCalleeSavedRegs(const MachineFunction *MF) const {
136 return CSR_ARC_SaveList;
139 BitVector ARCRegisterInfo::getReservedRegs(const MachineFunction &MF) const {
140 BitVector Reserved(getNumRegs());
142 Reserved.set(ARC::ILINK);
143 Reserved.set(ARC::SP);
144 Reserved.set(ARC::GP);
145 Reserved.set(ARC::R25);
146 Reserved.set(ARC::BLINK);
147 Reserved.set(ARC::FP);
148 return Reserved;
151 bool ARCRegisterInfo::requiresRegisterScavenging(
152 const MachineFunction &MF) const {
153 return true;
156 bool ARCRegisterInfo::trackLivenessAfterRegAlloc(
157 const MachineFunction &MF) const {
158 return true;
161 bool ARCRegisterInfo::useFPForScavengingIndex(const MachineFunction &MF) const {
162 return true;
165 void ARCRegisterInfo::eliminateFrameIndex(MachineBasicBlock::iterator II,
166 int SPAdj, unsigned FIOperandNum,
167 RegScavenger *RS) const {
168 assert(SPAdj == 0 && "Unexpected");
169 MachineInstr &MI = *II;
170 MachineOperand &FrameOp = MI.getOperand(FIOperandNum);
171 int FrameIndex = FrameOp.getIndex();
173 MachineFunction &MF = *MI.getParent()->getParent();
174 const ARCInstrInfo &TII = *MF.getSubtarget<ARCSubtarget>().getInstrInfo();
175 const ARCFrameLowering *TFI = getFrameLowering(MF);
176 int Offset = MF.getFrameInfo().getObjectOffset(FrameIndex);
177 int ObjSize = MF.getFrameInfo().getObjectSize(FrameIndex);
178 int StackSize = MF.getFrameInfo().getStackSize();
179 int LocalFrameSize = MF.getFrameInfo().getLocalFrameSize();
181 LLVM_DEBUG(dbgs() << "\nFunction : " << MF.getName() << "\n");
182 LLVM_DEBUG(dbgs() << "<--------->\n");
183 LLVM_DEBUG(dbgs() << MI << "\n");
184 LLVM_DEBUG(dbgs() << "FrameIndex : " << FrameIndex << "\n");
185 LLVM_DEBUG(dbgs() << "ObjSize : " << ObjSize << "\n");
186 LLVM_DEBUG(dbgs() << "FrameOffset : " << Offset << "\n");
187 LLVM_DEBUG(dbgs() << "StackSize : " << StackSize << "\n");
188 LLVM_DEBUG(dbgs() << "LocalFrameSize : " << LocalFrameSize << "\n");
189 (void)LocalFrameSize;
191 // Special handling of DBG_VALUE instructions.
192 if (MI.isDebugValue()) {
193 Register FrameReg = getFrameRegister(MF);
194 MI.getOperand(FIOperandNum).ChangeToRegister(FrameReg, false /*isDef*/);
195 MI.getOperand(FIOperandNum + 1).ChangeToImmediate(Offset);
196 return;
199 // fold constant into offset.
200 Offset += MI.getOperand(FIOperandNum + 1).getImm();
202 // TODO: assert based on the load type:
203 // ldb needs no alignment,
204 // ldh needs 2 byte alignment
205 // ld needs 4 byte alignment
206 LLVM_DEBUG(dbgs() << "Offset : " << Offset << "\n"
207 << "<--------->\n");
209 Register Reg = MI.getOperand(0).getReg();
210 assert(ARC::GPR32RegClass.contains(Reg) && "Unexpected register operand");
212 if (!TFI->hasFP(MF)) {
213 Offset = StackSize + Offset;
214 if (FrameIndex >= 0)
215 assert((Offset >= 0 && Offset < StackSize) && "SP Offset not in bounds.");
216 } else {
217 if (FrameIndex >= 0) {
218 assert((Offset < 0 && -Offset <= StackSize) &&
219 "FP Offset not in bounds.");
222 ReplaceFrameIndex(II, TII, Reg, getFrameRegister(MF), Offset, StackSize,
223 ObjSize, RS, SPAdj);
226 Register ARCRegisterInfo::getFrameRegister(const MachineFunction &MF) const {
227 const ARCFrameLowering *TFI = getFrameLowering(MF);
228 return TFI->hasFP(MF) ? ARC::FP : ARC::SP;
231 const uint32_t *
232 ARCRegisterInfo::getCallPreservedMask(const MachineFunction &MF,
233 CallingConv::ID CC) const {
234 return CSR_ARC_RegMask;