AMDGPU: Split x8 and x16 vector loads instead of scalarize
[oota-llvm.git] / lib / Target / AMDGPU / SIFixControlFlowLiveIntervals.cpp
1 //===-- SIFixControlFlowLiveIntervals.cpp - Fix CF live intervals ---------===//
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 /// \file
11 /// \brief Spilling of EXEC masks used for control flow messes up control flow
12 /// lowering, so mark all live intervals associated with CF instructions as
13 /// non-spillable.
14 ///
15 //===----------------------------------------------------------------------===//
16
17 #include "AMDGPU.h"
18 #include "SIInstrInfo.h"
19 #include "llvm/CodeGen/LiveIntervalAnalysis.h"
20 #include "llvm/CodeGen/MachineFunctionPass.h"
21 #include "llvm/CodeGen/MachineRegisterInfo.h"
22
23 using namespace llvm;
24
25 #define DEBUG_TYPE "si-fix-cf-live-intervals"
26
27 namespace {
28
29 class SIFixControlFlowLiveIntervals : public MachineFunctionPass {
30 public:
31   static char ID;
32
33 public:
34   SIFixControlFlowLiveIntervals() : MachineFunctionPass(ID) {
35     initializeSIFixControlFlowLiveIntervalsPass(*PassRegistry::getPassRegistry());
36   }
37
38   bool runOnMachineFunction(MachineFunction &MF) override;
39
40   const char *getPassName() const override {
41     return "SI Fix CF Live Intervals";
42   }
43
44   void getAnalysisUsage(AnalysisUsage &AU) const override {
45     AU.addRequired<LiveIntervals>();
46     AU.setPreservesAll();
47     MachineFunctionPass::getAnalysisUsage(AU);
48   }
49 };
50
51 } // End anonymous namespace.
52
53 INITIALIZE_PASS_BEGIN(SIFixControlFlowLiveIntervals, DEBUG_TYPE,
54                       "SI Fix CF Live Intervals", false, false)
55 INITIALIZE_PASS_DEPENDENCY(LiveIntervals)
56 INITIALIZE_PASS_END(SIFixControlFlowLiveIntervals, DEBUG_TYPE,
57                     "SI Fix CF Live Intervals", false, false)
58
59 char SIFixControlFlowLiveIntervals::ID = 0;
60
61 char &llvm::SIFixControlFlowLiveIntervalsID = SIFixControlFlowLiveIntervals::ID;
62
63 FunctionPass *llvm::createSIFixControlFlowLiveIntervalsPass() {
64   return new SIFixControlFlowLiveIntervals();
65 }
66
67 bool SIFixControlFlowLiveIntervals::runOnMachineFunction(MachineFunction &MF) {
68   LiveIntervals *LIS = &getAnalysis<LiveIntervals>();
69
70   for (const MachineBasicBlock &MBB : MF) {
71     for (const MachineInstr &MI : MBB) {
72       switch (MI.getOpcode()) {
73         case AMDGPU::SI_IF:
74         case AMDGPU::SI_ELSE:
75         case AMDGPU::SI_BREAK:
76         case AMDGPU::SI_IF_BREAK:
77         case AMDGPU::SI_ELSE_BREAK:
78         case AMDGPU::SI_END_CF: {
79           unsigned Reg = MI.getOperand(0).getReg();
80           LIS->getInterval(Reg).markNotSpillable();
81           break;
82         }
83         default:
84           break;
85       }
86     }
87   }
88
89   return false;
90 }