Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 1 | //===---- 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 Carruth | 9a67b07 | 2017-06-06 11:06:56 +0000 | [diff] [blame] | 10 | #include "llvm/IR/PatternMatch.h" |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 11 | #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 Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 17 | #include "llvm/IR/Function.h" |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 18 | #include "llvm/IR/IRBuilder.h" |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 19 | #include "llvm/IR/Instructions.h" |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 20 | #include "llvm/IR/LLVMContext.h" |
| 21 | #include "llvm/IR/MDBuilder.h" |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 22 | #include "llvm/IR/Module.h" |
Chandler Carruth | 64396b0 | 2014-03-04 12:05:47 +0000 | [diff] [blame] | 23 | #include "llvm/IR/NoFolder.h" |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 24 | #include "llvm/IR/Operator.h" |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 25 | #include "llvm/IR/Type.h" |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 26 | #include "gtest/gtest.h" |
| 27 | |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 28 | using namespace llvm; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 29 | using namespace llvm::PatternMatch; |
| 30 | |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 31 | namespace { |
| 32 | |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 33 | struct PatternMatchTest : ::testing::Test { |
| 34 | LLVMContext Ctx; |
Ahmed Charles | 56440fd | 2014-03-06 05:51:42 +0000 | [diff] [blame] | 35 | std::unique_ptr<Module> M; |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 36 | Function *F; |
| 37 | BasicBlock *BB; |
Mehdi Amini | ba9fba8 | 2016-03-13 21:05:13 +0000 | [diff] [blame] | 38 | IRBuilder<NoFolder> IRB; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 39 | |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 40 | 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 Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 45 | BB(BasicBlock::Create(Ctx, "entry", F)), IRB(BB) {} |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 46 | }; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 47 | |
Chandler Carruth | cde91b4 | 2014-01-05 09:14:53 +0000 | [diff] [blame] | 48 | TEST_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 Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 68 | TEST_F(PatternMatchTest, FloatingPointOrderedMin) { |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 69 | Type *FltTy = IRB.getFloatTy(); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 70 | Value *L = ConstantFP::get(FltTy, 1.0); |
| 71 | Value *R = ConstantFP::get(FltTy, 2.0); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 72 | Value *MatchL, *MatchR; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 73 | |
| 74 | // Test OLT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 75 | EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
| 76 | .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 77 | EXPECT_EQ(L, MatchL); |
| 78 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 79 | |
| 80 | // Test OLE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 81 | EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
| 82 | .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 83 | EXPECT_EQ(L, MatchL); |
| 84 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 85 | |
| 86 | // Test no match on OGE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 87 | EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
| 88 | .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 89 | |
| 90 | // Test no match on OGT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 91 | EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
| 92 | .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 93 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 94 | // 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 Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 103 | .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), R, L))); |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 104 | EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
| 105 | .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 106 | EXPECT_EQ(L, MatchL); |
| 107 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 108 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 109 | // [OU]GT with inverted select. |
| 110 | EXPECT_FALSE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 111 | .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), R, L))); |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 112 | EXPECT_TRUE(m_OrdFMin(m_Value(MatchL), m_Value(MatchR)) |
| 113 | .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 114 | EXPECT_EQ(L, MatchL); |
| 115 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 116 | } |
| 117 | |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 118 | TEST_F(PatternMatchTest, FloatingPointOrderedMax) { |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 119 | Type *FltTy = IRB.getFloatTy(); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 120 | Value *L = ConstantFP::get(FltTy, 1.0); |
| 121 | Value *R = ConstantFP::get(FltTy, 2.0); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 122 | Value *MatchL, *MatchR; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 123 | |
| 124 | // Test OGT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 125 | EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR)) |
| 126 | .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 127 | EXPECT_EQ(L, MatchL); |
| 128 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 129 | |
| 130 | // Test OGE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 131 | EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR)) |
| 132 | .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 133 | EXPECT_EQ(L, MatchL); |
| 134 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 135 | |
| 136 | // Test no match on OLE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 137 | EXPECT_FALSE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR)) |
| 138 | .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 139 | |
| 140 | // Test no match on OLT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 141 | EXPECT_FALSE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR)) |
| 142 | .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 143 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 144 | |
| 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 Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 155 | EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR)) |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 156 | .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 157 | EXPECT_EQ(L, MatchL); |
| 158 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 159 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 160 | // [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 Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 163 | EXPECT_TRUE(m_OrdFMax(m_Value(MatchL), m_Value(MatchR)) |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 164 | .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), R, L))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 165 | EXPECT_EQ(L, MatchL); |
| 166 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 167 | } |
| 168 | |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 169 | TEST_F(PatternMatchTest, FloatingPointUnorderedMin) { |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 170 | Type *FltTy = IRB.getFloatTy(); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 171 | Value *L = ConstantFP::get(FltTy, 1.0); |
| 172 | Value *R = ConstantFP::get(FltTy, 2.0); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 173 | Value *MatchL, *MatchR; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 174 | |
| 175 | // Test ULT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 176 | EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
| 177 | .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 178 | EXPECT_EQ(L, MatchL); |
| 179 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 180 | |
| 181 | // Test ULE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 182 | EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
| 183 | .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 184 | EXPECT_EQ(L, MatchL); |
| 185 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 186 | |
| 187 | // Test no match on UGE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 188 | EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
| 189 | .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 190 | |
| 191 | // Test no match on UGT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 192 | EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
| 193 | .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 194 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 195 | // 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 Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 204 | .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), R, L))); |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 205 | EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
| 206 | .match(IRB.CreateSelect(IRB.CreateFCmpOGE(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 207 | EXPECT_EQ(L, MatchL); |
| 208 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 209 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 210 | // [UO]GT with inverted select. |
| 211 | EXPECT_FALSE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 212 | .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), R, L))); |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 213 | EXPECT_TRUE(m_UnordFMin(m_Value(MatchL), m_Value(MatchR)) |
| 214 | .match(IRB.CreateSelect(IRB.CreateFCmpOGT(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 215 | EXPECT_EQ(L, MatchL); |
| 216 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 217 | } |
| 218 | |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 219 | TEST_F(PatternMatchTest, FloatingPointUnorderedMax) { |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 220 | Type *FltTy = IRB.getFloatTy(); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 221 | Value *L = ConstantFP::get(FltTy, 1.0); |
| 222 | Value *R = ConstantFP::get(FltTy, 2.0); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 223 | Value *MatchL, *MatchR; |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 224 | |
| 225 | // Test UGT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 226 | EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
| 227 | .match(IRB.CreateSelect(IRB.CreateFCmpUGT(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 228 | EXPECT_EQ(L, MatchL); |
| 229 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 230 | |
| 231 | // Test UGE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 232 | EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
| 233 | .match(IRB.CreateSelect(IRB.CreateFCmpUGE(L, R), L, R))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 234 | EXPECT_EQ(L, MatchL); |
| 235 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 236 | |
| 237 | // Test no match on ULE. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 238 | EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
| 239 | .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 240 | |
| 241 | // Test no match on ULT. |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 242 | EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
| 243 | .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), L, R))); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 244 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 245 | // 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 Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 254 | .match(IRB.CreateSelect(IRB.CreateFCmpULE(L, R), R, L))); |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 255 | EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
| 256 | .match(IRB.CreateSelect(IRB.CreateFCmpOLE(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 257 | EXPECT_EQ(L, MatchL); |
| 258 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 259 | |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 260 | // [UO]LT with inverted select. |
| 261 | EXPECT_FALSE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
Chandler Carruth | 91f4e60 | 2014-01-05 02:23:11 +0000 | [diff] [blame] | 262 | .match(IRB.CreateSelect(IRB.CreateFCmpULT(L, R), R, L))); |
Craig Topper | c663552 | 2017-06-13 17:18:45 +0000 | [diff] [blame] | 263 | EXPECT_TRUE(m_UnordFMax(m_Value(MatchL), m_Value(MatchR)) |
| 264 | .match(IRB.CreateSelect(IRB.CreateFCmpOLT(L, R), R, L))); |
Chandler Carruth | 4603e96a | 2014-01-05 02:07:20 +0000 | [diff] [blame] | 265 | EXPECT_EQ(L, MatchL); |
| 266 | EXPECT_EQ(R, MatchR); |
Arnold Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 267 | } |
| 268 | |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 269 | TEST_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 Topper | 66f09ad | 2014-06-08 22:29:17 +0000 | [diff] [blame] | 278 | MatchL = MatchR = nullptr; |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 279 | 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 Topper | 66f09ad | 2014-06-08 22:29:17 +0000 | [diff] [blame] | 283 | MatchL = MatchR = nullptr; |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 284 | 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 Topper | 66f09ad | 2014-06-08 22:29:17 +0000 | [diff] [blame] | 288 | MatchL = MatchR = nullptr; |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 289 | 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 Topper | 66f09ad | 2014-06-08 22:29:17 +0000 | [diff] [blame] | 298 | MatchL = MatchR = nullptr; |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 299 | 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 Topper | 66f09ad | 2014-06-08 22:29:17 +0000 | [diff] [blame] | 303 | MatchL = MatchR = nullptr; |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 304 | 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 Topper | 66f09ad | 2014-06-08 22:29:17 +0000 | [diff] [blame] | 308 | MatchL = MatchR = nullptr; |
Chandler Carruth | c77d50a | 2014-01-05 03:28:29 +0000 | [diff] [blame] | 309 | 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 Neilson | 45796f6 | 2018-03-28 15:39:00 +0000 | [diff] [blame] | 343 | TEST_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 Cooper | ab47fa6 | 2016-08-12 22:16:05 +0000 | [diff] [blame] | 451 | template <typename T> struct MutableConstTest : PatternMatchTest { }; |
| 452 | |
| 453 | typedef ::testing::Types<std::tuple<Value*, Instruction*>, |
| 454 | std::tuple<const Value*, const Instruction *>> |
| 455 | MutableConstTestTypes; |
| 456 | TYPED_TEST_CASE(MutableConstTest, MutableConstTestTypes); |
| 457 | |
| 458 | TYPED_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 Schwaighofer | e972d03 | 2013-05-05 01:54:46 +0000 | [diff] [blame] | 478 | } // anonymous namespace. |