blob: 6bdcf4de7e233126fc4e20343c2cd576dcdcab4d [file] [log] [blame]
Arnold Schwaighofere972d032013-05-05 01:54:46 +00001//===---- llvm/unittest/IR/PatternMatch.cpp - PatternMatch unit tests ----===//
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
Chandler Carruth9a67b072017-06-06 11:06:56 +000010#include "llvm/IR/PatternMatch.h"
Arnold Schwaighofere972d032013-05-05 01:54:46 +000011#include "llvm/ADT/STLExtras.h"
12#include "llvm/Analysis/ValueTracking.h"
13#include "llvm/IR/BasicBlock.h"
14#include "llvm/IR/Constants.h"
15#include "llvm/IR/DataLayout.h"
16#include "llvm/IR/DerivedTypes.h"
Chandler Carruth4603e96a2014-01-05 02:07:20 +000017#include "llvm/IR/Function.h"
Arnold Schwaighofere972d032013-05-05 01:54:46 +000018#include "llvm/IR/IRBuilder.h"
Chandler Carruth4603e96a2014-01-05 02:07:20 +000019#include "llvm/IR/Instructions.h"
Arnold Schwaighofere972d032013-05-05 01:54:46 +000020#include "llvm/IR/LLVMContext.h"
21#include "llvm/IR/MDBuilder.h"
Chandler Carruth4603e96a2014-01-05 02:07:20 +000022#include "llvm/IR/Module.h"
Chandler Carruth64396b02014-03-04 12:05:47 +000023#include "llvm/IR/NoFolder.h"
Arnold Schwaighofere972d032013-05-05 01:54:46 +000024#include "llvm/IR/Operator.h"
Chandler Carruth4603e96a2014-01-05 02:07:20 +000025#include "llvm/IR/Type.h"
Arnold Schwaighofere972d032013-05-05 01:54:46 +000026#include "gtest/gtest.h"
27
Chandler Carruth4603e96a2014-01-05 02:07:20 +000028using namespace llvm;
Arnold Schwaighofere972d032013-05-05 01:54:46 +000029using namespace llvm::PatternMatch;
30
Arnold Schwaighofere972d032013-05-05 01:54:46 +000031namespace {
32
Chandler Carruth4603e96a2014-01-05 02:07:20 +000033struct PatternMatchTest : ::testing::Test {
34 LLVMContext Ctx;
Ahmed Charles56440fd2014-03-06 05:51:42 +000035 std::unique_ptr<Module> M;
Chandler Carruth4603e96a2014-01-05 02:07:20 +000036 Function *F;
37 BasicBlock *BB;
Mehdi Aminiba9fba82016-03-13 21:05:13 +000038 IRBuilder<NoFolder> IRB;
Arnold Schwaighofere972d032013-05-05 01:54:46 +000039
Chandler Carruth4603e96a2014-01-05 02:07:20 +000040 PatternMatchTest()
41 : M(new Module("PatternMatchTestModule", Ctx)),
42 F(Function::Create(
43 FunctionType::get(Type::getVoidTy(Ctx), /* IsVarArg */ false),
44 Function::ExternalLinkage, "f", M.get())),
Chandler Carruth91f4e602014-01-05 02:23:11 +000045 BB(BasicBlock::Create(Ctx, "entry", F)), IRB(BB) {}
Chandler Carruth4603e96a2014-01-05 02:07:20 +000046};
Arnold Schwaighofere972d032013-05-05 01:54:46 +000047
Chandler Carruthcde91b42014-01-05 09:14:53 +000048TEST_F(PatternMatchTest, OneUse) {
49 // Build up a little tree of values:
50 //
51 // One = (1 + 2) + 42
52 // Two = One + 42
53 // Leaf = (Two + 8) + (Two + 13)
54 Value *One = IRB.CreateAdd(IRB.CreateAdd(IRB.getInt32(1), IRB.getInt32(2)),
55 IRB.getInt32(42));
56 Value *Two = IRB.CreateAdd(One, IRB.getInt32(42));
57 Value *Leaf = IRB.CreateAdd(IRB.CreateAdd(Two, IRB.getInt32(8)),
58 IRB.CreateAdd(Two, IRB.getInt32(13)));
59 Value *V;
60
61 EXPECT_TRUE(m_OneUse(m_Value(V)).match(One));
62 EXPECT_EQ(One, V);
63
64 EXPECT_FALSE(m_OneUse(m_Value()).match(Two));
65 EXPECT_FALSE(m_OneUse(m_Value()).match(Leaf));
66}
67
Chandler Carruth4603e96a2014-01-05 02:07:20 +000068TEST_F(PatternMatchTest, FloatingPointOrderedMin) {
Chandler Carruth91f4e602014-01-05 02:23:11 +000069 Type *FltTy = IRB.getFloatTy();
Arnold Schwaighofere972d032013-05-05 01:54:46 +000070 Value *L = ConstantFP::get(FltTy, 1.0);
71 Value *R = ConstantFP::get(FltTy, 2.0);
Chandler Carruth4603e96a2014-01-05 02:07:20 +000072 Value *MatchL, *MatchR;
Arnold Schwaighofere972d032013-05-05 01:54:46 +000073
74 // Test OLT.
Chandler Carruth91f4e602014-01-05 02:23:11 +000075 EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
76 .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +000077 EXPECT_EQ(L, MatchL);
78 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +000079
80 // Test OLE.
Chandler Carruth91f4e602014-01-05 02:23:11 +000081 EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
82 .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +000083 EXPECT_EQ(L, MatchL);
84 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +000085
86 // Test no match on OGE.
Chandler Carruth91f4e602014-01-05 02:23:11 +000087 EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
88 .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +000089
90 // Test no match on OGT.
Chandler Carruth91f4e602014-01-05 02:23:11 +000091 EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
92 .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +000093
Craig Topperc6635522017-06-13 17:18:45 +000094 // Test inverted selects. Note, that this "inverts" the ordering, e.g.:
95 // %cmp = fcmp oge L, R
96 // %min = select %cmp R, L
97 // Given L == NaN
98 // the above is expanded to %cmp == false ==> %min = L
99 // which is true for UnordFMin, not OrdFMin, so test that:
100
101 // [OU]GE with inverted select.
102 EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
Chandler Carruth91f4e602014-01-05 02:23:11 +0000103 .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), R, L)));
Craig Topperc6635522017-06-13 17:18:45 +0000104 EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
105 .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000106 EXPECT_EQ(L, MatchL);
107 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000108
Craig Topperc6635522017-06-13 17:18:45 +0000109 // [OU]GT with inverted select.
110 EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
Chandler Carruth91f4e602014-01-05 02:23:11 +0000111 .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), R, L)));
Craig Topperc6635522017-06-13 17:18:45 +0000112 EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR))
113 .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000114 EXPECT_EQ(L, MatchL);
115 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000116}
117
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000118TEST_F(PatternMatchTest, FloatingPointOrderedMax) {
Chandler Carruth91f4e602014-01-05 02:23:11 +0000119 Type *FltTy = IRB.getFloatTy();
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000120 Value *L = ConstantFP::get(FltTy, 1.0);
121 Value *R = ConstantFP::get(FltTy, 2.0);
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000122 Value *MatchL, *MatchR;
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000123
124 // Test OGT.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000125 EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
126 .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000127 EXPECT_EQ(L, MatchL);
128 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000129
130 // Test OGE.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000131 EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
132 .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000133 EXPECT_EQ(L, MatchL);
134 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000135
136 // Test no match on OLE.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000137 EXPECT_FALSE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
138 .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000139
140 // Test no match on OLT.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000141 EXPECT_FALSE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
142 .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000143
Craig Topperc6635522017-06-13 17:18:45 +0000144
145 // Test inverted selects. Note, that this "inverts" the ordering, e.g.:
146 // %cmp = fcmp ole L, R
147 // %max = select %cmp, R, L
148 // Given L == NaN,
149 // the above is expanded to %cmp == false ==> %max == L
150 // which is true for UnordFMax, not OrdFMax, so test that:
151
152 // [OU]LE with inverted select.
153 EXPECT_FALSE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
154 .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), R, L)));
Chandler Carruth91f4e602014-01-05 02:23:11 +0000155 EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
Craig Topperc6635522017-06-13 17:18:45 +0000156 .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000157 EXPECT_EQ(L, MatchL);
158 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000159
Craig Topperc6635522017-06-13 17:18:45 +0000160 // [OUT]LT with inverted select.
161 EXPECT_FALSE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
162 .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), R, L)));
Chandler Carruth91f4e602014-01-05 02:23:11 +0000163 EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR))
Craig Topperc6635522017-06-13 17:18:45 +0000164 .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), R, L)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000165 EXPECT_EQ(L, MatchL);
166 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000167}
168
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000169TEST_F(PatternMatchTest, FloatingPointUnorderedMin) {
Chandler Carruth91f4e602014-01-05 02:23:11 +0000170 Type *FltTy = IRB.getFloatTy();
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000171 Value *L = ConstantFP::get(FltTy, 1.0);
172 Value *R = ConstantFP::get(FltTy, 2.0);
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000173 Value *MatchL, *MatchR;
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000174
175 // Test ULT.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000176 EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
177 .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000178 EXPECT_EQ(L, MatchL);
179 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000180
181 // Test ULE.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000182 EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
183 .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000184 EXPECT_EQ(L, MatchL);
185 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000186
187 // Test no match on UGE.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000188 EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
189 .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000190
191 // Test no match on UGT.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000192 EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
193 .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000194
Craig Topperc6635522017-06-13 17:18:45 +0000195 // Test inverted selects. Note, that this "inverts" the ordering, e.g.:
196 // %cmp = fcmp uge L, R
197 // %min = select %cmp R, L
198 // Given L == NaN
199 // the above is expanded to %cmp == true ==> %min = R
200 // which is true for OrdFMin, not UnordFMin, so test that:
201
202 // [UO]GE with inverted select.
203 EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
Chandler Carruth91f4e602014-01-05 02:23:11 +0000204 .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), R, L)));
Craig Topperc6635522017-06-13 17:18:45 +0000205 EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
206 .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000207 EXPECT_EQ(L, MatchL);
208 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000209
Craig Topperc6635522017-06-13 17:18:45 +0000210 // [UO]GT with inverted select.
211 EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
Chandler Carruth91f4e602014-01-05 02:23:11 +0000212 .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), R, L)));
Craig Topperc6635522017-06-13 17:18:45 +0000213 EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR))
214 .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000215 EXPECT_EQ(L, MatchL);
216 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000217}
218
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000219TEST_F(PatternMatchTest, FloatingPointUnorderedMax) {
Chandler Carruth91f4e602014-01-05 02:23:11 +0000220 Type *FltTy = IRB.getFloatTy();
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000221 Value *L = ConstantFP::get(FltTy, 1.0);
222 Value *R = ConstantFP::get(FltTy, 2.0);
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000223 Value *MatchL, *MatchR;
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000224
225 // Test UGT.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000226 EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
227 .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000228 EXPECT_EQ(L, MatchL);
229 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000230
231 // Test UGE.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000232 EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
233 .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), L, R)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000234 EXPECT_EQ(L, MatchL);
235 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000236
237 // Test no match on ULE.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000238 EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
239 .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000240
241 // Test no match on ULT.
Chandler Carruth91f4e602014-01-05 02:23:11 +0000242 EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
243 .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), L, R)));
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000244
Craig Topperc6635522017-06-13 17:18:45 +0000245 // Test inverted selects. Note, that this "inverts" the ordering, e.g.:
246 // %cmp = fcmp ule L, R
247 // %max = select %cmp R, L
248 // Given L == NaN
249 // the above is expanded to %cmp == true ==> %max = R
250 // which is true for OrdFMax, not UnordFMax, so test that:
251
252 // [UO]LE with inverted select.
253 EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
Chandler Carruth91f4e602014-01-05 02:23:11 +0000254 .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), R, L)));
Craig Topperc6635522017-06-13 17:18:45 +0000255 EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
256 .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000257 EXPECT_EQ(L, MatchL);
258 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000259
Craig Topperc6635522017-06-13 17:18:45 +0000260 // [UO]LT with inverted select.
261 EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
Chandler Carruth91f4e602014-01-05 02:23:11 +0000262 .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), R, L)));
Craig Topperc6635522017-06-13 17:18:45 +0000263 EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR))
264 .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), R, L)));
Chandler Carruth4603e96a2014-01-05 02:07:20 +0000265 EXPECT_EQ(L, MatchL);
266 EXPECT_EQ(R, MatchR);
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000267}
268
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000269TEST_F(PatternMatchTest, OverflowingBinOps) {
270 Value *L = IRB.getInt32(1);
271 Value *R = IRB.getInt32(2);
272 Value *MatchL, *MatchR;
273
274 EXPECT_TRUE(
275 m_NSWAdd(m_Value(MatchL), m_Value(MatchR)).match(IRB.CreateNSWAdd(L, R)));
276 EXPECT_EQ(L, MatchL);
277 EXPECT_EQ(R, MatchR);
Craig Topper66f09ad2014-06-08 22:29:17 +0000278 MatchL = MatchR = nullptr;
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000279 EXPECT_TRUE(
280 m_NSWSub(m_Value(MatchL), m_Value(MatchR)).match(IRB.CreateNSWSub(L, R)));
281 EXPECT_EQ(L, MatchL);
282 EXPECT_EQ(R, MatchR);
Craig Topper66f09ad2014-06-08 22:29:17 +0000283 MatchL = MatchR = nullptr;
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000284 EXPECT_TRUE(
285 m_NSWMul(m_Value(MatchL), m_Value(MatchR)).match(IRB.CreateNSWMul(L, R)));
286 EXPECT_EQ(L, MatchL);
287 EXPECT_EQ(R, MatchR);
Craig Topper66f09ad2014-06-08 22:29:17 +0000288 MatchL = MatchR = nullptr;
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000289 EXPECT_TRUE(m_NSWShl(m_Value(MatchL), m_Value(MatchR)).match(
290 IRB.CreateShl(L, R, "", /* NUW */ false, /* NSW */ true)));
291 EXPECT_EQ(L, MatchL);
292 EXPECT_EQ(R, MatchR);
293
294 EXPECT_TRUE(
295 m_NUWAdd(m_Value(MatchL), m_Value(MatchR)).match(IRB.CreateNUWAdd(L, R)));
296 EXPECT_EQ(L, MatchL);
297 EXPECT_EQ(R, MatchR);
Craig Topper66f09ad2014-06-08 22:29:17 +0000298 MatchL = MatchR = nullptr;
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000299 EXPECT_TRUE(
300 m_NUWSub(m_Value(MatchL), m_Value(MatchR)).match(IRB.CreateNUWSub(L, R)));
301 EXPECT_EQ(L, MatchL);
302 EXPECT_EQ(R, MatchR);
Craig Topper66f09ad2014-06-08 22:29:17 +0000303 MatchL = MatchR = nullptr;
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000304 EXPECT_TRUE(
305 m_NUWMul(m_Value(MatchL), m_Value(MatchR)).match(IRB.CreateNUWMul(L, R)));
306 EXPECT_EQ(L, MatchL);
307 EXPECT_EQ(R, MatchR);
Craig Topper66f09ad2014-06-08 22:29:17 +0000308 MatchL = MatchR = nullptr;
Chandler Carruthc77d50a2014-01-05 03:28:29 +0000309 EXPECT_TRUE(m_NUWShl(m_Value(MatchL), m_Value(MatchR)).match(
310 IRB.CreateShl(L, R, "", /* NUW */ true, /* NSW */ false)));
311 EXPECT_EQ(L, MatchL);
312 EXPECT_EQ(R, MatchR);
313
314 EXPECT_FALSE(m_NSWAdd(m_Value(), m_Value()).match(IRB.CreateAdd(L, R)));
315 EXPECT_FALSE(m_NSWAdd(m_Value(), m_Value()).match(IRB.CreateNUWAdd(L, R)));
316 EXPECT_FALSE(m_NSWAdd(m_Value(), m_Value()).match(IRB.CreateNSWSub(L, R)));
317 EXPECT_FALSE(m_NSWSub(m_Value(), m_Value()).match(IRB.CreateSub(L, R)));
318 EXPECT_FALSE(m_NSWSub(m_Value(), m_Value()).match(IRB.CreateNUWSub(L, R)));
319 EXPECT_FALSE(m_NSWSub(m_Value(), m_Value()).match(IRB.CreateNSWAdd(L, R)));
320 EXPECT_FALSE(m_NSWMul(m_Value(), m_Value()).match(IRB.CreateMul(L, R)));
321 EXPECT_FALSE(m_NSWMul(m_Value(), m_Value()).match(IRB.CreateNUWMul(L, R)));
322 EXPECT_FALSE(m_NSWMul(m_Value(), m_Value()).match(IRB.CreateNSWAdd(L, R)));
323 EXPECT_FALSE(m_NSWShl(m_Value(), m_Value()).match(IRB.CreateShl(L, R)));
324 EXPECT_FALSE(m_NSWShl(m_Value(), m_Value()).match(
325 IRB.CreateShl(L, R, "", /* NUW */ true, /* NSW */ false)));
326 EXPECT_FALSE(m_NSWShl(m_Value(), m_Value()).match(IRB.CreateNSWAdd(L, R)));
327
328 EXPECT_FALSE(m_NUWAdd(m_Value(), m_Value()).match(IRB.CreateAdd(L, R)));
329 EXPECT_FALSE(m_NUWAdd(m_Value(), m_Value()).match(IRB.CreateNSWAdd(L, R)));
330 EXPECT_FALSE(m_NUWAdd(m_Value(), m_Value()).match(IRB.CreateNUWSub(L, R)));
331 EXPECT_FALSE(m_NUWSub(m_Value(), m_Value()).match(IRB.CreateSub(L, R)));
332 EXPECT_FALSE(m_NUWSub(m_Value(), m_Value()).match(IRB.CreateNSWSub(L, R)));
333 EXPECT_FALSE(m_NUWSub(m_Value(), m_Value()).match(IRB.CreateNUWAdd(L, R)));
334 EXPECT_FALSE(m_NUWMul(m_Value(), m_Value()).match(IRB.CreateMul(L, R)));
335 EXPECT_FALSE(m_NUWMul(m_Value(), m_Value()).match(IRB.CreateNSWMul(L, R)));
336 EXPECT_FALSE(m_NUWMul(m_Value(), m_Value()).match(IRB.CreateNUWAdd(L, R)));
337 EXPECT_FALSE(m_NUWShl(m_Value(), m_Value()).match(IRB.CreateShl(L, R)));
338 EXPECT_FALSE(m_NUWShl(m_Value(), m_Value()).match(
339 IRB.CreateShl(L, R, "", /* NUW */ false, /* NSW */ true)));
340 EXPECT_FALSE(m_NUWShl(m_Value(), m_Value()).match(IRB.CreateNUWAdd(L, R)));
341}
342
Daniel Neilson45796f62018-03-28 15:39:00 +0000343TEST_F(PatternMatchTest, VectorOps) {
344 // Build up small tree of vector operations
345 //
346 // Val = 0 + 1
347 // Val2 = Val + 3
348 // VI1 = insertelement <2 x i8> undef, i8 1, i32 0 = <1, undef>
349 // VI2 = insertelement <2 x i8> %VI1, i8 %Val2, i8 %Val = <1, 4>
350 // VI3 = insertelement <2 x i8> %VI1, i8 %Val2, i32 1 = <1, 4>
351 // VI4 = insertelement <2 x i8> %VI1, i8 2, i8 %Val = <1, 2>
352 //
353 // SI1 = shufflevector <2 x i8> %VI1, <2 x i8> undef, zeroinitializer
354 // SI2 = shufflevector <2 x i8> %VI3, <2 x i8> %VI4, <2 x i8> <i8 0, i8 2>
355 // SI3 = shufflevector <2 x i8> %VI3, <2 x i8> undef, zeroinitializer
356 // SI4 = shufflevector <2 x i8> %VI4, <2 x i8> undef, zeroinitializer
357 //
358 // SP1 = VectorSplat(2, i8 2)
359 // SP2 = VectorSplat(2, i8 %Val)
360 Type *VecTy = VectorType::get(IRB.getInt8Ty(), 2);
361 Type *i32 = IRB.getInt32Ty();
362 Type *i32VecTy = VectorType::get(i32, 2);
363
364 Value *Val = IRB.CreateAdd(IRB.getInt8(0), IRB.getInt8(1));
365 Value *Val2 = IRB.CreateAdd(Val, IRB.getInt8(3));
366
367 SmallVector<Constant *, 2> VecElemIdxs;
368 VecElemIdxs.push_back(ConstantInt::get(i32, 0));
369 VecElemIdxs.push_back(ConstantInt::get(i32, 2));
370 auto *IdxVec = ConstantVector::get(VecElemIdxs);
371
372 Value *UndefVec = UndefValue::get(VecTy);
373 Value *VI1 = IRB.CreateInsertElement(UndefVec, IRB.getInt8(1), (uint64_t)0);
374 Value *VI2 = IRB.CreateInsertElement(VI1, Val2, Val);
375 Value *VI3 = IRB.CreateInsertElement(VI1, Val2, (uint64_t)1);
376 Value *VI4 = IRB.CreateInsertElement(VI1, IRB.getInt8(2), Val);
377
378 Value *EX1 = IRB.CreateExtractElement(VI4, Val);
379 Value *EX2 = IRB.CreateExtractElement(VI4, (uint64_t)0);
380 Value *EX3 = IRB.CreateExtractElement(IdxVec, (uint64_t)1);
381
382 Value *Zero = ConstantAggregateZero::get(i32VecTy);
383 Value *SI1 = IRB.CreateShuffleVector(VI1, UndefVec, Zero);
384 Value *SI2 = IRB.CreateShuffleVector(VI3, VI4, IdxVec);
385 Value *SI3 = IRB.CreateShuffleVector(VI3, UndefVec, Zero);
386 Value *SI4 = IRB.CreateShuffleVector(VI4, UndefVec, Zero);
387
388 Value *SP1 = IRB.CreateVectorSplat(2, IRB.getInt8(2));
389 Value *SP2 = IRB.CreateVectorSplat(2, Val);
390
391 Value *A = nullptr, *B = nullptr, *C = nullptr;
392
393 // Test matching insertelement
394 EXPECT_TRUE(match(VI1, m_InsertElement(m_Value(), m_Value(), m_Value())));
395 EXPECT_TRUE(
396 match(VI1, m_InsertElement(m_Undef(), m_ConstantInt(), m_ConstantInt())));
397 EXPECT_TRUE(
398 match(VI1, m_InsertElement(m_Undef(), m_ConstantInt(), m_Zero())));
399 EXPECT_TRUE(
400 match(VI1, m_InsertElement(m_Undef(), m_SpecificInt(1), m_Zero())));
401 EXPECT_TRUE(match(VI2, m_InsertElement(m_Value(), m_Value(), m_Value())));
402 EXPECT_FALSE(
403 match(VI2, m_InsertElement(m_Value(), m_Value(), m_ConstantInt())));
404 EXPECT_FALSE(
405 match(VI2, m_InsertElement(m_Value(), m_ConstantInt(), m_Value())));
406 EXPECT_FALSE(match(VI2, m_InsertElement(m_Constant(), m_Value(), m_Value())));
407 EXPECT_TRUE(match(VI3, m_InsertElement(m_Value(A), m_Value(B), m_Value(C))));
408 EXPECT_TRUE(A == VI1);
409 EXPECT_TRUE(B == Val2);
410 EXPECT_TRUE(isa<ConstantInt>(C));
411 A = B = C = nullptr; // reset
412
413 // Test matching extractelement
414 EXPECT_TRUE(match(EX1, m_ExtractElement(m_Value(A), m_Value(B))));
415 EXPECT_TRUE(A == VI4);
416 EXPECT_TRUE(B == Val);
417 A = B = C = nullptr; // reset
418 EXPECT_FALSE(match(EX1, m_ExtractElement(m_Value(), m_ConstantInt())));
419 EXPECT_TRUE(match(EX2, m_ExtractElement(m_Value(), m_ConstantInt())));
420 EXPECT_TRUE(match(EX3, m_ExtractElement(m_Constant(), m_ConstantInt())));
421
422 // Test matching shufflevector
423 EXPECT_TRUE(match(SI1, m_ShuffleVector(m_Value(), m_Undef(), m_Zero())));
424 EXPECT_TRUE(match(SI2, m_ShuffleVector(m_Value(A), m_Value(B), m_Value(C))));
425 EXPECT_TRUE(A == VI3);
426 EXPECT_TRUE(B == VI4);
427 EXPECT_TRUE(C == IdxVec);
428 A = B = C = nullptr; // reset
429
430 // Test matching the vector splat pattern
431 EXPECT_TRUE(match(
432 SI1,
433 m_ShuffleVector(m_InsertElement(m_Undef(), m_SpecificInt(1), m_Zero()),
434 m_Undef(), m_Zero())));
435 EXPECT_FALSE(match(
436 SI3, m_ShuffleVector(m_InsertElement(m_Undef(), m_Value(), m_Zero()),
437 m_Undef(), m_Zero())));
438 EXPECT_FALSE(match(
439 SI4, m_ShuffleVector(m_InsertElement(m_Undef(), m_Value(), m_Zero()),
440 m_Undef(), m_Zero())));
441 EXPECT_TRUE(match(
442 SP1,
443 m_ShuffleVector(m_InsertElement(m_Undef(), m_SpecificInt(2), m_Zero()),
444 m_Undef(), m_Zero())));
445 EXPECT_TRUE(match(
446 SP2, m_ShuffleVector(m_InsertElement(m_Undef(), m_Value(A), m_Zero()),
447 m_Undef(), m_Zero())));
448 EXPECT_TRUE(A == Val);
449}
450
Pete Cooperab47fa62016-08-12 22:16:05 +0000451template <typename T> struct MutableConstTest : PatternMatchTest { };
452
453typedef ::testing::Types<std::tuple<Value*, Instruction*>,
454 std::tuple<const Value*, const Instruction *>>
455 MutableConstTestTypes;
456TYPED_TEST_CASE(MutableConstTest, MutableConstTestTypes);
457
458TYPED_TEST(MutableConstTest, ICmp) {
459 auto &IRB = PatternMatchTest::IRB;
460
461 typedef typename std::tuple_element<0, TypeParam>::type ValueType;
462 typedef typename std::tuple_element<1, TypeParam>::type InstructionType;
463
464 Value *L = IRB.getInt32(1);
465 Value *R = IRB.getInt32(2);
466 ICmpInst::Predicate Pred = ICmpInst::ICMP_UGT;
467
468 ValueType MatchL;
469 ValueType MatchR;
470 ICmpInst::Predicate MatchPred;
471
472 EXPECT_TRUE(m_ICmp(MatchPred, m_Value(MatchL), m_Value(MatchR))
473 .match((InstructionType)IRB.CreateICmp(Pred, L, R)));
474 EXPECT_EQ(L, MatchL);
475 EXPECT_EQ(R, MatchR);
476}
477
Arnold Schwaighofere972d032013-05-05 01:54:46 +0000478} // anonymous namespace.