blob: f7dda9607f881565bec447b167ea9b7ee4e891e1 [file] [log] [blame]
Aditya Nandakumar2036f442018-01-25 02:53:06 +00001//===- PatternMatchTest.cpp -----------------------------------------------===//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// 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
Aditya Nandakumar2036f442018-01-25 02:53:06 +00006//
7//===----------------------------------------------------------------------===//
8
Aditya Nandakumarb1c467d2018-04-09 17:30:56 +00009#include "llvm/CodeGen/GlobalISel/ConstantFoldingMIRBuilder.h"
Aditya Nandakumar2036f442018-01-25 02:53:06 +000010#include "llvm/CodeGen/GlobalISel/MIPatternMatch.h"
11#include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h"
12#include "llvm/CodeGen/GlobalISel/Utils.h"
13#include "llvm/CodeGen/MIRParser/MIRParser.h"
14#include "llvm/CodeGen/MachineFunction.h"
15#include "llvm/CodeGen/MachineModuleInfo.h"
16#include "llvm/CodeGen/TargetFrameLowering.h"
17#include "llvm/CodeGen/TargetInstrInfo.h"
18#include "llvm/CodeGen/TargetLowering.h"
19#include "llvm/CodeGen/TargetSubtargetInfo.h"
20#include "llvm/Support/SourceMgr.h"
21#include "llvm/Support/TargetRegistry.h"
22#include "llvm/Support/TargetSelect.h"
23#include "llvm/Target/TargetMachine.h"
24#include "llvm/Target/TargetOptions.h"
25#include "gtest/gtest.h"
26
27using namespace llvm;
28using namespace MIPatternMatch;
29
30namespace {
31
32void initLLVM() {
33 InitializeAllTargets();
34 InitializeAllTargetMCs();
35 InitializeAllAsmPrinters();
36 InitializeAllAsmParsers();
37
38 PassRegistry *Registry = PassRegistry::getPassRegistry();
39 initializeCore(*Registry);
40 initializeCodeGen(*Registry);
41}
42
43/// Create a TargetMachine. As we lack a dedicated always available target for
44/// unittests, we go for "AArch64".
Matthias Braun3d849f62018-11-05 23:49:13 +000045std::unique_ptr<LLVMTargetMachine> createTargetMachine() {
Aditya Nandakumar2036f442018-01-25 02:53:06 +000046 Triple TargetTriple("aarch64--");
47 std::string Error;
48 const Target *T = TargetRegistry::lookupTarget("", TargetTriple, Error);
49 if (!T)
50 return nullptr;
51
52 TargetOptions Options;
Matthias Braun3d849f62018-11-05 23:49:13 +000053 return std::unique_ptr<LLVMTargetMachine>(static_cast<LLVMTargetMachine*>(
54 T->createTargetMachine("AArch64", "", "", Options, None, None,
55 CodeGenOpt::Aggressive)));
Aditya Nandakumar2036f442018-01-25 02:53:06 +000056}
57
58std::unique_ptr<Module> parseMIR(LLVMContext &Context,
59 std::unique_ptr<MIRParser> &MIR,
60 const TargetMachine &TM, StringRef MIRCode,
61 const char *FuncName, MachineModuleInfo &MMI) {
62 SMDiagnostic Diagnostic;
63 std::unique_ptr<MemoryBuffer> MBuffer = MemoryBuffer::getMemBuffer(MIRCode);
64 MIR = createMIRParser(std::move(MBuffer), Context);
65 if (!MIR)
66 return nullptr;
67
68 std::unique_ptr<Module> M = MIR->parseIRModule();
69 if (!M)
70 return nullptr;
71
72 M->setDataLayout(TM.createDataLayout());
73
74 if (MIR->parseMachineFunctions(*M, MMI))
75 return nullptr;
76
77 return M;
78}
79
80std::pair<std::unique_ptr<Module>, std::unique_ptr<MachineModuleInfo>>
Matthias Braun3d849f62018-11-05 23:49:13 +000081createDummyModule(LLVMContext &Context, const LLVMTargetMachine &TM,
Aditya Nandakumar2036f442018-01-25 02:53:06 +000082 StringRef MIRFunc) {
83 SmallString<512> S;
84 StringRef MIRString = (Twine(R"MIR(
85---
86...
87name: func
88registers:
89 - { id: 0, class: _ }
90 - { id: 1, class: _ }
91 - { id: 2, class: _ }
92 - { id: 3, class: _ }
93body: |
94 bb.1:
Puyan Lotfi43e94b12018-01-31 22:04:26 +000095 %0(s64) = COPY $x0
96 %1(s64) = COPY $x1
97 %2(s64) = COPY $x2
Aditya Nandakumar2036f442018-01-25 02:53:06 +000098)MIR") + Twine(MIRFunc) + Twine("...\n"))
99 .toNullTerminatedStringRef(S);
100 std::unique_ptr<MIRParser> MIR;
101 auto MMI = make_unique<MachineModuleInfo>(&TM);
102 std::unique_ptr<Module> M =
103 parseMIR(Context, MIR, TM, MIRString, "func", *MMI);
104 return make_pair(std::move(M), std::move(MMI));
105}
106
107static MachineFunction *getMFFromMMI(const Module *M,
108 const MachineModuleInfo *MMI) {
109 Function *F = M->getFunction("func");
110 auto *MF = MMI->getMachineFunction(*F);
111 return MF;
112}
113
114static void collectCopies(SmallVectorImpl<unsigned> &Copies,
115 MachineFunction *MF) {
116 for (auto &MBB : *MF)
117 for (MachineInstr &MI : MBB) {
118 if (MI.getOpcode() == TargetOpcode::COPY)
119 Copies.push_back(MI.getOperand(0).getReg());
120 }
121}
122
123TEST(PatternMatchInstr, MatchIntConstant) {
124 LLVMContext Context;
Matthias Braun3d849f62018-11-05 23:49:13 +0000125 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000126 if (!TM)
127 return;
128 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
129 MachineFunction *MF =
130 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
131 SmallVector<unsigned, 4> Copies;
132 collectCopies(Copies, MF);
133 MachineBasicBlock *EntryMBB = &*MF->begin();
134 MachineIRBuilder B(*MF);
135 MachineRegisterInfo &MRI = MF->getRegInfo();
136 B.setInsertPt(*EntryMBB, EntryMBB->end());
137 auto MIBCst = B.buildConstant(LLT::scalar(64), 42);
Aditya Nandakumarb808e3a2018-03-13 23:21:13 +0000138 int64_t Cst;
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000139 bool match = mi_match(MIBCst->getOperand(0).getReg(), MRI, m_ICst(Cst));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000140 EXPECT_TRUE(match);
141 EXPECT_EQ(Cst, 42);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000142}
143
144TEST(PatternMatchInstr, MatchBinaryOp) {
145 LLVMContext Context;
Matthias Braun3d849f62018-11-05 23:49:13 +0000146 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000147 if (!TM)
148 return;
149 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
150 MachineFunction *MF =
151 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
152 SmallVector<unsigned, 4> Copies;
153 collectCopies(Copies, MF);
154 MachineBasicBlock *EntryMBB = &*MF->begin();
155 MachineIRBuilder B(*MF);
156 MachineRegisterInfo &MRI = MF->getRegInfo();
157 B.setInsertPt(*EntryMBB, EntryMBB->end());
158 LLT s64 = LLT::scalar(64);
159 auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
160 // Test case for no bind.
161 bool match =
162 mi_match(MIBAdd->getOperand(0).getReg(), MRI, m_GAdd(m_Reg(), m_Reg()));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000163 EXPECT_TRUE(match);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000164 unsigned Src0, Src1, Src2;
165 match = mi_match(MIBAdd->getOperand(0).getReg(), MRI,
166 m_GAdd(m_Reg(Src0), m_Reg(Src1)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000167 EXPECT_TRUE(match);
168 EXPECT_EQ(Src0, Copies[0]);
169 EXPECT_EQ(Src1, Copies[1]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000170
171 // Build MUL(ADD %0, %1), %2
172 auto MIBMul = B.buildMul(s64, MIBAdd, Copies[2]);
173
174 // Try to match MUL.
175 match = mi_match(MIBMul->getOperand(0).getReg(), MRI,
176 m_GMul(m_Reg(Src0), m_Reg(Src1)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000177 EXPECT_TRUE(match);
178 EXPECT_EQ(Src0, MIBAdd->getOperand(0).getReg());
179 EXPECT_EQ(Src1, Copies[2]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000180
181 // Try to match MUL(ADD)
182 match = mi_match(MIBMul->getOperand(0).getReg(), MRI,
183 m_GMul(m_GAdd(m_Reg(Src0), m_Reg(Src1)), m_Reg(Src2)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000184 EXPECT_TRUE(match);
185 EXPECT_EQ(Src0, Copies[0]);
186 EXPECT_EQ(Src1, Copies[1]);
187 EXPECT_EQ(Src2, Copies[2]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000188
189 // Test Commutativity.
190 auto MIBMul2 = B.buildMul(s64, Copies[0], B.buildConstant(s64, 42));
191 // Try to match MUL(Cst, Reg) on src of MUL(Reg, Cst) to validate
192 // commutativity.
Aditya Nandakumarb808e3a2018-03-13 23:21:13 +0000193 int64_t Cst;
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000194 match = mi_match(MIBMul2->getOperand(0).getReg(), MRI,
195 m_GMul(m_ICst(Cst), m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000196 EXPECT_TRUE(match);
197 EXPECT_EQ(Cst, 42);
198 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000199
200 // Make sure commutative doesn't work with something like SUB.
201 auto MIBSub = B.buildSub(s64, Copies[0], B.buildConstant(s64, 42));
202 match = mi_match(MIBSub->getOperand(0).getReg(), MRI,
203 m_GSub(m_ICst(Cst), m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000204 EXPECT_FALSE(match);
Aditya Nandakumar6250b182018-02-13 20:09:13 +0000205
Aditya Nandakumarcef44a22018-12-11 00:48:50 +0000206 auto MIBFMul = B.buildInstr(TargetOpcode::G_FMUL, {s64},
207 {Copies[0], B.buildConstant(s64, 42)});
Aditya Nandakumar6250b182018-02-13 20:09:13 +0000208 // Match and test commutativity for FMUL.
209 match = mi_match(MIBFMul->getOperand(0).getReg(), MRI,
210 m_GFMul(m_ICst(Cst), m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000211 EXPECT_TRUE(match);
212 EXPECT_EQ(Cst, 42);
213 EXPECT_EQ(Src0, Copies[0]);
Volkan Keles02bb1742018-02-14 19:58:36 +0000214
Aditya Nandakumar2980b012018-05-31 19:30:01 +0000215 // FSUB
Aditya Nandakumarcef44a22018-12-11 00:48:50 +0000216 auto MIBFSub = B.buildInstr(TargetOpcode::G_FSUB, {s64},
217 {Copies[0], B.buildConstant(s64, 42)});
Aditya Nandakumar2980b012018-05-31 19:30:01 +0000218 match = mi_match(MIBFSub->getOperand(0).getReg(), MRI,
219 m_GFSub(m_Reg(Src0), m_Reg()));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000220 EXPECT_TRUE(match);
221 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2980b012018-05-31 19:30:01 +0000222
Volkan Keles02bb1742018-02-14 19:58:36 +0000223 // Build AND %0, %1
224 auto MIBAnd = B.buildAnd(s64, Copies[0], Copies[1]);
225 // Try to match AND.
226 match = mi_match(MIBAnd->getOperand(0).getReg(), MRI,
227 m_GAnd(m_Reg(Src0), m_Reg(Src1)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000228 EXPECT_TRUE(match);
229 EXPECT_EQ(Src0, Copies[0]);
230 EXPECT_EQ(Src1, Copies[1]);
Volkan Keles02bb1742018-02-14 19:58:36 +0000231
232 // Build OR %0, %1
233 auto MIBOr = B.buildOr(s64, Copies[0], Copies[1]);
234 // Try to match OR.
235 match = mi_match(MIBOr->getOperand(0).getReg(), MRI,
236 m_GOr(m_Reg(Src0), m_Reg(Src1)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000237 EXPECT_TRUE(match);
238 EXPECT_EQ(Src0, Copies[0]);
239 EXPECT_EQ(Src1, Copies[1]);
Aditya Nandakumarb1c467d2018-04-09 17:30:56 +0000240
241 // Try to use the FoldableInstructionsBuilder to build binary ops.
242 ConstantFoldingMIRBuilder CFB(B.getState());
243 LLT s32 = LLT::scalar(32);
244 auto MIBCAdd =
245 CFB.buildAdd(s32, CFB.buildConstant(s32, 0), CFB.buildConstant(s32, 1));
246 // This should be a constant now.
247 match = mi_match(MIBCAdd->getOperand(0).getReg(), MRI, m_ICst(Cst));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000248 EXPECT_TRUE(match);
249 EXPECT_EQ(Cst, 1);
Aditya Nandakumarb1c467d2018-04-09 17:30:56 +0000250 auto MIBCAdd1 =
Aditya Nandakumarcef44a22018-12-11 00:48:50 +0000251 CFB.buildInstr(TargetOpcode::G_ADD, {s32},
252 {CFB.buildConstant(s32, 0), CFB.buildConstant(s32, 1)});
Aditya Nandakumarb1c467d2018-04-09 17:30:56 +0000253 // This should be a constant now.
254 match = mi_match(MIBCAdd1->getOperand(0).getReg(), MRI, m_ICst(Cst));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000255 EXPECT_TRUE(match);
256 EXPECT_EQ(Cst, 1);
Aditya Nandakumarb1c467d2018-04-09 17:30:56 +0000257
258 // Try one of the other constructors of MachineIRBuilder to make sure it's
259 // compatible.
260 ConstantFoldingMIRBuilder CFB1(*MF);
261 CFB1.setInsertPt(*EntryMBB, EntryMBB->end());
262 auto MIBCSub =
Aditya Nandakumarcef44a22018-12-11 00:48:50 +0000263 CFB1.buildInstr(TargetOpcode::G_SUB, {s32},
264 {CFB1.buildConstant(s32, 1), CFB1.buildConstant(s32, 1)});
Aditya Nandakumarb1c467d2018-04-09 17:30:56 +0000265 // This should be a constant now.
266 match = mi_match(MIBCSub->getOperand(0).getReg(), MRI, m_ICst(Cst));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000267 EXPECT_TRUE(match);
268 EXPECT_EQ(Cst, 0);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000269}
270
Volkan Keles2bc42e92018-03-05 22:31:55 +0000271TEST(PatternMatchInstr, MatchFPUnaryOp) {
272 LLVMContext Context;
Matthias Braun3d849f62018-11-05 23:49:13 +0000273 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
Volkan Keles2bc42e92018-03-05 22:31:55 +0000274 if (!TM)
275 return;
276 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
277 MachineFunction *MF =
278 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
279 SmallVector<unsigned, 4> Copies;
280 collectCopies(Copies, MF);
281 MachineBasicBlock *EntryMBB = &*MF->begin();
282 MachineIRBuilder B(*MF);
283 MachineRegisterInfo &MRI = MF->getRegInfo();
284 B.setInsertPt(*EntryMBB, EntryMBB->end());
285
286 // Truncate s64 to s32.
287 LLT s32 = LLT::scalar(32);
288 auto Copy0s32 = B.buildFPTrunc(s32, Copies[0]);
289
290 // Match G_FABS.
Aditya Nandakumarcef44a22018-12-11 00:48:50 +0000291 auto MIBFabs = B.buildInstr(TargetOpcode::G_FABS, {s32}, {Copy0s32});
Volkan Keles2bc42e92018-03-05 22:31:55 +0000292 bool match = mi_match(MIBFabs->getOperand(0).getReg(), MRI, m_GFabs(m_Reg()));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000293 EXPECT_TRUE(match);
Aditya Nandakumar2980b012018-05-31 19:30:01 +0000294
Volkan Keles2bc42e92018-03-05 22:31:55 +0000295 unsigned Src;
Aditya Nandakumarcef44a22018-12-11 00:48:50 +0000296 auto MIBFNeg = B.buildInstr(TargetOpcode::G_FNEG, {s32}, {Copy0s32});
Aditya Nandakumar2980b012018-05-31 19:30:01 +0000297 match = mi_match(MIBFNeg->getOperand(0).getReg(), MRI, m_GFNeg(m_Reg(Src)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000298 EXPECT_TRUE(match);
299 EXPECT_EQ(Src, Copy0s32->getOperand(0).getReg());
Aditya Nandakumar2980b012018-05-31 19:30:01 +0000300
Volkan Keles2bc42e92018-03-05 22:31:55 +0000301 match = mi_match(MIBFabs->getOperand(0).getReg(), MRI, m_GFabs(m_Reg(Src)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000302 EXPECT_TRUE(match);
303 EXPECT_EQ(Src, Copy0s32->getOperand(0).getReg());
Aditya Nandakumar91fc4e02018-03-09 17:31:51 +0000304
305 // Build and match FConstant.
306 auto MIBFCst = B.buildFConstant(s32, .5);
307 const ConstantFP *TmpFP{};
308 match = mi_match(MIBFCst->getOperand(0).getReg(), MRI, m_GFCst(TmpFP));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000309 EXPECT_TRUE(match);
310 EXPECT_TRUE(TmpFP);
Aditya Nandakumar91fc4e02018-03-09 17:31:51 +0000311 APFloat APF((float).5);
312 auto *CFP = ConstantFP::get(Context, APF);
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000313 EXPECT_EQ(CFP, TmpFP);
Aditya Nandakumar91fc4e02018-03-09 17:31:51 +0000314
315 // Build double float.
316 LLT s64 = LLT::scalar(64);
317 auto MIBFCst64 = B.buildFConstant(s64, .5);
318 const ConstantFP *TmpFP64{};
319 match = mi_match(MIBFCst64->getOperand(0).getReg(), MRI, m_GFCst(TmpFP64));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000320 EXPECT_TRUE(match);
321 EXPECT_TRUE(TmpFP64);
Aditya Nandakumar91fc4e02018-03-09 17:31:51 +0000322 APFloat APF64(.5);
323 auto CFP64 = ConstantFP::get(Context, APF64);
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000324 EXPECT_EQ(CFP64, TmpFP64);
325 EXPECT_NE(TmpFP64, TmpFP);
Aditya Nandakumar91fc4e02018-03-09 17:31:51 +0000326
327 // Build half float.
328 LLT s16 = LLT::scalar(16);
329 auto MIBFCst16 = B.buildFConstant(s16, .5);
330 const ConstantFP *TmpFP16{};
331 match = mi_match(MIBFCst16->getOperand(0).getReg(), MRI, m_GFCst(TmpFP16));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000332 EXPECT_TRUE(match);
333 EXPECT_TRUE(TmpFP16);
Aditya Nandakumar91fc4e02018-03-09 17:31:51 +0000334 bool Ignored;
335 APFloat APF16(.5);
336 APF16.convert(APFloat::IEEEhalf(), APFloat::rmNearestTiesToEven, &Ignored);
337 auto CFP16 = ConstantFP::get(Context, APF16);
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000338 EXPECT_EQ(TmpFP16, CFP16);
339 EXPECT_NE(TmpFP16, TmpFP);
Volkan Keles2bc42e92018-03-05 22:31:55 +0000340}
341
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000342TEST(PatternMatchInstr, MatchExtendsTrunc) {
343 LLVMContext Context;
Matthias Braun3d849f62018-11-05 23:49:13 +0000344 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000345 if (!TM)
346 return;
347 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
348 MachineFunction *MF =
349 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
350 SmallVector<unsigned, 4> Copies;
351 collectCopies(Copies, MF);
352 MachineBasicBlock *EntryMBB = &*MF->begin();
353 MachineIRBuilder B(*MF);
354 MachineRegisterInfo &MRI = MF->getRegInfo();
355 B.setInsertPt(*EntryMBB, EntryMBB->end());
356 LLT s64 = LLT::scalar(64);
357 LLT s32 = LLT::scalar(32);
358
359 auto MIBTrunc = B.buildTrunc(s32, Copies[0]);
360 auto MIBAExt = B.buildAnyExt(s64, MIBTrunc);
361 auto MIBZExt = B.buildZExt(s64, MIBTrunc);
362 auto MIBSExt = B.buildSExt(s64, MIBTrunc);
363 unsigned Src0;
364 bool match =
365 mi_match(MIBTrunc->getOperand(0).getReg(), MRI, m_GTrunc(m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000366 EXPECT_TRUE(match);
367 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000368 match =
369 mi_match(MIBAExt->getOperand(0).getReg(), MRI, m_GAnyExt(m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000370 EXPECT_TRUE(match);
371 EXPECT_EQ(Src0, MIBTrunc->getOperand(0).getReg());
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000372
373 match = mi_match(MIBSExt->getOperand(0).getReg(), MRI, m_GSExt(m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000374 EXPECT_TRUE(match);
375 EXPECT_EQ(Src0, MIBTrunc->getOperand(0).getReg());
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000376
377 match = mi_match(MIBZExt->getOperand(0).getReg(), MRI, m_GZExt(m_Reg(Src0)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000378 EXPECT_TRUE(match);
379 EXPECT_EQ(Src0, MIBTrunc->getOperand(0).getReg());
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000380
381 // Match ext(trunc src)
382 match = mi_match(MIBAExt->getOperand(0).getReg(), MRI,
383 m_GAnyExt(m_GTrunc(m_Reg(Src0))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000384 EXPECT_TRUE(match);
385 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000386
387 match = mi_match(MIBSExt->getOperand(0).getReg(), MRI,
388 m_GSExt(m_GTrunc(m_Reg(Src0))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000389 EXPECT_TRUE(match);
390 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000391
392 match = mi_match(MIBZExt->getOperand(0).getReg(), MRI,
393 m_GZExt(m_GTrunc(m_Reg(Src0))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000394 EXPECT_TRUE(match);
395 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000396}
397
398TEST(PatternMatchInstr, MatchSpecificType) {
399 LLVMContext Context;
Matthias Braun3d849f62018-11-05 23:49:13 +0000400 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000401 if (!TM)
402 return;
403 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
404 MachineFunction *MF =
405 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
406 SmallVector<unsigned, 4> Copies;
407 collectCopies(Copies, MF);
408 MachineBasicBlock *EntryMBB = &*MF->begin();
409 MachineIRBuilder B(*MF);
410 MachineRegisterInfo &MRI = MF->getRegInfo();
411 B.setInsertPt(*EntryMBB, EntryMBB->end());
Volkan Keles02bb1742018-02-14 19:58:36 +0000412
413 // Try to match a 64bit add.
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000414 LLT s64 = LLT::scalar(64);
415 LLT s32 = LLT::scalar(32);
416 auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000417 EXPECT_FALSE(mi_match(MIBAdd->getOperand(0).getReg(), MRI,
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000418 m_GAdd(m_SpecificType(s32), m_Reg())));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000419 EXPECT_TRUE(mi_match(MIBAdd->getOperand(0).getReg(), MRI,
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000420 m_GAdd(m_SpecificType(s64), m_Reg())));
Volkan Keles02bb1742018-02-14 19:58:36 +0000421
422 // Try to match the destination type of a bitcast.
423 LLT v2s32 = LLT::vector(2, 32);
424 auto MIBCast = B.buildCast(v2s32, Copies[0]);
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000425 EXPECT_TRUE(
Aditya Nandakumarbab2d3e2018-02-19 23:11:53 +0000426 mi_match(MIBCast->getOperand(0).getReg(), MRI, m_GBitcast(m_Reg())));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000427 EXPECT_TRUE(
Volkan Keles02bb1742018-02-14 19:58:36 +0000428 mi_match(MIBCast->getOperand(0).getReg(), MRI, m_SpecificType(v2s32)));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000429 EXPECT_TRUE(
Volkan Keles02bb1742018-02-14 19:58:36 +0000430 mi_match(MIBCast->getOperand(1).getReg(), MRI, m_SpecificType(s64)));
Aditya Nandakumarbab2d3e2018-02-19 23:11:53 +0000431
432 // Build a PTRToInt and INTTOPTR and match and test them.
433 LLT PtrTy = LLT::pointer(0, 64);
434 auto MIBIntToPtr = B.buildCast(PtrTy, Copies[0]);
435 auto MIBPtrToInt = B.buildCast(s64, MIBIntToPtr);
436 unsigned Src0;
437
438 // match the ptrtoint(inttoptr reg)
439 bool match = mi_match(MIBPtrToInt->getOperand(0).getReg(), MRI,
440 m_GPtrToInt(m_GIntToPtr(m_Reg(Src0))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000441 EXPECT_TRUE(match);
442 EXPECT_EQ(Src0, Copies[0]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000443}
444
445TEST(PatternMatchInstr, MatchCombinators) {
446 LLVMContext Context;
Matthias Braun3d849f62018-11-05 23:49:13 +0000447 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000448 if (!TM)
449 return;
450 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
451 MachineFunction *MF =
452 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
453 SmallVector<unsigned, 4> Copies;
454 collectCopies(Copies, MF);
455 MachineBasicBlock *EntryMBB = &*MF->begin();
456 MachineIRBuilder B(*MF);
457 MachineRegisterInfo &MRI = MF->getRegInfo();
458 B.setInsertPt(*EntryMBB, EntryMBB->end());
459 LLT s64 = LLT::scalar(64);
460 LLT s32 = LLT::scalar(32);
461 auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
462 unsigned Src0, Src1;
463 bool match =
464 mi_match(MIBAdd->getOperand(0).getReg(), MRI,
465 m_all_of(m_SpecificType(s64), m_GAdd(m_Reg(Src0), m_Reg(Src1))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000466 EXPECT_TRUE(match);
467 EXPECT_EQ(Src0, Copies[0]);
468 EXPECT_EQ(Src1, Copies[1]);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000469 // Check for s32 (which should fail).
470 match =
471 mi_match(MIBAdd->getOperand(0).getReg(), MRI,
472 m_all_of(m_SpecificType(s32), m_GAdd(m_Reg(Src0), m_Reg(Src1))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000473 EXPECT_FALSE(match);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000474 match =
475 mi_match(MIBAdd->getOperand(0).getReg(), MRI,
476 m_any_of(m_SpecificType(s32), m_GAdd(m_Reg(Src0), m_Reg(Src1))));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000477 EXPECT_TRUE(match);
478 EXPECT_EQ(Src0, Copies[0]);
479 EXPECT_EQ(Src1, Copies[1]);
Aditya Nandakumarcf85f312018-02-23 01:01:59 +0000480
481 // Match a case where none of the predicates hold true.
482 match = mi_match(
483 MIBAdd->getOperand(0).getReg(), MRI,
484 m_any_of(m_SpecificType(LLT::scalar(16)), m_GSub(m_Reg(), m_Reg())));
Matt Arsenaultb3e86702019-02-04 18:58:27 +0000485 EXPECT_FALSE(match);
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000486}
Aditya Nandakumar5c7fcbd2019-06-14 17:19:37 +0000487
488TEST(PatternMatchInstr, MatchMiscellaneous) {
489 LLVMContext Context;
490 std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
491 if (!TM)
492 return;
493 auto ModuleMMIPair = createDummyModule(Context, *TM, "");
494 MachineFunction *MF =
495 getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
496 SmallVector<unsigned, 4> Copies;
497 collectCopies(Copies, MF);
498 MachineBasicBlock *EntryMBB = &*MF->begin();
499 MachineIRBuilder B(*MF);
500 MachineRegisterInfo &MRI = MF->getRegInfo();
501 B.setInsertPt(*EntryMBB, EntryMBB->end());
502 LLT s64 = LLT::scalar(64);
503 auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
504 // Make multiple uses of this add.
505 B.buildCast(LLT::pointer(0, 32), MIBAdd);
506 B.buildCast(LLT::pointer(1, 32), MIBAdd);
507 bool match = mi_match(MIBAdd.getReg(0), MRI, m_GAdd(m_Reg(), m_Reg()));
508 EXPECT_TRUE(match);
509 match = mi_match(MIBAdd.getReg(0), MRI, m_OneUse(m_GAdd(m_Reg(), m_Reg())));
510 EXPECT_FALSE(match);
511}
Aditya Nandakumar2036f442018-01-25 02:53:06 +0000512} // namespace
513
514int main(int argc, char **argv) {
515 ::testing::InitGoogleTest(&argc, argv);
516 initLLVM();
517 return RUN_ALL_TESTS();
518}