blob: 94333d34454aa5df1cc250ddc643f351a2b61d09 [file] [log] [blame]
Dan Gohman94b8d7e2008-09-03 16:01:59 +00001//===---- ScheduleDAGEmit.cpp - Emit routines for the ScheduleDAG class ---===//
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 implements the Emit routines for the ScheduleDAG class, which creates
11// MachineInstrs according to the computed schedule.
12//
13//===----------------------------------------------------------------------===//
14
15#define DEBUG_TYPE "pre-RA-sched"
Dan Gohman84fbac52009-02-06 17:22:58 +000016#include "ScheduleDAGSDNodes.h"
Dan Gohman94b8d7e2008-09-03 16:01:59 +000017#include "llvm/CodeGen/MachineConstantPool.h"
18#include "llvm/CodeGen/MachineFunction.h"
19#include "llvm/CodeGen/MachineInstrBuilder.h"
20#include "llvm/CodeGen/MachineRegisterInfo.h"
21#include "llvm/Target/TargetData.h"
22#include "llvm/Target/TargetMachine.h"
23#include "llvm/Target/TargetInstrInfo.h"
24#include "llvm/Target/TargetLowering.h"
25#include "llvm/ADT/Statistic.h"
26#include "llvm/Support/CommandLine.h"
27#include "llvm/Support/Debug.h"
28#include "llvm/Support/MathExtras.h"
29using namespace llvm;
30
Dan Gohman94b8d7e2008-09-03 16:01:59 +000031/// getInstrOperandRegClass - Return register class of the operand of an
32/// instruction of the specified TargetInstrDesc.
33static const TargetRegisterClass*
34getInstrOperandRegClass(const TargetRegisterInfo *TRI,
Evan Cheng770bcc72009-02-06 17:43:24 +000035 const TargetInstrDesc &II, unsigned Op) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +000036 if (Op >= II.getNumOperands()) {
37 assert(II.isVariadic() && "Invalid operand # of instruction");
38 return NULL;
39 }
40 if (II.OpInfo[Op].isLookupPtrRegClass())
Evan Cheng770bcc72009-02-06 17:43:24 +000041 return TRI->getPointerRegClass();
Dan Gohman94b8d7e2008-09-03 16:01:59 +000042 return TRI->getRegClass(II.OpInfo[Op].RegClass);
43}
44
45/// EmitCopyFromReg - Generate machine code for an CopyFromReg node or an
46/// implicit physical register output.
Dan Gohman343f0c02008-11-19 23:18:57 +000047void ScheduleDAGSDNodes::EmitCopyFromReg(SDNode *Node, unsigned ResNo,
Evan Chenge57187c2009-01-16 20:57:18 +000048 bool IsClone, bool IsCloned,
49 unsigned SrcReg,
Dan Gohman343f0c02008-11-19 23:18:57 +000050 DenseMap<SDValue, unsigned> &VRBaseMap) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +000051 unsigned VRBase = 0;
52 if (TargetRegisterInfo::isVirtualRegister(SrcReg)) {
53 // Just use the input register directly!
54 SDValue Op(Node, ResNo);
55 if (IsClone)
56 VRBaseMap.erase(Op);
57 bool isNew = VRBaseMap.insert(std::make_pair(Op, SrcReg)).second;
58 isNew = isNew; // Silence compiler warning.
59 assert(isNew && "Node emitted out of order - early");
60 return;
61 }
62
63 // If the node is only used by a CopyToReg and the dest reg is a vreg, use
64 // the CopyToReg'd destination register instead of creating a new vreg.
65 bool MatchReg = true;
Evan Cheng1cd33272008-09-16 23:12:11 +000066 const TargetRegisterClass *UseRC = NULL;
Evan Chenge57187c2009-01-16 20:57:18 +000067 if (!IsClone && !IsCloned)
68 for (SDNode::use_iterator UI = Node->use_begin(), E = Node->use_end();
69 UI != E; ++UI) {
70 SDNode *User = *UI;
71 bool Match = true;
72 if (User->getOpcode() == ISD::CopyToReg &&
73 User->getOperand(2).getNode() == Node &&
74 User->getOperand(2).getResNo() == ResNo) {
75 unsigned DestReg = cast<RegisterSDNode>(User->getOperand(1))->getReg();
76 if (TargetRegisterInfo::isVirtualRegister(DestReg)) {
77 VRBase = DestReg;
78 Match = false;
79 } else if (DestReg != SrcReg)
80 Match = false;
81 } else {
82 for (unsigned i = 0, e = User->getNumOperands(); i != e; ++i) {
83 SDValue Op = User->getOperand(i);
84 if (Op.getNode() != Node || Op.getResNo() != ResNo)
85 continue;
86 MVT VT = Node->getValueType(Op.getResNo());
87 if (VT == MVT::Other || VT == MVT::Flag)
88 continue;
89 Match = false;
90 if (User->isMachineOpcode()) {
91 const TargetInstrDesc &II = TII->get(User->getMachineOpcode());
92 const TargetRegisterClass *RC =
Evan Cheng770bcc72009-02-06 17:43:24 +000093 getInstrOperandRegClass(TRI, II, i+II.getNumDefs());
Evan Chenge57187c2009-01-16 20:57:18 +000094 if (!UseRC)
95 UseRC = RC;
Dan Gohmanf8c73942009-04-13 15:38:05 +000096 else if (RC) {
97 if (UseRC->hasSuperClass(RC))
98 UseRC = RC;
99 else
100 assert((UseRC == RC || RC->hasSuperClass(UseRC)) &&
101 "Multiple uses expecting different register classes!");
102 }
Evan Chenge57187c2009-01-16 20:57:18 +0000103 }
Evan Cheng1cd33272008-09-16 23:12:11 +0000104 }
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000105 }
Evan Chenge57187c2009-01-16 20:57:18 +0000106 MatchReg &= Match;
107 if (VRBase)
108 break;
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000109 }
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000110
Evan Cheng1cd33272008-09-16 23:12:11 +0000111 MVT VT = Node->getValueType(ResNo);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000112 const TargetRegisterClass *SrcRC = 0, *DstRC = 0;
Evan Cheng1cd33272008-09-16 23:12:11 +0000113 SrcRC = TRI->getPhysicalRegisterRegClass(SrcReg, VT);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000114
115 // Figure out the register class to create for the destreg.
116 if (VRBase) {
117 DstRC = MRI.getRegClass(VRBase);
Evan Cheng1cd33272008-09-16 23:12:11 +0000118 } else if (UseRC) {
119 assert(UseRC->hasType(VT) && "Incompatible phys register def and uses!");
120 DstRC = UseRC;
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000121 } else {
Evan Cheng1cd33272008-09-16 23:12:11 +0000122 DstRC = TLI->getRegClassFor(VT);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000123 }
124
125 // If all uses are reading from the src physical register and copying the
126 // register is either impossible or very expensive, then don't create a copy.
127 if (MatchReg && SrcRC->getCopyCost() < 0) {
128 VRBase = SrcReg;
129 } else {
130 // Create the reg, emit the copy.
131 VRBase = MRI.createVirtualRegister(DstRC);
Dan Gohman47ac0f02009-02-11 04:27:20 +0000132 bool Emitted = TII->copyRegToReg(*BB, InsertPos, VRBase, SrcReg,
133 DstRC, SrcRC);
Dan Gohmanf8c73942009-04-13 15:38:05 +0000134 // If the target didn't handle the copy with different register
135 // classes and the destination is a subset of the source,
136 // try a normal same-RC copy.
137 if (!Emitted && DstRC->hasSuperClass(SrcRC))
138 Emitted = TII->copyRegToReg(*BB, InsertPos, VRBase, SrcReg,
139 SrcRC, SrcRC);
140
141 assert(Emitted && "Unable to issue a copy instruction!\n");
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000142 }
143
144 SDValue Op(Node, ResNo);
145 if (IsClone)
146 VRBaseMap.erase(Op);
147 bool isNew = VRBaseMap.insert(std::make_pair(Op, VRBase)).second;
148 isNew = isNew; // Silence compiler warning.
149 assert(isNew && "Node emitted out of order - early");
150}
151
152/// getDstOfCopyToRegUse - If the only use of the specified result number of
153/// node is a CopyToReg, return its destination register. Return 0 otherwise.
Dan Gohman343f0c02008-11-19 23:18:57 +0000154unsigned ScheduleDAGSDNodes::getDstOfOnlyCopyToRegUse(SDNode *Node,
155 unsigned ResNo) const {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000156 if (!Node->hasOneUse())
157 return 0;
158
159 SDNode *User = *Node->use_begin();
160 if (User->getOpcode() == ISD::CopyToReg &&
161 User->getOperand(2).getNode() == Node &&
162 User->getOperand(2).getResNo() == ResNo) {
163 unsigned Reg = cast<RegisterSDNode>(User->getOperand(1))->getReg();
164 if (TargetRegisterInfo::isVirtualRegister(Reg))
165 return Reg;
166 }
167 return 0;
168}
169
Dan Gohman343f0c02008-11-19 23:18:57 +0000170void ScheduleDAGSDNodes::CreateVirtualRegisters(SDNode *Node, MachineInstr *MI,
Evan Chenge57187c2009-01-16 20:57:18 +0000171 const TargetInstrDesc &II,
172 bool IsClone, bool IsCloned,
Evan Cheng5c3c5a42009-01-09 22:44:02 +0000173 DenseMap<SDValue, unsigned> &VRBaseMap) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000174 assert(Node->getMachineOpcode() != TargetInstrInfo::IMPLICIT_DEF &&
175 "IMPLICIT_DEF should have been handled as a special case elsewhere!");
176
177 for (unsigned i = 0; i < II.getNumDefs(); ++i) {
178 // If the specific node value is only used by a CopyToReg and the dest reg
Dan Gohmanf8c73942009-04-13 15:38:05 +0000179 // is a vreg in the same register class, use the CopyToReg'd destination
180 // register instead of creating a new vreg.
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000181 unsigned VRBase = 0;
Dan Gohmanf8c73942009-04-13 15:38:05 +0000182 const TargetRegisterClass *RC = getInstrOperandRegClass(TRI, II, i);
Evan Chenge57187c2009-01-16 20:57:18 +0000183
184 if (!IsClone && !IsCloned)
185 for (SDNode::use_iterator UI = Node->use_begin(), E = Node->use_end();
186 UI != E; ++UI) {
187 SDNode *User = *UI;
188 if (User->getOpcode() == ISD::CopyToReg &&
189 User->getOperand(2).getNode() == Node &&
190 User->getOperand(2).getResNo() == i) {
191 unsigned Reg = cast<RegisterSDNode>(User->getOperand(1))->getReg();
192 if (TargetRegisterInfo::isVirtualRegister(Reg)) {
Dan Gohmanf8c73942009-04-13 15:38:05 +0000193 const TargetRegisterClass *RegRC = MRI.getRegClass(Reg);
194 if (RegRC == RC) {
195 VRBase = Reg;
196 MI->addOperand(MachineOperand::CreateReg(Reg, true));
197 break;
198 }
Evan Chenge57187c2009-01-16 20:57:18 +0000199 }
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000200 }
201 }
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000202
203 // Create the result registers for this node and add the result regs to
204 // the machine instruction.
205 if (VRBase == 0) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000206 assert(RC && "Isn't a register operand!");
207 VRBase = MRI.createVirtualRegister(RC);
208 MI->addOperand(MachineOperand::CreateReg(VRBase, true));
209 }
210
211 SDValue Op(Node, i);
Evan Cheng5c3c5a42009-01-09 22:44:02 +0000212 if (IsClone)
213 VRBaseMap.erase(Op);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000214 bool isNew = VRBaseMap.insert(std::make_pair(Op, VRBase)).second;
215 isNew = isNew; // Silence compiler warning.
216 assert(isNew && "Node emitted out of order - early");
217 }
218}
219
220/// getVR - Return the virtual register corresponding to the specified result
221/// of the specified node.
Dan Gohman343f0c02008-11-19 23:18:57 +0000222unsigned ScheduleDAGSDNodes::getVR(SDValue Op,
223 DenseMap<SDValue, unsigned> &VRBaseMap) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000224 if (Op.isMachineOpcode() &&
225 Op.getMachineOpcode() == TargetInstrInfo::IMPLICIT_DEF) {
226 // Add an IMPLICIT_DEF instruction before every use.
227 unsigned VReg = getDstOfOnlyCopyToRegUse(Op.getNode(), Op.getResNo());
228 // IMPLICIT_DEF can produce any type of result so its TargetInstrDesc
229 // does not include operand register class info.
230 if (!VReg) {
231 const TargetRegisterClass *RC = TLI->getRegClassFor(Op.getValueType());
232 VReg = MRI.createVirtualRegister(RC);
233 }
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000234 BuildMI(BB, Op.getDebugLoc(), TII->get(TargetInstrInfo::IMPLICIT_DEF),VReg);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000235 return VReg;
236 }
237
238 DenseMap<SDValue, unsigned>::iterator I = VRBaseMap.find(Op);
239 assert(I != VRBaseMap.end() && "Node emitted out of order - late");
240 return I->second;
241}
242
243
Dan Gohmanf8c73942009-04-13 15:38:05 +0000244/// AddRegisterOperand - Add the specified register as an operand to the
245/// specified machine instr. Insert register copies if the register is
246/// not in the required register class.
247void
248ScheduleDAGSDNodes::AddRegisterOperand(MachineInstr *MI, SDValue Op,
249 unsigned IIOpNum,
250 const TargetInstrDesc *II,
251 DenseMap<SDValue, unsigned> &VRBaseMap) {
252 assert(Op.getValueType() != MVT::Other &&
253 Op.getValueType() != MVT::Flag &&
254 "Chain and flag operands should occur at end of operand list!");
255 // Get/emit the operand.
256 unsigned VReg = getVR(Op, VRBaseMap);
257 assert(TargetRegisterInfo::isVirtualRegister(VReg) && "Not a vreg?");
258
259 const TargetInstrDesc &TID = MI->getDesc();
260 bool isOptDef = IIOpNum < TID.getNumOperands() &&
261 TID.OpInfo[IIOpNum].isOptionalDef();
262
263 // If the instruction requires a register in a different class, create
264 // a new virtual register and copy the value into it.
265 if (II) {
266 const TargetRegisterClass *SrcRC =
267 MRI.getRegClass(VReg);
268 const TargetRegisterClass *DstRC =
269 getInstrOperandRegClass(TRI, *II, IIOpNum);
270 assert((DstRC || (TID.isVariadic() && IIOpNum >= TID.getNumOperands())) &&
271 "Don't have operand info for this instruction!");
272 if (DstRC && SrcRC != DstRC && !SrcRC->hasSuperClass(DstRC)) {
273 unsigned NewVReg = MRI.createVirtualRegister(DstRC);
274 bool Emitted = TII->copyRegToReg(*BB, InsertPos, NewVReg, VReg,
275 DstRC, SrcRC);
276 // If the target didn't handle the copy with different register
277 // classes and the destination is a subset of the source,
278 // try a normal same-RC copy.
279 if (!Emitted && DstRC->hasSuperClass(SrcRC))
280 Emitted = TII->copyRegToReg(*BB, InsertPos, NewVReg, VReg,
281 SrcRC, SrcRC);
282 assert(Emitted && "Unable to issue a copy instruction!\n");
283 VReg = NewVReg;
284 }
285 }
286
287 MI->addOperand(MachineOperand::CreateReg(VReg, isOptDef));
288}
289
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000290/// AddOperand - Add the specified operand to the specified machine instr. II
291/// specifies the instruction information for the node, and IIOpNum is the
292/// operand number (in the II) that we are adding. IIOpNum and II are used for
293/// assertions only.
Dan Gohman343f0c02008-11-19 23:18:57 +0000294void ScheduleDAGSDNodes::AddOperand(MachineInstr *MI, SDValue Op,
295 unsigned IIOpNum,
296 const TargetInstrDesc *II,
297 DenseMap<SDValue, unsigned> &VRBaseMap) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000298 if (Op.isMachineOpcode()) {
Dan Gohmanf8c73942009-04-13 15:38:05 +0000299 AddRegisterOperand(MI, Op, IIOpNum, II, VRBaseMap);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000300 } else if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op)) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000301 MI->addOperand(MachineOperand::CreateImm(C->getZExtValue()));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000302 } else if (ConstantFPSDNode *F = dyn_cast<ConstantFPSDNode>(Op)) {
Dan Gohman4fbd7962008-09-12 18:08:03 +0000303 const ConstantFP *CFP = F->getConstantFPValue();
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000304 MI->addOperand(MachineOperand::CreateFPImm(CFP));
305 } else if (RegisterSDNode *R = dyn_cast<RegisterSDNode>(Op)) {
Dale Johannesen86b49f82008-09-24 01:07:17 +0000306 MI->addOperand(MachineOperand::CreateReg(R->getReg(), false));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000307 } else if (GlobalAddressSDNode *TGA = dyn_cast<GlobalAddressSDNode>(Op)) {
308 MI->addOperand(MachineOperand::CreateGA(TGA->getGlobal(),TGA->getOffset()));
Dan Gohmanf8c73942009-04-13 15:38:05 +0000309 } else if (BasicBlockSDNode *BBNode = dyn_cast<BasicBlockSDNode>(Op)) {
310 MI->addOperand(MachineOperand::CreateMBB(BBNode->getBasicBlock()));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000311 } else if (FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Op)) {
312 MI->addOperand(MachineOperand::CreateFI(FI->getIndex()));
313 } else if (JumpTableSDNode *JT = dyn_cast<JumpTableSDNode>(Op)) {
314 MI->addOperand(MachineOperand::CreateJTI(JT->getIndex()));
315 } else if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(Op)) {
316 int Offset = CP->getOffset();
317 unsigned Align = CP->getAlignment();
318 const Type *Type = CP->getType();
319 // MachineConstantPool wants an explicit alignment.
320 if (Align == 0) {
Evan Cheng1606e8e2009-03-13 07:51:59 +0000321 Align = TM.getTargetData()->getPrefTypeAlignment(Type);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000322 if (Align == 0) {
323 // Alignment of vector types. FIXME!
Duncan Sandsceb4d1a2009-01-12 20:38:59 +0000324 Align = TM.getTargetData()->getTypePaddedSize(Type);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000325 }
326 }
327
328 unsigned Idx;
329 if (CP->isMachineConstantPoolEntry())
330 Idx = ConstPool->getConstantPoolIndex(CP->getMachineCPVal(), Align);
331 else
332 Idx = ConstPool->getConstantPoolIndex(CP->getConstVal(), Align);
333 MI->addOperand(MachineOperand::CreateCPI(Idx, Offset));
Bill Wendling056292f2008-09-16 21:48:12 +0000334 } else if (ExternalSymbolSDNode *ES = dyn_cast<ExternalSymbolSDNode>(Op)) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000335 MI->addOperand(MachineOperand::CreateES(ES->getSymbol()));
336 } else {
337 assert(Op.getValueType() != MVT::Other &&
338 Op.getValueType() != MVT::Flag &&
339 "Chain and flag operands should occur at end of operand list!");
Dan Gohmanf8c73942009-04-13 15:38:05 +0000340 AddRegisterOperand(MI, Op, IIOpNum, II, VRBaseMap);
341 }
342}
343
344/// getSubRegisterRegClass - Returns the register class of specified register
345/// class' "SubIdx"'th sub-register class.
346static const TargetRegisterClass*
347getSubRegisterRegClass(const TargetRegisterClass *TRC, unsigned SubIdx) {
348 // Pick the register class of the subregister
349 TargetRegisterInfo::regclass_iterator I =
350 TRC->subregclasses_begin() + SubIdx-1;
351 assert(I < TRC->subregclasses_end() &&
352 "Invalid subregister index for register class");
353 return *I;
354}
355
356/// getSuperRegisterRegClass - Returns the register class of a superreg A whose
357/// "SubIdx"'th sub-register class is the specified register class and whose
358/// type matches the specified type.
359static const TargetRegisterClass*
360getSuperRegisterRegClass(const TargetRegisterClass *TRC,
361 unsigned SubIdx, MVT VT) {
362 // Pick the register class of the superegister for this type
363 for (TargetRegisterInfo::regclass_iterator I = TRC->superregclasses_begin(),
364 E = TRC->superregclasses_end(); I != E; ++I)
365 if ((*I)->hasType(VT) && getSubRegisterRegClass(*I, SubIdx) == TRC)
366 return *I;
367 assert(false && "Couldn't find the register class");
368 return 0;
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000369}
370
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000371/// EmitSubregNode - Generate machine code for subreg nodes.
372///
Dan Gohman343f0c02008-11-19 23:18:57 +0000373void ScheduleDAGSDNodes::EmitSubregNode(SDNode *Node,
374 DenseMap<SDValue, unsigned> &VRBaseMap) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000375 unsigned VRBase = 0;
376 unsigned Opc = Node->getMachineOpcode();
377
378 // If the node is only used by a CopyToReg and the dest reg is a vreg, use
379 // the CopyToReg'd destination register instead of creating a new vreg.
380 for (SDNode::use_iterator UI = Node->use_begin(), E = Node->use_end();
381 UI != E; ++UI) {
382 SDNode *User = *UI;
383 if (User->getOpcode() == ISD::CopyToReg &&
384 User->getOperand(2).getNode() == Node) {
385 unsigned DestReg = cast<RegisterSDNode>(User->getOperand(1))->getReg();
386 if (TargetRegisterInfo::isVirtualRegister(DestReg)) {
387 VRBase = DestReg;
388 break;
389 }
390 }
391 }
392
393 if (Opc == TargetInstrInfo::EXTRACT_SUBREG) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000394 unsigned SubIdx = cast<ConstantSDNode>(Node->getOperand(1))->getZExtValue();
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000395
396 // Create the extract_subreg machine instruction.
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000397 MachineInstr *MI = BuildMI(MF, Node->getDebugLoc(),
398 TII->get(TargetInstrInfo::EXTRACT_SUBREG));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000399
400 // Figure out the register class to create for the destreg.
Dan Gohmanf8c73942009-04-13 15:38:05 +0000401 unsigned VReg = getVR(Node->getOperand(0), VRBaseMap);
402 const TargetRegisterClass *TRC = MRI.getRegClass(VReg);
403 const TargetRegisterClass *SRC = getSubRegisterRegClass(TRC, SubIdx);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000404
405 if (VRBase) {
406 // Grab the destination register
407#ifndef NDEBUG
408 const TargetRegisterClass *DRC = MRI.getRegClass(VRBase);
Dan Gohmanf8c73942009-04-13 15:38:05 +0000409 assert(SRC && DRC && (SRC == DRC || DRC->hasSubClass(SRC)) &&
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000410 "Source subregister and destination must have the same class");
411#endif
412 } else {
413 // Create the reg
414 assert(SRC && "Couldn't find source register class");
415 VRBase = MRI.createVirtualRegister(SRC);
416 }
417
418 // Add def, source, and subreg index
419 MI->addOperand(MachineOperand::CreateReg(VRBase, true));
420 AddOperand(MI, Node->getOperand(0), 0, 0, VRBaseMap);
421 MI->addOperand(MachineOperand::CreateImm(SubIdx));
Dan Gohman47ac0f02009-02-11 04:27:20 +0000422 BB->insert(InsertPos, MI);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000423 } else if (Opc == TargetInstrInfo::INSERT_SUBREG ||
424 Opc == TargetInstrInfo::SUBREG_TO_REG) {
425 SDValue N0 = Node->getOperand(0);
426 SDValue N1 = Node->getOperand(1);
427 SDValue N2 = Node->getOperand(2);
Dan Gohmanf8c73942009-04-13 15:38:05 +0000428 unsigned SubReg = getVR(N1, VRBaseMap);
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000429 unsigned SubIdx = cast<ConstantSDNode>(N2)->getZExtValue();
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000430
431
432 // Figure out the register class to create for the destreg.
433 const TargetRegisterClass *TRC = 0;
434 if (VRBase) {
435 TRC = MRI.getRegClass(VRBase);
436 } else {
Dan Gohmanf8c73942009-04-13 15:38:05 +0000437 TRC = getSuperRegisterRegClass(MRI.getRegClass(SubReg), SubIdx,
438 Node->getValueType(0));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000439 assert(TRC && "Couldn't determine register class for insert_subreg");
440 VRBase = MRI.createVirtualRegister(TRC); // Create the reg
441 }
442
443 // Create the insert_subreg or subreg_to_reg machine instruction.
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000444 MachineInstr *MI = BuildMI(MF, Node->getDebugLoc(), TII->get(Opc));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000445 MI->addOperand(MachineOperand::CreateReg(VRBase, true));
446
447 // If creating a subreg_to_reg, then the first input operand
448 // is an implicit value immediate, otherwise it's a register
449 if (Opc == TargetInstrInfo::SUBREG_TO_REG) {
450 const ConstantSDNode *SD = cast<ConstantSDNode>(N0);
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000451 MI->addOperand(MachineOperand::CreateImm(SD->getZExtValue()));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000452 } else
453 AddOperand(MI, N0, 0, 0, VRBaseMap);
454 // Add the subregster being inserted
455 AddOperand(MI, N1, 0, 0, VRBaseMap);
456 MI->addOperand(MachineOperand::CreateImm(SubIdx));
Dan Gohman47ac0f02009-02-11 04:27:20 +0000457 BB->insert(InsertPos, MI);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000458 } else
459 assert(0 && "Node is not insert_subreg, extract_subreg, or subreg_to_reg");
460
461 SDValue Op(Node, 0);
462 bool isNew = VRBaseMap.insert(std::make_pair(Op, VRBase)).second;
463 isNew = isNew; // Silence compiler warning.
464 assert(isNew && "Node emitted out of order - early");
465}
466
Dan Gohman88c7af02009-04-13 21:06:25 +0000467/// EmitCopyToRegClassNode - Generate machine code for COPY_TO_REGCLASS nodes.
468/// COPY_TO_REGCLASS is just a normal copy, except that the destination
Dan Gohmanf8c73942009-04-13 15:38:05 +0000469/// register is constrained to be in a particular register class.
470///
471void
Dan Gohman88c7af02009-04-13 21:06:25 +0000472ScheduleDAGSDNodes::EmitCopyToRegClassNode(SDNode *Node,
Dan Gohmanf8c73942009-04-13 15:38:05 +0000473 DenseMap<SDValue, unsigned> &VRBaseMap) {
474 unsigned VReg = getVR(Node->getOperand(0), VRBaseMap);
475 const TargetRegisterClass *SrcRC = MRI.getRegClass(VReg);
476
477 unsigned DstRCIdx = cast<ConstantSDNode>(Node->getOperand(1))->getZExtValue();
478 const TargetRegisterClass *DstRC = TRI->getRegClass(DstRCIdx);
479
Dan Gohmanf8c73942009-04-13 15:38:05 +0000480 // Create the new VReg in the destination class and emit a copy.
481 unsigned NewVReg = MRI.createVirtualRegister(DstRC);
482 bool Emitted = TII->copyRegToReg(*BB, InsertPos, NewVReg, VReg,
483 DstRC, SrcRC);
Dan Gohman88c7af02009-04-13 21:06:25 +0000484 // If the target didn't handle the copy with different register
485 // classes and the destination is a subset of the source,
486 // try a normal same-RC copy.
487 if (!Emitted && SrcRC->hasSubClass(DstRC))
Dan Gohmanf8c73942009-04-13 15:38:05 +0000488 Emitted = TII->copyRegToReg(*BB, InsertPos, NewVReg, VReg,
489 SrcRC, SrcRC);
490 assert(Emitted &&
Dan Gohman88c7af02009-04-13 21:06:25 +0000491 "Unable to issue a copy instruction for a COPY_TO_REGCLASS node!\n");
Dan Gohmanf8c73942009-04-13 15:38:05 +0000492
493 SDValue Op(Node, 0);
494 bool isNew = VRBaseMap.insert(std::make_pair(Op, NewVReg)).second;
495 isNew = isNew; // Silence compiler warning.
496 assert(isNew && "Node emitted out of order - early");
497}
498
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000499/// EmitNode - Generate machine code for an node and needed dependencies.
500///
Evan Chenge57187c2009-01-16 20:57:18 +0000501void ScheduleDAGSDNodes::EmitNode(SDNode *Node, bool IsClone, bool IsCloned,
Dan Gohman343f0c02008-11-19 23:18:57 +0000502 DenseMap<SDValue, unsigned> &VRBaseMap) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000503 // If machine instruction
504 if (Node->isMachineOpcode()) {
505 unsigned Opc = Node->getMachineOpcode();
506
507 // Handle subreg insert/extract specially
508 if (Opc == TargetInstrInfo::EXTRACT_SUBREG ||
509 Opc == TargetInstrInfo::INSERT_SUBREG ||
510 Opc == TargetInstrInfo::SUBREG_TO_REG) {
511 EmitSubregNode(Node, VRBaseMap);
512 return;
513 }
514
Dan Gohman88c7af02009-04-13 21:06:25 +0000515 // Handle COPY_TO_REGCLASS specially.
516 if (Opc == TargetInstrInfo::COPY_TO_REGCLASS) {
517 EmitCopyToRegClassNode(Node, VRBaseMap);
Dan Gohmanf8c73942009-04-13 15:38:05 +0000518 return;
519 }
520
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000521 if (Opc == TargetInstrInfo::IMPLICIT_DEF)
522 // We want a unique VR for each IMPLICIT_DEF use.
523 return;
524
525 const TargetInstrDesc &II = TII->get(Opc);
526 unsigned NumResults = CountResults(Node);
527 unsigned NodeOperands = CountOperands(Node);
528 unsigned MemOperandsEnd = ComputeMemOperandsEnd(Node);
529 bool HasPhysRegOuts = (NumResults > II.getNumDefs()) &&
530 II.getImplicitDefs() != 0;
531#ifndef NDEBUG
532 unsigned NumMIOperands = NodeOperands + NumResults;
533 assert((II.getNumOperands() == NumMIOperands ||
534 HasPhysRegOuts || II.isVariadic()) &&
535 "#operands for dag node doesn't match .td file!");
536#endif
537
538 // Create the new machine instruction.
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000539 MachineInstr *MI = BuildMI(MF, Node->getDebugLoc(), II);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000540
541 // Add result register values for things that are defined by this
542 // instruction.
543 if (NumResults)
Evan Chenge57187c2009-01-16 20:57:18 +0000544 CreateVirtualRegisters(Node, MI, II, IsClone, IsCloned, VRBaseMap);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000545
546 // Emit all of the actual operands of this instruction, adding them to the
547 // instruction as appropriate.
548 for (unsigned i = 0; i != NodeOperands; ++i)
549 AddOperand(MI, Node->getOperand(i), i+II.getNumDefs(), &II, VRBaseMap);
550
551 // Emit all of the memory operands of this instruction
552 for (unsigned i = NodeOperands; i != MemOperandsEnd; ++i)
553 AddMemOperand(MI, cast<MemOperandSDNode>(Node->getOperand(i))->MO);
554
Dan Gohmanf7119392009-01-16 22:10:20 +0000555 if (II.usesCustomDAGSchedInsertionHook()) {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000556 // Insert this instruction into the basic block using a target
557 // specific inserter which may returns a new basic block.
558 BB = TLI->EmitInstrWithCustomInserter(MI, BB);
Dan Gohman47ac0f02009-02-11 04:27:20 +0000559 InsertPos = BB->end();
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000560 } else {
Dan Gohman47ac0f02009-02-11 04:27:20 +0000561 BB->insert(InsertPos, MI);
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000562 }
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000563
564 // Additional results must be an physical register def.
565 if (HasPhysRegOuts) {
566 for (unsigned i = II.getNumDefs(); i < NumResults; ++i) {
567 unsigned Reg = II.getImplicitDefs()[i - II.getNumDefs()];
568 if (Node->hasAnyUseOfValue(i))
Evan Chenge57187c2009-01-16 20:57:18 +0000569 EmitCopyFromReg(Node, i, IsClone, IsCloned, Reg, VRBaseMap);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000570 }
571 }
572 return;
573 }
574
575 switch (Node->getOpcode()) {
576 default:
577#ifndef NDEBUG
Dan Gohmana23b3b82008-11-13 21:21:28 +0000578 Node->dump(DAG);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000579#endif
580 assert(0 && "This target-independent node should have been selected!");
581 break;
582 case ISD::EntryToken:
583 assert(0 && "EntryToken should have been excluded from the schedule!");
584 break;
585 case ISD::TokenFactor: // fall thru
586 break;
587 case ISD::CopyToReg: {
588 unsigned SrcReg;
589 SDValue SrcVal = Node->getOperand(2);
590 if (RegisterSDNode *R = dyn_cast<RegisterSDNode>(SrcVal))
591 SrcReg = R->getReg();
592 else
593 SrcReg = getVR(SrcVal, VRBaseMap);
594
595 unsigned DestReg = cast<RegisterSDNode>(Node->getOperand(1))->getReg();
596 if (SrcReg == DestReg) // Coalesced away the copy? Ignore.
597 break;
598
599 const TargetRegisterClass *SrcTRC = 0, *DstTRC = 0;
600 // Get the register classes of the src/dst.
601 if (TargetRegisterInfo::isVirtualRegister(SrcReg))
602 SrcTRC = MRI.getRegClass(SrcReg);
603 else
604 SrcTRC = TRI->getPhysicalRegisterRegClass(SrcReg,SrcVal.getValueType());
605
606 if (TargetRegisterInfo::isVirtualRegister(DestReg))
607 DstTRC = MRI.getRegClass(DestReg);
608 else
609 DstTRC = TRI->getPhysicalRegisterRegClass(DestReg,
610 Node->getOperand(1).getValueType());
Dan Gohmanf8c73942009-04-13 15:38:05 +0000611
Dan Gohman47ac0f02009-02-11 04:27:20 +0000612 bool Emitted = TII->copyRegToReg(*BB, InsertPos, DestReg, SrcReg,
613 DstTRC, SrcTRC);
Dan Gohmanf8c73942009-04-13 15:38:05 +0000614 // If the target didn't handle the copy with different register
615 // classes and the destination is a subset of the source,
616 // try a normal same-RC copy.
617 if (!Emitted && DstTRC->hasSubClass(SrcTRC))
618 Emitted = TII->copyRegToReg(*BB, InsertPos, DestReg, SrcReg,
619 DstTRC, DstTRC);
620
621 assert(Emitted && "Unable to issue a copy instruction!\n");
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000622 break;
623 }
624 case ISD::CopyFromReg: {
625 unsigned SrcReg = cast<RegisterSDNode>(Node->getOperand(1))->getReg();
Evan Chenge57187c2009-01-16 20:57:18 +0000626 EmitCopyFromReg(Node, 0, IsClone, IsCloned, SrcReg, VRBaseMap);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000627 break;
628 }
629 case ISD::INLINEASM: {
630 unsigned NumOps = Node->getNumOperands();
631 if (Node->getOperand(NumOps-1).getValueType() == MVT::Flag)
632 --NumOps; // Ignore the flag operand.
633
634 // Create the inline asm machine instruction.
Bill Wendlingf2ad58d2009-02-03 01:02:39 +0000635 MachineInstr *MI = BuildMI(MF, Node->getDebugLoc(),
636 TII->get(TargetInstrInfo::INLINEASM));
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000637
638 // Add the asm string as an external symbol operand.
Bill Wendling056292f2008-09-16 21:48:12 +0000639 const char *AsmStr =
640 cast<ExternalSymbolSDNode>(Node->getOperand(1))->getSymbol();
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000641 MI->addOperand(MachineOperand::CreateES(AsmStr));
642
643 // Add all of the operand registers to the instruction.
644 for (unsigned i = 2; i != NumOps;) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000645 unsigned Flags =
646 cast<ConstantSDNode>(Node->getOperand(i))->getZExtValue();
Evan Cheng697cbbf2009-03-20 18:03:34 +0000647 unsigned NumVals = InlineAsm::getNumOperandRegisters(Flags);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000648
649 MI->addOperand(MachineOperand::CreateImm(Flags));
650 ++i; // Skip the ID value.
651
652 switch (Flags & 7) {
653 default: assert(0 && "Bad flags!");
654 case 2: // Def of register.
655 for (; NumVals; --NumVals, ++i) {
656 unsigned Reg = cast<RegisterSDNode>(Node->getOperand(i))->getReg();
657 MI->addOperand(MachineOperand::CreateReg(Reg, true));
658 }
659 break;
Dale Johannesen913d3df2008-09-12 17:49:03 +0000660 case 6: // Def of earlyclobber register.
661 for (; NumVals; --NumVals, ++i) {
662 unsigned Reg = cast<RegisterSDNode>(Node->getOperand(i))->getReg();
663 MI->addOperand(MachineOperand::CreateReg(Reg, true, false, false,
664 false, 0, true));
665 }
666 break;
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000667 case 1: // Use of register.
668 case 3: // Immediate.
669 case 4: // Addressing mode.
670 // The addressing mode has been selected, just add all of the
671 // operands to the machine instruction.
672 for (; NumVals; --NumVals, ++i)
Dale Johannesen86b49f82008-09-24 01:07:17 +0000673 AddOperand(MI, Node->getOperand(i), 0, 0, VRBaseMap);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000674 break;
675 }
676 }
Dan Gohman47ac0f02009-02-11 04:27:20 +0000677 BB->insert(InsertPos, MI);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000678 break;
679 }
680 }
681}
682
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000683/// EmitSchedule - Emit the machine code in scheduled order.
Dan Gohman343f0c02008-11-19 23:18:57 +0000684MachineBasicBlock *ScheduleDAGSDNodes::EmitSchedule() {
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000685 DenseMap<SDValue, unsigned> VRBaseMap;
686 DenseMap<SUnit*, unsigned> CopyVRBaseMap;
687 for (unsigned i = 0, e = Sequence.size(); i != e; i++) {
688 SUnit *SU = Sequence[i];
689 if (!SU) {
690 // Null SUnit* is a noop.
691 EmitNoop();
692 continue;
693 }
Dan Gohmanf449bf32008-11-14 00:06:09 +0000694
Dan Gohmanf449bf32008-11-14 00:06:09 +0000695 // For pre-regalloc scheduling, create instructions corresponding to the
696 // SDNode and any flagged SDNodes and append them to the block.
Evan Chengc29a56d2009-01-12 03:19:55 +0000697 if (!SU->getNode()) {
698 // Emit a copy.
699 EmitPhysRegCopy(SU, CopyVRBaseMap);
700 continue;
701 }
702
Dan Gohmand23e0f82008-11-13 23:24:17 +0000703 SmallVector<SDNode *, 4> FlaggedNodes;
Evan Chenge57187c2009-01-16 20:57:18 +0000704 for (SDNode *N = SU->getNode()->getFlaggedNode(); N;
705 N = N->getFlaggedNode())
Dan Gohmand23e0f82008-11-13 23:24:17 +0000706 FlaggedNodes.push_back(N);
707 while (!FlaggedNodes.empty()) {
Evan Chenge57187c2009-01-16 20:57:18 +0000708 EmitNode(FlaggedNodes.back(), SU->OrigNode != SU, SU->isCloned,VRBaseMap);
Dan Gohmand23e0f82008-11-13 23:24:17 +0000709 FlaggedNodes.pop_back();
710 }
Evan Chenge57187c2009-01-16 20:57:18 +0000711 EmitNode(SU->getNode(), SU->OrigNode != SU, SU->isCloned, VRBaseMap);
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000712 }
713
Dan Gohman94b8d7e2008-09-03 16:01:59 +0000714 return BB;
715}