1134bec27SJames Molloy //===- ValueTrackingTest.cpp - ValueTracking tests ------------------------===//
2134bec27SJames Molloy //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6134bec27SJames Molloy //
7134bec27SJames Molloy //===----------------------------------------------------------------------===//
8134bec27SJames Molloy 
9134bec27SJames Molloy #include "llvm/Analysis/ValueTracking.h"
10*d86fff6aSNikita Popov #include "llvm/Analysis/AssumptionCache.h"
11134bec27SJames Molloy #include "llvm/AsmParser/Parser.h"
12134bec27SJames Molloy #include "llvm/IR/Function.h"
13134bec27SJames Molloy #include "llvm/IR/InstIterator.h"
14be04aba6SCraig Topper #include "llvm/IR/Instructions.h"
15134bec27SJames Molloy #include "llvm/IR/LLVMContext.h"
16134bec27SJames Molloy #include "llvm/IR/Module.h"
17134bec27SJames Molloy #include "llvm/Support/ErrorHandling.h"
1870dbd5fbSSimon Dardis #include "llvm/Support/KnownBits.h"
19329d15b5SVitaly Buka #include "llvm/Support/SourceMgr.h"
20134bec27SJames Molloy #include "gtest/gtest.h"
21134bec27SJames Molloy 
22134bec27SJames Molloy using namespace llvm;
23134bec27SJames Molloy 
24134bec27SJames Molloy namespace {
25134bec27SJames Molloy 
26219e5367SNikita Popov class ValueTrackingTest : public testing::Test {
27134bec27SJames Molloy protected:
28329d15b5SVitaly Buka   std::unique_ptr<Module> parseModule(StringRef Assembly) {
29134bec27SJames Molloy     SMDiagnostic Error;
30329d15b5SVitaly Buka     std::unique_ptr<Module> M = parseAssemblyString(Assembly, Error, Context);
31134bec27SJames Molloy 
32134bec27SJames Molloy     std::string errMsg;
33134bec27SJames Molloy     raw_string_ostream os(errMsg);
34134bec27SJames Molloy     Error.print("", os);
35329d15b5SVitaly Buka     EXPECT_TRUE(M) << os.str();
36134bec27SJames Molloy 
37329d15b5SVitaly Buka     return M;
38329d15b5SVitaly Buka   }
39329d15b5SVitaly Buka 
40329d15b5SVitaly Buka   void parseAssembly(StringRef Assembly) {
41329d15b5SVitaly Buka     M = parseModule(Assembly);
42329d15b5SVitaly Buka     ASSERT_TRUE(M);
43134bec27SJames Molloy 
44*d86fff6aSNikita Popov     F = M->getFunction("test");
45329d15b5SVitaly Buka     ASSERT_TRUE(F) << "Test must have a function @test";
46329d15b5SVitaly Buka     if (!F)
47329d15b5SVitaly Buka       return;
48134bec27SJames Molloy 
49134bec27SJames Molloy     A = nullptr;
50134bec27SJames Molloy     for (inst_iterator I = inst_begin(F), E = inst_end(F); I != E; ++I) {
51134bec27SJames Molloy       if (I->hasName()) {
52134bec27SJames Molloy         if (I->getName() == "A")
53134bec27SJames Molloy           A = &*I;
54134bec27SJames Molloy       }
55134bec27SJames Molloy     }
56329d15b5SVitaly Buka     ASSERT_TRUE(A) << "@test must have an instruction %A";
57134bec27SJames Molloy   }
58134bec27SJames Molloy 
59219e5367SNikita Popov   LLVMContext Context;
60219e5367SNikita Popov   std::unique_ptr<Module> M;
61*d86fff6aSNikita Popov   Function *F = nullptr;
62329d15b5SVitaly Buka   Instruction *A = nullptr;
63219e5367SNikita Popov };
64219e5367SNikita Popov 
65219e5367SNikita Popov class MatchSelectPatternTest : public ValueTrackingTest {
66219e5367SNikita Popov protected:
67134bec27SJames Molloy   void expectPattern(const SelectPatternResult &P) {
68134bec27SJames Molloy     Value *LHS, *RHS;
69134bec27SJames Molloy     Instruction::CastOps CastOp;
70134bec27SJames Molloy     SelectPatternResult R = matchSelectPattern(A, LHS, RHS, &CastOp);
71134bec27SJames Molloy     EXPECT_EQ(P.Flavor, R.Flavor);
72134bec27SJames Molloy     EXPECT_EQ(P.NaNBehavior, R.NaNBehavior);
73134bec27SJames Molloy     EXPECT_EQ(P.Ordered, R.Ordered);
74134bec27SJames Molloy   }
75219e5367SNikita Popov };
76134bec27SJames Molloy 
77219e5367SNikita Popov class ComputeKnownBitsTest : public ValueTrackingTest {
78219e5367SNikita Popov protected:
79219e5367SNikita Popov   void expectKnownBits(uint64_t Zero, uint64_t One) {
80219e5367SNikita Popov     auto Known = computeKnownBits(A, M->getDataLayout());
81219e5367SNikita Popov     ASSERT_FALSE(Known.hasConflict());
82219e5367SNikita Popov     EXPECT_EQ(Known.One.getZExtValue(), One);
83219e5367SNikita Popov     EXPECT_EQ(Known.Zero.getZExtValue(), Zero);
84219e5367SNikita Popov   }
85134bec27SJames Molloy };
86134bec27SJames Molloy 
87134bec27SJames Molloy }
88134bec27SJames Molloy 
89134bec27SJames Molloy TEST_F(MatchSelectPatternTest, SimpleFMin) {
90134bec27SJames Molloy   parseAssembly(
91134bec27SJames Molloy       "define float @test(float %a) {\n"
92134bec27SJames Molloy       "  %1 = fcmp ult float %a, 5.0\n"
93134bec27SJames Molloy       "  %A = select i1 %1, float %a, float 5.0\n"
94134bec27SJames Molloy       "  ret float %A\n"
95134bec27SJames Molloy       "}\n");
96134bec27SJames Molloy   expectPattern({SPF_FMINNUM, SPNB_RETURNS_NAN, false});
97134bec27SJames Molloy }
98134bec27SJames Molloy 
99134bec27SJames Molloy TEST_F(MatchSelectPatternTest, SimpleFMax) {
100134bec27SJames Molloy   parseAssembly(
101134bec27SJames Molloy       "define float @test(float %a) {\n"
102134bec27SJames Molloy       "  %1 = fcmp ogt float %a, 5.0\n"
103134bec27SJames Molloy       "  %A = select i1 %1, float %a, float 5.0\n"
104134bec27SJames Molloy       "  ret float %A\n"
105134bec27SJames Molloy       "}\n");
106134bec27SJames Molloy   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, true});
107134bec27SJames Molloy }
108134bec27SJames Molloy 
109134bec27SJames Molloy TEST_F(MatchSelectPatternTest, SwappedFMax) {
110134bec27SJames Molloy   parseAssembly(
111134bec27SJames Molloy       "define float @test(float %a) {\n"
112134bec27SJames Molloy       "  %1 = fcmp olt float 5.0, %a\n"
113134bec27SJames Molloy       "  %A = select i1 %1, float %a, float 5.0\n"
114134bec27SJames Molloy       "  ret float %A\n"
115134bec27SJames Molloy       "}\n");
116134bec27SJames Molloy   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, false});
117134bec27SJames Molloy }
118134bec27SJames Molloy 
119134bec27SJames Molloy TEST_F(MatchSelectPatternTest, SwappedFMax2) {
120134bec27SJames Molloy   parseAssembly(
121134bec27SJames Molloy       "define float @test(float %a) {\n"
122134bec27SJames Molloy       "  %1 = fcmp olt float %a, 5.0\n"
123134bec27SJames Molloy       "  %A = select i1 %1, float 5.0, float %a\n"
124134bec27SJames Molloy       "  ret float %A\n"
125134bec27SJames Molloy       "}\n");
126134bec27SJames Molloy   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_NAN, false});
127134bec27SJames Molloy }
128134bec27SJames Molloy 
129134bec27SJames Molloy TEST_F(MatchSelectPatternTest, SwappedFMax3) {
130134bec27SJames Molloy   parseAssembly(
131134bec27SJames Molloy       "define float @test(float %a) {\n"
132134bec27SJames Molloy       "  %1 = fcmp ult float %a, 5.0\n"
133134bec27SJames Molloy       "  %A = select i1 %1, float 5.0, float %a\n"
134134bec27SJames Molloy       "  ret float %A\n"
135134bec27SJames Molloy       "}\n");
136134bec27SJames Molloy   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, true});
137134bec27SJames Molloy }
138134bec27SJames Molloy 
139134bec27SJames Molloy TEST_F(MatchSelectPatternTest, FastFMin) {
140134bec27SJames Molloy   parseAssembly(
141134bec27SJames Molloy       "define float @test(float %a) {\n"
142134bec27SJames Molloy       "  %1 = fcmp nnan olt float %a, 5.0\n"
143134bec27SJames Molloy       "  %A = select i1 %1, float %a, float 5.0\n"
144134bec27SJames Molloy       "  ret float %A\n"
145134bec27SJames Molloy       "}\n");
146134bec27SJames Molloy   expectPattern({SPF_FMINNUM, SPNB_RETURNS_ANY, false});
147134bec27SJames Molloy }
148134bec27SJames Molloy 
149134bec27SJames Molloy TEST_F(MatchSelectPatternTest, FMinConstantZero) {
150134bec27SJames Molloy   parseAssembly(
151134bec27SJames Molloy       "define float @test(float %a) {\n"
152134bec27SJames Molloy       "  %1 = fcmp ole float %a, 0.0\n"
153134bec27SJames Molloy       "  %A = select i1 %1, float %a, float 0.0\n"
154134bec27SJames Molloy       "  ret float %A\n"
155134bec27SJames Molloy       "}\n");
156134bec27SJames Molloy   // This shouldn't be matched, as %a could be -0.0.
157134bec27SJames Molloy   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
158134bec27SJames Molloy }
159134bec27SJames Molloy 
160134bec27SJames Molloy TEST_F(MatchSelectPatternTest, FMinConstantZeroNsz) {
161134bec27SJames Molloy   parseAssembly(
162134bec27SJames Molloy       "define float @test(float %a) {\n"
163134bec27SJames Molloy       "  %1 = fcmp nsz ole float %a, 0.0\n"
164134bec27SJames Molloy       "  %A = select i1 %1, float %a, float 0.0\n"
165134bec27SJames Molloy       "  ret float %A\n"
166134bec27SJames Molloy       "}\n");
167134bec27SJames Molloy   // But this should be, because we've ignored signed zeroes.
168134bec27SJames Molloy   expectPattern({SPF_FMINNUM, SPNB_RETURNS_OTHER, true});
169134bec27SJames Molloy }
170569cea65SJames Molloy 
17103da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero1) {
17203da6e6aSSanjay Patel   parseAssembly(
17303da6e6aSSanjay Patel       "define float @test(float %a) {\n"
17403da6e6aSSanjay Patel       "  %1 = fcmp olt float -0.0, %a\n"
17503da6e6aSSanjay Patel       "  %A = select i1 %1, float 0.0, float %a\n"
17603da6e6aSSanjay Patel       "  ret float %A\n"
17703da6e6aSSanjay Patel       "}\n");
178e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
179e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_NAN, true});
18003da6e6aSSanjay Patel }
18103da6e6aSSanjay Patel 
18203da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero2) {
18303da6e6aSSanjay Patel   parseAssembly(
18403da6e6aSSanjay Patel       "define float @test(float %a) {\n"
18503da6e6aSSanjay Patel       "  %1 = fcmp ogt float %a, -0.0\n"
18603da6e6aSSanjay Patel       "  %A = select i1 %1, float 0.0, float %a\n"
18703da6e6aSSanjay Patel       "  ret float %A\n"
18803da6e6aSSanjay Patel       "}\n");
189e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
190e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_NAN, false});
19103da6e6aSSanjay Patel }
19203da6e6aSSanjay Patel 
19303da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero3) {
19403da6e6aSSanjay Patel   parseAssembly(
19503da6e6aSSanjay Patel       "define float @test(float %a) {\n"
19603da6e6aSSanjay Patel       "  %1 = fcmp olt float 0.0, %a\n"
19703da6e6aSSanjay Patel       "  %A = select i1 %1, float -0.0, float %a\n"
19803da6e6aSSanjay Patel       "  ret float %A\n"
19903da6e6aSSanjay Patel       "}\n");
200e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
201e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_NAN, true});
20203da6e6aSSanjay Patel }
20303da6e6aSSanjay Patel 
20403da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero4) {
20503da6e6aSSanjay Patel   parseAssembly(
20603da6e6aSSanjay Patel       "define float @test(float %a) {\n"
20703da6e6aSSanjay Patel       "  %1 = fcmp ogt float %a, 0.0\n"
20803da6e6aSSanjay Patel       "  %A = select i1 %1, float -0.0, float %a\n"
20903da6e6aSSanjay Patel       "  ret float %A\n"
21003da6e6aSSanjay Patel       "}\n");
211e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
212e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_NAN, false});
21303da6e6aSSanjay Patel }
21403da6e6aSSanjay Patel 
21503da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero5) {
21603da6e6aSSanjay Patel   parseAssembly(
21703da6e6aSSanjay Patel       "define float @test(float %a) {\n"
21803da6e6aSSanjay Patel       "  %1 = fcmp ogt float -0.0, %a\n"
21903da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float 0.0\n"
22003da6e6aSSanjay Patel       "  ret float %A\n"
22103da6e6aSSanjay Patel       "}\n");
222e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
223e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_OTHER, false});
22403da6e6aSSanjay Patel }
22503da6e6aSSanjay Patel 
22603da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero6) {
22703da6e6aSSanjay Patel   parseAssembly(
22803da6e6aSSanjay Patel       "define float @test(float %a) {\n"
22903da6e6aSSanjay Patel       "  %1 = fcmp olt float %a, -0.0\n"
23003da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float 0.0\n"
23103da6e6aSSanjay Patel       "  ret float %A\n"
23203da6e6aSSanjay Patel       "}\n");
233e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
234e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_OTHER, true});
23503da6e6aSSanjay Patel }
23603da6e6aSSanjay Patel 
23703da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero7) {
23803da6e6aSSanjay Patel   parseAssembly(
23903da6e6aSSanjay Patel       "define float @test(float %a) {\n"
24003da6e6aSSanjay Patel       "  %1 = fcmp ogt float 0.0, %a\n"
24103da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float -0.0\n"
24203da6e6aSSanjay Patel       "  ret float %A\n"
24303da6e6aSSanjay Patel       "}\n");
244e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
245e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_OTHER, false});
24603da6e6aSSanjay Patel }
24703da6e6aSSanjay Patel 
24803da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZero8) {
24903da6e6aSSanjay Patel   parseAssembly(
25003da6e6aSSanjay Patel       "define float @test(float %a) {\n"
25103da6e6aSSanjay Patel       "  %1 = fcmp olt float %a, 0.0\n"
25203da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float -0.0\n"
25303da6e6aSSanjay Patel       "  ret float %A\n"
25403da6e6aSSanjay Patel       "}\n");
255e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
256e7c94ef1SSanjay Patel   expectPattern({SPF_FMINNUM, SPNB_RETURNS_OTHER, true});
25703da6e6aSSanjay Patel }
25803da6e6aSSanjay Patel 
25903da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero1) {
26003da6e6aSSanjay Patel   parseAssembly(
26103da6e6aSSanjay Patel       "define float @test(float %a) {\n"
26203da6e6aSSanjay Patel       "  %1 = fcmp ogt float -0.0, %a\n"
26303da6e6aSSanjay Patel       "  %A = select i1 %1, float 0.0, float %a\n"
26403da6e6aSSanjay Patel       "  ret float %A\n"
26503da6e6aSSanjay Patel       "}\n");
266e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
267e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_NAN, true});
26803da6e6aSSanjay Patel }
26903da6e6aSSanjay Patel 
27003da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero2) {
27103da6e6aSSanjay Patel   parseAssembly(
27203da6e6aSSanjay Patel       "define float @test(float %a) {\n"
27303da6e6aSSanjay Patel       "  %1 = fcmp olt float %a, -0.0\n"
27403da6e6aSSanjay Patel       "  %A = select i1 %1, float 0.0, float %a\n"
27503da6e6aSSanjay Patel       "  ret float %A\n"
27603da6e6aSSanjay Patel       "}\n");
277e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
278e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_NAN, false});
27903da6e6aSSanjay Patel }
28003da6e6aSSanjay Patel 
28103da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero3) {
28203da6e6aSSanjay Patel   parseAssembly(
28303da6e6aSSanjay Patel       "define float @test(float %a) {\n"
28403da6e6aSSanjay Patel       "  %1 = fcmp ogt float 0.0, %a\n"
28503da6e6aSSanjay Patel       "  %A = select i1 %1, float -0.0, float %a\n"
28603da6e6aSSanjay Patel       "  ret float %A\n"
28703da6e6aSSanjay Patel       "}\n");
288e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
289e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_NAN, true});
29003da6e6aSSanjay Patel }
29103da6e6aSSanjay Patel 
29203da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero4) {
29303da6e6aSSanjay Patel   parseAssembly(
29403da6e6aSSanjay Patel       "define float @test(float %a) {\n"
29503da6e6aSSanjay Patel       "  %1 = fcmp olt float %a, 0.0\n"
29603da6e6aSSanjay Patel       "  %A = select i1 %1, float -0.0, float %a\n"
29703da6e6aSSanjay Patel       "  ret float %A\n"
29803da6e6aSSanjay Patel       "}\n");
299e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
300e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_NAN, false});
30103da6e6aSSanjay Patel }
30203da6e6aSSanjay Patel 
30303da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero5) {
30403da6e6aSSanjay Patel   parseAssembly(
30503da6e6aSSanjay Patel       "define float @test(float %a) {\n"
30603da6e6aSSanjay Patel       "  %1 = fcmp olt float -0.0, %a\n"
30703da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float 0.0\n"
30803da6e6aSSanjay Patel       "  ret float %A\n"
30903da6e6aSSanjay Patel       "}\n");
310e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
311e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, false});
31203da6e6aSSanjay Patel }
31303da6e6aSSanjay Patel 
31403da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero6) {
31503da6e6aSSanjay Patel   parseAssembly(
31603da6e6aSSanjay Patel       "define float @test(float %a) {\n"
31703da6e6aSSanjay Patel       "  %1 = fcmp ogt float %a, -0.0\n"
31803da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float 0.0\n"
31903da6e6aSSanjay Patel       "  ret float %A\n"
32003da6e6aSSanjay Patel       "}\n");
321e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
322e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, true});
32303da6e6aSSanjay Patel }
32403da6e6aSSanjay Patel 
32503da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero7) {
32603da6e6aSSanjay Patel   parseAssembly(
32703da6e6aSSanjay Patel       "define float @test(float %a) {\n"
32803da6e6aSSanjay Patel       "  %1 = fcmp olt float 0.0, %a\n"
32903da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float -0.0\n"
33003da6e6aSSanjay Patel       "  ret float %A\n"
33103da6e6aSSanjay Patel       "}\n");
332e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
333e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, false});
33403da6e6aSSanjay Patel }
33503da6e6aSSanjay Patel 
33603da6e6aSSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZero8) {
33703da6e6aSSanjay Patel   parseAssembly(
33803da6e6aSSanjay Patel       "define float @test(float %a) {\n"
33903da6e6aSSanjay Patel       "  %1 = fcmp ogt float %a, 0.0\n"
34003da6e6aSSanjay Patel       "  %A = select i1 %1, float %a, float -0.0\n"
34103da6e6aSSanjay Patel       "  ret float %A\n"
34203da6e6aSSanjay Patel       "}\n");
343e7c94ef1SSanjay Patel   // The sign of zero doesn't matter in fcmp.
344e7c94ef1SSanjay Patel   expectPattern({SPF_FMAXNUM, SPNB_RETURNS_OTHER, true});
345e7c94ef1SSanjay Patel }
346e7c94ef1SSanjay Patel 
347e7c94ef1SSanjay Patel TEST_F(MatchSelectPatternTest, FMinMismatchConstantZeroVecUndef) {
348e7c94ef1SSanjay Patel   parseAssembly(
349e7c94ef1SSanjay Patel       "define <2 x float> @test(<2 x float> %a) {\n"
350e7c94ef1SSanjay Patel       "  %1 = fcmp ogt <2 x float> %a, <float -0.0, float -0.0>\n"
351e7c94ef1SSanjay Patel       "  %A = select <2 x i1> %1, <2 x float> <float undef, float 0.0>, <2 x float> %a\n"
352e7c94ef1SSanjay Patel       "  ret <2 x float> %A\n"
353e7c94ef1SSanjay Patel       "}\n");
354e7c94ef1SSanjay Patel   // An undef in a vector constant can not be back-propagated for this analysis.
355e7c94ef1SSanjay Patel   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
356e7c94ef1SSanjay Patel }
357e7c94ef1SSanjay Patel 
358e7c94ef1SSanjay Patel TEST_F(MatchSelectPatternTest, FMaxMismatchConstantZeroVecUndef) {
359e7c94ef1SSanjay Patel   parseAssembly(
360e7c94ef1SSanjay Patel       "define <2 x float> @test(<2 x float> %a) {\n"
361e7c94ef1SSanjay Patel       "  %1 = fcmp ogt <2 x float> %a, zeroinitializer\n"
362e7c94ef1SSanjay Patel       "  %A = select <2 x i1> %1, <2 x float> %a, <2 x float> <float -0.0, float undef>\n"
363e7c94ef1SSanjay Patel       "  ret <2 x float> %A\n"
364e7c94ef1SSanjay Patel       "}\n");
365e7c94ef1SSanjay Patel   // An undef in a vector constant can not be back-propagated for this analysis.
36603da6e6aSSanjay Patel   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
36703da6e6aSSanjay Patel }
36803da6e6aSSanjay Patel 
36930f1d691SThomas Lively TEST_F(MatchSelectPatternTest, VectorFMinimum) {
370d47b5c7bSThomas Lively   parseAssembly(
371d47b5c7bSThomas Lively       "define <4 x float> @test(<4 x float> %a) {\n"
372d47b5c7bSThomas Lively       "  %1 = fcmp ule <4 x float> %a, \n"
373d47b5c7bSThomas Lively       "    <float 5.0, float 5.0, float 5.0, float 5.0>\n"
374d47b5c7bSThomas Lively       "  %A = select <4 x i1> %1, <4 x float> %a,\n"
375d47b5c7bSThomas Lively       "     <4 x float> <float 5.0, float 5.0, float 5.0, float 5.0>\n"
376d47b5c7bSThomas Lively       "  ret <4 x float> %A\n"
377d47b5c7bSThomas Lively       "}\n");
378d47b5c7bSThomas Lively   // Check that pattern matching works on vectors where each lane has the same
379d47b5c7bSThomas Lively   // unordered pattern.
380d47b5c7bSThomas Lively   expectPattern({SPF_FMINNUM, SPNB_RETURNS_NAN, false});
381d47b5c7bSThomas Lively }
382d47b5c7bSThomas Lively 
383d47b5c7bSThomas Lively TEST_F(MatchSelectPatternTest, VectorFMinOtherOrdered) {
384d47b5c7bSThomas Lively   parseAssembly(
385d47b5c7bSThomas Lively       "define <4 x float> @test(<4 x float> %a) {\n"
386d47b5c7bSThomas Lively       "  %1 = fcmp ole <4 x float> %a, \n"
387d47b5c7bSThomas Lively       "    <float 5.0, float 5.0, float 5.0, float 5.0>\n"
388d47b5c7bSThomas Lively       "  %A = select <4 x i1> %1, <4 x float> %a,\n"
389d47b5c7bSThomas Lively       "     <4 x float> <float 5.0, float 5.0, float 5.0, float 5.0>\n"
390d47b5c7bSThomas Lively       "  ret <4 x float> %A\n"
391d47b5c7bSThomas Lively       "}\n");
392d47b5c7bSThomas Lively   // Check that pattern matching works on vectors where each lane has the same
393d47b5c7bSThomas Lively   // ordered pattern.
394d47b5c7bSThomas Lively   expectPattern({SPF_FMINNUM, SPNB_RETURNS_OTHER, true});
395d47b5c7bSThomas Lively }
396d47b5c7bSThomas Lively 
39730f1d691SThomas Lively TEST_F(MatchSelectPatternTest, VectorNotFMinimum) {
398d47b5c7bSThomas Lively   parseAssembly(
399d47b5c7bSThomas Lively       "define <4 x float> @test(<4 x float> %a) {\n"
400d47b5c7bSThomas Lively       "  %1 = fcmp ule <4 x float> %a, \n"
401d47b5c7bSThomas Lively       "    <float 5.0, float 0x7ff8000000000000, float 5.0, float 5.0>\n"
402d47b5c7bSThomas Lively       "  %A = select <4 x i1> %1, <4 x float> %a,\n"
403d47b5c7bSThomas Lively       "     <4 x float> <float 5.0, float 0x7ff8000000000000, float 5.0, float "
404d47b5c7bSThomas Lively       "5.0>\n"
405d47b5c7bSThomas Lively       "  ret <4 x float> %A\n"
406d47b5c7bSThomas Lively       "}\n");
407d47b5c7bSThomas Lively   // The lane that contains a NaN (0x7ff80...) behaves like a
408d47b5c7bSThomas Lively   // non-NaN-propagating min and the other lines behave like a NaN-propagating
409d47b5c7bSThomas Lively   // min, so check that neither is returned.
410d47b5c7bSThomas Lively   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
411d47b5c7bSThomas Lively }
412d47b5c7bSThomas Lively 
413d47b5c7bSThomas Lively TEST_F(MatchSelectPatternTest, VectorNotFMinZero) {
414d47b5c7bSThomas Lively   parseAssembly(
415d47b5c7bSThomas Lively       "define <4 x float> @test(<4 x float> %a) {\n"
416d47b5c7bSThomas Lively       "  %1 = fcmp ule <4 x float> %a, \n"
417d47b5c7bSThomas Lively       "    <float 5.0, float -0.0, float 5.0, float 5.0>\n"
418d47b5c7bSThomas Lively       "  %A = select <4 x i1> %1, <4 x float> %a,\n"
419d47b5c7bSThomas Lively       "     <4 x float> <float 5.0, float 0.0, float 5.0, float 5.0>\n"
420d47b5c7bSThomas Lively       "  ret <4 x float> %A\n"
421d47b5c7bSThomas Lively       "}\n");
422d47b5c7bSThomas Lively   // Always selects the second lane of %a if it is positive or negative zero, so
423d47b5c7bSThomas Lively   // this is stricter than a min.
424d47b5c7bSThomas Lively   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
425d47b5c7bSThomas Lively }
426d47b5c7bSThomas Lively 
427569cea65SJames Molloy TEST_F(MatchSelectPatternTest, DoubleCastU) {
428569cea65SJames Molloy   parseAssembly(
429569cea65SJames Molloy       "define i32 @test(i8 %a, i8 %b) {\n"
430569cea65SJames Molloy       "  %1 = icmp ult i8 %a, %b\n"
431569cea65SJames Molloy       "  %2 = zext i8 %a to i32\n"
432569cea65SJames Molloy       "  %3 = zext i8 %b to i32\n"
433569cea65SJames Molloy       "  %A = select i1 %1, i32 %2, i32 %3\n"
434569cea65SJames Molloy       "  ret i32 %A\n"
435569cea65SJames Molloy       "}\n");
436569cea65SJames Molloy   // We should be able to look through the situation where we cast both operands
437569cea65SJames Molloy   // to the select.
438569cea65SJames Molloy   expectPattern({SPF_UMIN, SPNB_NA, false});
439569cea65SJames Molloy }
440569cea65SJames Molloy 
441569cea65SJames Molloy TEST_F(MatchSelectPatternTest, DoubleCastS) {
442569cea65SJames Molloy   parseAssembly(
443569cea65SJames Molloy       "define i32 @test(i8 %a, i8 %b) {\n"
444569cea65SJames Molloy       "  %1 = icmp slt i8 %a, %b\n"
445569cea65SJames Molloy       "  %2 = sext i8 %a to i32\n"
446569cea65SJames Molloy       "  %3 = sext i8 %b to i32\n"
447569cea65SJames Molloy       "  %A = select i1 %1, i32 %2, i32 %3\n"
448569cea65SJames Molloy       "  ret i32 %A\n"
449569cea65SJames Molloy       "}\n");
450569cea65SJames Molloy   // We should be able to look through the situation where we cast both operands
451569cea65SJames Molloy   // to the select.
452569cea65SJames Molloy   expectPattern({SPF_SMIN, SPNB_NA, false});
453569cea65SJames Molloy }
454569cea65SJames Molloy 
455569cea65SJames Molloy TEST_F(MatchSelectPatternTest, DoubleCastBad) {
456569cea65SJames Molloy   parseAssembly(
457569cea65SJames Molloy       "define i32 @test(i8 %a, i8 %b) {\n"
458569cea65SJames Molloy       "  %1 = icmp ult i8 %a, %b\n"
459569cea65SJames Molloy       "  %2 = zext i8 %a to i32\n"
460569cea65SJames Molloy       "  %3 = sext i8 %b to i32\n"
461569cea65SJames Molloy       "  %A = select i1 %1, i32 %2, i32 %3\n"
462569cea65SJames Molloy       "  ret i32 %A\n"
463569cea65SJames Molloy       "}\n");
464687a8448SJames Molloy   // The cast types here aren't the same, so we cannot match an UMIN.
465569cea65SJames Molloy   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
466569cea65SJames Molloy }
4673bb2dbd6SSanjoy Das 
4686d343740SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotSMin) {
4696d343740SSanjay Patel   parseAssembly(
4706d343740SSanjay Patel       "define i8 @test(i8 %a, i8 %b) {\n"
4716d343740SSanjay Patel       "  %cmp = icmp sgt i8 %a, %b\n"
4726d343740SSanjay Patel       "  %an = xor i8 %a, -1\n"
4736d343740SSanjay Patel       "  %bn = xor i8 %b, -1\n"
4746d343740SSanjay Patel       "  %A = select i1 %cmp, i8 %an, i8 %bn\n"
4756d343740SSanjay Patel       "  ret i8 %A\n"
4766d343740SSanjay Patel       "}\n");
477867f0c3cSSanjay Patel   expectPattern({SPF_SMIN, SPNB_NA, false});
4786d343740SSanjay Patel }
4796d343740SSanjay Patel 
4806d343740SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotSMinSwap) {
4816d343740SSanjay Patel   parseAssembly(
482463143f0SSanjay Patel       "define <2 x i8> @test(<2 x i8> %a, <2 x i8> %b) {\n"
483463143f0SSanjay Patel       "  %cmp = icmp slt <2 x i8> %a, %b\n"
484463143f0SSanjay Patel       "  %an = xor <2 x i8> %a, <i8 -1, i8-1>\n"
485463143f0SSanjay Patel       "  %bn = xor <2 x i8> %b, <i8 -1, i8-1>\n"
486463143f0SSanjay Patel       "  %A = select <2 x i1> %cmp, <2 x i8> %bn, <2 x i8> %an\n"
487463143f0SSanjay Patel       "  ret <2 x i8> %A\n"
4886d343740SSanjay Patel       "}\n");
489867f0c3cSSanjay Patel   expectPattern({SPF_SMIN, SPNB_NA, false});
4906d343740SSanjay Patel }
4916d343740SSanjay Patel 
4926d343740SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotSMax) {
4936d343740SSanjay Patel   parseAssembly(
4946d343740SSanjay Patel       "define i8 @test(i8 %a, i8 %b) {\n"
4956d343740SSanjay Patel       "  %cmp = icmp slt i8 %a, %b\n"
4966d343740SSanjay Patel       "  %an = xor i8 %a, -1\n"
4976d343740SSanjay Patel       "  %bn = xor i8 %b, -1\n"
4986d343740SSanjay Patel       "  %A = select i1 %cmp, i8 %an, i8 %bn\n"
4996d343740SSanjay Patel       "  ret i8 %A\n"
5006d343740SSanjay Patel       "}\n");
501867f0c3cSSanjay Patel   expectPattern({SPF_SMAX, SPNB_NA, false});
5026d343740SSanjay Patel }
5036d343740SSanjay Patel 
5046d343740SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotSMaxSwap) {
5056d343740SSanjay Patel   parseAssembly(
506463143f0SSanjay Patel       "define <2 x i8> @test(<2 x i8> %a, <2 x i8> %b) {\n"
507463143f0SSanjay Patel       "  %cmp = icmp sgt <2 x i8> %a, %b\n"
508463143f0SSanjay Patel       "  %an = xor <2 x i8> %a, <i8 -1, i8-1>\n"
509463143f0SSanjay Patel       "  %bn = xor <2 x i8> %b, <i8 -1, i8-1>\n"
510463143f0SSanjay Patel       "  %A = select <2 x i1> %cmp, <2 x i8> %bn, <2 x i8> %an\n"
511463143f0SSanjay Patel       "  ret <2 x i8> %A\n"
512463143f0SSanjay Patel       "}\n");
513463143f0SSanjay Patel   expectPattern({SPF_SMAX, SPNB_NA, false});
514463143f0SSanjay Patel }
515463143f0SSanjay Patel 
516463143f0SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotUMin) {
517463143f0SSanjay Patel   parseAssembly(
518463143f0SSanjay Patel       "define <2 x i8> @test(<2 x i8> %a, <2 x i8> %b) {\n"
519463143f0SSanjay Patel       "  %cmp = icmp ugt <2 x i8> %a, %b\n"
520463143f0SSanjay Patel       "  %an = xor <2 x i8> %a, <i8 -1, i8-1>\n"
521463143f0SSanjay Patel       "  %bn = xor <2 x i8> %b, <i8 -1, i8-1>\n"
522463143f0SSanjay Patel       "  %A = select <2 x i1> %cmp, <2 x i8> %an, <2 x i8> %bn\n"
523463143f0SSanjay Patel       "  ret <2 x i8> %A\n"
524463143f0SSanjay Patel       "}\n");
525fbb1b43fSSanjay Patel   expectPattern({SPF_UMIN, SPNB_NA, false});
526463143f0SSanjay Patel }
527463143f0SSanjay Patel 
528463143f0SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotUMinSwap) {
529463143f0SSanjay Patel   parseAssembly(
5306d343740SSanjay Patel       "define i8 @test(i8 %a, i8 %b) {\n"
531463143f0SSanjay Patel       "  %cmp = icmp ult i8 %a, %b\n"
5326d343740SSanjay Patel       "  %an = xor i8 %a, -1\n"
5336d343740SSanjay Patel       "  %bn = xor i8 %b, -1\n"
5346d343740SSanjay Patel       "  %A = select i1 %cmp, i8 %bn, i8 %an\n"
5356d343740SSanjay Patel       "  ret i8 %A\n"
5366d343740SSanjay Patel       "}\n");
537fbb1b43fSSanjay Patel   expectPattern({SPF_UMIN, SPNB_NA, false});
538463143f0SSanjay Patel }
539463143f0SSanjay Patel 
540463143f0SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotUMax) {
541463143f0SSanjay Patel   parseAssembly(
542463143f0SSanjay Patel       "define <2 x i8> @test(<2 x i8> %a, <2 x i8> %b) {\n"
543463143f0SSanjay Patel       "  %cmp = icmp ult <2 x i8> %a, %b\n"
544463143f0SSanjay Patel       "  %an = xor <2 x i8> %a, <i8 -1, i8-1>\n"
545463143f0SSanjay Patel       "  %bn = xor <2 x i8> %b, <i8 -1, i8-1>\n"
546463143f0SSanjay Patel       "  %A = select <2 x i1> %cmp, <2 x i8> %an, <2 x i8> %bn\n"
547463143f0SSanjay Patel       "  ret <2 x i8> %A\n"
548463143f0SSanjay Patel       "}\n");
549fbb1b43fSSanjay Patel   expectPattern({SPF_UMAX, SPNB_NA, false});
550463143f0SSanjay Patel }
551463143f0SSanjay Patel 
552463143f0SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotUMaxSwap) {
553463143f0SSanjay Patel   parseAssembly(
554463143f0SSanjay Patel       "define i8 @test(i8 %a, i8 %b) {\n"
555463143f0SSanjay Patel       "  %cmp = icmp ugt i8 %a, %b\n"
556463143f0SSanjay Patel       "  %an = xor i8 %a, -1\n"
557463143f0SSanjay Patel       "  %bn = xor i8 %b, -1\n"
558463143f0SSanjay Patel       "  %A = select i1 %cmp, i8 %bn, i8 %an\n"
559463143f0SSanjay Patel       "  ret i8 %A\n"
560463143f0SSanjay Patel       "}\n");
561fbb1b43fSSanjay Patel   expectPattern({SPF_UMAX, SPNB_NA, false});
562463143f0SSanjay Patel }
563463143f0SSanjay Patel 
564463143f0SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotEq) {
565463143f0SSanjay Patel   parseAssembly(
566463143f0SSanjay Patel       "define i8 @test(i8 %a, i8 %b) {\n"
567463143f0SSanjay Patel       "  %cmp = icmp eq i8 %a, %b\n"
568463143f0SSanjay Patel       "  %an = xor i8 %a, -1\n"
569463143f0SSanjay Patel       "  %bn = xor i8 %b, -1\n"
570463143f0SSanjay Patel       "  %A = select i1 %cmp, i8 %bn, i8 %an\n"
571463143f0SSanjay Patel       "  ret i8 %A\n"
572463143f0SSanjay Patel       "}\n");
573463143f0SSanjay Patel   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
574463143f0SSanjay Patel }
575463143f0SSanjay Patel 
576463143f0SSanjay Patel TEST_F(MatchSelectPatternTest, NotNotNe) {
577463143f0SSanjay Patel   parseAssembly(
578463143f0SSanjay Patel       "define i8 @test(i8 %a, i8 %b) {\n"
579463143f0SSanjay Patel       "  %cmp = icmp ne i8 %a, %b\n"
580463143f0SSanjay Patel       "  %an = xor i8 %a, -1\n"
581463143f0SSanjay Patel       "  %bn = xor i8 %b, -1\n"
582463143f0SSanjay Patel       "  %A = select i1 %cmp, i8 %bn, i8 %an\n"
583463143f0SSanjay Patel       "  ret i8 %A\n"
584463143f0SSanjay Patel       "}\n");
585463143f0SSanjay Patel   expectPattern({SPF_UNKNOWN, SPNB_NA, false});
5866d343740SSanjay Patel }
5876d343740SSanjay Patel 
5883bb2dbd6SSanjoy Das TEST(ValueTracking, GuaranteedToTransferExecutionToSuccessor) {
5893bb2dbd6SSanjoy Das   StringRef Assembly =
5903bb2dbd6SSanjoy Das       "declare void @nounwind_readonly(i32*) nounwind readonly "
5913bb2dbd6SSanjoy Das       "declare void @nounwind_argmemonly(i32*) nounwind argmemonly "
5923bb2dbd6SSanjoy Das       "declare void @throws_but_readonly(i32*) readonly "
5933bb2dbd6SSanjoy Das       "declare void @throws_but_argmemonly(i32*) argmemonly "
5946ed459fdSJohannes Doerfert       "declare void @nounwind_willreturn(i32*) nounwind willreturn"
5953bb2dbd6SSanjoy Das       " "
5963bb2dbd6SSanjoy Das       "declare void @unknown(i32*) "
5973bb2dbd6SSanjoy Das       " "
5983bb2dbd6SSanjoy Das       "define void @f(i32* %p) { "
5993bb2dbd6SSanjoy Das       "  call void @nounwind_readonly(i32* %p) "
6003bb2dbd6SSanjoy Das       "  call void @nounwind_argmemonly(i32* %p) "
6013bb2dbd6SSanjoy Das       "  call void @throws_but_readonly(i32* %p) "
6023bb2dbd6SSanjoy Das       "  call void @throws_but_argmemonly(i32* %p) "
6033bb2dbd6SSanjoy Das       "  call void @unknown(i32* %p) nounwind readonly "
6043bb2dbd6SSanjoy Das       "  call void @unknown(i32* %p) nounwind argmemonly "
6053bb2dbd6SSanjoy Das       "  call void @unknown(i32* %p) readonly "
6063bb2dbd6SSanjoy Das       "  call void @unknown(i32* %p) argmemonly "
6076ed459fdSJohannes Doerfert       "  call void @nounwind_willreturn(i32* %p)"
6083bb2dbd6SSanjoy Das       "  ret void "
6093bb2dbd6SSanjoy Das       "} ";
6103bb2dbd6SSanjoy Das 
6113bb2dbd6SSanjoy Das   LLVMContext Context;
6123bb2dbd6SSanjoy Das   SMDiagnostic Error;
6133bb2dbd6SSanjoy Das   auto M = parseAssemblyString(Assembly, Error, Context);
6143bb2dbd6SSanjoy Das   assert(M && "Bad assembly?");
6153bb2dbd6SSanjoy Das 
6163bb2dbd6SSanjoy Das   auto *F = M->getFunction("f");
6173bb2dbd6SSanjoy Das   assert(F && "Bad assembly?");
6183bb2dbd6SSanjoy Das 
6193bb2dbd6SSanjoy Das   auto &BB = F->getEntryBlock();
6208580cd4eSCraig Topper   bool ExpectedAnswers[] = {
6213bb2dbd6SSanjoy Das       true,  // call void @nounwind_readonly(i32* %p)
6223bb2dbd6SSanjoy Das       true,  // call void @nounwind_argmemonly(i32* %p)
6233bb2dbd6SSanjoy Das       false, // call void @throws_but_readonly(i32* %p)
6243bb2dbd6SSanjoy Das       false, // call void @throws_but_argmemonly(i32* %p)
6253bb2dbd6SSanjoy Das       true,  // call void @unknown(i32* %p) nounwind readonly
6263bb2dbd6SSanjoy Das       true,  // call void @unknown(i32* %p) nounwind argmemonly
6273bb2dbd6SSanjoy Das       false, // call void @unknown(i32* %p) readonly
6283bb2dbd6SSanjoy Das       false, // call void @unknown(i32* %p) argmemonly
6296ed459fdSJohannes Doerfert       true,  // call void @nounwind_willreturn(i32* %p)
6303bb2dbd6SSanjoy Das       false, // ret void
6313bb2dbd6SSanjoy Das   };
6323bb2dbd6SSanjoy Das 
6333bb2dbd6SSanjoy Das   int Index = 0;
6343bb2dbd6SSanjoy Das   for (auto &I : BB) {
6353bb2dbd6SSanjoy Das     EXPECT_EQ(isGuaranteedToTransferExecutionToSuccessor(&I),
6363bb2dbd6SSanjoy Das               ExpectedAnswers[Index])
6373bb2dbd6SSanjoy Das         << "Incorrect answer at instruction " << Index << " = " << I;
6383bb2dbd6SSanjoy Das     Index++;
6393bb2dbd6SSanjoy Das   }
6403bb2dbd6SSanjoy Das }
64139a684d1SSanjoy Das 
642219e5367SNikita Popov TEST_F(ValueTrackingTest, ComputeNumSignBits_PR32045) {
643219e5367SNikita Popov   parseAssembly(
644219e5367SNikita Popov       "define i32 @test(i32 %a) {\n"
645219e5367SNikita Popov       "  %A = ashr i32 %a, -1\n"
646219e5367SNikita Popov       "  ret i32 %A\n"
647219e5367SNikita Popov       "}\n");
648219e5367SNikita Popov   EXPECT_EQ(ComputeNumSignBits(A, M->getDataLayout()), 1u);
64939a684d1SSanjoy Das }
65070dbd5fbSSimon Dardis 
651a68096c7SSanjay Patel // No guarantees for canonical IR in this analysis, so this just bails out.
652219e5367SNikita Popov TEST_F(ValueTrackingTest, ComputeNumSignBits_Shuffle) {
653219e5367SNikita Popov   parseAssembly(
654219e5367SNikita Popov       "define <2 x i32> @test() {\n"
655219e5367SNikita Popov       "  %A = shufflevector <2 x i32> undef, <2 x i32> undef, <2 x i32> <i32 0, i32 0>\n"
656219e5367SNikita Popov       "  ret <2 x i32> %A\n"
657219e5367SNikita Popov       "}\n");
658219e5367SNikita Popov   EXPECT_EQ(ComputeNumSignBits(A, M->getDataLayout()), 1u);
659a68096c7SSanjay Patel }
660a68096c7SSanjay Patel 
66145a148deSSanjay Patel // No guarantees for canonical IR in this analysis, so a shuffle element that
66245a148deSSanjay Patel // references an undef value means this can't return any extra information.
663219e5367SNikita Popov TEST_F(ValueTrackingTest, ComputeNumSignBits_Shuffle2) {
664219e5367SNikita Popov   parseAssembly(
665219e5367SNikita Popov       "define <2 x i32> @test(<2 x i1> %x) {\n"
666219e5367SNikita Popov       "  %sext = sext <2 x i1> %x to <2 x i32>\n"
667219e5367SNikita Popov       "  %A = shufflevector <2 x i32> %sext, <2 x i32> undef, <2 x i32> <i32 0, i32 2>\n"
668219e5367SNikita Popov       "  ret <2 x i32> %A\n"
669219e5367SNikita Popov       "}\n");
670219e5367SNikita Popov   EXPECT_EQ(ComputeNumSignBits(A, M->getDataLayout()), 1u);
671994543abSJuneyoung Lee }
672994543abSJuneyoung Lee 
673e5f602d8SJuneyoung Lee TEST(ValueTracking, propagatesPoison) {
674e5f602d8SJuneyoung Lee   std::string AsmHead = "declare i32 @g(i32)\n"
675e5f602d8SJuneyoung Lee                         "define void @f(i32 %x, i32 %y, float %fx, float %fy, "
676e5f602d8SJuneyoung Lee                         "i1 %cond, i8* %p) {\n";
677e5f602d8SJuneyoung Lee   std::string AsmTail = "  ret void\n}";
678e5f602d8SJuneyoung Lee   // (propagates poison?, IR instruction)
679e5f602d8SJuneyoung Lee   SmallVector<std::pair<bool, std::string>, 32> Data = {
680e5f602d8SJuneyoung Lee       {true, "add i32 %x, %y"},
681e5f602d8SJuneyoung Lee       {true, "add nsw nuw i32 %x, %y"},
682e5f602d8SJuneyoung Lee       {true, "ashr i32 %x, %y"},
683e5f602d8SJuneyoung Lee       {true, "lshr exact i32 %x, 31"},
684e5f602d8SJuneyoung Lee       {true, "fcmp oeq float %fx, %fy"},
685e5f602d8SJuneyoung Lee       {true, "icmp eq i32 %x, %y"},
686e5f602d8SJuneyoung Lee       {true, "getelementptr i8, i8* %p, i32 %x"},
687e5f602d8SJuneyoung Lee       {true, "getelementptr inbounds i8, i8* %p, i32 %x"},
688e5f602d8SJuneyoung Lee       {true, "bitcast float %fx to i32"},
689e5f602d8SJuneyoung Lee       {false, "select i1 %cond, i32 %x, i32 %y"},
690e5f602d8SJuneyoung Lee       {false, "freeze i32 %x"},
691e5f602d8SJuneyoung Lee       {true, "udiv i32 %x, %y"},
692e5f602d8SJuneyoung Lee       {true, "urem i32 %x, %y"},
693e5f602d8SJuneyoung Lee       {true, "sdiv exact i32 %x, %y"},
694e5f602d8SJuneyoung Lee       {true, "srem i32 %x, %y"},
695e5f602d8SJuneyoung Lee       {false, "call i32 @g(i32 %x)"}};
696e5f602d8SJuneyoung Lee 
697e5f602d8SJuneyoung Lee   std::string AssemblyStr = AsmHead;
698e5f602d8SJuneyoung Lee   for (auto &Itm : Data)
699e5f602d8SJuneyoung Lee     AssemblyStr += Itm.second + "\n";
700e5f602d8SJuneyoung Lee   AssemblyStr += AsmTail;
701e5f602d8SJuneyoung Lee 
702e5f602d8SJuneyoung Lee   LLVMContext Context;
703e5f602d8SJuneyoung Lee   SMDiagnostic Error;
704e5f602d8SJuneyoung Lee   auto M = parseAssemblyString(AssemblyStr, Error, Context);
705e5f602d8SJuneyoung Lee   assert(M && "Bad assembly?");
706e5f602d8SJuneyoung Lee 
707e5f602d8SJuneyoung Lee   auto *F = M->getFunction("f");
708e5f602d8SJuneyoung Lee   assert(F && "Bad assembly?");
709e5f602d8SJuneyoung Lee 
710e5f602d8SJuneyoung Lee   auto &BB = F->getEntryBlock();
711e5f602d8SJuneyoung Lee 
712e5f602d8SJuneyoung Lee   int Index = 0;
713e5f602d8SJuneyoung Lee   for (auto &I : BB) {
714e5f602d8SJuneyoung Lee     if (isa<ReturnInst>(&I))
715e5f602d8SJuneyoung Lee       break;
716e5f602d8SJuneyoung Lee     EXPECT_EQ(propagatesPoison(&I), Data[Index].first)
717e5f602d8SJuneyoung Lee         << "Incorrect answer at instruction " << Index << " = " << I;
718e5f602d8SJuneyoung Lee     Index++;
719e5f602d8SJuneyoung Lee   }
720e5f602d8SJuneyoung Lee }
721e5f602d8SJuneyoung Lee 
722994543abSJuneyoung Lee TEST(ValueTracking, canCreatePoison) {
723994543abSJuneyoung Lee   std::string AsmHead =
724994543abSJuneyoung Lee       "declare i32 @g(i32)\n"
725994543abSJuneyoung Lee       "define void @f(i32 %x, i32 %y, float %fx, float %fy, i1 %cond, "
726994543abSJuneyoung Lee       "<4 x i32> %vx, <4 x i32> %vx2, <vscale x 4 x i32> %svx, i8* %p) {\n";
727994543abSJuneyoung Lee   std::string AsmTail = "  ret void\n}";
728994543abSJuneyoung Lee   // (can create poison?, IR instruction)
729994543abSJuneyoung Lee   SmallVector<std::pair<bool, std::string>, 32> Data = {
730994543abSJuneyoung Lee       {false, "add i32 %x, %y"},
731994543abSJuneyoung Lee       {true, "add nsw nuw i32 %x, %y"},
732994543abSJuneyoung Lee       {true, "shl i32 %x, %y"},
733994543abSJuneyoung Lee       {true, "shl <4 x i32> %vx, %vx2"},
734994543abSJuneyoung Lee       {true, "shl nsw i32 %x, %y"},
735994543abSJuneyoung Lee       {true, "shl nsw <4 x i32> %vx, <i32 0, i32 1, i32 2, i32 3>"},
736994543abSJuneyoung Lee       {false, "shl i32 %x, 31"},
737994543abSJuneyoung Lee       {true, "shl i32 %x, 32"},
738994543abSJuneyoung Lee       {false, "shl <4 x i32> %vx, <i32 0, i32 1, i32 2, i32 3>"},
739994543abSJuneyoung Lee       {true, "shl <4 x i32> %vx, <i32 0, i32 1, i32 2, i32 32>"},
740994543abSJuneyoung Lee       {true, "ashr i32 %x, %y"},
741994543abSJuneyoung Lee       {true, "ashr exact i32 %x, %y"},
742994543abSJuneyoung Lee       {false, "ashr i32 %x, 31"},
743994543abSJuneyoung Lee       {true, "ashr exact i32 %x, 31"},
744994543abSJuneyoung Lee       {false, "ashr <4 x i32> %vx, <i32 0, i32 1, i32 2, i32 3>"},
745994543abSJuneyoung Lee       {true, "ashr <4 x i32> %vx, <i32 0, i32 1, i32 2, i32 32>"},
746994543abSJuneyoung Lee       {true, "ashr exact <4 x i32> %vx, <i32 0, i32 1, i32 2, i32 3>"},
747994543abSJuneyoung Lee       {true, "lshr i32 %x, %y"},
748994543abSJuneyoung Lee       {true, "lshr exact i32 %x, 31"},
749994543abSJuneyoung Lee       {false, "udiv i32 %x, %y"},
750994543abSJuneyoung Lee       {true, "udiv exact i32 %x, %y"},
751994543abSJuneyoung Lee       {false, "getelementptr i8, i8* %p, i32 %x"},
752994543abSJuneyoung Lee       {true, "getelementptr inbounds i8, i8* %p, i32 %x"},
753994543abSJuneyoung Lee       {true, "fneg nnan float %fx"},
754994543abSJuneyoung Lee       {false, "fneg float %fx"},
755994543abSJuneyoung Lee       {false, "fadd float %fx, %fy"},
756994543abSJuneyoung Lee       {true, "fadd nnan float %fx, %fy"},
757994543abSJuneyoung Lee       {false, "urem i32 %x, %y"},
758994543abSJuneyoung Lee       {true, "fptoui float %fx to i32"},
759994543abSJuneyoung Lee       {true, "fptosi float %fx to i32"},
760994543abSJuneyoung Lee       {false, "bitcast float %fx to i32"},
761994543abSJuneyoung Lee       {false, "select i1 %cond, i32 %x, i32 %y"},
762994543abSJuneyoung Lee       {true, "select nnan i1 %cond, float %fx, float %fy"},
763994543abSJuneyoung Lee       {true, "extractelement <4 x i32> %vx, i32 %x"},
764994543abSJuneyoung Lee       {false, "extractelement <4 x i32> %vx, i32 3"},
765994543abSJuneyoung Lee       {true, "extractelement <vscale x 4 x i32> %svx, i32 4"},
766994543abSJuneyoung Lee       {true, "insertelement <4 x i32> %vx, i32 %x, i32 %y"},
767994543abSJuneyoung Lee       {false, "insertelement <4 x i32> %vx, i32 %x, i32 3"},
768994543abSJuneyoung Lee       {true, "insertelement <vscale x 4 x i32> %svx, i32 %x, i32 4"},
769994543abSJuneyoung Lee       {false, "freeze i32 %x"},
770994543abSJuneyoung Lee       {true, "call i32 @g(i32 %x)"},
771994543abSJuneyoung Lee       {true, "fcmp nnan oeq float %fx, %fy"},
772994543abSJuneyoung Lee       {false, "fcmp oeq float %fx, %fy"}};
773994543abSJuneyoung Lee 
774994543abSJuneyoung Lee   std::string AssemblyStr = AsmHead;
775994543abSJuneyoung Lee   for (auto &Itm : Data)
776994543abSJuneyoung Lee     AssemblyStr += Itm.second + "\n";
777994543abSJuneyoung Lee   AssemblyStr += AsmTail;
778994543abSJuneyoung Lee 
779994543abSJuneyoung Lee   LLVMContext Context;
780994543abSJuneyoung Lee   SMDiagnostic Error;
781994543abSJuneyoung Lee   auto M = parseAssemblyString(AssemblyStr, Error, Context);
782994543abSJuneyoung Lee   assert(M && "Bad assembly?");
783994543abSJuneyoung Lee 
784994543abSJuneyoung Lee   auto *F = M->getFunction("f");
785994543abSJuneyoung Lee   assert(F && "Bad assembly?");
786994543abSJuneyoung Lee 
787994543abSJuneyoung Lee   auto &BB = F->getEntryBlock();
788994543abSJuneyoung Lee 
789994543abSJuneyoung Lee   int Index = 0;
790994543abSJuneyoung Lee   for (auto &I : BB) {
791994543abSJuneyoung Lee     if (isa<ReturnInst>(&I))
792994543abSJuneyoung Lee       break;
793994543abSJuneyoung Lee     EXPECT_EQ(canCreatePoison(&I), Data[Index].first)
794994543abSJuneyoung Lee         << "Incorrect answer at instruction " << Index << " = " << I;
795994543abSJuneyoung Lee     Index++;
796994543abSJuneyoung Lee   }
79745a148deSSanjay Patel }
79845a148deSSanjay Patel 
799219e5367SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownBits) {
800219e5367SNikita Popov   parseAssembly(
801219e5367SNikita Popov       "define i32 @test(i32 %a, i32 %b) {\n"
802219e5367SNikita Popov       "  %ash = mul i32 %a, 8\n"
803219e5367SNikita Popov       "  %aad = add i32 %ash, 7\n"
804219e5367SNikita Popov       "  %aan = and i32 %aad, 4095\n"
805219e5367SNikita Popov       "  %bsh = shl i32 %b, 4\n"
806219e5367SNikita Popov       "  %bad = or i32 %bsh, 6\n"
807219e5367SNikita Popov       "  %ban = and i32 %bad, 4095\n"
808219e5367SNikita Popov       "  %A = mul i32 %aan, %ban\n"
809219e5367SNikita Popov       "  ret i32 %A\n"
810219e5367SNikita Popov       "}\n");
811219e5367SNikita Popov   expectKnownBits(/*zero*/ 4278190085u, /*one*/ 10u);
81270dbd5fbSSimon Dardis }
81370dbd5fbSSimon Dardis 
814219e5367SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownMulBits) {
815219e5367SNikita Popov   parseAssembly(
816219e5367SNikita Popov       "define i32 @test(i32 %a, i32 %b) {\n"
817219e5367SNikita Popov       "  %aa = shl i32 %a, 5\n"
818219e5367SNikita Popov       "  %bb = shl i32 %b, 5\n"
819219e5367SNikita Popov       "  %aaa = or i32 %aa, 24\n"
820219e5367SNikita Popov       "  %bbb = or i32 %bb, 28\n"
821219e5367SNikita Popov       "  %A = mul i32 %aaa, %bbb\n"
822219e5367SNikita Popov       "  ret i32 %A\n"
823219e5367SNikita Popov       "}\n");
824219e5367SNikita Popov   expectKnownBits(/*zero*/ 95u, /*one*/ 32u);
82570dbd5fbSSimon Dardis }
826687b92cdSNikita Popov 
8277cf0f856SNikita Popov TEST_F(ComputeKnownBitsTest, KnownNonZeroShift) {
8287cf0f856SNikita Popov   // %q is known nonzero without known bits.
8297cf0f856SNikita Popov   // Because %q is nonzero, %A[0] is known to be zero.
8307cf0f856SNikita Popov   parseAssembly(
8317cf0f856SNikita Popov       "define i8 @test(i8 %p, i8* %pq) {\n"
8327cf0f856SNikita Popov       "  %q = load i8, i8* %pq, !range !0\n"
8337cf0f856SNikita Popov       "  %A = shl i8 %p, %q\n"
8347cf0f856SNikita Popov       "  ret i8 %A\n"
8357cf0f856SNikita Popov       "}\n"
8367cf0f856SNikita Popov       "!0 = !{ i8 1, i8 5 }\n");
8377cf0f856SNikita Popov   expectKnownBits(/*zero*/ 1u, /*one*/ 0u);
8387cf0f856SNikita Popov }
8397cf0f856SNikita Popov 
840687b92cdSNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownFshl) {
841687b92cdSNikita Popov   // fshl(....1111....0000, 00..1111........, 6)
842687b92cdSNikita Popov   // = 11....000000..11
843687b92cdSNikita Popov   parseAssembly(
844687b92cdSNikita Popov       "define i16 @test(i16 %a, i16 %b) {\n"
845687b92cdSNikita Popov       "  %aa = shl i16 %a, 4\n"
846687b92cdSNikita Popov       "  %bb = lshr i16 %b, 2\n"
847687b92cdSNikita Popov       "  %aaa = or i16 %aa, 3840\n"
848687b92cdSNikita Popov       "  %bbb = or i16 %bb, 3840\n"
849687b92cdSNikita Popov       "  %A = call i16 @llvm.fshl.i16(i16 %aaa, i16 %bbb, i16 6)\n"
850687b92cdSNikita Popov       "  ret i16 %A\n"
851687b92cdSNikita Popov       "}\n"
852687b92cdSNikita Popov       "declare i16 @llvm.fshl.i16(i16, i16, i16)\n");
853687b92cdSNikita Popov   expectKnownBits(/*zero*/ 1008u, /*one*/ 49155u);
854687b92cdSNikita Popov }
855687b92cdSNikita Popov 
856687b92cdSNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownFshr) {
857687b92cdSNikita Popov   // fshr(....1111....0000, 00..1111........, 26)
858687b92cdSNikita Popov   // = 11....000000..11
859687b92cdSNikita Popov   parseAssembly(
860687b92cdSNikita Popov       "define i16 @test(i16 %a, i16 %b) {\n"
861687b92cdSNikita Popov       "  %aa = shl i16 %a, 4\n"
862687b92cdSNikita Popov       "  %bb = lshr i16 %b, 2\n"
863687b92cdSNikita Popov       "  %aaa = or i16 %aa, 3840\n"
864687b92cdSNikita Popov       "  %bbb = or i16 %bb, 3840\n"
865687b92cdSNikita Popov       "  %A = call i16 @llvm.fshr.i16(i16 %aaa, i16 %bbb, i16 26)\n"
866687b92cdSNikita Popov       "  ret i16 %A\n"
867687b92cdSNikita Popov       "}\n"
868687b92cdSNikita Popov       "declare i16 @llvm.fshr.i16(i16, i16, i16)\n");
869687b92cdSNikita Popov   expectKnownBits(/*zero*/ 1008u, /*one*/ 49155u);
870687b92cdSNikita Popov }
871687b92cdSNikita Popov 
872687b92cdSNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownFshlZero) {
873687b92cdSNikita Popov   // fshl(....1111....0000, 00..1111........, 0)
874687b92cdSNikita Popov   // = ....1111....0000
875687b92cdSNikita Popov   parseAssembly(
876687b92cdSNikita Popov       "define i16 @test(i16 %a, i16 %b) {\n"
877687b92cdSNikita Popov       "  %aa = shl i16 %a, 4\n"
878687b92cdSNikita Popov       "  %bb = lshr i16 %b, 2\n"
879687b92cdSNikita Popov       "  %aaa = or i16 %aa, 3840\n"
880687b92cdSNikita Popov       "  %bbb = or i16 %bb, 3840\n"
881687b92cdSNikita Popov       "  %A = call i16 @llvm.fshl.i16(i16 %aaa, i16 %bbb, i16 0)\n"
882687b92cdSNikita Popov       "  ret i16 %A\n"
883687b92cdSNikita Popov       "}\n"
884687b92cdSNikita Popov       "declare i16 @llvm.fshl.i16(i16, i16, i16)\n");
885687b92cdSNikita Popov   expectKnownBits(/*zero*/ 15u, /*one*/ 3840u);
886687b92cdSNikita Popov }
887ed3ca927SNikita Popov 
888ed3ca927SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownUAddSatLeadingOnes) {
889ed3ca927SNikita Popov   // uadd.sat(1111...1, ........)
890ed3ca927SNikita Popov   // = 1111....
891ed3ca927SNikita Popov   parseAssembly(
892ed3ca927SNikita Popov       "define i8 @test(i8 %a, i8 %b) {\n"
893ed3ca927SNikita Popov       "  %aa = or i8 %a, 241\n"
894ed3ca927SNikita Popov       "  %A = call i8 @llvm.uadd.sat.i8(i8 %aa, i8 %b)\n"
895ed3ca927SNikita Popov       "  ret i8 %A\n"
896ed3ca927SNikita Popov       "}\n"
897ed3ca927SNikita Popov       "declare i8 @llvm.uadd.sat.i8(i8, i8)\n");
898ed3ca927SNikita Popov   expectKnownBits(/*zero*/ 0u, /*one*/ 240u);
899ed3ca927SNikita Popov }
900ed3ca927SNikita Popov 
901ed3ca927SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownUAddSatOnesPreserved) {
902ed3ca927SNikita Popov   // uadd.sat(00...011, .1...110)
903ed3ca927SNikita Popov   // = .......1
904ed3ca927SNikita Popov   parseAssembly(
905ed3ca927SNikita Popov       "define i8 @test(i8 %a, i8 %b) {\n"
906ed3ca927SNikita Popov       "  %aa = or i8 %a, 3\n"
907ed3ca927SNikita Popov       "  %aaa = and i8 %aa, 59\n"
908ed3ca927SNikita Popov       "  %bb = or i8 %b, 70\n"
909ed3ca927SNikita Popov       "  %bbb = and i8 %bb, 254\n"
910ed3ca927SNikita Popov       "  %A = call i8 @llvm.uadd.sat.i8(i8 %aaa, i8 %bbb)\n"
911ed3ca927SNikita Popov       "  ret i8 %A\n"
912ed3ca927SNikita Popov       "}\n"
913ed3ca927SNikita Popov       "declare i8 @llvm.uadd.sat.i8(i8, i8)\n");
914ed3ca927SNikita Popov   expectKnownBits(/*zero*/ 0u, /*one*/ 1u);
915ed3ca927SNikita Popov }
916ed3ca927SNikita Popov 
917ed3ca927SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownUSubSatLHSLeadingZeros) {
918ed3ca927SNikita Popov   // usub.sat(0000...0, ........)
919ed3ca927SNikita Popov   // = 0000....
920ed3ca927SNikita Popov   parseAssembly(
921ed3ca927SNikita Popov       "define i8 @test(i8 %a, i8 %b) {\n"
922ed3ca927SNikita Popov       "  %aa = and i8 %a, 14\n"
923ed3ca927SNikita Popov       "  %A = call i8 @llvm.usub.sat.i8(i8 %aa, i8 %b)\n"
924ed3ca927SNikita Popov       "  ret i8 %A\n"
925ed3ca927SNikita Popov       "}\n"
926ed3ca927SNikita Popov       "declare i8 @llvm.usub.sat.i8(i8, i8)\n");
927ed3ca927SNikita Popov   expectKnownBits(/*zero*/ 240u, /*one*/ 0u);
928ed3ca927SNikita Popov }
929ed3ca927SNikita Popov 
930ed3ca927SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownUSubSatRHSLeadingOnes) {
931ed3ca927SNikita Popov   // usub.sat(........, 1111...1)
932ed3ca927SNikita Popov   // = 0000....
933ed3ca927SNikita Popov   parseAssembly(
934ed3ca927SNikita Popov       "define i8 @test(i8 %a, i8 %b) {\n"
935ed3ca927SNikita Popov       "  %bb = or i8 %a, 241\n"
936ed3ca927SNikita Popov       "  %A = call i8 @llvm.usub.sat.i8(i8 %a, i8 %bb)\n"
937ed3ca927SNikita Popov       "  ret i8 %A\n"
938ed3ca927SNikita Popov       "}\n"
939ed3ca927SNikita Popov       "declare i8 @llvm.usub.sat.i8(i8, i8)\n");
940ed3ca927SNikita Popov   expectKnownBits(/*zero*/ 240u, /*one*/ 0u);
941ed3ca927SNikita Popov }
942ed3ca927SNikita Popov 
943ed3ca927SNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownUSubSatZerosPreserved) {
944ed3ca927SNikita Popov   // usub.sat(11...011, .1...110)
945ed3ca927SNikita Popov   // = ......0.
946ed3ca927SNikita Popov   parseAssembly(
947ed3ca927SNikita Popov       "define i8 @test(i8 %a, i8 %b) {\n"
948ed3ca927SNikita Popov       "  %aa = or i8 %a, 195\n"
949ed3ca927SNikita Popov       "  %aaa = and i8 %aa, 251\n"
950ed3ca927SNikita Popov       "  %bb = or i8 %b, 70\n"
951ed3ca927SNikita Popov       "  %bbb = and i8 %bb, 254\n"
952ed3ca927SNikita Popov       "  %A = call i8 @llvm.usub.sat.i8(i8 %aaa, i8 %bbb)\n"
953ed3ca927SNikita Popov       "  ret i8 %A\n"
954ed3ca927SNikita Popov       "}\n"
955ed3ca927SNikita Popov       "declare i8 @llvm.usub.sat.i8(i8, i8)\n");
956ed3ca927SNikita Popov   expectKnownBits(/*zero*/ 2u, /*one*/ 0u);
957ed3ca927SNikita Popov }
958329d15b5SVitaly Buka 
959*d86fff6aSNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownBitsPtrToIntTrunc) {
960*d86fff6aSNikita Popov   // ptrtoint truncates the pointer type.
961*d86fff6aSNikita Popov   parseAssembly(
962*d86fff6aSNikita Popov       "define void @test(i8** %p) {\n"
963*d86fff6aSNikita Popov       "  %A = load i8*, i8** %p\n"
964*d86fff6aSNikita Popov       "  %i = ptrtoint i8* %A to i32\n"
965*d86fff6aSNikita Popov       "  %m = and i32 %i, 31\n"
966*d86fff6aSNikita Popov       "  %c = icmp eq i32 %m, 0\n"
967*d86fff6aSNikita Popov       "  call void @llvm.assume(i1 %c)\n"
968*d86fff6aSNikita Popov       "  ret void\n"
969*d86fff6aSNikita Popov       "}\n"
970*d86fff6aSNikita Popov       "declare void @llvm.assume(i1)\n");
971*d86fff6aSNikita Popov   AssumptionCache AC(*F);
972*d86fff6aSNikita Popov   KnownBits Known = computeKnownBits(
973*d86fff6aSNikita Popov       A, M->getDataLayout(), /* Depth */ 0, &AC, F->front().getTerminator());
974*d86fff6aSNikita Popov   EXPECT_EQ(Known.Zero.getZExtValue(), 31u);
975*d86fff6aSNikita Popov   EXPECT_EQ(Known.One.getZExtValue(), 0u);
976*d86fff6aSNikita Popov }
977*d86fff6aSNikita Popov 
978*d86fff6aSNikita Popov TEST_F(ComputeKnownBitsTest, ComputeKnownBitsPtrToIntZext) {
979*d86fff6aSNikita Popov   // ptrtoint zero extends the pointer type.
980*d86fff6aSNikita Popov   parseAssembly(
981*d86fff6aSNikita Popov       "define void @test(i8** %p) {\n"
982*d86fff6aSNikita Popov       "  %A = load i8*, i8** %p\n"
983*d86fff6aSNikita Popov       "  %i = ptrtoint i8* %A to i128\n"
984*d86fff6aSNikita Popov       "  %m = and i128 %i, 31\n"
985*d86fff6aSNikita Popov       "  %c = icmp eq i128 %m, 0\n"
986*d86fff6aSNikita Popov       "  call void @llvm.assume(i1 %c)\n"
987*d86fff6aSNikita Popov       "  ret void\n"
988*d86fff6aSNikita Popov       "}\n"
989*d86fff6aSNikita Popov       "declare void @llvm.assume(i1)\n");
990*d86fff6aSNikita Popov   AssumptionCache AC(*F);
991*d86fff6aSNikita Popov   KnownBits Known = computeKnownBits(
992*d86fff6aSNikita Popov       A, M->getDataLayout(), /* Depth */ 0, &AC, F->front().getTerminator());
993*d86fff6aSNikita Popov   EXPECT_EQ(Known.Zero.getZExtValue(), 31u);
994*d86fff6aSNikita Popov   EXPECT_EQ(Known.One.getZExtValue(), 0u);
995*d86fff6aSNikita Popov }
996*d86fff6aSNikita Popov 
997329d15b5SVitaly Buka class IsBytewiseValueTest : public ValueTrackingTest,
998329d15b5SVitaly Buka                             public ::testing::WithParamInterface<
999329d15b5SVitaly Buka                                 std::pair<const char *, const char *>> {
1000329d15b5SVitaly Buka protected:
1001329d15b5SVitaly Buka };
1002329d15b5SVitaly Buka 
1003329d15b5SVitaly Buka const std::pair<const char *, const char *> IsBytewiseValueTests[] = {
1004329d15b5SVitaly Buka     {
1005329d15b5SVitaly Buka         "i8 0",
1006329d15b5SVitaly Buka         "i48* null",
1007329d15b5SVitaly Buka     },
1008329d15b5SVitaly Buka     {
1009329d15b5SVitaly Buka         "i8 undef",
1010329d15b5SVitaly Buka         "i48* undef",
1011329d15b5SVitaly Buka     },
1012329d15b5SVitaly Buka     {
1013329d15b5SVitaly Buka         "i8 0",
1014329d15b5SVitaly Buka         "i8 zeroinitializer",
1015329d15b5SVitaly Buka     },
1016329d15b5SVitaly Buka     {
1017329d15b5SVitaly Buka         "i8 0",
1018329d15b5SVitaly Buka         "i8 0",
1019329d15b5SVitaly Buka     },
1020329d15b5SVitaly Buka     {
1021329d15b5SVitaly Buka         "i8 -86",
1022329d15b5SVitaly Buka         "i8 -86",
1023329d15b5SVitaly Buka     },
1024329d15b5SVitaly Buka     {
1025329d15b5SVitaly Buka         "i8 -1",
1026329d15b5SVitaly Buka         "i8 -1",
1027329d15b5SVitaly Buka     },
1028329d15b5SVitaly Buka     {
1029329d15b5SVitaly Buka         "i8 undef",
1030329d15b5SVitaly Buka         "i16 undef",
1031329d15b5SVitaly Buka     },
1032329d15b5SVitaly Buka     {
1033329d15b5SVitaly Buka         "i8 0",
1034329d15b5SVitaly Buka         "i16 0",
1035329d15b5SVitaly Buka     },
1036329d15b5SVitaly Buka     {
1037329d15b5SVitaly Buka         "",
1038329d15b5SVitaly Buka         "i16 7",
1039329d15b5SVitaly Buka     },
1040329d15b5SVitaly Buka     {
1041329d15b5SVitaly Buka         "i8 -86",
1042329d15b5SVitaly Buka         "i16 -21846",
1043329d15b5SVitaly Buka     },
1044329d15b5SVitaly Buka     {
1045329d15b5SVitaly Buka         "i8 -1",
1046329d15b5SVitaly Buka         "i16 -1",
1047329d15b5SVitaly Buka     },
1048329d15b5SVitaly Buka     {
1049329d15b5SVitaly Buka         "i8 0",
1050329d15b5SVitaly Buka         "i48 0",
1051329d15b5SVitaly Buka     },
1052329d15b5SVitaly Buka     {
1053329d15b5SVitaly Buka         "i8 -1",
1054329d15b5SVitaly Buka         "i48 -1",
1055329d15b5SVitaly Buka     },
1056329d15b5SVitaly Buka     {
1057329d15b5SVitaly Buka         "i8 0",
1058329d15b5SVitaly Buka         "i49 0",
1059329d15b5SVitaly Buka     },
1060329d15b5SVitaly Buka     {
1061329d15b5SVitaly Buka         "",
1062329d15b5SVitaly Buka         "i49 -1",
1063329d15b5SVitaly Buka     },
1064329d15b5SVitaly Buka     {
1065329d15b5SVitaly Buka         "i8 0",
1066329d15b5SVitaly Buka         "half 0xH0000",
1067329d15b5SVitaly Buka     },
1068329d15b5SVitaly Buka     {
1069329d15b5SVitaly Buka         "i8 -85",
1070329d15b5SVitaly Buka         "half 0xHABAB",
1071329d15b5SVitaly Buka     },
1072329d15b5SVitaly Buka     {
1073329d15b5SVitaly Buka         "i8 0",
1074329d15b5SVitaly Buka         "float 0.0",
1075329d15b5SVitaly Buka     },
1076329d15b5SVitaly Buka     {
1077329d15b5SVitaly Buka         "i8 -1",
1078329d15b5SVitaly Buka         "float 0xFFFFFFFFE0000000",
1079329d15b5SVitaly Buka     },
1080329d15b5SVitaly Buka     {
1081329d15b5SVitaly Buka         "i8 0",
1082329d15b5SVitaly Buka         "double 0.0",
1083329d15b5SVitaly Buka     },
1084329d15b5SVitaly Buka     {
1085329d15b5SVitaly Buka         "i8 -15",
1086329d15b5SVitaly Buka         "double 0xF1F1F1F1F1F1F1F1",
1087329d15b5SVitaly Buka     },
1088329d15b5SVitaly Buka     {
1089329d15b5SVitaly Buka         "i8 undef",
1090329d15b5SVitaly Buka         "i16* undef",
1091329d15b5SVitaly Buka     },
1092329d15b5SVitaly Buka     {
1093329d15b5SVitaly Buka         "i8 0",
1094329d15b5SVitaly Buka         "i16* inttoptr (i64 0 to i16*)",
1095329d15b5SVitaly Buka     },
1096329d15b5SVitaly Buka     {
1097c559e637SVitaly Buka         "i8 -1",
1098329d15b5SVitaly Buka         "i16* inttoptr (i64 -1 to i16*)",
1099329d15b5SVitaly Buka     },
1100329d15b5SVitaly Buka     {
1101c559e637SVitaly Buka         "i8 -86",
1102329d15b5SVitaly Buka         "i16* inttoptr (i64 -6148914691236517206 to i16*)",
1103329d15b5SVitaly Buka     },
1104329d15b5SVitaly Buka     {
1105329d15b5SVitaly Buka         "",
1106329d15b5SVitaly Buka         "i16* inttoptr (i48 -1 to i16*)",
1107329d15b5SVitaly Buka     },
1108329d15b5SVitaly Buka     {
1109c559e637SVitaly Buka         "i8 -1",
1110329d15b5SVitaly Buka         "i16* inttoptr (i96 -1 to i16*)",
1111329d15b5SVitaly Buka     },
1112329d15b5SVitaly Buka     {
111352096ee9SVitaly Buka         "i8 undef",
1114329d15b5SVitaly Buka         "[0 x i8] zeroinitializer",
1115329d15b5SVitaly Buka     },
1116329d15b5SVitaly Buka     {
1117329d15b5SVitaly Buka         "i8 undef",
1118329d15b5SVitaly Buka         "[0 x i8] undef",
1119329d15b5SVitaly Buka     },
1120329d15b5SVitaly Buka     {
112152096ee9SVitaly Buka         "i8 undef",
1122329d15b5SVitaly Buka         "[5 x [0 x i8]] zeroinitializer",
1123329d15b5SVitaly Buka     },
1124329d15b5SVitaly Buka     {
1125329d15b5SVitaly Buka         "i8 undef",
1126329d15b5SVitaly Buka         "[5 x [0 x i8]] undef",
1127329d15b5SVitaly Buka     },
1128329d15b5SVitaly Buka     {
1129329d15b5SVitaly Buka         "i8 0",
1130329d15b5SVitaly Buka         "[6 x i8] zeroinitializer",
1131329d15b5SVitaly Buka     },
1132329d15b5SVitaly Buka     {
1133329d15b5SVitaly Buka         "i8 undef",
1134329d15b5SVitaly Buka         "[6 x i8] undef",
1135329d15b5SVitaly Buka     },
1136329d15b5SVitaly Buka     {
1137329d15b5SVitaly Buka         "i8 1",
1138329d15b5SVitaly Buka         "[5 x i8] [i8 1, i8 1, i8 1, i8 1, i8 1]",
1139329d15b5SVitaly Buka     },
1140329d15b5SVitaly Buka     {
1141329d15b5SVitaly Buka         "",
1142329d15b5SVitaly Buka         "[5 x i64] [i64 1, i64 1, i64 1, i64 1, i64 1]",
1143329d15b5SVitaly Buka     },
1144329d15b5SVitaly Buka     {
1145329d15b5SVitaly Buka         "i8 -1",
1146329d15b5SVitaly Buka         "[5 x i64] [i64 -1, i64 -1, i64 -1, i64 -1, i64 -1]",
1147329d15b5SVitaly Buka     },
1148329d15b5SVitaly Buka     {
1149329d15b5SVitaly Buka         "",
1150329d15b5SVitaly Buka         "[4 x i8] [i8 1, i8 2, i8 1, i8 1]",
1151329d15b5SVitaly Buka     },
1152329d15b5SVitaly Buka     {
1153329d15b5SVitaly Buka         "i8 1",
1154329d15b5SVitaly Buka         "[4 x i8] [i8 1, i8 undef, i8 1, i8 1]",
1155329d15b5SVitaly Buka     },
1156329d15b5SVitaly Buka     {
1157329d15b5SVitaly Buka         "i8 0",
1158329d15b5SVitaly Buka         "<6 x i8> zeroinitializer",
1159329d15b5SVitaly Buka     },
1160329d15b5SVitaly Buka     {
1161329d15b5SVitaly Buka         "i8 undef",
1162329d15b5SVitaly Buka         "<6 x i8> undef",
1163329d15b5SVitaly Buka     },
1164329d15b5SVitaly Buka     {
1165329d15b5SVitaly Buka         "i8 1",
1166329d15b5SVitaly Buka         "<5 x i8> <i8 1, i8 1, i8 1, i8 1, i8 1>",
1167329d15b5SVitaly Buka     },
1168329d15b5SVitaly Buka     {
1169329d15b5SVitaly Buka         "",
1170329d15b5SVitaly Buka         "<5 x i64> <i64 1, i64 1, i64 1, i64 1, i64 1>",
1171329d15b5SVitaly Buka     },
1172329d15b5SVitaly Buka     {
1173329d15b5SVitaly Buka         "i8 -1",
1174329d15b5SVitaly Buka         "<5 x i64> <i64 -1, i64 -1, i64 -1, i64 -1, i64 -1>",
1175329d15b5SVitaly Buka     },
1176329d15b5SVitaly Buka     {
1177329d15b5SVitaly Buka         "",
1178329d15b5SVitaly Buka         "<4 x i8> <i8 1, i8 1, i8 2, i8 1>",
1179329d15b5SVitaly Buka     },
1180329d15b5SVitaly Buka     {
1181b1bff76eSVitaly Buka         "i8 5",
1182329d15b5SVitaly Buka         "<2 x i8> < i8 5, i8 undef >",
1183329d15b5SVitaly Buka     },
1184329d15b5SVitaly Buka     {
1185329d15b5SVitaly Buka         "i8 0",
1186329d15b5SVitaly Buka         "[2 x [2 x i16]] zeroinitializer",
1187329d15b5SVitaly Buka     },
1188329d15b5SVitaly Buka     {
1189329d15b5SVitaly Buka         "i8 undef",
1190329d15b5SVitaly Buka         "[2 x [2 x i16]] undef",
1191329d15b5SVitaly Buka     },
1192329d15b5SVitaly Buka     {
1193329d15b5SVitaly Buka         "i8 -86",
1194329d15b5SVitaly Buka         "[2 x [2 x i16]] [[2 x i16] [i16 -21846, i16 -21846], "
1195329d15b5SVitaly Buka         "[2 x i16] [i16 -21846, i16 -21846]]",
1196329d15b5SVitaly Buka     },
1197329d15b5SVitaly Buka     {
1198329d15b5SVitaly Buka         "",
1199329d15b5SVitaly Buka         "[2 x [2 x i16]] [[2 x i16] [i16 -21846, i16 -21846], "
1200329d15b5SVitaly Buka         "[2 x i16] [i16 -21836, i16 -21846]]",
1201329d15b5SVitaly Buka     },
1202329d15b5SVitaly Buka     {
120352096ee9SVitaly Buka         "i8 undef",
1204329d15b5SVitaly Buka         "{ } zeroinitializer",
1205329d15b5SVitaly Buka     },
1206329d15b5SVitaly Buka     {
1207329d15b5SVitaly Buka         "i8 undef",
1208329d15b5SVitaly Buka         "{ } undef",
1209329d15b5SVitaly Buka     },
1210329d15b5SVitaly Buka     {
121152096ee9SVitaly Buka         "i8 undef",
1212329d15b5SVitaly Buka         "{ {}, {} } zeroinitializer",
1213329d15b5SVitaly Buka     },
1214329d15b5SVitaly Buka     {
1215329d15b5SVitaly Buka         "i8 undef",
1216329d15b5SVitaly Buka         "{ {}, {} } undef",
1217329d15b5SVitaly Buka     },
1218329d15b5SVitaly Buka     {
1219329d15b5SVitaly Buka         "i8 0",
1220329d15b5SVitaly Buka         "{i8, i64, i16*} zeroinitializer",
1221329d15b5SVitaly Buka     },
1222329d15b5SVitaly Buka     {
1223329d15b5SVitaly Buka         "i8 undef",
1224329d15b5SVitaly Buka         "{i8, i64, i16*} undef",
1225329d15b5SVitaly Buka     },
1226329d15b5SVitaly Buka     {
1227329d15b5SVitaly Buka         "i8 -86",
1228329d15b5SVitaly Buka         "{i8, i64, i16*} {i8 -86, i64 -6148914691236517206, i16* undef}",
1229329d15b5SVitaly Buka     },
1230329d15b5SVitaly Buka     {
1231329d15b5SVitaly Buka         "",
1232329d15b5SVitaly Buka         "{i8, i64, i16*} {i8 86, i64 -6148914691236517206, i16* undef}",
1233329d15b5SVitaly Buka     },
1234329d15b5SVitaly Buka };
1235329d15b5SVitaly Buka 
1236329d15b5SVitaly Buka INSTANTIATE_TEST_CASE_P(IsBytewiseValueParamTests, IsBytewiseValueTest,
12377618b2a5SMikael Holmen                         ::testing::ValuesIn(IsBytewiseValueTests),);
1238329d15b5SVitaly Buka 
1239329d15b5SVitaly Buka TEST_P(IsBytewiseValueTest, IsBytewiseValue) {
1240329d15b5SVitaly Buka   auto M = parseModule(std::string("@test = global ") + GetParam().second);
1241329d15b5SVitaly Buka   GlobalVariable *GV = dyn_cast<GlobalVariable>(M->getNamedValue("test"));
1242329d15b5SVitaly Buka   Value *Actual = isBytewiseValue(GV->getInitializer(), M->getDataLayout());
1243329d15b5SVitaly Buka   std::string Buff;
1244329d15b5SVitaly Buka   raw_string_ostream S(Buff);
1245329d15b5SVitaly Buka   if (Actual)
1246329d15b5SVitaly Buka     S << *Actual;
1247329d15b5SVitaly Buka   EXPECT_EQ(GetParam().first, S.str());
1248329d15b5SVitaly Buka }
1249