1*0b57cec5SDimitry Andric //===- R600ControlFlowFinalizer.cpp - Finalize Control Flow Inst ----------===//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric //
9*0b57cec5SDimitry Andric /// \file
10*0b57cec5SDimitry Andric /// This pass compute turns all control flow pseudo instructions into native one
11*0b57cec5SDimitry Andric /// computing their address on the fly; it also sets STACK_SIZE info.
12*0b57cec5SDimitry Andric //
13*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
14*0b57cec5SDimitry Andric 
15*0b57cec5SDimitry Andric #include "MCTargetDesc/R600MCTargetDesc.h"
16*0b57cec5SDimitry Andric #include "R600.h"
17*0b57cec5SDimitry Andric #include "R600MachineFunctionInfo.h"
18*0b57cec5SDimitry Andric #include "R600Subtarget.h"
19*0b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunctionPass.h"
20*0b57cec5SDimitry Andric #include <set>
21*0b57cec5SDimitry Andric 
22*0b57cec5SDimitry Andric using namespace llvm;
23*0b57cec5SDimitry Andric 
24*0b57cec5SDimitry Andric #define DEBUG_TYPE "r600cf"
25*0b57cec5SDimitry Andric 
26*0b57cec5SDimitry Andric namespace {
27*0b57cec5SDimitry Andric 
28*0b57cec5SDimitry Andric struct CFStack {
29*0b57cec5SDimitry Andric   enum StackItem {
30*0b57cec5SDimitry Andric     ENTRY = 0,
31*0b57cec5SDimitry Andric     SUB_ENTRY = 1,
32*0b57cec5SDimitry Andric     FIRST_NON_WQM_PUSH = 2,
33*0b57cec5SDimitry Andric     FIRST_NON_WQM_PUSH_W_FULL_ENTRY = 3
34*0b57cec5SDimitry Andric   };
35*0b57cec5SDimitry Andric 
36*0b57cec5SDimitry Andric   const R600Subtarget *ST;
37*0b57cec5SDimitry Andric   std::vector<StackItem> BranchStack;
38*0b57cec5SDimitry Andric   std::vector<StackItem> LoopStack;
39*0b57cec5SDimitry Andric   unsigned MaxStackSize;
40*0b57cec5SDimitry Andric   unsigned CurrentEntries = 0;
41*0b57cec5SDimitry Andric   unsigned CurrentSubEntries = 0;
42*0b57cec5SDimitry Andric 
CFStack__anon81db2f4e0111::CFStack43*0b57cec5SDimitry Andric   CFStack(const R600Subtarget *st, CallingConv::ID cc) : ST(st),
44*0b57cec5SDimitry Andric       // We need to reserve a stack entry for CALL_FS in vertex shaders.
45*0b57cec5SDimitry Andric       MaxStackSize(cc == CallingConv::AMDGPU_VS ? 1 : 0) {}
46*0b57cec5SDimitry Andric 
47*0b57cec5SDimitry Andric   unsigned getLoopDepth();
48*0b57cec5SDimitry Andric   bool branchStackContains(CFStack::StackItem);
49*0b57cec5SDimitry Andric   bool requiresWorkAroundForInst(unsigned Opcode);
50*0b57cec5SDimitry Andric   unsigned getSubEntrySize(CFStack::StackItem Item);
51*0b57cec5SDimitry Andric   void updateMaxStackSize();
52*0b57cec5SDimitry Andric   void pushBranch(unsigned Opcode, bool isWQM = false);
53*0b57cec5SDimitry Andric   void pushLoop();
54*0b57cec5SDimitry Andric   void popBranch();
55*0b57cec5SDimitry Andric   void popLoop();
56*0b57cec5SDimitry Andric };
57*0b57cec5SDimitry Andric 
getLoopDepth()58*0b57cec5SDimitry Andric unsigned CFStack::getLoopDepth() {
59*0b57cec5SDimitry Andric   return LoopStack.size();
60*0b57cec5SDimitry Andric }
61*0b57cec5SDimitry Andric 
branchStackContains(CFStack::StackItem Item)62*0b57cec5SDimitry Andric bool CFStack::branchStackContains(CFStack::StackItem Item) {
63*0b57cec5SDimitry Andric   return llvm::is_contained(BranchStack, Item);
64*0b57cec5SDimitry Andric }
65*0b57cec5SDimitry Andric 
requiresWorkAroundForInst(unsigned Opcode)66*0b57cec5SDimitry Andric bool CFStack::requiresWorkAroundForInst(unsigned Opcode) {
67*0b57cec5SDimitry Andric   if (Opcode == R600::CF_ALU_PUSH_BEFORE && ST->hasCaymanISA() &&
68*0b57cec5SDimitry Andric       getLoopDepth() > 1)
69*0b57cec5SDimitry Andric     return true;
70*0b57cec5SDimitry Andric 
71*0b57cec5SDimitry Andric   if (!ST->hasCFAluBug())
72*0b57cec5SDimitry Andric     return false;
73*0b57cec5SDimitry Andric 
74*0b57cec5SDimitry Andric   switch(Opcode) {
75*0b57cec5SDimitry Andric   default: return false;
76*0b57cec5SDimitry Andric   case R600::CF_ALU_PUSH_BEFORE:
77*0b57cec5SDimitry Andric   case R600::CF_ALU_ELSE_AFTER:
78*0b57cec5SDimitry Andric   case R600::CF_ALU_BREAK:
79*0b57cec5SDimitry Andric   case R600::CF_ALU_CONTINUE:
80*0b57cec5SDimitry Andric     if (CurrentSubEntries == 0)
81*0b57cec5SDimitry Andric       return false;
82*0b57cec5SDimitry Andric     if (ST->getWavefrontSize() == 64) {
83*0b57cec5SDimitry Andric       // We are being conservative here.  We only require this work-around if
84*0b57cec5SDimitry Andric       // CurrentSubEntries > 3 &&
85*0b57cec5SDimitry Andric       // (CurrentSubEntries % 4 == 3 || CurrentSubEntries % 4 == 0)
86*0b57cec5SDimitry Andric       //
87*0b57cec5SDimitry Andric       // We have to be conservative, because we don't know for certain that
88*0b57cec5SDimitry Andric       // our stack allocation algorithm for Evergreen/NI is correct.  Applying this
89*0b57cec5SDimitry Andric       // work-around when CurrentSubEntries > 3 allows us to over-allocate stack
90*0b57cec5SDimitry Andric       // resources without any problems.
91*0b57cec5SDimitry Andric       return CurrentSubEntries > 3;
92*0b57cec5SDimitry Andric     } else {
93*0b57cec5SDimitry Andric       assert(ST->getWavefrontSize() == 32);
94*0b57cec5SDimitry Andric       // We are being conservative here.  We only require the work-around if
95*0b57cec5SDimitry Andric       // CurrentSubEntries > 7 &&
96*0b57cec5SDimitry Andric       // (CurrentSubEntries % 8 == 7 || CurrentSubEntries % 8 == 0)
97*0b57cec5SDimitry Andric       // See the comment on the wavefront size == 64 case for why we are
98*0b57cec5SDimitry Andric       // being conservative.
99*0b57cec5SDimitry Andric       return CurrentSubEntries > 7;
100*0b57cec5SDimitry Andric     }
101*0b57cec5SDimitry Andric   }
102*0b57cec5SDimitry Andric }
103*0b57cec5SDimitry Andric 
getSubEntrySize(CFStack::StackItem Item)104*0b57cec5SDimitry Andric unsigned CFStack::getSubEntrySize(CFStack::StackItem Item) {
105*0b57cec5SDimitry Andric   switch(Item) {
106*0b57cec5SDimitry Andric   default:
107*0b57cec5SDimitry Andric     return 0;
108*0b57cec5SDimitry Andric   case CFStack::FIRST_NON_WQM_PUSH:
109*0b57cec5SDimitry Andric   assert(!ST->hasCaymanISA());
110*0b57cec5SDimitry Andric   if (ST->getGeneration() <= AMDGPUSubtarget::R700) {
111*0b57cec5SDimitry Andric     // +1 For the push operation.
112*0b57cec5SDimitry Andric     // +2 Extra space required.
113*0b57cec5SDimitry Andric     return 3;
114*0b57cec5SDimitry Andric   } else {
115*0b57cec5SDimitry Andric     // Some documentation says that this is not necessary on Evergreen,
116*0b57cec5SDimitry Andric     // but experimentation has show that we need to allocate 1 extra
117*0b57cec5SDimitry Andric     // sub-entry for the first non-WQM push.
118*0b57cec5SDimitry Andric     // +1 For the push operation.
119*0b57cec5SDimitry Andric     // +1 Extra space required.
120*0b57cec5SDimitry Andric     return 2;
121*0b57cec5SDimitry Andric   }
122*0b57cec5SDimitry Andric   case CFStack::FIRST_NON_WQM_PUSH_W_FULL_ENTRY:
123*0b57cec5SDimitry Andric     assert(ST->getGeneration() >= AMDGPUSubtarget::EVERGREEN);
124*0b57cec5SDimitry Andric     // +1 For the push operation.
125*0b57cec5SDimitry Andric     // +1 Extra space required.
126*0b57cec5SDimitry Andric     return 2;
127*0b57cec5SDimitry Andric   case CFStack::SUB_ENTRY:
128*0b57cec5SDimitry Andric     return 1;
129*0b57cec5SDimitry Andric   }
130*0b57cec5SDimitry Andric }
131*0b57cec5SDimitry Andric 
updateMaxStackSize()132*0b57cec5SDimitry Andric void CFStack::updateMaxStackSize() {
133*0b57cec5SDimitry Andric   unsigned CurrentStackSize = CurrentEntries + divideCeil(CurrentSubEntries, 4);
134*0b57cec5SDimitry Andric   MaxStackSize = std::max(CurrentStackSize, MaxStackSize);
135*0b57cec5SDimitry Andric }
136*0b57cec5SDimitry Andric 
pushBranch(unsigned Opcode,bool isWQM)137*0b57cec5SDimitry Andric void CFStack::pushBranch(unsigned Opcode, bool isWQM) {
138*0b57cec5SDimitry Andric   CFStack::StackItem Item = CFStack::ENTRY;
139*0b57cec5SDimitry Andric   switch(Opcode) {
140*0b57cec5SDimitry Andric   case R600::CF_PUSH_EG:
141*0b57cec5SDimitry Andric   case R600::CF_ALU_PUSH_BEFORE:
142*0b57cec5SDimitry Andric     if (!isWQM) {
143*0b57cec5SDimitry Andric       if (!ST->hasCaymanISA() &&
144*0b57cec5SDimitry Andric           !branchStackContains(CFStack::FIRST_NON_WQM_PUSH))
145*0b57cec5SDimitry Andric         Item = CFStack::FIRST_NON_WQM_PUSH;  // May not be required on Evergreen/NI
146*0b57cec5SDimitry Andric                                              // See comment in
147*0b57cec5SDimitry Andric                                              // CFStack::getSubEntrySize()
148*0b57cec5SDimitry Andric       else if (CurrentEntries > 0 &&
149*0b57cec5SDimitry Andric                ST->getGeneration() > AMDGPUSubtarget::EVERGREEN &&
150*0b57cec5SDimitry Andric                !ST->hasCaymanISA() &&
151*0b57cec5SDimitry Andric                !branchStackContains(CFStack::FIRST_NON_WQM_PUSH_W_FULL_ENTRY))
152*0b57cec5SDimitry Andric         Item = CFStack::FIRST_NON_WQM_PUSH_W_FULL_ENTRY;
153*0b57cec5SDimitry Andric       else
154*0b57cec5SDimitry Andric         Item = CFStack::SUB_ENTRY;
155*0b57cec5SDimitry Andric     } else
156*0b57cec5SDimitry Andric       Item = CFStack::ENTRY;
157*0b57cec5SDimitry Andric     break;
158*0b57cec5SDimitry Andric   }
159*0b57cec5SDimitry Andric   BranchStack.push_back(Item);
160*0b57cec5SDimitry Andric   if (Item == CFStack::ENTRY)
161*0b57cec5SDimitry Andric     CurrentEntries++;
162*0b57cec5SDimitry Andric   else
163*0b57cec5SDimitry Andric     CurrentSubEntries += getSubEntrySize(Item);
164*0b57cec5SDimitry Andric   updateMaxStackSize();
165*0b57cec5SDimitry Andric }
166*0b57cec5SDimitry Andric 
pushLoop()167*0b57cec5SDimitry Andric void CFStack::pushLoop() {
168*0b57cec5SDimitry Andric   LoopStack.push_back(CFStack::ENTRY);
169*0b57cec5SDimitry Andric   CurrentEntries++;
170*0b57cec5SDimitry Andric   updateMaxStackSize();
171*0b57cec5SDimitry Andric }
172*0b57cec5SDimitry Andric 
popBranch()173*0b57cec5SDimitry Andric void CFStack::popBranch() {
174*0b57cec5SDimitry Andric   CFStack::StackItem Top = BranchStack.back();
175*0b57cec5SDimitry Andric   if (Top == CFStack::ENTRY)
176*0b57cec5SDimitry Andric     CurrentEntries--;
177*0b57cec5SDimitry Andric   else
178*0b57cec5SDimitry Andric     CurrentSubEntries-= getSubEntrySize(Top);
179*0b57cec5SDimitry Andric   BranchStack.pop_back();
180*0b57cec5SDimitry Andric }
181*0b57cec5SDimitry Andric 
popLoop()182*0b57cec5SDimitry Andric void CFStack::popLoop() {
183*0b57cec5SDimitry Andric   CurrentEntries--;
184*0b57cec5SDimitry Andric   LoopStack.pop_back();
185*0b57cec5SDimitry Andric }
186*0b57cec5SDimitry Andric 
187*0b57cec5SDimitry Andric class R600ControlFlowFinalizer : public MachineFunctionPass {
188*0b57cec5SDimitry Andric private:
189*0b57cec5SDimitry Andric   using ClauseFile = std::pair<MachineInstr *, std::vector<MachineInstr *>>;
190*0b57cec5SDimitry Andric 
191*0b57cec5SDimitry Andric   enum ControlFlowInstruction {
192*0b57cec5SDimitry Andric     CF_TC,
193*0b57cec5SDimitry Andric     CF_VC,
194*0b57cec5SDimitry Andric     CF_CALL_FS,
195*0b57cec5SDimitry Andric     CF_WHILE_LOOP,
196*0b57cec5SDimitry Andric     CF_END_LOOP,
197*0b57cec5SDimitry Andric     CF_LOOP_BREAK,
198*0b57cec5SDimitry Andric     CF_LOOP_CONTINUE,
199*0b57cec5SDimitry Andric     CF_JUMP,
200*0b57cec5SDimitry Andric     CF_ELSE,
201*0b57cec5SDimitry Andric     CF_POP,
202*0b57cec5SDimitry Andric     CF_END
203*0b57cec5SDimitry Andric   };
204*0b57cec5SDimitry Andric 
205*0b57cec5SDimitry Andric   const R600InstrInfo *TII = nullptr;
206*0b57cec5SDimitry Andric   const R600RegisterInfo *TRI = nullptr;
207*0b57cec5SDimitry Andric   unsigned MaxFetchInst;
208*0b57cec5SDimitry Andric   const R600Subtarget *ST = nullptr;
209*0b57cec5SDimitry Andric 
IsTrivialInst(MachineInstr & MI) const210*0b57cec5SDimitry Andric   bool IsTrivialInst(MachineInstr &MI) const {
211*0b57cec5SDimitry Andric     switch (MI.getOpcode()) {
212*0b57cec5SDimitry Andric     case R600::KILL:
213*0b57cec5SDimitry Andric     case R600::RETURN:
214*0b57cec5SDimitry Andric       return true;
215*0b57cec5SDimitry Andric     default:
216*0b57cec5SDimitry Andric       return false;
217*0b57cec5SDimitry Andric     }
218*0b57cec5SDimitry Andric   }
219*0b57cec5SDimitry Andric 
getHWInstrDesc(ControlFlowInstruction CFI) const220*0b57cec5SDimitry Andric   const MCInstrDesc &getHWInstrDesc(ControlFlowInstruction CFI) const {
221*0b57cec5SDimitry Andric     unsigned Opcode = 0;
222*0b57cec5SDimitry Andric     bool isEg = (ST->getGeneration() >= AMDGPUSubtarget::EVERGREEN);
223*0b57cec5SDimitry Andric     switch (CFI) {
224*0b57cec5SDimitry Andric     case CF_TC:
225*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_TC_EG : R600::CF_TC_R600;
226*0b57cec5SDimitry Andric       break;
227*0b57cec5SDimitry Andric     case CF_VC:
228*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_VC_EG : R600::CF_VC_R600;
229*0b57cec5SDimitry Andric       break;
230*0b57cec5SDimitry Andric     case CF_CALL_FS:
231*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_CALL_FS_EG : R600::CF_CALL_FS_R600;
232*0b57cec5SDimitry Andric       break;
233*0b57cec5SDimitry Andric     case CF_WHILE_LOOP:
234*0b57cec5SDimitry Andric       Opcode = isEg ? R600::WHILE_LOOP_EG : R600::WHILE_LOOP_R600;
235*0b57cec5SDimitry Andric       break;
236*0b57cec5SDimitry Andric     case CF_END_LOOP:
237*0b57cec5SDimitry Andric       Opcode = isEg ? R600::END_LOOP_EG : R600::END_LOOP_R600;
238*0b57cec5SDimitry Andric       break;
239*0b57cec5SDimitry Andric     case CF_LOOP_BREAK:
240*0b57cec5SDimitry Andric       Opcode = isEg ? R600::LOOP_BREAK_EG : R600::LOOP_BREAK_R600;
241*0b57cec5SDimitry Andric       break;
242*0b57cec5SDimitry Andric     case CF_LOOP_CONTINUE:
243*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_CONTINUE_EG : R600::CF_CONTINUE_R600;
244*0b57cec5SDimitry Andric       break;
245*0b57cec5SDimitry Andric     case CF_JUMP:
246*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_JUMP_EG : R600::CF_JUMP_R600;
247*0b57cec5SDimitry Andric       break;
248*0b57cec5SDimitry Andric     case CF_ELSE:
249*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_ELSE_EG : R600::CF_ELSE_R600;
250*0b57cec5SDimitry Andric       break;
251*0b57cec5SDimitry Andric     case CF_POP:
252*0b57cec5SDimitry Andric       Opcode = isEg ? R600::POP_EG : R600::POP_R600;
253*0b57cec5SDimitry Andric       break;
254*0b57cec5SDimitry Andric     case CF_END:
255*0b57cec5SDimitry Andric       if (ST->hasCaymanISA()) {
256*0b57cec5SDimitry Andric         Opcode = R600::CF_END_CM;
257*0b57cec5SDimitry Andric         break;
258*0b57cec5SDimitry Andric       }
259*0b57cec5SDimitry Andric       Opcode = isEg ? R600::CF_END_EG : R600::CF_END_R600;
260*0b57cec5SDimitry Andric       break;
261*0b57cec5SDimitry Andric     }
262*0b57cec5SDimitry Andric     assert (Opcode && "No opcode selected");
263*0b57cec5SDimitry Andric     return TII->get(Opcode);
264*0b57cec5SDimitry Andric   }
265*0b57cec5SDimitry Andric 
isCompatibleWithClause(const MachineInstr & MI,std::set<unsigned> & DstRegs) const266*0b57cec5SDimitry Andric   bool isCompatibleWithClause(const MachineInstr &MI,
267*0b57cec5SDimitry Andric                               std::set<unsigned> &DstRegs) const {
268*0b57cec5SDimitry Andric     unsigned DstMI, SrcMI;
269*0b57cec5SDimitry Andric     for (MachineInstr::const_mop_iterator I = MI.operands_begin(),
270*0b57cec5SDimitry Andric                                           E = MI.operands_end();
271*0b57cec5SDimitry Andric          I != E; ++I) {
272*0b57cec5SDimitry Andric       const MachineOperand &MO = *I;
273*0b57cec5SDimitry Andric       if (!MO.isReg())
274*0b57cec5SDimitry Andric         continue;
275*0b57cec5SDimitry Andric       if (MO.isDef()) {
276*0b57cec5SDimitry Andric         Register Reg = MO.getReg();
277*0b57cec5SDimitry Andric         if (R600::R600_Reg128RegClass.contains(Reg))
278*0b57cec5SDimitry Andric           DstMI = Reg;
279*0b57cec5SDimitry Andric         else
280*0b57cec5SDimitry Andric           DstMI = TRI->getMatchingSuperReg(Reg,
281*0b57cec5SDimitry Andric               R600RegisterInfo::getSubRegFromChannel(TRI->getHWRegChan(Reg)),
282*0b57cec5SDimitry Andric               &R600::R600_Reg128RegClass);
283*0b57cec5SDimitry Andric       }
284*0b57cec5SDimitry Andric       if (MO.isUse()) {
285*0b57cec5SDimitry Andric         Register Reg = MO.getReg();
286*0b57cec5SDimitry Andric         if (R600::R600_Reg128RegClass.contains(Reg))
287*0b57cec5SDimitry Andric           SrcMI = Reg;
288*0b57cec5SDimitry Andric         else
289*0b57cec5SDimitry Andric           SrcMI = TRI->getMatchingSuperReg(Reg,
290*0b57cec5SDimitry Andric               R600RegisterInfo::getSubRegFromChannel(TRI->getHWRegChan(Reg)),
291*0b57cec5SDimitry Andric               &R600::R600_Reg128RegClass);
292*0b57cec5SDimitry Andric       }
293*0b57cec5SDimitry Andric     }
294*0b57cec5SDimitry Andric     if ((DstRegs.find(SrcMI) == DstRegs.end())) {
295*0b57cec5SDimitry Andric       DstRegs.insert(DstMI);
296*0b57cec5SDimitry Andric       return true;
297*0b57cec5SDimitry Andric     } else
298*0b57cec5SDimitry Andric       return false;
299*0b57cec5SDimitry Andric   }
300*0b57cec5SDimitry Andric 
301*0b57cec5SDimitry Andric   ClauseFile
MakeFetchClause(MachineBasicBlock & MBB,MachineBasicBlock::iterator & I) const302*0b57cec5SDimitry Andric   MakeFetchClause(MachineBasicBlock &MBB, MachineBasicBlock::iterator &I)
303*0b57cec5SDimitry Andric       const {
304*0b57cec5SDimitry Andric     MachineBasicBlock::iterator ClauseHead = I;
305*0b57cec5SDimitry Andric     std::vector<MachineInstr *> ClauseContent;
306*0b57cec5SDimitry Andric     unsigned AluInstCount = 0;
307*0b57cec5SDimitry Andric     bool IsTex = TII->usesTextureCache(*ClauseHead);
308*0b57cec5SDimitry Andric     std::set<unsigned> DstRegs;
309*0b57cec5SDimitry Andric     for (MachineBasicBlock::iterator E = MBB.end(); I != E; ++I) {
310*0b57cec5SDimitry Andric       if (IsTrivialInst(*I))
311*0b57cec5SDimitry Andric         continue;
312*0b57cec5SDimitry Andric       if (AluInstCount >= MaxFetchInst)
313*0b57cec5SDimitry Andric         break;
314*0b57cec5SDimitry Andric       if ((IsTex && !TII->usesTextureCache(*I)) ||
315*0b57cec5SDimitry Andric           (!IsTex && !TII->usesVertexCache(*I)))
316*0b57cec5SDimitry Andric         break;
317*0b57cec5SDimitry Andric       if (!isCompatibleWithClause(*I, DstRegs))
318*0b57cec5SDimitry Andric         break;
319*0b57cec5SDimitry Andric       AluInstCount ++;
320*0b57cec5SDimitry Andric       ClauseContent.push_back(&*I);
321*0b57cec5SDimitry Andric     }
322*0b57cec5SDimitry Andric     MachineInstr *MIb = BuildMI(MBB, ClauseHead, MBB.findDebugLoc(ClauseHead),
323*0b57cec5SDimitry Andric         getHWInstrDesc(IsTex?CF_TC:CF_VC))
324*0b57cec5SDimitry Andric         .addImm(0) // ADDR
325*0b57cec5SDimitry Andric         .addImm(AluInstCount - 1); // COUNT
326*0b57cec5SDimitry Andric     return ClauseFile(MIb, std::move(ClauseContent));
327*0b57cec5SDimitry Andric   }
328*0b57cec5SDimitry Andric 
getLiteral(MachineInstr & MI,std::vector<MachineOperand * > & Lits) const329*0b57cec5SDimitry Andric   void getLiteral(MachineInstr &MI, std::vector<MachineOperand *> &Lits) const {
330*0b57cec5SDimitry Andric     static const unsigned LiteralRegs[] = {
331*0b57cec5SDimitry Andric       R600::ALU_LITERAL_X,
332*0b57cec5SDimitry Andric       R600::ALU_LITERAL_Y,
333*0b57cec5SDimitry Andric       R600::ALU_LITERAL_Z,
334*0b57cec5SDimitry Andric       R600::ALU_LITERAL_W
335*0b57cec5SDimitry Andric     };
336*0b57cec5SDimitry Andric     const SmallVector<std::pair<MachineOperand *, int64_t>, 3> Srcs =
337*0b57cec5SDimitry Andric         TII->getSrcs(MI);
338*0b57cec5SDimitry Andric     for (const auto &Src:Srcs) {
339*0b57cec5SDimitry Andric       if (Src.first->getReg() != R600::ALU_LITERAL_X)
340*0b57cec5SDimitry Andric         continue;
341*0b57cec5SDimitry Andric       int64_t Imm = Src.second;
342*0b57cec5SDimitry Andric       std::vector<MachineOperand *>::iterator It =
343*0b57cec5SDimitry Andric           llvm::find_if(Lits, [&](MachineOperand *val) {
344*0b57cec5SDimitry Andric             return val->isImm() && (val->getImm() == Imm);
345*0b57cec5SDimitry Andric           });
346*0b57cec5SDimitry Andric 
347*0b57cec5SDimitry Andric       // Get corresponding Operand
348*0b57cec5SDimitry Andric       MachineOperand &Operand = MI.getOperand(
349*0b57cec5SDimitry Andric           TII->getOperandIdx(MI.getOpcode(), R600::OpName::literal));
350*0b57cec5SDimitry Andric 
351*0b57cec5SDimitry Andric       if (It != Lits.end()) {
352*0b57cec5SDimitry Andric         // Reuse existing literal reg
353*0b57cec5SDimitry Andric         unsigned Index = It - Lits.begin();
354*0b57cec5SDimitry Andric         Src.first->setReg(LiteralRegs[Index]);
355*0b57cec5SDimitry Andric       } else {
356*0b57cec5SDimitry Andric         // Allocate new literal reg
357*0b57cec5SDimitry Andric         assert(Lits.size() < 4 && "Too many literals in Instruction Group");
358*0b57cec5SDimitry Andric         Src.first->setReg(LiteralRegs[Lits.size()]);
359*0b57cec5SDimitry Andric         Lits.push_back(&Operand);
360*0b57cec5SDimitry Andric       }
361*0b57cec5SDimitry Andric     }
362*0b57cec5SDimitry Andric   }
363*0b57cec5SDimitry Andric 
insertLiterals(MachineBasicBlock::iterator InsertPos,const std::vector<unsigned> & Literals) const364*0b57cec5SDimitry Andric   MachineBasicBlock::iterator insertLiterals(
365*0b57cec5SDimitry Andric       MachineBasicBlock::iterator InsertPos,
366*0b57cec5SDimitry Andric       const std::vector<unsigned> &Literals) const {
367*0b57cec5SDimitry Andric     MachineBasicBlock *MBB = InsertPos->getParent();
368*0b57cec5SDimitry Andric     for (unsigned i = 0, e = Literals.size(); i < e; i+=2) {
369*0b57cec5SDimitry Andric       unsigned LiteralPair0 = Literals[i];
370*0b57cec5SDimitry Andric       unsigned LiteralPair1 = (i + 1 < e)?Literals[i + 1]:0;
371*0b57cec5SDimitry Andric       InsertPos = BuildMI(MBB, InsertPos->getDebugLoc(),
372*0b57cec5SDimitry Andric           TII->get(R600::LITERALS))
373*0b57cec5SDimitry Andric           .addImm(LiteralPair0)
374*0b57cec5SDimitry Andric           .addImm(LiteralPair1);
375*0b57cec5SDimitry Andric     }
376*0b57cec5SDimitry Andric     return InsertPos;
377*0b57cec5SDimitry Andric   }
378*0b57cec5SDimitry Andric 
379*0b57cec5SDimitry Andric   ClauseFile
MakeALUClause(MachineBasicBlock & MBB,MachineBasicBlock::iterator & I) const380*0b57cec5SDimitry Andric   MakeALUClause(MachineBasicBlock &MBB, MachineBasicBlock::iterator &I)
381*0b57cec5SDimitry Andric       const {
382*0b57cec5SDimitry Andric     MachineInstr &ClauseHead = *I;
383*0b57cec5SDimitry Andric     std::vector<MachineInstr *> ClauseContent;
384*0b57cec5SDimitry Andric     I++;
385*0b57cec5SDimitry Andric     for (MachineBasicBlock::instr_iterator E = MBB.instr_end(); I != E;) {
386*0b57cec5SDimitry Andric       if (IsTrivialInst(*I)) {
387*0b57cec5SDimitry Andric         ++I;
388*0b57cec5SDimitry Andric         continue;
389*0b57cec5SDimitry Andric       }
390*0b57cec5SDimitry Andric       if (!I->isBundle() && !TII->isALUInstr(I->getOpcode()))
391*0b57cec5SDimitry Andric         break;
392*0b57cec5SDimitry Andric       std::vector<MachineOperand *>Literals;
393*0b57cec5SDimitry Andric       if (I->isBundle()) {
394*0b57cec5SDimitry Andric         MachineInstr &DeleteMI = *I;
395*0b57cec5SDimitry Andric         MachineBasicBlock::instr_iterator BI = I.getInstrIterator();
396*0b57cec5SDimitry Andric         while (++BI != E && BI->isBundledWithPred()) {
397*0b57cec5SDimitry Andric           BI->unbundleFromPred();
398*0b57cec5SDimitry Andric           for (MachineOperand &MO : BI->operands()) {
399*0b57cec5SDimitry Andric             if (MO.isReg() && MO.isInternalRead())
400*0b57cec5SDimitry Andric               MO.setIsInternalRead(false);
401*0b57cec5SDimitry Andric           }
402*0b57cec5SDimitry Andric           getLiteral(*BI, Literals);
403*0b57cec5SDimitry Andric           ClauseContent.push_back(&*BI);
404*0b57cec5SDimitry Andric         }
405*0b57cec5SDimitry Andric         I = BI;
406*0b57cec5SDimitry Andric         DeleteMI.eraseFromParent();
407*0b57cec5SDimitry Andric       } else {
408*0b57cec5SDimitry Andric         getLiteral(*I, Literals);
409*0b57cec5SDimitry Andric         ClauseContent.push_back(&*I);
410*0b57cec5SDimitry Andric         I++;
411*0b57cec5SDimitry Andric       }
412*0b57cec5SDimitry Andric       for (unsigned i = 0, e = Literals.size(); i < e; i += 2) {
413*0b57cec5SDimitry Andric         MachineInstrBuilder MILit = BuildMI(MBB, I, I->getDebugLoc(),
414*0b57cec5SDimitry Andric             TII->get(R600::LITERALS));
415*0b57cec5SDimitry Andric         if (Literals[i]->isImm()) {
416*0b57cec5SDimitry Andric             MILit.addImm(Literals[i]->getImm());
417*0b57cec5SDimitry Andric         } else {
418*0b57cec5SDimitry Andric             MILit.addGlobalAddress(Literals[i]->getGlobal(),
419*0b57cec5SDimitry Andric                                    Literals[i]->getOffset());
420*0b57cec5SDimitry Andric         }
421*0b57cec5SDimitry Andric         if (i + 1 < e) {
422*0b57cec5SDimitry Andric           if (Literals[i + 1]->isImm()) {
423*0b57cec5SDimitry Andric             MILit.addImm(Literals[i + 1]->getImm());
424*0b57cec5SDimitry Andric           } else {
425*0b57cec5SDimitry Andric             MILit.addGlobalAddress(Literals[i + 1]->getGlobal(),
426*0b57cec5SDimitry Andric                                    Literals[i + 1]->getOffset());
427*0b57cec5SDimitry Andric           }
428*0b57cec5SDimitry Andric         } else
429*0b57cec5SDimitry Andric           MILit.addImm(0);
430*0b57cec5SDimitry Andric         ClauseContent.push_back(MILit);
431*0b57cec5SDimitry Andric       }
432*0b57cec5SDimitry Andric     }
433*0b57cec5SDimitry Andric     assert(ClauseContent.size() < 128 && "ALU clause is too big");
434*0b57cec5SDimitry Andric     ClauseHead.getOperand(7).setImm(ClauseContent.size() - 1);
435*0b57cec5SDimitry Andric     return ClauseFile(&ClauseHead, std::move(ClauseContent));
436*0b57cec5SDimitry Andric   }
437*0b57cec5SDimitry Andric 
EmitFetchClause(MachineBasicBlock::iterator InsertPos,const DebugLoc & DL,ClauseFile & Clause,unsigned & CfCount)438*0b57cec5SDimitry Andric   void EmitFetchClause(MachineBasicBlock::iterator InsertPos,
439*0b57cec5SDimitry Andric                        const DebugLoc &DL, ClauseFile &Clause,
440*0b57cec5SDimitry Andric                        unsigned &CfCount) {
441*0b57cec5SDimitry Andric     CounterPropagateAddr(*Clause.first, CfCount);
442*0b57cec5SDimitry Andric     MachineBasicBlock *BB = Clause.first->getParent();
443*0b57cec5SDimitry Andric     BuildMI(BB, DL, TII->get(R600::FETCH_CLAUSE)).addImm(CfCount);
444*0b57cec5SDimitry Andric     for (MachineInstr *MI : Clause.second)
445*0b57cec5SDimitry Andric       BB->splice(InsertPos, BB, MI);
446*0b57cec5SDimitry Andric     CfCount += 2 * Clause.second.size();
447*0b57cec5SDimitry Andric   }
448*0b57cec5SDimitry Andric 
EmitALUClause(MachineBasicBlock::iterator InsertPos,const DebugLoc & DL,ClauseFile & Clause,unsigned & CfCount)449*0b57cec5SDimitry Andric   void EmitALUClause(MachineBasicBlock::iterator InsertPos, const DebugLoc &DL,
450*0b57cec5SDimitry Andric                      ClauseFile &Clause, unsigned &CfCount) {
451*0b57cec5SDimitry Andric     Clause.first->getOperand(0).setImm(0);
452*0b57cec5SDimitry Andric     CounterPropagateAddr(*Clause.first, CfCount);
453*0b57cec5SDimitry Andric     MachineBasicBlock *BB = Clause.first->getParent();
454*0b57cec5SDimitry Andric     BuildMI(BB, DL, TII->get(R600::ALU_CLAUSE)).addImm(CfCount);
455*0b57cec5SDimitry Andric     for (MachineInstr *MI : Clause.second)
456*0b57cec5SDimitry Andric       BB->splice(InsertPos, BB, MI);
457*0b57cec5SDimitry Andric     CfCount += Clause.second.size();
458*0b57cec5SDimitry Andric   }
459*0b57cec5SDimitry Andric 
CounterPropagateAddr(MachineInstr & MI,unsigned Addr) const460*0b57cec5SDimitry Andric   void CounterPropagateAddr(MachineInstr &MI, unsigned Addr) const {
461*0b57cec5SDimitry Andric     MI.getOperand(0).setImm(Addr + MI.getOperand(0).getImm());
462*0b57cec5SDimitry Andric   }
CounterPropagateAddr(const std::set<MachineInstr * > & MIs,unsigned Addr) const463*0b57cec5SDimitry Andric   void CounterPropagateAddr(const std::set<MachineInstr *> &MIs,
464*0b57cec5SDimitry Andric                             unsigned Addr) const {
465*0b57cec5SDimitry Andric     for (MachineInstr *MI : MIs) {
466*0b57cec5SDimitry Andric       CounterPropagateAddr(*MI, Addr);
467*0b57cec5SDimitry Andric     }
468*0b57cec5SDimitry Andric   }
469*0b57cec5SDimitry Andric 
470*0b57cec5SDimitry Andric public:
471*0b57cec5SDimitry Andric   static char ID;
472*0b57cec5SDimitry Andric 
R600ControlFlowFinalizer()473*0b57cec5SDimitry Andric   R600ControlFlowFinalizer() : MachineFunctionPass(ID) {}
474*0b57cec5SDimitry Andric 
runOnMachineFunction(MachineFunction & MF)475*0b57cec5SDimitry Andric   bool runOnMachineFunction(MachineFunction &MF) override {
476*0b57cec5SDimitry Andric     ST = &MF.getSubtarget<R600Subtarget>();
477*0b57cec5SDimitry Andric     MaxFetchInst = ST->getTexVTXClauseSize();
478*0b57cec5SDimitry Andric     TII = ST->getInstrInfo();
479*0b57cec5SDimitry Andric     TRI = ST->getRegisterInfo();
480*0b57cec5SDimitry Andric 
481*0b57cec5SDimitry Andric     R600MachineFunctionInfo *MFI = MF.getInfo<R600MachineFunctionInfo>();
482*0b57cec5SDimitry Andric 
483*0b57cec5SDimitry Andric     CFStack CFStack(ST, MF.getFunction().getCallingConv());
484*0b57cec5SDimitry Andric     for (MachineFunction::iterator MB = MF.begin(), ME = MF.end(); MB != ME;
485*0b57cec5SDimitry Andric         ++MB) {
486*0b57cec5SDimitry Andric       MachineBasicBlock &MBB = *MB;
487*0b57cec5SDimitry Andric       unsigned CfCount = 0;
488*0b57cec5SDimitry Andric       std::vector<std::pair<unsigned, std::set<MachineInstr *>>> LoopStack;
489*0b57cec5SDimitry Andric       std::vector<MachineInstr * > IfThenElseStack;
490*0b57cec5SDimitry Andric       if (MF.getFunction().getCallingConv() == CallingConv::AMDGPU_VS) {
491*0b57cec5SDimitry Andric         BuildMI(MBB, MBB.begin(), MBB.findDebugLoc(MBB.begin()),
492*0b57cec5SDimitry Andric             getHWInstrDesc(CF_CALL_FS));
493*0b57cec5SDimitry Andric         CfCount++;
494*0b57cec5SDimitry Andric       }
495*0b57cec5SDimitry Andric       std::vector<ClauseFile> FetchClauses, AluClauses;
496*0b57cec5SDimitry Andric       std::vector<MachineInstr *> LastAlu(1);
497*0b57cec5SDimitry Andric       std::vector<MachineInstr *> ToPopAfter;
498*0b57cec5SDimitry Andric 
499*0b57cec5SDimitry Andric       for (MachineBasicBlock::iterator I = MBB.begin(), E = MBB.end();
500*0b57cec5SDimitry Andric           I != E;) {
501*0b57cec5SDimitry Andric         if (TII->usesTextureCache(*I) || TII->usesVertexCache(*I)) {
502*0b57cec5SDimitry Andric           LLVM_DEBUG(dbgs() << CfCount << ":"; I->dump(););
503*0b57cec5SDimitry Andric           FetchClauses.push_back(MakeFetchClause(MBB, I));
504*0b57cec5SDimitry Andric           CfCount++;
505*0b57cec5SDimitry Andric           LastAlu.back() = nullptr;
506*0b57cec5SDimitry Andric           continue;
507*0b57cec5SDimitry Andric         }
508*0b57cec5SDimitry Andric 
509*0b57cec5SDimitry Andric         MachineBasicBlock::iterator MI = I;
510*0b57cec5SDimitry Andric         if (MI->getOpcode() != R600::ENDIF)
511*0b57cec5SDimitry Andric           LastAlu.back() = nullptr;
512*0b57cec5SDimitry Andric         if (MI->getOpcode() == R600::CF_ALU)
513*0b57cec5SDimitry Andric           LastAlu.back() = &*MI;
514*0b57cec5SDimitry Andric         I++;
515*0b57cec5SDimitry Andric         bool RequiresWorkAround =
516*0b57cec5SDimitry Andric             CFStack.requiresWorkAroundForInst(MI->getOpcode());
517*0b57cec5SDimitry Andric         switch (MI->getOpcode()) {
518*0b57cec5SDimitry Andric         case R600::CF_ALU_PUSH_BEFORE:
519*0b57cec5SDimitry Andric           if (RequiresWorkAround) {
520*0b57cec5SDimitry Andric             LLVM_DEBUG(dbgs()
521*0b57cec5SDimitry Andric                        << "Applying bug work-around for ALU_PUSH_BEFORE\n");
522*0b57cec5SDimitry Andric             BuildMI(MBB, MI, MBB.findDebugLoc(MI), TII->get(R600::CF_PUSH_EG))
523*0b57cec5SDimitry Andric                 .addImm(CfCount + 1)
524*0b57cec5SDimitry Andric                 .addImm(1);
525*0b57cec5SDimitry Andric             MI->setDesc(TII->get(R600::CF_ALU));
526*0b57cec5SDimitry Andric             CfCount++;
527*0b57cec5SDimitry Andric             CFStack.pushBranch(R600::CF_PUSH_EG);
528*0b57cec5SDimitry Andric           } else
529*0b57cec5SDimitry Andric             CFStack.pushBranch(R600::CF_ALU_PUSH_BEFORE);
530*0b57cec5SDimitry Andric           [[fallthrough]];
531*0b57cec5SDimitry Andric         case R600::CF_ALU:
532*0b57cec5SDimitry Andric           I = MI;
533*0b57cec5SDimitry Andric           AluClauses.push_back(MakeALUClause(MBB, I));
534*0b57cec5SDimitry Andric           LLVM_DEBUG(dbgs() << CfCount << ":"; MI->dump(););
535*0b57cec5SDimitry Andric           CfCount++;
536*0b57cec5SDimitry Andric           break;
537*0b57cec5SDimitry Andric         case R600::WHILELOOP: {
538*0b57cec5SDimitry Andric           CFStack.pushLoop();
539*0b57cec5SDimitry Andric           MachineInstr *MIb = BuildMI(MBB, MI, MBB.findDebugLoc(MI),
540*0b57cec5SDimitry Andric               getHWInstrDesc(CF_WHILE_LOOP))
541*0b57cec5SDimitry Andric               .addImm(1);
542*0b57cec5SDimitry Andric           std::pair<unsigned, std::set<MachineInstr *>> Pair(CfCount,
543*0b57cec5SDimitry Andric               std::set<MachineInstr *>());
544*0b57cec5SDimitry Andric           Pair.second.insert(MIb);
545*0b57cec5SDimitry Andric           LoopStack.push_back(std::move(Pair));
546*0b57cec5SDimitry Andric           MI->eraseFromParent();
547*0b57cec5SDimitry Andric           CfCount++;
548*0b57cec5SDimitry Andric           break;
549*0b57cec5SDimitry Andric         }
550*0b57cec5SDimitry Andric         case R600::ENDLOOP: {
551*0b57cec5SDimitry Andric           CFStack.popLoop();
552*0b57cec5SDimitry Andric           std::pair<unsigned, std::set<MachineInstr *>> Pair =
553*0b57cec5SDimitry Andric               std::move(LoopStack.back());
554*0b57cec5SDimitry Andric           LoopStack.pop_back();
555*0b57cec5SDimitry Andric           CounterPropagateAddr(Pair.second, CfCount);
556*0b57cec5SDimitry Andric           BuildMI(MBB, MI, MBB.findDebugLoc(MI), getHWInstrDesc(CF_END_LOOP))
557*0b57cec5SDimitry Andric               .addImm(Pair.first + 1);
558*0b57cec5SDimitry Andric           MI->eraseFromParent();
559*0b57cec5SDimitry Andric           CfCount++;
560*0b57cec5SDimitry Andric           break;
561*0b57cec5SDimitry Andric         }
562*0b57cec5SDimitry Andric         case R600::IF_PREDICATE_SET: {
563*0b57cec5SDimitry Andric           LastAlu.push_back(nullptr);
564*0b57cec5SDimitry Andric           MachineInstr *MIb = BuildMI(MBB, MI, MBB.findDebugLoc(MI),
565*0b57cec5SDimitry Andric               getHWInstrDesc(CF_JUMP))
566*0b57cec5SDimitry Andric               .addImm(0)
567*0b57cec5SDimitry Andric               .addImm(0);
568*0b57cec5SDimitry Andric           IfThenElseStack.push_back(MIb);
569*0b57cec5SDimitry Andric           LLVM_DEBUG(dbgs() << CfCount << ":"; MIb->dump(););
570*0b57cec5SDimitry Andric           MI->eraseFromParent();
571*0b57cec5SDimitry Andric           CfCount++;
572*0b57cec5SDimitry Andric           break;
573*0b57cec5SDimitry Andric         }
574*0b57cec5SDimitry Andric         case R600::ELSE: {
575*0b57cec5SDimitry Andric           MachineInstr * JumpInst = IfThenElseStack.back();
576*0b57cec5SDimitry Andric           IfThenElseStack.pop_back();
577*0b57cec5SDimitry Andric           CounterPropagateAddr(*JumpInst, CfCount);
578*0b57cec5SDimitry Andric           MachineInstr *MIb = BuildMI(MBB, MI, MBB.findDebugLoc(MI),
579*0b57cec5SDimitry Andric               getHWInstrDesc(CF_ELSE))
580*0b57cec5SDimitry Andric               .addImm(0)
581*0b57cec5SDimitry Andric               .addImm(0);
582*0b57cec5SDimitry Andric           LLVM_DEBUG(dbgs() << CfCount << ":"; MIb->dump(););
583*0b57cec5SDimitry Andric           IfThenElseStack.push_back(MIb);
584*0b57cec5SDimitry Andric           MI->eraseFromParent();
585*0b57cec5SDimitry Andric           CfCount++;
586*0b57cec5SDimitry Andric           break;
587*0b57cec5SDimitry Andric         }
588*0b57cec5SDimitry Andric         case R600::ENDIF: {
589*0b57cec5SDimitry Andric           CFStack.popBranch();
590*0b57cec5SDimitry Andric           if (LastAlu.back()) {
591*0b57cec5SDimitry Andric             ToPopAfter.push_back(LastAlu.back());
592*0b57cec5SDimitry Andric           } else {
593*0b57cec5SDimitry Andric             MachineInstr *MIb = BuildMI(MBB, MI, MBB.findDebugLoc(MI),
594*0b57cec5SDimitry Andric                 getHWInstrDesc(CF_POP))
595*0b57cec5SDimitry Andric                 .addImm(CfCount + 1)
596*0b57cec5SDimitry Andric                 .addImm(1);
597*0b57cec5SDimitry Andric             (void)MIb;
598*0b57cec5SDimitry Andric             LLVM_DEBUG(dbgs() << CfCount << ":"; MIb->dump(););
599*0b57cec5SDimitry Andric             CfCount++;
600*0b57cec5SDimitry Andric           }
601*0b57cec5SDimitry Andric 
602*0b57cec5SDimitry Andric           MachineInstr *IfOrElseInst = IfThenElseStack.back();
603*0b57cec5SDimitry Andric           IfThenElseStack.pop_back();
604*0b57cec5SDimitry Andric           CounterPropagateAddr(*IfOrElseInst, CfCount);
605*0b57cec5SDimitry Andric           IfOrElseInst->getOperand(1).setImm(1);
606*0b57cec5SDimitry Andric           LastAlu.pop_back();
607*0b57cec5SDimitry Andric           MI->eraseFromParent();
608*0b57cec5SDimitry Andric           break;
609*0b57cec5SDimitry Andric         }
610*0b57cec5SDimitry Andric         case R600::BREAK: {
611*0b57cec5SDimitry Andric           CfCount ++;
612*0b57cec5SDimitry Andric           MachineInstr *MIb = BuildMI(MBB, MI, MBB.findDebugLoc(MI),
613*0b57cec5SDimitry Andric               getHWInstrDesc(CF_LOOP_BREAK))
614*0b57cec5SDimitry Andric               .addImm(0);
615*0b57cec5SDimitry Andric           LoopStack.back().second.insert(MIb);
616*0b57cec5SDimitry Andric           MI->eraseFromParent();
617*0b57cec5SDimitry Andric           break;
618*0b57cec5SDimitry Andric         }
619*0b57cec5SDimitry Andric         case R600::CONTINUE: {
620*0b57cec5SDimitry Andric           MachineInstr *MIb = BuildMI(MBB, MI, MBB.findDebugLoc(MI),
621*0b57cec5SDimitry Andric               getHWInstrDesc(CF_LOOP_CONTINUE))
622*0b57cec5SDimitry Andric               .addImm(0);
623*0b57cec5SDimitry Andric           LoopStack.back().second.insert(MIb);
624*0b57cec5SDimitry Andric           MI->eraseFromParent();
625*0b57cec5SDimitry Andric           CfCount++;
626*0b57cec5SDimitry Andric           break;
627*0b57cec5SDimitry Andric         }
628*0b57cec5SDimitry Andric         case R600::RETURN: {
629*0b57cec5SDimitry Andric           DebugLoc DL = MBB.findDebugLoc(MI);
630*0b57cec5SDimitry Andric           BuildMI(MBB, MI, DL, getHWInstrDesc(CF_END));
631*0b57cec5SDimitry Andric           CfCount++;
632*0b57cec5SDimitry Andric           if (CfCount % 2) {
633*0b57cec5SDimitry Andric             BuildMI(MBB, I, DL, TII->get(R600::PAD));
634*0b57cec5SDimitry Andric             CfCount++;
635*0b57cec5SDimitry Andric           }
636*0b57cec5SDimitry Andric           MI->eraseFromParent();
637*0b57cec5SDimitry Andric           for (ClauseFile &CF : FetchClauses)
638*0b57cec5SDimitry Andric             EmitFetchClause(I, DL, CF, CfCount);
639*0b57cec5SDimitry Andric           for (ClauseFile &CF : AluClauses)
640*0b57cec5SDimitry Andric             EmitALUClause(I, DL, CF, CfCount);
641*0b57cec5SDimitry Andric           break;
642*0b57cec5SDimitry Andric         }
643*0b57cec5SDimitry Andric         default:
644*0b57cec5SDimitry Andric           if (TII->isExport(MI->getOpcode())) {
645*0b57cec5SDimitry Andric             LLVM_DEBUG(dbgs() << CfCount << ":"; MI->dump(););
646*0b57cec5SDimitry Andric             CfCount++;
647*0b57cec5SDimitry Andric           }
648*0b57cec5SDimitry Andric           break;
649*0b57cec5SDimitry Andric         }
650*0b57cec5SDimitry Andric       }
651*0b57cec5SDimitry Andric       for (MachineInstr *Alu : ToPopAfter) {
652*0b57cec5SDimitry Andric         BuildMI(MBB, Alu, MBB.findDebugLoc((MachineBasicBlock::iterator)Alu),
653*0b57cec5SDimitry Andric             TII->get(R600::CF_ALU_POP_AFTER))
654*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(0).getImm())
655*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(1).getImm())
656*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(2).getImm())
657*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(3).getImm())
658*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(4).getImm())
659*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(5).getImm())
660*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(6).getImm())
661*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(7).getImm())
662*0b57cec5SDimitry Andric             .addImm(Alu->getOperand(8).getImm());
663*0b57cec5SDimitry Andric         Alu->eraseFromParent();
664*0b57cec5SDimitry Andric       }
665*0b57cec5SDimitry Andric       MFI->CFStackSize = CFStack.MaxStackSize;
666*0b57cec5SDimitry Andric     }
667*0b57cec5SDimitry Andric 
668*0b57cec5SDimitry Andric     return false;
669*0b57cec5SDimitry Andric   }
670*0b57cec5SDimitry Andric 
getPassName() const671*0b57cec5SDimitry Andric   StringRef getPassName() const override {
672*0b57cec5SDimitry Andric     return "R600 Control Flow Finalizer Pass";
673*0b57cec5SDimitry Andric   }
674*0b57cec5SDimitry Andric };
675*0b57cec5SDimitry Andric 
676*0b57cec5SDimitry Andric } // end anonymous namespace
677*0b57cec5SDimitry Andric 
678*0b57cec5SDimitry Andric INITIALIZE_PASS_BEGIN(R600ControlFlowFinalizer, DEBUG_TYPE,
679*0b57cec5SDimitry Andric                      "R600 Control Flow Finalizer", false, false)
680*0b57cec5SDimitry Andric INITIALIZE_PASS_END(R600ControlFlowFinalizer, DEBUG_TYPE,
681*0b57cec5SDimitry Andric                     "R600 Control Flow Finalizer", false, false)
682*0b57cec5SDimitry Andric 
683*0b57cec5SDimitry Andric char R600ControlFlowFinalizer::ID = 0;
684*0b57cec5SDimitry Andric 
685*0b57cec5SDimitry Andric char &llvm::R600ControlFlowFinalizerID = R600ControlFlowFinalizer::ID;
686*0b57cec5SDimitry Andric 
createR600ControlFlowFinalizer()687*0b57cec5SDimitry Andric FunctionPass *llvm::createR600ControlFlowFinalizer() {
688*0b57cec5SDimitry Andric   return new R600ControlFlowFinalizer();
689*0b57cec5SDimitry Andric }
690*0b57cec5SDimitry Andric