1 //===- HexagonVectorPrint.cpp - Generate vector printing instructions -----===//
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 pass adds the capability to generate pseudo vector/predicate register
10 // printing instructions. These pseudo instructions should be used with the
11 // simulator, NEVER on hardware.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "HexagonInstrInfo.h"
16 #include "HexagonSubtarget.h"
17 #include "llvm/ADT/StringRef.h"
18 #include "llvm/CodeGen/MachineBasicBlock.h"
19 #include "llvm/CodeGen/MachineFunction.h"
20 #include "llvm/CodeGen/MachineFunctionPass.h"
21 #include "llvm/CodeGen/MachineInstr.h"
22 #include "llvm/CodeGen/MachineInstrBuilder.h"
23 #include "llvm/CodeGen/MachineOperand.h"
24 #include "llvm/CodeGen/TargetOpcodes.h"
25 #include "llvm/IR/DebugLoc.h"
26 #include "llvm/IR/InlineAsm.h"
27 #include "llvm/Pass.h"
28 #include "llvm/Support/CommandLine.h"
29 #include "llvm/Support/Debug.h"
30 #include "llvm/Support/ErrorHandling.h"
31 #include "llvm/Support/raw_ostream.h"
32 #include <string>
33 #include <vector>
34
35 using namespace llvm;
36
37 #define DEBUG_TYPE "hexagon-vector-print"
38
39 static cl::opt<bool> TraceHexVectorStoresOnly("trace-hex-vector-stores-only",
40 cl::Hidden, cl::ZeroOrMore, cl::init(false),
41 cl::desc("Enables tracing of vector stores"));
42
43 namespace llvm {
44
45 FunctionPass *createHexagonVectorPrint();
46 void initializeHexagonVectorPrintPass(PassRegistry&);
47
48 } // end namespace llvm
49
50 namespace {
51
52 class HexagonVectorPrint : public MachineFunctionPass {
53 const HexagonSubtarget *QST = nullptr;
54 const HexagonInstrInfo *QII = nullptr;
55 const HexagonRegisterInfo *QRI = nullptr;
56
57 public:
58 static char ID;
59
HexagonVectorPrint()60 HexagonVectorPrint() : MachineFunctionPass(ID) {
61 initializeHexagonVectorPrintPass(*PassRegistry::getPassRegistry());
62 }
63
getPassName() const64 StringRef getPassName() const override { return "Hexagon VectorPrint pass"; }
65
66 bool runOnMachineFunction(MachineFunction &Fn) override;
67 };
68
69 } // end anonymous namespace
70
71 char HexagonVectorPrint::ID = 0;
72
isVecReg(unsigned Reg)73 static bool isVecReg(unsigned Reg) {
74 return (Reg >= Hexagon::V0 && Reg <= Hexagon::V31)
75 || (Reg >= Hexagon::W0 && Reg <= Hexagon::W15)
76 || (Reg >= Hexagon::Q0 && Reg <= Hexagon::Q3);
77 }
78
getStringReg(unsigned R)79 static std::string getStringReg(unsigned R) {
80 if (R >= Hexagon::V0 && R <= Hexagon::V31) {
81 static const char* S[] = { "20", "21", "22", "23", "24", "25", "26", "27",
82 "28", "29", "2a", "2b", "2c", "2d", "2e", "2f",
83 "30", "31", "32", "33", "34", "35", "36", "37",
84 "38", "39", "3a", "3b", "3c", "3d", "3e", "3f"};
85 return S[R-Hexagon::V0];
86 }
87 if (R >= Hexagon::Q0 && R <= Hexagon::Q3) {
88 static const char* S[] = { "00", "01", "02", "03"};
89 return S[R-Hexagon::Q0];
90
91 }
92 llvm_unreachable("valid vreg");
93 }
94
addAsmInstr(MachineBasicBlock * MBB,unsigned Reg,MachineBasicBlock::instr_iterator I,const DebugLoc & DL,const HexagonInstrInfo * QII,MachineFunction & Fn)95 static void addAsmInstr(MachineBasicBlock *MBB, unsigned Reg,
96 MachineBasicBlock::instr_iterator I,
97 const DebugLoc &DL, const HexagonInstrInfo *QII,
98 MachineFunction &Fn) {
99 std::string VDescStr = ".long 0x1dffe0" + getStringReg(Reg);
100 const char *cstr = Fn.createExternalSymbolName(VDescStr);
101 unsigned ExtraInfo = InlineAsm::Extra_HasSideEffects;
102 BuildMI(*MBB, I, DL, QII->get(TargetOpcode::INLINEASM))
103 .addExternalSymbol(cstr)
104 .addImm(ExtraInfo);
105 }
106
getInstrVecReg(const MachineInstr & MI,unsigned & Reg)107 static bool getInstrVecReg(const MachineInstr &MI, unsigned &Reg) {
108 if (MI.getNumOperands() < 1) return false;
109 // Vec load or compute.
110 if (MI.getOperand(0).isReg() && MI.getOperand(0).isDef()) {
111 Reg = MI.getOperand(0).getReg();
112 if (isVecReg(Reg))
113 return !TraceHexVectorStoresOnly;
114 }
115 // Vec store.
116 if (MI.mayStore() && MI.getNumOperands() >= 3 && MI.getOperand(2).isReg()) {
117 Reg = MI.getOperand(2).getReg();
118 if (isVecReg(Reg))
119 return true;
120 }
121 // Vec store post increment.
122 if (MI.mayStore() && MI.getNumOperands() >= 4 && MI.getOperand(3).isReg()) {
123 Reg = MI.getOperand(3).getReg();
124 if (isVecReg(Reg))
125 return true;
126 }
127 return false;
128 }
129
runOnMachineFunction(MachineFunction & Fn)130 bool HexagonVectorPrint::runOnMachineFunction(MachineFunction &Fn) {
131 bool Changed = false;
132 QST = &Fn.getSubtarget<HexagonSubtarget>();
133 QRI = QST->getRegisterInfo();
134 QII = QST->getInstrInfo();
135 std::vector<MachineInstr *> VecPrintList;
136 for (auto &MBB : Fn)
137 for (auto &MI : MBB) {
138 if (MI.isBundle()) {
139 MachineBasicBlock::instr_iterator MII = MI.getIterator();
140 for (++MII; MII != MBB.instr_end() && MII->isInsideBundle(); ++MII) {
141 if (MII->getNumOperands() < 1)
142 continue;
143 unsigned Reg = 0;
144 if (getInstrVecReg(*MII, Reg)) {
145 VecPrintList.push_back((&*MII));
146 LLVM_DEBUG(dbgs() << "Found vector reg inside bundle \n";
147 MII->dump());
148 }
149 }
150 } else {
151 unsigned Reg = 0;
152 if (getInstrVecReg(MI, Reg)) {
153 VecPrintList.push_back(&MI);
154 LLVM_DEBUG(dbgs() << "Found vector reg \n"; MI.dump());
155 }
156 }
157 }
158
159 Changed = !VecPrintList.empty();
160 if (!Changed)
161 return Changed;
162
163 for (auto *I : VecPrintList) {
164 DebugLoc DL = I->getDebugLoc();
165 MachineBasicBlock *MBB = I->getParent();
166 LLVM_DEBUG(dbgs() << "Evaluating V MI\n"; I->dump());
167 unsigned Reg = 0;
168 if (!getInstrVecReg(*I, Reg))
169 llvm_unreachable("Need a vector reg");
170 MachineBasicBlock::instr_iterator MII = I->getIterator();
171 if (I->isInsideBundle()) {
172 LLVM_DEBUG(dbgs() << "add to end of bundle\n"; I->dump());
173 while (MBB->instr_end() != MII && MII->isInsideBundle())
174 MII++;
175 } else {
176 LLVM_DEBUG(dbgs() << "add after instruction\n"; I->dump());
177 MII++;
178 }
179 if (MBB->instr_end() == MII)
180 continue;
181
182 if (Reg >= Hexagon::V0 && Reg <= Hexagon::V31) {
183 LLVM_DEBUG(dbgs() << "adding dump for V" << Reg - Hexagon::V0 << '\n');
184 addAsmInstr(MBB, Reg, MII, DL, QII, Fn);
185 } else if (Reg >= Hexagon::W0 && Reg <= Hexagon::W15) {
186 LLVM_DEBUG(dbgs() << "adding dump for W" << Reg - Hexagon::W0 << '\n');
187 addAsmInstr(MBB, Hexagon::V0 + (Reg - Hexagon::W0) * 2 + 1,
188 MII, DL, QII, Fn);
189 addAsmInstr(MBB, Hexagon::V0 + (Reg - Hexagon::W0) * 2,
190 MII, DL, QII, Fn);
191 } else if (Reg >= Hexagon::Q0 && Reg <= Hexagon::Q3) {
192 LLVM_DEBUG(dbgs() << "adding dump for Q" << Reg - Hexagon::Q0 << '\n');
193 addAsmInstr(MBB, Reg, MII, DL, QII, Fn);
194 } else
195 llvm_unreachable("Bad Vector reg");
196 }
197 return Changed;
198 }
199
200 //===----------------------------------------------------------------------===//
201 // Public Constructor Functions
202 //===----------------------------------------------------------------------===//
203 INITIALIZE_PASS(HexagonVectorPrint, "hexagon-vector-print",
204 "Hexagon VectorPrint pass", false, false)
205
createHexagonVectorPrint()206 FunctionPass *llvm::createHexagonVectorPrint() {
207 return new HexagonVectorPrint();
208 }
209