1 //===-------------- PPCVSXCopy.cpp - VSX Copy Legalization ----------------===//
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 // A pass which deals with the complexity of generating legal VSX register
11 // copies to/from register classes which partially overlap with the VSX
12 // register file.
13 //
14 //===----------------------------------------------------------------------===//
15 
16 #include "PPC.h"
17 #include "MCTargetDesc/PPCPredicates.h"
18 #include "PPCHazardRecognizers.h"
19 #include "PPCInstrBuilder.h"
20 #include "PPCInstrInfo.h"
21 #include "PPCMachineFunctionInfo.h"
22 #include "PPCTargetMachine.h"
23 #include "llvm/ADT/STLExtras.h"
24 #include "llvm/ADT/Statistic.h"
25 #include "llvm/CodeGen/MachineFrameInfo.h"
26 #include "llvm/CodeGen/MachineFunctionPass.h"
27 #include "llvm/CodeGen/MachineInstrBuilder.h"
28 #include "llvm/CodeGen/MachineMemOperand.h"
29 #include "llvm/CodeGen/MachineRegisterInfo.h"
30 #include "llvm/MC/MCAsmInfo.h"
31 #include "llvm/Support/Debug.h"
32 #include "llvm/Support/ErrorHandling.h"
33 #include "llvm/Support/TargetRegistry.h"
34 #include "llvm/Support/raw_ostream.h"
35 
36 using namespace llvm;
37 
38 #define DEBUG_TYPE "ppc-vsx-copy"
39 
40 namespace llvm {
41   void initializePPCVSXCopyPass(PassRegistry&);
42 }
43 
44 namespace {
45   // PPCVSXCopy pass - For copies between VSX registers and non-VSX registers
46   // (Altivec and scalar floating-point registers), we need to transform the
47   // copies into subregister copies with other restrictions.
48   struct PPCVSXCopy : public MachineFunctionPass {
49     static char ID;
50     PPCVSXCopy() : MachineFunctionPass(ID) {
51       initializePPCVSXCopyPass(*PassRegistry::getPassRegistry());
52     }
53 
54     const TargetInstrInfo *TII;
55 
56     bool IsRegInClass(unsigned Reg, const TargetRegisterClass *RC,
57                       MachineRegisterInfo &MRI) {
58       if (TargetRegisterInfo::isVirtualRegister(Reg)) {
59         return RC->hasSubClassEq(MRI.getRegClass(Reg));
60       } else if (RC->contains(Reg)) {
61         return true;
62       }
63 
64       return false;
65     }
66 
67     bool IsVSReg(unsigned Reg, MachineRegisterInfo &MRI) {
68       return IsRegInClass(Reg, &PPC::VSRCRegClass, MRI);
69     }
70 
71     bool IsVRReg(unsigned Reg, MachineRegisterInfo &MRI) {
72       return IsRegInClass(Reg, &PPC::VRRCRegClass, MRI);
73     }
74 
75     bool IsF8Reg(unsigned Reg, MachineRegisterInfo &MRI) {
76       return IsRegInClass(Reg, &PPC::F8RCRegClass, MRI);
77     }
78 
79     bool IsVSFReg(unsigned Reg, MachineRegisterInfo &MRI) {
80       return IsRegInClass(Reg, &PPC::VSFRCRegClass, MRI);
81     }
82 
83     bool IsVSSReg(unsigned Reg, MachineRegisterInfo &MRI) {
84       return IsRegInClass(Reg, &PPC::VSSRCRegClass, MRI);
85     }
86 
87 protected:
88     bool processBlock(MachineBasicBlock &MBB) {
89       bool Changed = false;
90 
91       MachineRegisterInfo &MRI = MBB.getParent()->getRegInfo();
92       for (MachineInstr &MI : MBB) {
93         if (!MI.isFullCopy())
94           continue;
95 
96         MachineOperand &DstMO = MI.getOperand(0);
97         MachineOperand &SrcMO = MI.getOperand(1);
98 
99         if ( IsVSReg(DstMO.getReg(), MRI) &&
100             !IsVSReg(SrcMO.getReg(), MRI)) {
101           // This is a copy *to* a VSX register from a non-VSX register.
102           Changed = true;
103 
104           const TargetRegisterClass *SrcRC =
105             IsVRReg(SrcMO.getReg(), MRI) ? &PPC::VSHRCRegClass :
106                                            &PPC::VSLRCRegClass;
107           assert((IsF8Reg(SrcMO.getReg(), MRI) ||
108                   IsVRReg(SrcMO.getReg(), MRI) ||
109                   IsVSSReg(SrcMO.getReg(), MRI) ||
110                   IsVSFReg(SrcMO.getReg(), MRI)) &&
111                  "Unknown source for a VSX copy");
112 
113           unsigned NewVReg = MRI.createVirtualRegister(SrcRC);
114           BuildMI(MBB, MI, MI.getDebugLoc(),
115                   TII->get(TargetOpcode::SUBREG_TO_REG), NewVReg)
116               .addImm(1) // add 1, not 0, because there is no implicit clearing
117                          // of the high bits.
118               .addOperand(SrcMO)
119               .addImm(IsVRReg(SrcMO.getReg(), MRI) ? PPC::sub_128
120                                                    : PPC::sub_64);
121 
122           // The source of the original copy is now the new virtual register.
123           SrcMO.setReg(NewVReg);
124         } else if (!IsVSReg(DstMO.getReg(), MRI) &&
125                     IsVSReg(SrcMO.getReg(), MRI)) {
126           // This is a copy *from* a VSX register to a non-VSX register.
127           Changed = true;
128 
129           const TargetRegisterClass *DstRC =
130             IsVRReg(DstMO.getReg(), MRI) ? &PPC::VSHRCRegClass :
131                                            &PPC::VSLRCRegClass;
132           assert((IsF8Reg(DstMO.getReg(), MRI) ||
133                   IsVSFReg(DstMO.getReg(), MRI) ||
134                   IsVSSReg(DstMO.getReg(), MRI) ||
135                   IsVRReg(DstMO.getReg(), MRI)) &&
136                  "Unknown destination for a VSX copy");
137 
138           // Copy the VSX value into a new VSX register of the correct subclass.
139           unsigned NewVReg = MRI.createVirtualRegister(DstRC);
140           BuildMI(MBB, MI, MI.getDebugLoc(), TII->get(TargetOpcode::COPY),
141                   NewVReg)
142               .addOperand(SrcMO);
143 
144           // Transform the original copy into a subregister extraction copy.
145           SrcMO.setReg(NewVReg);
146           SrcMO.setSubReg(IsVRReg(DstMO.getReg(), MRI) ? PPC::sub_128 :
147                                                          PPC::sub_64);
148         }
149       }
150 
151       return Changed;
152     }
153 
154 public:
155     bool runOnMachineFunction(MachineFunction &MF) override {
156       // If we don't have VSX on the subtarget, don't do anything.
157       const PPCSubtarget &STI = MF.getSubtarget<PPCSubtarget>();
158       if (!STI.hasVSX())
159         return false;
160       TII = STI.getInstrInfo();
161 
162       bool Changed = false;
163 
164       for (MachineFunction::iterator I = MF.begin(); I != MF.end();) {
165         MachineBasicBlock &B = *I++;
166         if (processBlock(B))
167           Changed = true;
168       }
169 
170       return Changed;
171     }
172 
173     void getAnalysisUsage(AnalysisUsage &AU) const override {
174       MachineFunctionPass::getAnalysisUsage(AU);
175     }
176   };
177 }
178 
179 INITIALIZE_PASS(PPCVSXCopy, DEBUG_TYPE,
180                 "PowerPC VSX Copy Legalization", false, false)
181 
182 char PPCVSXCopy::ID = 0;
183 FunctionPass*
184 llvm::createPPCVSXCopyPass() { return new PPCVSXCopy(); }
185 
186