blob: 8877799b6ae4f9409c0682cad62ce681b8b1c5f8 [file] [log] [blame]
Chris Lattner7e044912010-01-04 07:17:19 +00001//===- InstCombineSimplifyDemanded.cpp ------------------------------------===//
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//
10// This file contains logic for simplifying instructions based on information
11// about how they are used.
12//
13//===----------------------------------------------------------------------===//
14
Chandler Carrutha9174582015-01-22 05:25:13 +000015#include "InstCombineInternal.h"
James Molloy2b21a7c2015-05-20 18:41:25 +000016#include "llvm/Analysis/ValueTracking.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000017#include "llvm/IR/IntrinsicInst.h"
Chandler Carruth820a9082014-03-04 11:08:18 +000018#include "llvm/IR/PatternMatch.h"
Craig Topperb45eabc2017-04-26 16:39:58 +000019#include "llvm/Support/KnownBits.h"
Chris Lattner7e044912010-01-04 07:17:19 +000020
21using namespace llvm;
Shuxin Yang63e999e2012-12-04 00:04:54 +000022using namespace llvm::PatternMatch;
Chris Lattner7e044912010-01-04 07:17:19 +000023
Chandler Carruth964daaa2014-04-22 02:55:47 +000024#define DEBUG_TYPE "instcombine"
25
Sanjay Patelbbbb3ce2016-07-14 20:54:43 +000026/// Check to see if the specified operand of the specified instruction is a
27/// constant integer. If so, check to see if there are any bits set in the
28/// constant that are not demanded. If so, shrink the constant and return true.
Craig Topper4c947752012-12-22 18:09:02 +000029static bool ShrinkDemandedConstant(Instruction *I, unsigned OpNo,
Craig Topper358cd9a2017-04-20 23:58:27 +000030 const APInt &Demanded) {
Chris Lattner7e044912010-01-04 07:17:19 +000031 assert(I && "No instruction?");
32 assert(OpNo < I->getNumOperands() && "Operand index too large");
33
Sanjay Patelae3b43e2017-02-09 21:43:06 +000034 // The operand must be a constant integer or splat integer.
35 Value *Op = I->getOperand(OpNo);
36 const APInt *C;
37 if (!match(Op, m_APInt(C)))
38 return false;
Chris Lattner7e044912010-01-04 07:17:19 +000039
40 // If there are no bits set that aren't demanded, nothing to do.
Craig Toppera8129a12017-04-20 16:17:13 +000041 if (C->isSubsetOf(Demanded))
Chris Lattner7e044912010-01-04 07:17:19 +000042 return false;
43
44 // This instruction is producing bits that are not demanded. Shrink the RHS.
Craig Topper358cd9a2017-04-20 23:58:27 +000045 I->setOperand(OpNo, ConstantInt::get(Op->getType(), *C & Demanded));
David Majnemer42b83a52014-08-22 07:56:32 +000046
Chris Lattner7e044912010-01-04 07:17:19 +000047 return true;
48}
49
50
51
Sanjay Patelbbbb3ce2016-07-14 20:54:43 +000052/// Inst is an integer instruction that SimplifyDemandedBits knows about. See if
53/// the instruction has any properties that allow us to simplify its operands.
Chris Lattner7e044912010-01-04 07:17:19 +000054bool InstCombiner::SimplifyDemandedInstructionBits(Instruction &Inst) {
55 unsigned BitWidth = Inst.getType()->getScalarSizeInBits();
Craig Topperb45eabc2017-04-26 16:39:58 +000056 KnownBits Known(BitWidth);
Chris Lattner7e044912010-01-04 07:17:19 +000057 APInt DemandedMask(APInt::getAllOnesValue(BitWidth));
Craig Topper4c947752012-12-22 18:09:02 +000058
Craig Topperb45eabc2017-04-26 16:39:58 +000059 Value *V = SimplifyDemandedUseBits(&Inst, DemandedMask, Known,
Mehdi Aminia28d91d2015-03-10 02:37:25 +000060 0, &Inst);
Craig Topperf40110f2014-04-25 05:29:35 +000061 if (!V) return false;
Chris Lattner7e044912010-01-04 07:17:19 +000062 if (V == &Inst) return true;
Sanjay Patel4b198802016-02-01 22:23:39 +000063 replaceInstUsesWith(Inst, V);
Chris Lattner7e044912010-01-04 07:17:19 +000064 return true;
65}
66
Sanjay Patelbbbb3ce2016-07-14 20:54:43 +000067/// This form of SimplifyDemandedBits simplifies the specified instruction
68/// operand if possible, updating it in place. It returns true if it made any
69/// change and false otherwise.
Craig Topper47596dd2017-03-25 06:52:52 +000070bool InstCombiner::SimplifyDemandedBits(Instruction *I, unsigned OpNo,
71 const APInt &DemandedMask,
Craig Topperb45eabc2017-04-26 16:39:58 +000072 KnownBits &Known,
Chris Lattner7e044912010-01-04 07:17:19 +000073 unsigned Depth) {
Craig Topper47596dd2017-03-25 06:52:52 +000074 Use &U = I->getOperandUse(OpNo);
Craig Topperb45eabc2017-04-26 16:39:58 +000075 Value *NewVal = SimplifyDemandedUseBits(U.get(), DemandedMask, Known,
76 Depth, I);
Craig Topperf40110f2014-04-25 05:29:35 +000077 if (!NewVal) return false;
Chris Lattner7e044912010-01-04 07:17:19 +000078 U = NewVal;
79 return true;
80}
81
82
Sanjay Patelbbbb3ce2016-07-14 20:54:43 +000083/// This function attempts to replace V with a simpler value based on the
84/// demanded bits. When this function is called, it is known that only the bits
85/// set in DemandedMask of the result of V are ever used downstream.
86/// Consequently, depending on the mask and V, it may be possible to replace V
87/// with a constant or one of its operands. In such cases, this function does
88/// the replacement and returns true. In all other cases, it returns false after
89/// analyzing the expression and setting KnownOne and known to be one in the
Craig Topperb45eabc2017-04-26 16:39:58 +000090/// expression. Known.Zero contains all the bits that are known to be zero in
91/// the expression. These are provided to potentially allow the caller (which
92/// might recursively be SimplifyDemandedBits itself) to simplify the
93/// expression.
94/// Known.One and Known.Zero always follow the invariant that:
95/// Known.One & Known.Zero == 0.
96/// That is, a bit can't be both 1 and 0. Note that the bits in Known.One and
97/// Known.Zero may only be accurate for those bits set in DemandedMask. Note
98/// also that the bitwidth of V, DemandedMask, Known.Zero and Known.One must all
99/// be the same.
Chris Lattner7e044912010-01-04 07:17:19 +0000100///
101/// This returns null if it did not change anything and it permits no
102/// simplification. This returns V itself if it did some simplification of V's
103/// operands based on the information about what bits are demanded. This returns
104/// some other non-null value if it found out that V is equal to another value
105/// in the context where the specified bits are demanded, but not for all users.
106Value *InstCombiner::SimplifyDemandedUseBits(Value *V, APInt DemandedMask,
Craig Topperb45eabc2017-04-26 16:39:58 +0000107 KnownBits &Known, unsigned Depth,
Hal Finkel60db0582014-09-07 18:57:58 +0000108 Instruction *CxtI) {
Craig Toppere73658d2014-04-28 04:05:08 +0000109 assert(V != nullptr && "Null pointer of Value???");
Chris Lattner7e044912010-01-04 07:17:19 +0000110 assert(Depth <= 6 && "Limit Search Depth");
111 uint32_t BitWidth = DemandedMask.getBitWidth();
Chris Lattner229907c2011-07-18 04:54:35 +0000112 Type *VTy = V->getType();
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000113 assert(
114 (!VTy->isIntOrIntVectorTy() || VTy->getScalarSizeInBits() == BitWidth) &&
Craig Topperb45eabc2017-04-26 16:39:58 +0000115 Known.getBitWidth() == BitWidth &&
116 "Value *V, DemandedMask and Known must have same BitWidth");
Craig Topper83dc1c62017-04-20 16:14:58 +0000117
118 if (isa<Constant>(V)) {
Craig Topperb45eabc2017-04-26 16:39:58 +0000119 computeKnownBits(V, Known, Depth, CxtI);
Craig Topperf40110f2014-04-25 05:29:35 +0000120 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000121 }
122
Craig Topperb45eabc2017-04-26 16:39:58 +0000123 Known.Zero.clearAllBits();
124 Known.One.clearAllBits();
Craig Topper83dc1c62017-04-20 16:14:58 +0000125 if (DemandedMask == 0) // Not demanding any bits from V.
Chris Lattner7e044912010-01-04 07:17:19 +0000126 return UndefValue::get(VTy);
Craig Topper4c947752012-12-22 18:09:02 +0000127
Chris Lattner7e044912010-01-04 07:17:19 +0000128 if (Depth == 6) // Limit search depth.
Craig Topperf40110f2014-04-25 05:29:35 +0000129 return nullptr;
Craig Topper4c947752012-12-22 18:09:02 +0000130
Chris Lattner7e044912010-01-04 07:17:19 +0000131 Instruction *I = dyn_cast<Instruction>(V);
132 if (!I) {
Craig Topperb45eabc2017-04-26 16:39:58 +0000133 computeKnownBits(V, Known, Depth, CxtI);
Craig Topperf40110f2014-04-25 05:29:35 +0000134 return nullptr; // Only analyze instructions.
Chris Lattner7e044912010-01-04 07:17:19 +0000135 }
136
137 // If there are multiple uses of this value and we aren't at the root, then
138 // we can't do any simplifications of the operands, because DemandedMask
139 // only reflects the bits demanded by *one* of the users.
Craig Topper7603dce2017-04-25 16:48:19 +0000140 if (Depth != 0 && !I->hasOneUse())
Craig Topperb45eabc2017-04-26 16:39:58 +0000141 return SimplifyMultipleUseDemandedBits(I, DemandedMask, Known, Depth, CxtI);
Craig Topper4c947752012-12-22 18:09:02 +0000142
Craig Topperb45eabc2017-04-26 16:39:58 +0000143 KnownBits LHSKnown(BitWidth), RHSKnown(BitWidth);
Craig Topperb0076fe2017-04-12 18:05:21 +0000144
Chris Lattner7e044912010-01-04 07:17:19 +0000145 // If this is the root being simplified, allow it to have multiple uses,
146 // just set the DemandedMask to all bits so that we can try to simplify the
147 // operands. This allows visitTruncInst (for example) to simplify the
148 // operand of a trunc without duplicating all the logic below.
149 if (Depth == 0 && !V->hasOneUse())
Craig Toppere06b6bc2017-04-04 05:03:02 +0000150 DemandedMask.setAllBits();
Craig Topper4c947752012-12-22 18:09:02 +0000151
Chris Lattner7e044912010-01-04 07:17:19 +0000152 switch (I->getOpcode()) {
153 default:
Craig Topperb45eabc2017-04-26 16:39:58 +0000154 computeKnownBits(I, Known, Depth, CxtI);
Chris Lattner7e044912010-01-04 07:17:19 +0000155 break;
Craig Topper9a458cd2017-04-14 22:34:14 +0000156 case Instruction::And: {
Chris Lattner7e044912010-01-04 07:17:19 +0000157 // If either the LHS or the RHS are Zero, the result is zero.
Craig Topperb45eabc2017-04-26 16:39:58 +0000158 if (SimplifyDemandedBits(I, 1, DemandedMask, RHSKnown, Depth + 1) ||
159 SimplifyDemandedBits(I, 0, DemandedMask & ~RHSKnown.Zero, LHSKnown,
160 Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000161 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000162 assert(!(RHSKnown.Zero & RHSKnown.One) && "Bits known to be one AND zero?");
163 assert(!(LHSKnown.Zero & LHSKnown.One) && "Bits known to be one AND zero?");
Chris Lattner7e044912010-01-04 07:17:19 +0000164
Craig Topper9a458cd2017-04-14 22:34:14 +0000165 // Output known-0 are known to be clear if zero in either the LHS | RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000166 APInt IKnownZero = RHSKnown.Zero | LHSKnown.Zero;
Craig Topper9a458cd2017-04-14 22:34:14 +0000167 // Output known-1 bits are only known if set in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000168 APInt IKnownOne = RHSKnown.One & LHSKnown.One;
Craig Topper9a458cd2017-04-14 22:34:14 +0000169
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000170 // If the client is only demanding bits that we know, return the known
171 // constant.
Craig Topper17f37ba2017-04-20 20:47:35 +0000172 if (DemandedMask.isSubsetOf(IKnownZero|IKnownOne))
Craig Topper9a458cd2017-04-14 22:34:14 +0000173 return Constant::getIntegerValue(VTy, IKnownOne);
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000174
Chris Lattner7e044912010-01-04 07:17:19 +0000175 // If all of the demanded bits are known 1 on one side, return the other.
176 // These bits cannot contribute to the result of the 'and'.
Craig Topperb45eabc2017-04-26 16:39:58 +0000177 if (DemandedMask.isSubsetOf(LHSKnown.Zero | RHSKnown.One))
Chris Lattner7e044912010-01-04 07:17:19 +0000178 return I->getOperand(0);
Craig Topperb45eabc2017-04-26 16:39:58 +0000179 if (DemandedMask.isSubsetOf(RHSKnown.Zero | LHSKnown.One))
Chris Lattner7e044912010-01-04 07:17:19 +0000180 return I->getOperand(1);
Craig Topper4c947752012-12-22 18:09:02 +0000181
Chris Lattner7e044912010-01-04 07:17:19 +0000182 // If the RHS is a constant, see if we can simplify it.
Craig Topperb45eabc2017-04-26 16:39:58 +0000183 if (ShrinkDemandedConstant(I, 1, DemandedMask & ~LHSKnown.Zero))
Chris Lattner7e044912010-01-04 07:17:19 +0000184 return I;
Craig Topper4c947752012-12-22 18:09:02 +0000185
Craig Topperb45eabc2017-04-26 16:39:58 +0000186 Known.Zero = std::move(IKnownZero);
187 Known.One = std::move(IKnownOne);
Chris Lattner7e044912010-01-04 07:17:19 +0000188 break;
Craig Topper9a458cd2017-04-14 22:34:14 +0000189 }
190 case Instruction::Or: {
Chris Lattner7e044912010-01-04 07:17:19 +0000191 // If either the LHS or the RHS are One, the result is One.
Craig Topperb45eabc2017-04-26 16:39:58 +0000192 if (SimplifyDemandedBits(I, 1, DemandedMask, RHSKnown, Depth + 1) ||
193 SimplifyDemandedBits(I, 0, DemandedMask & ~RHSKnown.One, LHSKnown,
194 Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000195 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000196 assert(!(RHSKnown.Zero & RHSKnown.One) && "Bits known to be one AND zero?");
197 assert(!(LHSKnown.Zero & LHSKnown.One) && "Bits known to be one AND zero?");
Craig Topper4c947752012-12-22 18:09:02 +0000198
Craig Topper9a458cd2017-04-14 22:34:14 +0000199 // Output known-0 bits are only known if clear in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000200 APInt IKnownZero = RHSKnown.Zero & LHSKnown.Zero;
201 // Output known-1 are known. to be set if s.et in either the LHS | RHS.
202 APInt IKnownOne = RHSKnown.One | LHSKnown.One;
Craig Topper9a458cd2017-04-14 22:34:14 +0000203
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000204 // If the client is only demanding bits that we know, return the known
205 // constant.
Craig Topper17f37ba2017-04-20 20:47:35 +0000206 if (DemandedMask.isSubsetOf(IKnownZero|IKnownOne))
Craig Topper9a458cd2017-04-14 22:34:14 +0000207 return Constant::getIntegerValue(VTy, IKnownOne);
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000208
Chris Lattner7e044912010-01-04 07:17:19 +0000209 // If all of the demanded bits are known zero on one side, return the other.
210 // These bits cannot contribute to the result of the 'or'.
Craig Topperb45eabc2017-04-26 16:39:58 +0000211 if (DemandedMask.isSubsetOf(LHSKnown.One | RHSKnown.Zero))
Chris Lattner7e044912010-01-04 07:17:19 +0000212 return I->getOperand(0);
Craig Topperb45eabc2017-04-26 16:39:58 +0000213 if (DemandedMask.isSubsetOf(RHSKnown.One | LHSKnown.Zero))
Chris Lattner7e044912010-01-04 07:17:19 +0000214 return I->getOperand(1);
215
Chris Lattner7e044912010-01-04 07:17:19 +0000216 // If the RHS is a constant, see if we can simplify it.
217 if (ShrinkDemandedConstant(I, 1, DemandedMask))
218 return I;
Craig Topper4c947752012-12-22 18:09:02 +0000219
Craig Topperb45eabc2017-04-26 16:39:58 +0000220 Known.Zero = std::move(IKnownZero);
221 Known.One = std::move(IKnownOne);
Chris Lattner7e044912010-01-04 07:17:19 +0000222 break;
Craig Topper9a458cd2017-04-14 22:34:14 +0000223 }
Chris Lattner7e044912010-01-04 07:17:19 +0000224 case Instruction::Xor: {
Craig Topperb45eabc2017-04-26 16:39:58 +0000225 if (SimplifyDemandedBits(I, 1, DemandedMask, RHSKnown, Depth + 1) ||
226 SimplifyDemandedBits(I, 0, DemandedMask, LHSKnown, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000227 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000228 assert(!(RHSKnown.Zero & RHSKnown.One) && "Bits known to be one AND zero?");
229 assert(!(LHSKnown.Zero & LHSKnown.One) && "Bits known to be one AND zero?");
Craig Topper4c947752012-12-22 18:09:02 +0000230
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000231 // Output known-0 bits are known if clear or set in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000232 APInt IKnownZero = (RHSKnown.Zero & LHSKnown.Zero) |
233 (RHSKnown.One & LHSKnown.One);
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000234 // Output known-1 are known to be set if set in only one of the LHS, RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000235 APInt IKnownOne = (RHSKnown.Zero & LHSKnown.One) |
236 (RHSKnown.One & LHSKnown.Zero);
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000237
238 // If the client is only demanding bits that we know, return the known
239 // constant.
Craig Topper17f37ba2017-04-20 20:47:35 +0000240 if (DemandedMask.isSubsetOf(IKnownZero|IKnownOne))
Hal Finkel15aeaaf2014-09-07 19:21:07 +0000241 return Constant::getIntegerValue(VTy, IKnownOne);
242
Chris Lattner7e044912010-01-04 07:17:19 +0000243 // If all of the demanded bits are known zero on one side, return the other.
244 // These bits cannot contribute to the result of the 'xor'.
Craig Topperb45eabc2017-04-26 16:39:58 +0000245 if (DemandedMask.isSubsetOf(RHSKnown.Zero))
Chris Lattner7e044912010-01-04 07:17:19 +0000246 return I->getOperand(0);
Craig Topperb45eabc2017-04-26 16:39:58 +0000247 if (DemandedMask.isSubsetOf(LHSKnown.Zero))
Chris Lattner7e044912010-01-04 07:17:19 +0000248 return I->getOperand(1);
Craig Topper4c947752012-12-22 18:09:02 +0000249
Chris Lattner7e044912010-01-04 07:17:19 +0000250 // If all of the demanded bits are known to be zero on one side or the
251 // other, turn this into an *inclusive* or.
Sylvestre Ledru91ce36c2012-09-27 10:14:43 +0000252 // e.g. (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1&C2 == 0
Craig Topperb45eabc2017-04-26 16:39:58 +0000253 if (DemandedMask.isSubsetOf(RHSKnown.Zero | LHSKnown.Zero)) {
Craig Topper4c947752012-12-22 18:09:02 +0000254 Instruction *Or =
Chris Lattner7e044912010-01-04 07:17:19 +0000255 BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1),
256 I->getName());
Eli Friedman6efb64e2011-05-19 01:20:42 +0000257 return InsertNewInstWith(Or, *I);
Chris Lattner7e044912010-01-04 07:17:19 +0000258 }
Craig Topper4c947752012-12-22 18:09:02 +0000259
Chris Lattner7e044912010-01-04 07:17:19 +0000260 // If all of the demanded bits on one side are known, and all of the set
261 // bits on that side are also known to be set on the other side, turn this
262 // into an AND, as we know the bits will be cleared.
Sylvestre Ledru91ce36c2012-09-27 10:14:43 +0000263 // e.g. (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2
Craig Topperb45eabc2017-04-26 16:39:58 +0000264 if (DemandedMask.isSubsetOf(RHSKnown.Zero|RHSKnown.One) &&
265 RHSKnown.One.isSubsetOf(LHSKnown.One)) {
Craig Topper17f37ba2017-04-20 20:47:35 +0000266 Constant *AndC = Constant::getIntegerValue(VTy,
Craig Topperb45eabc2017-04-26 16:39:58 +0000267 ~RHSKnown.One & DemandedMask);
Craig Topper17f37ba2017-04-20 20:47:35 +0000268 Instruction *And = BinaryOperator::CreateAnd(I->getOperand(0), AndC);
269 return InsertNewInstWith(And, *I);
Chris Lattner7e044912010-01-04 07:17:19 +0000270 }
Craig Topper4c947752012-12-22 18:09:02 +0000271
Sanjay Patel8ce1d4c2017-04-21 20:29:17 +0000272 // If the RHS is a constant, see if we can simplify it.
273 // FIXME: for XOR, we prefer to force bits to 1 if they will make a -1.
274 if (ShrinkDemandedConstant(I, 1, DemandedMask))
275 return I;
Craig Topper4c947752012-12-22 18:09:02 +0000276
Chris Lattner7e044912010-01-04 07:17:19 +0000277 // If our LHS is an 'and' and if it has one use, and if any of the bits we
278 // are flipping are known to be set, then the xor is just resetting those
279 // bits to zero. We can just knock out bits from the 'and' and the 'xor',
280 // simplifying both of them.
281 if (Instruction *LHSInst = dyn_cast<Instruction>(I->getOperand(0)))
282 if (LHSInst->getOpcode() == Instruction::And && LHSInst->hasOneUse() &&
283 isa<ConstantInt>(I->getOperand(1)) &&
284 isa<ConstantInt>(LHSInst->getOperand(1)) &&
Craig Topperb45eabc2017-04-26 16:39:58 +0000285 (LHSKnown.One & RHSKnown.One & DemandedMask) != 0) {
Chris Lattner7e044912010-01-04 07:17:19 +0000286 ConstantInt *AndRHS = cast<ConstantInt>(LHSInst->getOperand(1));
287 ConstantInt *XorRHS = cast<ConstantInt>(I->getOperand(1));
Craig Topperb45eabc2017-04-26 16:39:58 +0000288 APInt NewMask = ~(LHSKnown.One & RHSKnown.One & DemandedMask);
Craig Topper4c947752012-12-22 18:09:02 +0000289
Chris Lattner7e044912010-01-04 07:17:19 +0000290 Constant *AndC =
291 ConstantInt::get(I->getType(), NewMask & AndRHS->getValue());
Benjamin Kramer547b6c52011-09-27 20:39:19 +0000292 Instruction *NewAnd = BinaryOperator::CreateAnd(I->getOperand(0), AndC);
Eli Friedman6efb64e2011-05-19 01:20:42 +0000293 InsertNewInstWith(NewAnd, *I);
Craig Topper4c947752012-12-22 18:09:02 +0000294
Chris Lattner7e044912010-01-04 07:17:19 +0000295 Constant *XorC =
296 ConstantInt::get(I->getType(), NewMask & XorRHS->getValue());
Benjamin Kramer547b6c52011-09-27 20:39:19 +0000297 Instruction *NewXor = BinaryOperator::CreateXor(NewAnd, XorC);
Eli Friedman6efb64e2011-05-19 01:20:42 +0000298 return InsertNewInstWith(NewXor, *I);
Chris Lattner7e044912010-01-04 07:17:19 +0000299 }
Duncan Sandsc8a3e562010-01-29 06:18:46 +0000300
301 // Output known-0 bits are known if clear or set in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000302 Known.Zero = std::move(IKnownZero);
Duncan Sandsc8a3e562010-01-29 06:18:46 +0000303 // Output known-1 are known to be set if set in only one of the LHS, RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000304 Known.One = std::move(IKnownOne);
Chris Lattner7e044912010-01-04 07:17:19 +0000305 break;
306 }
307 case Instruction::Select:
James Molloy2b21a7c2015-05-20 18:41:25 +0000308 // If this is a select as part of a min/max pattern, don't simplify any
309 // further in case we break the structure.
310 Value *LHS, *RHS;
James Molloy134bec22015-08-11 09:12:57 +0000311 if (matchSelectPattern(I, LHS, RHS).Flavor != SPF_UNKNOWN)
James Molloy2b21a7c2015-05-20 18:41:25 +0000312 return nullptr;
Simon Pilgrim61116dd2015-09-17 20:32:45 +0000313
Craig Topperb45eabc2017-04-26 16:39:58 +0000314 if (SimplifyDemandedBits(I, 2, DemandedMask, RHSKnown, Depth + 1) ||
315 SimplifyDemandedBits(I, 1, DemandedMask, LHSKnown, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000316 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000317 assert(!(RHSKnown.Zero & RHSKnown.One) && "Bits known to be one AND zero?");
318 assert(!(LHSKnown.Zero & LHSKnown.One) && "Bits known to be one AND zero?");
Craig Topper4c947752012-12-22 18:09:02 +0000319
Chris Lattner7e044912010-01-04 07:17:19 +0000320 // If the operands are constants, see if we can simplify them.
321 if (ShrinkDemandedConstant(I, 1, DemandedMask) ||
322 ShrinkDemandedConstant(I, 2, DemandedMask))
323 return I;
Craig Topper4c947752012-12-22 18:09:02 +0000324
Chris Lattner7e044912010-01-04 07:17:19 +0000325 // Only known if known in both the LHS and RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000326 Known.One = RHSKnown.One & LHSKnown.One;
327 Known.Zero = RHSKnown.Zero & LHSKnown.Zero;
Chris Lattner7e044912010-01-04 07:17:19 +0000328 break;
329 case Instruction::Trunc: {
330 unsigned truncBf = I->getOperand(0)->getType()->getScalarSizeInBits();
Jay Foad583abbc2010-12-07 08:25:19 +0000331 DemandedMask = DemandedMask.zext(truncBf);
Craig Topperb45eabc2017-04-26 16:39:58 +0000332 Known.Zero = Known.Zero.zext(truncBf);
333 Known.One = Known.One.zext(truncBf);
334 if (SimplifyDemandedBits(I, 0, DemandedMask, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000335 return I;
Jay Foad583abbc2010-12-07 08:25:19 +0000336 DemandedMask = DemandedMask.trunc(BitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000337 Known.Zero = Known.Zero.trunc(BitWidth);
338 Known.One = Known.One.trunc(BitWidth);
339 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
Chris Lattner7e044912010-01-04 07:17:19 +0000340 break;
341 }
342 case Instruction::BitCast:
Duncan Sands9dff9be2010-02-15 16:12:20 +0000343 if (!I->getOperand(0)->getType()->isIntOrIntVectorTy())
Craig Topperf40110f2014-04-25 05:29:35 +0000344 return nullptr; // vector->int or fp->int?
Chris Lattner7e044912010-01-04 07:17:19 +0000345
Chris Lattner229907c2011-07-18 04:54:35 +0000346 if (VectorType *DstVTy = dyn_cast<VectorType>(I->getType())) {
347 if (VectorType *SrcVTy =
Chris Lattner7e044912010-01-04 07:17:19 +0000348 dyn_cast<VectorType>(I->getOperand(0)->getType())) {
349 if (DstVTy->getNumElements() != SrcVTy->getNumElements())
350 // Don't touch a bitcast between vectors of different element counts.
Craig Topperf40110f2014-04-25 05:29:35 +0000351 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000352 } else
353 // Don't touch a scalar-to-vector bitcast.
Craig Topperf40110f2014-04-25 05:29:35 +0000354 return nullptr;
Duncan Sands19d0b472010-02-16 11:11:14 +0000355 } else if (I->getOperand(0)->getType()->isVectorTy())
Chris Lattner7e044912010-01-04 07:17:19 +0000356 // Don't touch a vector-to-scalar bitcast.
Craig Topperf40110f2014-04-25 05:29:35 +0000357 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000358
Craig Topperb45eabc2017-04-26 16:39:58 +0000359 if (SimplifyDemandedBits(I, 0, DemandedMask, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000360 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000361 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
Chris Lattner7e044912010-01-04 07:17:19 +0000362 break;
363 case Instruction::ZExt: {
364 // Compute the bits in the result that are not present in the input.
365 unsigned SrcBitWidth =I->getOperand(0)->getType()->getScalarSizeInBits();
Craig Topper4c947752012-12-22 18:09:02 +0000366
Jay Foad583abbc2010-12-07 08:25:19 +0000367 DemandedMask = DemandedMask.trunc(SrcBitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000368 Known.Zero = Known.Zero.trunc(SrcBitWidth);
369 Known.One = Known.One.trunc(SrcBitWidth);
370 if (SimplifyDemandedBits(I, 0, DemandedMask, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000371 return I;
Jay Foad583abbc2010-12-07 08:25:19 +0000372 DemandedMask = DemandedMask.zext(BitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000373 Known.Zero = Known.Zero.zext(BitWidth);
374 Known.One = Known.One.zext(BitWidth);
375 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
Chris Lattner7e044912010-01-04 07:17:19 +0000376 // The top bits are known to be zero.
Craig Topperb45eabc2017-04-26 16:39:58 +0000377 Known.Zero.setBitsFrom(SrcBitWidth);
Chris Lattner7e044912010-01-04 07:17:19 +0000378 break;
379 }
380 case Instruction::SExt: {
381 // Compute the bits in the result that are not present in the input.
382 unsigned SrcBitWidth =I->getOperand(0)->getType()->getScalarSizeInBits();
Craig Topper4c947752012-12-22 18:09:02 +0000383
384 APInt InputDemandedBits = DemandedMask &
Chris Lattner7e044912010-01-04 07:17:19 +0000385 APInt::getLowBitsSet(BitWidth, SrcBitWidth);
386
Craig Topper3a86a042017-03-19 05:49:16 +0000387 APInt NewBits(APInt::getBitsSetFrom(BitWidth, SrcBitWidth));
Chris Lattner7e044912010-01-04 07:17:19 +0000388 // If any of the sign extended bits are demanded, we know that the sign
389 // bit is demanded.
390 if ((NewBits & DemandedMask) != 0)
Jay Foad25a5e4c2010-12-01 08:53:58 +0000391 InputDemandedBits.setBit(SrcBitWidth-1);
Craig Topper4c947752012-12-22 18:09:02 +0000392
Jay Foad583abbc2010-12-07 08:25:19 +0000393 InputDemandedBits = InputDemandedBits.trunc(SrcBitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000394 Known.Zero = Known.Zero.trunc(SrcBitWidth);
395 Known.One = Known.One.trunc(SrcBitWidth);
396 if (SimplifyDemandedBits(I, 0, InputDemandedBits, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000397 return I;
Jay Foad583abbc2010-12-07 08:25:19 +0000398 InputDemandedBits = InputDemandedBits.zext(BitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000399 Known.Zero = Known.Zero.zext(BitWidth);
400 Known.One = Known.One.zext(BitWidth);
401 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
Craig Topper4c947752012-12-22 18:09:02 +0000402
Chris Lattner7e044912010-01-04 07:17:19 +0000403 // If the sign bit of the input is known set or clear, then we know the
404 // top bits of the result.
405
406 // If the input sign bit is known zero, or if the NewBits are not demanded
407 // convert this into a zero extension.
Craig Topperb45eabc2017-04-26 16:39:58 +0000408 if (Known.Zero[SrcBitWidth-1] || (NewBits & ~DemandedMask) == NewBits) {
Chris Lattner7e044912010-01-04 07:17:19 +0000409 // Convert to ZExt cast
410 CastInst *NewCast = new ZExtInst(I->getOperand(0), VTy, I->getName());
Eli Friedman6efb64e2011-05-19 01:20:42 +0000411 return InsertNewInstWith(NewCast, *I);
Craig Topperb45eabc2017-04-26 16:39:58 +0000412 } else if (Known.One[SrcBitWidth-1]) { // Input sign bit known set
413 Known.One |= NewBits;
Chris Lattner7e044912010-01-04 07:17:19 +0000414 }
415 break;
416 }
Matthias Braune48484c2015-04-30 22:05:30 +0000417 case Instruction::Add:
418 case Instruction::Sub: {
419 /// If the high-bits of an ADD/SUB are not demanded, then we do not care
420 /// about the high bits of the operands.
Chris Lattner7e044912010-01-04 07:17:19 +0000421 unsigned NLZ = DemandedMask.countLeadingZeros();
Matthias Braune48484c2015-04-30 22:05:30 +0000422 if (NLZ > 0) {
423 // Right fill the mask of bits for this ADD/SUB to demand the most
Chris Lattner7e044912010-01-04 07:17:19 +0000424 // significant bit and all those below it.
Chris Lattner7e044912010-01-04 07:17:19 +0000425 APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ));
Craig Topper07f29152017-03-22 04:03:53 +0000426 if (ShrinkDemandedConstant(I, 0, DemandedFromOps) ||
Craig Topperb45eabc2017-04-26 16:39:58 +0000427 SimplifyDemandedBits(I, 0, DemandedFromOps, LHSKnown, Depth + 1) ||
Matthias Braune48484c2015-04-30 22:05:30 +0000428 ShrinkDemandedConstant(I, 1, DemandedFromOps) ||
Craig Topperb45eabc2017-04-26 16:39:58 +0000429 SimplifyDemandedBits(I, 1, DemandedFromOps, RHSKnown, Depth + 1)) {
Matthias Braune48484c2015-04-30 22:05:30 +0000430 // Disable the nsw and nuw flags here: We can no longer guarantee that
431 // we won't wrap after simplification. Removing the nsw/nuw flags is
432 // legal here because the top bit is not demanded.
433 BinaryOperator &BinOP = *cast<BinaryOperator>(I);
434 BinOP.setHasNoSignedWrap(false);
435 BinOP.setHasNoUnsignedWrap(false);
Chris Lattner7e044912010-01-04 07:17:19 +0000436 return I;
David Majnemer7d0e99c2015-04-22 22:42:05 +0000437 }
Craig Topper845033a2017-04-12 16:49:59 +0000438
439 // If we are known to be adding/subtracting zeros to every bit below
440 // the highest demanded bit, we just return the other side.
Craig Topperb45eabc2017-04-26 16:39:58 +0000441 if (DemandedFromOps.isSubsetOf(RHSKnown.Zero))
Craig Topper845033a2017-04-12 16:49:59 +0000442 return I->getOperand(0);
443 // We can't do this with the LHS for subtraction.
444 if (I->getOpcode() == Instruction::Add &&
Craig Topperb45eabc2017-04-26 16:39:58 +0000445 DemandedFromOps.isSubsetOf(LHSKnown.Zero))
Craig Topper845033a2017-04-12 16:49:59 +0000446 return I->getOperand(1);
Chris Lattner7e044912010-01-04 07:17:19 +0000447 }
Benjamin Kramer010337c2011-12-24 17:31:38 +0000448
Craig Topper8fbb74b2017-03-24 22:12:10 +0000449 // Otherwise just hand the add/sub off to computeKnownBits to fill in
450 // the known zeros and ones.
Craig Topperb45eabc2017-04-26 16:39:58 +0000451 computeKnownBits(V, Known, Depth, CxtI);
Chris Lattner7e044912010-01-04 07:17:19 +0000452 break;
Matthias Braune48484c2015-04-30 22:05:30 +0000453 }
Sanjay Patel3e1ae722017-04-20 21:33:02 +0000454 case Instruction::Shl: {
455 const APInt *SA;
456 if (match(I->getOperand(1), m_APInt(SA))) {
Sanjay Patelc9485ca2017-04-20 22:33:54 +0000457 const APInt *ShrAmt;
458 if (match(I->getOperand(0), m_Shr(m_Value(), m_APInt(ShrAmt)))) {
459 Instruction *Shr = cast<Instruction>(I->getOperand(0));
Sanjay Patelcc663b82017-04-20 22:37:01 +0000460 if (Value *R = simplifyShrShlDemandedBits(
Craig Topperb45eabc2017-04-26 16:39:58 +0000461 Shr, *ShrAmt, I, *SA, DemandedMask, Known))
Sanjay Patelc9485ca2017-04-20 22:33:54 +0000462 return R;
Shuxin Yang63e999e2012-12-04 00:04:54 +0000463 }
464
Chris Lattner768003c2011-02-10 05:09:34 +0000465 uint64_t ShiftAmt = SA->getLimitedValue(BitWidth-1);
Chris Lattner7e044912010-01-04 07:17:19 +0000466 APInt DemandedMaskIn(DemandedMask.lshr(ShiftAmt));
Craig Topper4c947752012-12-22 18:09:02 +0000467
Chris Lattner768003c2011-02-10 05:09:34 +0000468 // If the shift is NUW/NSW, then it does demand the high bits.
469 ShlOperator *IOp = cast<ShlOperator>(I);
470 if (IOp->hasNoSignedWrap())
Craig Topper3a86a042017-03-19 05:49:16 +0000471 DemandedMaskIn.setHighBits(ShiftAmt+1);
Chris Lattner768003c2011-02-10 05:09:34 +0000472 else if (IOp->hasNoUnsignedWrap())
Craig Topper3a86a042017-03-19 05:49:16 +0000473 DemandedMaskIn.setHighBits(ShiftAmt);
Craig Topper4c947752012-12-22 18:09:02 +0000474
Craig Topperb45eabc2017-04-26 16:39:58 +0000475 if (SimplifyDemandedBits(I, 0, DemandedMaskIn, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000476 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000477 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
478 Known.Zero <<= ShiftAmt;
479 Known.One <<= ShiftAmt;
Chris Lattner7e044912010-01-04 07:17:19 +0000480 // low bits known zero.
481 if (ShiftAmt)
Craig Topperb45eabc2017-04-26 16:39:58 +0000482 Known.Zero.setLowBits(ShiftAmt);
Chris Lattner7e044912010-01-04 07:17:19 +0000483 }
484 break;
Sanjay Patel3e1ae722017-04-20 21:33:02 +0000485 }
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000486 case Instruction::LShr: {
487 const APInt *SA;
488 if (match(I->getOperand(1), m_APInt(SA))) {
Chris Lattner768003c2011-02-10 05:09:34 +0000489 uint64_t ShiftAmt = SA->getLimitedValue(BitWidth-1);
Craig Topper4c947752012-12-22 18:09:02 +0000490
Chris Lattner7e044912010-01-04 07:17:19 +0000491 // Unsigned shift right.
492 APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt));
Craig Topper4c947752012-12-22 18:09:02 +0000493
Chris Lattner768003c2011-02-10 05:09:34 +0000494 // If the shift is exact, then it does demand the low bits (and knows that
495 // they are zero).
496 if (cast<LShrOperator>(I)->isExact())
Craig Topper3a86a042017-03-19 05:49:16 +0000497 DemandedMaskIn.setLowBits(ShiftAmt);
Craig Topper4c947752012-12-22 18:09:02 +0000498
Craig Topperb45eabc2017-04-26 16:39:58 +0000499 if (SimplifyDemandedBits(I, 0, DemandedMaskIn, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000500 return I;
Craig Topperb45eabc2017-04-26 16:39:58 +0000501 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
502 Known.Zero.lshrInPlace(ShiftAmt);
503 Known.One.lshrInPlace(ShiftAmt);
Craig Topper3a86a042017-03-19 05:49:16 +0000504 if (ShiftAmt)
Craig Topperb45eabc2017-04-26 16:39:58 +0000505 Known.Zero.setHighBits(ShiftAmt); // high bits known zero.
Chris Lattner7e044912010-01-04 07:17:19 +0000506 }
507 break;
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000508 }
509 case Instruction::AShr: {
Chris Lattner7e044912010-01-04 07:17:19 +0000510 // If this is an arithmetic shift right and only the low-bit is set, we can
511 // always convert this into a logical shr, even if the shift amount is
512 // variable. The low bit of the shift cannot be an input sign bit unless
513 // the shift amount is >= the size of the datatype, which is undefined.
514 if (DemandedMask == 1) {
515 // Perform the logical shift right.
516 Instruction *NewVal = BinaryOperator::CreateLShr(
517 I->getOperand(0), I->getOperand(1), I->getName());
Eli Friedman6efb64e2011-05-19 01:20:42 +0000518 return InsertNewInstWith(NewVal, *I);
Craig Topper4c947752012-12-22 18:09:02 +0000519 }
Chris Lattner7e044912010-01-04 07:17:19 +0000520
521 // If the sign bit is the only bit demanded by this ashr, then there is no
522 // need to do it, the shift doesn't change the high bit.
Craig Topperbcfd2d12017-04-20 16:56:25 +0000523 if (DemandedMask.isSignMask())
Chris Lattner7e044912010-01-04 07:17:19 +0000524 return I->getOperand(0);
Craig Topper4c947752012-12-22 18:09:02 +0000525
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000526 const APInt *SA;
527 if (match(I->getOperand(1), m_APInt(SA))) {
Chris Lattner768003c2011-02-10 05:09:34 +0000528 uint32_t ShiftAmt = SA->getLimitedValue(BitWidth-1);
Craig Topper4c947752012-12-22 18:09:02 +0000529
Chris Lattner7e044912010-01-04 07:17:19 +0000530 // Signed shift right.
531 APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt));
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000532 // If any of the high bits are demanded, we should set the sign bit as
Chris Lattner7e044912010-01-04 07:17:19 +0000533 // demanded.
534 if (DemandedMask.countLeadingZeros() <= ShiftAmt)
Craig Topperc9a4fc02017-04-14 05:09:04 +0000535 DemandedMaskIn.setSignBit();
Craig Topper4c947752012-12-22 18:09:02 +0000536
Chris Lattner768003c2011-02-10 05:09:34 +0000537 // If the shift is exact, then it does demand the low bits (and knows that
538 // they are zero).
539 if (cast<AShrOperator>(I)->isExact())
Craig Topper3a86a042017-03-19 05:49:16 +0000540 DemandedMaskIn.setLowBits(ShiftAmt);
Craig Topper4c947752012-12-22 18:09:02 +0000541
Craig Topperb45eabc2017-04-26 16:39:58 +0000542 if (SimplifyDemandedBits(I, 0, DemandedMaskIn, Known, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000543 return I;
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000544
Craig Topperb45eabc2017-04-26 16:39:58 +0000545 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
Chris Lattner7e044912010-01-04 07:17:19 +0000546 // Compute the new bits that are at the top now.
547 APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt));
Craig Topperb45eabc2017-04-26 16:39:58 +0000548 Known.Zero.lshrInPlace(ShiftAmt);
549 Known.One.lshrInPlace(ShiftAmt);
Craig Topper4c947752012-12-22 18:09:02 +0000550
Chris Lattner7e044912010-01-04 07:17:19 +0000551 // Handle the sign bits.
Craig Topperbcfd2d12017-04-20 16:56:25 +0000552 APInt SignMask(APInt::getSignMask(BitWidth));
Chris Lattner7e044912010-01-04 07:17:19 +0000553 // Adjust to where it is now in the mask.
Craig Topperbcfd2d12017-04-20 16:56:25 +0000554 SignMask.lshrInPlace(ShiftAmt);
Craig Topper4c947752012-12-22 18:09:02 +0000555
Chris Lattner7e044912010-01-04 07:17:19 +0000556 // If the input sign bit is known to be zero, or if none of the top bits
557 // are demanded, turn this into an unsigned shift right.
Craig Topperb45eabc2017-04-26 16:39:58 +0000558 if (BitWidth <= ShiftAmt || Known.Zero[BitWidth-ShiftAmt-1] ||
Craig Topperff238892017-04-20 21:24:37 +0000559 !DemandedMask.intersects(HighBits)) {
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000560 BinaryOperator *LShr = BinaryOperator::CreateLShr(I->getOperand(0),
561 I->getOperand(1));
562 LShr->setIsExact(cast<BinaryOperator>(I)->isExact());
563 return InsertNewInstWith(LShr, *I);
Craig Topperb45eabc2017-04-26 16:39:58 +0000564 } else if (Known.One.intersects(SignMask)) { // New bits are known one.
565 Known.One |= HighBits;
Chris Lattner7e044912010-01-04 07:17:19 +0000566 }
567 }
568 break;
Sanjay Patelfb5b3e72017-04-20 20:59:02 +0000569 }
Chris Lattner7e044912010-01-04 07:17:19 +0000570 case Instruction::SRem:
571 if (ConstantInt *Rem = dyn_cast<ConstantInt>(I->getOperand(1))) {
Eli Friedmana81a82d2011-03-09 01:28:35 +0000572 // X % -1 demands all the bits because we don't want to introduce
573 // INT_MIN % -1 (== undef) by accident.
574 if (Rem->isAllOnesValue())
575 break;
Chris Lattner7e044912010-01-04 07:17:19 +0000576 APInt RA = Rem->getValue().abs();
577 if (RA.isPowerOf2()) {
578 if (DemandedMask.ult(RA)) // srem won't affect demanded bits
579 return I->getOperand(0);
580
581 APInt LowBits = RA - 1;
Craig Topperbcfd2d12017-04-20 16:56:25 +0000582 APInt Mask2 = LowBits | APInt::getSignMask(BitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000583 if (SimplifyDemandedBits(I, 0, Mask2, LHSKnown, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000584 return I;
585
Duncan Sands3a48b872010-01-28 17:22:42 +0000586 // The low bits of LHS are unchanged by the srem.
Craig Topperb45eabc2017-04-26 16:39:58 +0000587 Known.Zero = LHSKnown.Zero & LowBits;
588 Known.One = LHSKnown.One & LowBits;
Chris Lattner7e044912010-01-04 07:17:19 +0000589
Duncan Sands3a48b872010-01-28 17:22:42 +0000590 // If LHS is non-negative or has all low bits zero, then the upper bits
591 // are all zero.
Craig Topperb45eabc2017-04-26 16:39:58 +0000592 if (LHSKnown.Zero.isSignBitSet() || LowBits.isSubsetOf(LHSKnown.Zero))
593 Known.Zero |= ~LowBits;
Duncan Sands3a48b872010-01-28 17:22:42 +0000594
595 // If LHS is negative and not all low bits are zero, then the upper bits
596 // are all one.
Craig Topperb45eabc2017-04-26 16:39:58 +0000597 if (LHSKnown.One.isSignBitSet() && LowBits.intersects(LHSKnown.One))
598 Known.One |= ~LowBits;
Chris Lattner7e044912010-01-04 07:17:19 +0000599
Craig Topperb45eabc2017-04-26 16:39:58 +0000600 assert(!(Known.Zero & Known.One) && "Bits known to be one AND zero?");
Craig Topperda886c62017-04-16 21:46:12 +0000601 break;
Chris Lattner7e044912010-01-04 07:17:19 +0000602 }
603 }
Nick Lewyckye4679792011-03-07 01:50:10 +0000604
605 // The sign bit is the LHS's sign bit, except when the result of the
606 // remainder is zero.
Craig Topperd23004c2017-04-17 16:38:20 +0000607 if (DemandedMask.isSignBitSet()) {
Craig Topperb45eabc2017-04-26 16:39:58 +0000608 computeKnownBits(I->getOperand(0), LHSKnown, Depth + 1, CxtI);
Nick Lewyckye4679792011-03-07 01:50:10 +0000609 // If it's known zero, our sign bit is also zero.
Craig Topperb45eabc2017-04-26 16:39:58 +0000610 if (LHSKnown.Zero.isSignBitSet())
611 Known.Zero.setSignBit();
Nick Lewyckye4679792011-03-07 01:50:10 +0000612 }
Chris Lattner7e044912010-01-04 07:17:19 +0000613 break;
614 case Instruction::URem: {
Craig Topperb45eabc2017-04-26 16:39:58 +0000615 KnownBits Known2(BitWidth);
Chris Lattner7e044912010-01-04 07:17:19 +0000616 APInt AllOnes = APInt::getAllOnesValue(BitWidth);
Craig Topperb45eabc2017-04-26 16:39:58 +0000617 if (SimplifyDemandedBits(I, 0, AllOnes, Known2, Depth + 1) ||
618 SimplifyDemandedBits(I, 1, AllOnes, Known2, Depth + 1))
Chris Lattner7e044912010-01-04 07:17:19 +0000619 return I;
620
Craig Topperb45eabc2017-04-26 16:39:58 +0000621 unsigned Leaders = Known2.Zero.countLeadingOnes();
622 Known.Zero = APInt::getHighBitsSet(BitWidth, Leaders) & DemandedMask;
Chris Lattner7e044912010-01-04 07:17:19 +0000623 break;
624 }
625 case Instruction::Call:
626 if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) {
627 switch (II->getIntrinsicID()) {
628 default: break;
629 case Intrinsic::bswap: {
630 // If the only bits demanded come from one byte of the bswap result,
631 // just shift the input byte into position to eliminate the bswap.
632 unsigned NLZ = DemandedMask.countLeadingZeros();
633 unsigned NTZ = DemandedMask.countTrailingZeros();
Craig Topper4c947752012-12-22 18:09:02 +0000634
Chris Lattner7e044912010-01-04 07:17:19 +0000635 // Round NTZ down to the next byte. If we have 11 trailing zeros, then
636 // we need all the bits down to bit 8. Likewise, round NLZ. If we
637 // have 14 leading zeros, round to 8.
638 NLZ &= ~7;
639 NTZ &= ~7;
640 // If we need exactly one byte, we can do this transformation.
641 if (BitWidth-NLZ-NTZ == 8) {
642 unsigned ResultBit = NTZ;
643 unsigned InputBit = BitWidth-NTZ-8;
Craig Topper4c947752012-12-22 18:09:02 +0000644
Chris Lattner7e044912010-01-04 07:17:19 +0000645 // Replace this with either a left or right shift to get the byte into
646 // the right place.
647 Instruction *NewVal;
648 if (InputBit > ResultBit)
Gabor Greif79430172010-06-24 12:35:13 +0000649 NewVal = BinaryOperator::CreateLShr(II->getArgOperand(0),
Chris Lattner7e044912010-01-04 07:17:19 +0000650 ConstantInt::get(I->getType(), InputBit-ResultBit));
651 else
Gabor Greif79430172010-06-24 12:35:13 +0000652 NewVal = BinaryOperator::CreateShl(II->getArgOperand(0),
Chris Lattner7e044912010-01-04 07:17:19 +0000653 ConstantInt::get(I->getType(), ResultBit-InputBit));
654 NewVal->takeName(I);
Eli Friedman6efb64e2011-05-19 01:20:42 +0000655 return InsertNewInstWith(NewVal, *I);
Chris Lattner7e044912010-01-04 07:17:19 +0000656 }
Craig Topper4c947752012-12-22 18:09:02 +0000657
Chris Lattner7e044912010-01-04 07:17:19 +0000658 // TODO: Could compute known zero/one bits based on the input.
659 break;
660 }
Simon Pilgrimfda22d62016-06-04 13:42:46 +0000661 case Intrinsic::x86_mmx_pmovmskb:
Simon Pilgrimbd4a3be2016-04-28 12:22:53 +0000662 case Intrinsic::x86_sse_movmsk_ps:
663 case Intrinsic::x86_sse2_movmsk_pd:
664 case Intrinsic::x86_sse2_pmovmskb_128:
665 case Intrinsic::x86_avx_movmsk_ps_256:
666 case Intrinsic::x86_avx_movmsk_pd_256:
667 case Intrinsic::x86_avx2_pmovmskb: {
668 // MOVMSK copies the vector elements' sign bits to the low bits
669 // and zeros the high bits.
Simon Pilgrimfda22d62016-06-04 13:42:46 +0000670 unsigned ArgWidth;
671 if (II->getIntrinsicID() == Intrinsic::x86_mmx_pmovmskb) {
672 ArgWidth = 8; // Arg is x86_mmx, but treated as <8 x i8>.
673 } else {
674 auto Arg = II->getArgOperand(0);
675 auto ArgType = cast<VectorType>(Arg->getType());
676 ArgWidth = ArgType->getNumElements();
677 }
Simon Pilgrimbd4a3be2016-04-28 12:22:53 +0000678
679 // If we don't need any of low bits then return zero,
680 // we know that DemandedMask is non-zero already.
681 APInt DemandedElts = DemandedMask.zextOrTrunc(ArgWidth);
682 if (DemandedElts == 0)
683 return ConstantInt::getNullValue(VTy);
684
Ahmed Bougacha17482a52016-04-28 14:36:07 +0000685 // We know that the upper bits are set to zero.
Craig Topperb45eabc2017-04-26 16:39:58 +0000686 Known.Zero.setBitsFrom(ArgWidth);
Simon Pilgrimbd4a3be2016-04-28 12:22:53 +0000687 return nullptr;
688 }
Chad Rosierb3628842011-05-26 23:13:19 +0000689 case Intrinsic::x86_sse42_crc32_64_64:
Craig Topperb45eabc2017-04-26 16:39:58 +0000690 Known.Zero.setBitsFrom(32);
Craig Topperf40110f2014-04-25 05:29:35 +0000691 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000692 }
693 }
Craig Topperb45eabc2017-04-26 16:39:58 +0000694 computeKnownBits(V, Known, Depth, CxtI);
Chris Lattner7e044912010-01-04 07:17:19 +0000695 break;
696 }
Craig Topper4c947752012-12-22 18:09:02 +0000697
Chris Lattner7e044912010-01-04 07:17:19 +0000698 // If the client is only demanding bits that we know, return the known
699 // constant.
Craig Topperb45eabc2017-04-26 16:39:58 +0000700 if (DemandedMask.isSubsetOf(Known.Zero|Known.One))
701 return Constant::getIntegerValue(VTy, Known.One);
Craig Topperf40110f2014-04-25 05:29:35 +0000702 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000703}
704
Craig Topperb45eabc2017-04-26 16:39:58 +0000705/// Helper routine of SimplifyDemandedUseBits. It computes Known
Craig Topperb0076fe2017-04-12 18:05:21 +0000706/// bits. It also tries to handle simplifications that can be done based on
707/// DemandedMask, but without modifying the Instruction.
708Value *InstCombiner::SimplifyMultipleUseDemandedBits(Instruction *I,
709 const APInt &DemandedMask,
Craig Topperb45eabc2017-04-26 16:39:58 +0000710 KnownBits &Known,
Craig Topperb0076fe2017-04-12 18:05:21 +0000711 unsigned Depth,
712 Instruction *CxtI) {
713 unsigned BitWidth = DemandedMask.getBitWidth();
714 Type *ITy = I->getType();
715
Craig Topperb45eabc2017-04-26 16:39:58 +0000716 KnownBits LHSKnown(BitWidth);
717 KnownBits RHSKnown(BitWidth);
Craig Topperb0076fe2017-04-12 18:05:21 +0000718
719 // Despite the fact that we can't simplify this instruction in all User's
Craig Topperb45eabc2017-04-26 16:39:58 +0000720 // context, we can at least compute the known bits, and we can
Craig Topperb0076fe2017-04-12 18:05:21 +0000721 // do simplifications that apply to *just* the one user if we know that
722 // this instruction has a simpler value in that context.
Craig Topperf35a7f72017-04-12 18:25:25 +0000723 switch (I->getOpcode()) {
Craig Topper9a458cd2017-04-14 22:34:14 +0000724 case Instruction::And: {
Craig Topperb0076fe2017-04-12 18:05:21 +0000725 // If either the LHS or the RHS are Zero, the result is zero.
Craig Topperb45eabc2017-04-26 16:39:58 +0000726 computeKnownBits(I->getOperand(1), RHSKnown, Depth + 1, CxtI);
727 computeKnownBits(I->getOperand(0), LHSKnown, Depth + 1,
Craig Topperb0076fe2017-04-12 18:05:21 +0000728 CxtI);
729
Craig Topper9a458cd2017-04-14 22:34:14 +0000730 // Output known-0 are known to be clear if zero in either the LHS | RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000731 APInt IKnownZero = RHSKnown.Zero | LHSKnown.Zero;
Craig Topper9a458cd2017-04-14 22:34:14 +0000732 // Output known-1 bits are only known if set in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000733 APInt IKnownOne = RHSKnown.One & LHSKnown.One;
Craig Topper9a458cd2017-04-14 22:34:14 +0000734
Craig Topperc75f94b2017-04-12 19:32:47 +0000735 // If the client is only demanding bits that we know, return the known
736 // constant.
Craig Topper17f37ba2017-04-20 20:47:35 +0000737 if (DemandedMask.isSubsetOf(IKnownZero|IKnownOne))
Craig Topper9a458cd2017-04-14 22:34:14 +0000738 return Constant::getIntegerValue(ITy, IKnownOne);
Craig Topperc75f94b2017-04-12 19:32:47 +0000739
Craig Topperb0076fe2017-04-12 18:05:21 +0000740 // If all of the demanded bits are known 1 on one side, return the other.
741 // These bits cannot contribute to the result of the 'and' in this
742 // context.
Craig Topperb45eabc2017-04-26 16:39:58 +0000743 if (DemandedMask.isSubsetOf(LHSKnown.Zero | RHSKnown.One))
Craig Topperb0076fe2017-04-12 18:05:21 +0000744 return I->getOperand(0);
Craig Topperb45eabc2017-04-26 16:39:58 +0000745 if (DemandedMask.isSubsetOf(RHSKnown.Zero | LHSKnown.One))
Craig Topperb0076fe2017-04-12 18:05:21 +0000746 return I->getOperand(1);
747
Craig Topperb45eabc2017-04-26 16:39:58 +0000748 Known.Zero = std::move(IKnownZero);
749 Known.One = std::move(IKnownOne);
Craig Topperf35a7f72017-04-12 18:25:25 +0000750 break;
Craig Topper9a458cd2017-04-14 22:34:14 +0000751 }
752 case Instruction::Or: {
Craig Topperb0076fe2017-04-12 18:05:21 +0000753 // We can simplify (X|Y) -> X or Y in the user's context if we know that
754 // only bits from X or Y are demanded.
755
756 // If either the LHS or the RHS are One, the result is One.
Craig Topperb45eabc2017-04-26 16:39:58 +0000757 computeKnownBits(I->getOperand(1), RHSKnown, Depth + 1, CxtI);
758 computeKnownBits(I->getOperand(0), LHSKnown, Depth + 1,
Craig Topperb0076fe2017-04-12 18:05:21 +0000759 CxtI);
760
Craig Topper9a458cd2017-04-14 22:34:14 +0000761 // Output known-0 bits are only known if clear in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000762 APInt IKnownZero = RHSKnown.Zero & LHSKnown.Zero;
Craig Topper9a458cd2017-04-14 22:34:14 +0000763 // Output known-1 are known to be set if set in either the LHS | RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000764 APInt IKnownOne = RHSKnown.One | LHSKnown.One;
Craig Topper9a458cd2017-04-14 22:34:14 +0000765
Craig Topperc75f94b2017-04-12 19:32:47 +0000766 // If the client is only demanding bits that we know, return the known
767 // constant.
Craig Topper17f37ba2017-04-20 20:47:35 +0000768 if (DemandedMask.isSubsetOf(IKnownZero|IKnownOne))
Craig Topper9a458cd2017-04-14 22:34:14 +0000769 return Constant::getIntegerValue(ITy, IKnownOne);
Craig Topperc75f94b2017-04-12 19:32:47 +0000770
Craig Topperb0076fe2017-04-12 18:05:21 +0000771 // If all of the demanded bits are known zero on one side, return the
772 // other. These bits cannot contribute to the result of the 'or' in this
773 // context.
Craig Topperb45eabc2017-04-26 16:39:58 +0000774 if (DemandedMask.isSubsetOf(LHSKnown.One | RHSKnown.Zero))
Craig Topperb0076fe2017-04-12 18:05:21 +0000775 return I->getOperand(0);
Craig Topperb45eabc2017-04-26 16:39:58 +0000776 if (DemandedMask.isSubsetOf(RHSKnown.One | LHSKnown.Zero))
Craig Topperb0076fe2017-04-12 18:05:21 +0000777 return I->getOperand(1);
778
Craig Topperb45eabc2017-04-26 16:39:58 +0000779 Known.Zero = std::move(IKnownZero);
780 Known.One = std::move(IKnownOne);
Craig Topperf35a7f72017-04-12 18:25:25 +0000781 break;
Craig Topper9a458cd2017-04-14 22:34:14 +0000782 }
Craig Topperc75f94b2017-04-12 19:32:47 +0000783 case Instruction::Xor: {
Craig Topperb0076fe2017-04-12 18:05:21 +0000784 // We can simplify (X^Y) -> X or Y in the user's context if we know that
785 // only bits from X or Y are demanded.
786
Craig Topperb45eabc2017-04-26 16:39:58 +0000787 computeKnownBits(I->getOperand(1), RHSKnown, Depth + 1, CxtI);
788 computeKnownBits(I->getOperand(0), LHSKnown, Depth + 1,
Craig Topperb0076fe2017-04-12 18:05:21 +0000789 CxtI);
790
Craig Topperc75f94b2017-04-12 19:32:47 +0000791 // Output known-0 bits are known if clear or set in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000792 APInt IKnownZero = (RHSKnown.Zero & LHSKnown.Zero) |
793 (RHSKnown.One & LHSKnown.One);
Craig Topperc75f94b2017-04-12 19:32:47 +0000794 // Output known-1 are known to be set if set in only one of the LHS, RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000795 APInt IKnownOne = (RHSKnown.Zero & LHSKnown.One) |
796 (RHSKnown.One & LHSKnown.Zero);
Craig Topperc75f94b2017-04-12 19:32:47 +0000797
798 // If the client is only demanding bits that we know, return the known
799 // constant.
Craig Topper17f37ba2017-04-20 20:47:35 +0000800 if (DemandedMask.isSubsetOf(IKnownZero|IKnownOne))
Craig Topperc75f94b2017-04-12 19:32:47 +0000801 return Constant::getIntegerValue(ITy, IKnownOne);
802
Craig Topperb0076fe2017-04-12 18:05:21 +0000803 // If all of the demanded bits are known zero on one side, return the
804 // other.
Craig Topperb45eabc2017-04-26 16:39:58 +0000805 if (DemandedMask.isSubsetOf(RHSKnown.Zero))
Craig Topperb0076fe2017-04-12 18:05:21 +0000806 return I->getOperand(0);
Craig Topperb45eabc2017-04-26 16:39:58 +0000807 if (DemandedMask.isSubsetOf(LHSKnown.Zero))
Craig Topperb0076fe2017-04-12 18:05:21 +0000808 return I->getOperand(1);
Craig Topperf35a7f72017-04-12 18:25:25 +0000809
Craig Topperc75f94b2017-04-12 19:32:47 +0000810 // Output known-0 bits are known if clear or set in both the LHS & RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000811 Known.Zero = std::move(IKnownZero);
Craig Topperc75f94b2017-04-12 19:32:47 +0000812 // Output known-1 are known to be set if set in only one of the LHS, RHS.
Craig Topperb45eabc2017-04-26 16:39:58 +0000813 Known.One = std::move(IKnownOne);
Craig Topperf35a7f72017-04-12 18:25:25 +0000814 break;
Craig Topperb0076fe2017-04-12 18:05:21 +0000815 }
Craig Topperc75f94b2017-04-12 19:32:47 +0000816 default:
Craig Topperb45eabc2017-04-26 16:39:58 +0000817 // Compute the Known bits to simplify things downstream.
818 computeKnownBits(I, Known, Depth, CxtI);
Craig Topperb0076fe2017-04-12 18:05:21 +0000819
Craig Topperc75f94b2017-04-12 19:32:47 +0000820 // If this user is only demanding bits that we know, return the known
821 // constant.
Craig Topperb45eabc2017-04-26 16:39:58 +0000822 if (DemandedMask.isSubsetOf(Known.Zero|Known.One))
823 return Constant::getIntegerValue(ITy, Known.One);
Craig Topper9a51c7f2017-04-12 18:17:46 +0000824
Craig Topperc75f94b2017-04-12 19:32:47 +0000825 break;
826 }
Craig Topper9a51c7f2017-04-12 18:17:46 +0000827
Craig Topperb0076fe2017-04-12 18:05:21 +0000828 return nullptr;
829}
830
831
Shuxin Yang63e999e2012-12-04 00:04:54 +0000832/// Helper routine of SimplifyDemandedUseBits. It tries to simplify
833/// "E1 = (X lsr C1) << C2", where the C1 and C2 are constant, into
834/// "E2 = X << (C2 - C1)" or "E2 = X >> (C1 - C2)", depending on the sign
835/// of "C2-C1".
836///
837/// Suppose E1 and E2 are generally different in bits S={bm, bm+1,
838/// ..., bn}, without considering the specific value X is holding.
839/// This transformation is legal iff one of following conditions is hold:
840/// 1) All the bit in S are 0, in this case E1 == E2.
841/// 2) We don't care those bits in S, per the input DemandedMask.
842/// 3) Combination of 1) and 2). Some bits in S are 0, and we don't care the
843/// rest bits.
844///
845/// Currently we only test condition 2).
846///
847/// As with SimplifyDemandedUseBits, it returns NULL if the simplification was
848/// not successful.
Sanjay Patelc9485ca2017-04-20 22:33:54 +0000849Value *
Sanjay Patelcc663b82017-04-20 22:37:01 +0000850InstCombiner::simplifyShrShlDemandedBits(Instruction *Shr, const APInt &ShrOp1,
Sanjay Patelc9485ca2017-04-20 22:33:54 +0000851 Instruction *Shl, const APInt &ShlOp1,
852 const APInt &DemandedMask,
Craig Topperb45eabc2017-04-26 16:39:58 +0000853 KnownBits &Known) {
Benjamin Kramer010f1082013-08-30 14:35:35 +0000854 if (!ShlOp1 || !ShrOp1)
Sanjay Patelc9485ca2017-04-20 22:33:54 +0000855 return nullptr; // No-op.
Benjamin Kramer010f1082013-08-30 14:35:35 +0000856
857 Value *VarX = Shr->getOperand(0);
858 Type *Ty = VarX->getType();
Sanjay Patelc9485ca2017-04-20 22:33:54 +0000859 unsigned BitWidth = Ty->getScalarSizeInBits();
Benjamin Kramer010f1082013-08-30 14:35:35 +0000860 if (ShlOp1.uge(BitWidth) || ShrOp1.uge(BitWidth))
Craig Topperf40110f2014-04-25 05:29:35 +0000861 return nullptr; // Undef.
Benjamin Kramer010f1082013-08-30 14:35:35 +0000862
863 unsigned ShlAmt = ShlOp1.getZExtValue();
864 unsigned ShrAmt = ShrOp1.getZExtValue();
Shuxin Yang63e999e2012-12-04 00:04:54 +0000865
Craig Topperb45eabc2017-04-26 16:39:58 +0000866 Known.One.clearAllBits();
867 Known.Zero.setLowBits(ShlAmt - 1);
868 Known.Zero &= DemandedMask;
Shuxin Yang63e999e2012-12-04 00:04:54 +0000869
Benjamin Kramer010f1082013-08-30 14:35:35 +0000870 APInt BitMask1(APInt::getAllOnesValue(BitWidth));
871 APInt BitMask2(APInt::getAllOnesValue(BitWidth));
Shuxin Yang63e999e2012-12-04 00:04:54 +0000872
873 bool isLshr = (Shr->getOpcode() == Instruction::LShr);
874 BitMask1 = isLshr ? (BitMask1.lshr(ShrAmt) << ShlAmt) :
875 (BitMask1.ashr(ShrAmt) << ShlAmt);
876
877 if (ShrAmt <= ShlAmt) {
878 BitMask2 <<= (ShlAmt - ShrAmt);
879 } else {
880 BitMask2 = isLshr ? BitMask2.lshr(ShrAmt - ShlAmt):
881 BitMask2.ashr(ShrAmt - ShlAmt);
882 }
883
884 // Check if condition-2 (see the comment to this function) is satified.
885 if ((BitMask1 & DemandedMask) == (BitMask2 & DemandedMask)) {
886 if (ShrAmt == ShlAmt)
887 return VarX;
888
889 if (!Shr->hasOneUse())
Craig Topperf40110f2014-04-25 05:29:35 +0000890 return nullptr;
Shuxin Yang63e999e2012-12-04 00:04:54 +0000891
892 BinaryOperator *New;
893 if (ShrAmt < ShlAmt) {
894 Constant *Amt = ConstantInt::get(VarX->getType(), ShlAmt - ShrAmt);
895 New = BinaryOperator::CreateShl(VarX, Amt);
896 BinaryOperator *Orig = cast<BinaryOperator>(Shl);
897 New->setHasNoSignedWrap(Orig->hasNoSignedWrap());
898 New->setHasNoUnsignedWrap(Orig->hasNoUnsignedWrap());
899 } else {
900 Constant *Amt = ConstantInt::get(VarX->getType(), ShrAmt - ShlAmt);
Shuxin Yang86c0e232012-12-04 03:28:32 +0000901 New = isLshr ? BinaryOperator::CreateLShr(VarX, Amt) :
902 BinaryOperator::CreateAShr(VarX, Amt);
Shuxin Yang81b36782012-12-12 00:29:03 +0000903 if (cast<BinaryOperator>(Shr)->isExact())
904 New->setIsExact(true);
Shuxin Yang63e999e2012-12-04 00:04:54 +0000905 }
906
907 return InsertNewInstWith(New, *Shl);
908 }
909
Craig Topperf40110f2014-04-25 05:29:35 +0000910 return nullptr;
Shuxin Yang63e999e2012-12-04 00:04:54 +0000911}
Chris Lattner7e044912010-01-04 07:17:19 +0000912
Sanjay Patelbbbb3ce2016-07-14 20:54:43 +0000913/// The specified value produces a vector with any number of elements.
914/// DemandedElts contains the set of elements that are actually used by the
915/// caller. This method analyzes which elements of the operand are undef and
916/// returns that information in UndefElts.
Chris Lattner7e044912010-01-04 07:17:19 +0000917///
918/// If the information about demanded elements can be used to simplify the
919/// operation, the operation is simplified, then the resultant value is
920/// returned. This returns null if no change was made.
921Value *InstCombiner::SimplifyDemandedVectorElts(Value *V, APInt DemandedElts,
Chris Lattnerb22423c2010-02-08 23:56:03 +0000922 APInt &UndefElts,
Chris Lattner7e044912010-01-04 07:17:19 +0000923 unsigned Depth) {
Sanjay Patel9190b4a2016-04-29 20:54:56 +0000924 unsigned VWidth = V->getType()->getVectorNumElements();
Chris Lattner7e044912010-01-04 07:17:19 +0000925 APInt EltMask(APInt::getAllOnesValue(VWidth));
926 assert((DemandedElts & ~EltMask) == 0 && "Invalid DemandedElts!");
927
928 if (isa<UndefValue>(V)) {
929 // If the entire vector is undefined, just return this info.
930 UndefElts = EltMask;
Craig Topperf40110f2014-04-25 05:29:35 +0000931 return nullptr;
Chris Lattnerb22423c2010-02-08 23:56:03 +0000932 }
Craig Topper4c947752012-12-22 18:09:02 +0000933
Chris Lattnerb22423c2010-02-08 23:56:03 +0000934 if (DemandedElts == 0) { // If nothing is demanded, provide undef.
Chris Lattner7e044912010-01-04 07:17:19 +0000935 UndefElts = EltMask;
936 return UndefValue::get(V->getType());
937 }
938
939 UndefElts = 0;
Craig Topper4c947752012-12-22 18:09:02 +0000940
Chris Lattner67058832012-01-25 06:48:06 +0000941 // Handle ConstantAggregateZero, ConstantVector, ConstantDataSequential.
942 if (Constant *C = dyn_cast<Constant>(V)) {
943 // Check if this is identity. If so, return 0 since we are not simplifying
944 // anything.
945 if (DemandedElts.isAllOnesValue())
Craig Topperf40110f2014-04-25 05:29:35 +0000946 return nullptr;
Chris Lattner67058832012-01-25 06:48:06 +0000947
Chris Lattner229907c2011-07-18 04:54:35 +0000948 Type *EltTy = cast<VectorType>(V->getType())->getElementType();
Chris Lattner7e044912010-01-04 07:17:19 +0000949 Constant *Undef = UndefValue::get(EltTy);
Craig Topper4c947752012-12-22 18:09:02 +0000950
Chris Lattner67058832012-01-25 06:48:06 +0000951 SmallVector<Constant*, 16> Elts;
952 for (unsigned i = 0; i != VWidth; ++i) {
Chris Lattner7e044912010-01-04 07:17:19 +0000953 if (!DemandedElts[i]) { // If not demanded, set to undef.
954 Elts.push_back(Undef);
Jay Foad25a5e4c2010-12-01 08:53:58 +0000955 UndefElts.setBit(i);
Chris Lattner67058832012-01-25 06:48:06 +0000956 continue;
957 }
Craig Topper4c947752012-12-22 18:09:02 +0000958
Chris Lattner67058832012-01-25 06:48:06 +0000959 Constant *Elt = C->getAggregateElement(i);
Craig Topperf40110f2014-04-25 05:29:35 +0000960 if (!Elt) return nullptr;
Craig Topper4c947752012-12-22 18:09:02 +0000961
Chris Lattner67058832012-01-25 06:48:06 +0000962 if (isa<UndefValue>(Elt)) { // Already undef.
Chris Lattner7e044912010-01-04 07:17:19 +0000963 Elts.push_back(Undef);
Jay Foad25a5e4c2010-12-01 08:53:58 +0000964 UndefElts.setBit(i);
Chris Lattner7e044912010-01-04 07:17:19 +0000965 } else { // Otherwise, defined.
Chris Lattner67058832012-01-25 06:48:06 +0000966 Elts.push_back(Elt);
Chris Lattner7e044912010-01-04 07:17:19 +0000967 }
Chris Lattner67058832012-01-25 06:48:06 +0000968 }
Craig Topper4c947752012-12-22 18:09:02 +0000969
Chris Lattner7e044912010-01-04 07:17:19 +0000970 // If we changed the constant, return it.
Chris Lattner47a86bd2012-01-25 06:02:56 +0000971 Constant *NewCV = ConstantVector::get(Elts);
Craig Topperf40110f2014-04-25 05:29:35 +0000972 return NewCV != C ? NewCV : nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000973 }
Craig Topper4c947752012-12-22 18:09:02 +0000974
Chris Lattner7e044912010-01-04 07:17:19 +0000975 // Limit search depth.
976 if (Depth == 10)
Craig Topperf40110f2014-04-25 05:29:35 +0000977 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000978
Stuart Hastings5bd18b62011-05-17 22:13:31 +0000979 // If multiple users are using the root value, proceed with
Chris Lattner7e044912010-01-04 07:17:19 +0000980 // simplification conservatively assuming that all elements
981 // are needed.
982 if (!V->hasOneUse()) {
983 // Quit if we find multiple users of a non-root value though.
984 // They'll be handled when it's their turn to be visited by
985 // the main instcombine process.
986 if (Depth != 0)
987 // TODO: Just compute the UndefElts information recursively.
Craig Topperf40110f2014-04-25 05:29:35 +0000988 return nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +0000989
990 // Conservatively assume that all elements are needed.
991 DemandedElts = EltMask;
992 }
Craig Topper4c947752012-12-22 18:09:02 +0000993
Chris Lattner7e044912010-01-04 07:17:19 +0000994 Instruction *I = dyn_cast<Instruction>(V);
Craig Topperf40110f2014-04-25 05:29:35 +0000995 if (!I) return nullptr; // Only analyze instructions.
Craig Topper4c947752012-12-22 18:09:02 +0000996
Chris Lattner7e044912010-01-04 07:17:19 +0000997 bool MadeChange = false;
998 APInt UndefElts2(VWidth, 0);
Craig Topper23ebd952016-12-11 08:54:52 +0000999 APInt UndefElts3(VWidth, 0);
Chris Lattner7e044912010-01-04 07:17:19 +00001000 Value *TmpV;
1001 switch (I->getOpcode()) {
1002 default: break;
Craig Topper4c947752012-12-22 18:09:02 +00001003
Chris Lattner7e044912010-01-04 07:17:19 +00001004 case Instruction::InsertElement: {
1005 // If this is a variable index, we don't know which element it overwrites.
1006 // demand exactly the same input as we produce.
1007 ConstantInt *Idx = dyn_cast<ConstantInt>(I->getOperand(2));
Craig Topperf40110f2014-04-25 05:29:35 +00001008 if (!Idx) {
Chris Lattner7e044912010-01-04 07:17:19 +00001009 // Note that we can't propagate undef elt info, because we don't know
1010 // which elt is getting updated.
1011 TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001012 UndefElts2, Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001013 if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; }
1014 break;
1015 }
Craig Topper4c947752012-12-22 18:09:02 +00001016
Chris Lattner7e044912010-01-04 07:17:19 +00001017 // If this is inserting an element that isn't demanded, remove this
1018 // insertelement.
1019 unsigned IdxNo = Idx->getZExtValue();
1020 if (IdxNo >= VWidth || !DemandedElts[IdxNo]) {
1021 Worklist.Add(I);
1022 return I->getOperand(0);
1023 }
Craig Topper4c947752012-12-22 18:09:02 +00001024
Chris Lattner7e044912010-01-04 07:17:19 +00001025 // Otherwise, the element inserted overwrites whatever was there, so the
1026 // input demanded set is simpler than the output set.
1027 APInt DemandedElts2 = DemandedElts;
Jay Foad25a5e4c2010-12-01 08:53:58 +00001028 DemandedElts2.clearBit(IdxNo);
Chris Lattner7e044912010-01-04 07:17:19 +00001029 TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts2,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001030 UndefElts, Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001031 if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; }
1032
1033 // The inserted element is defined.
Jay Foad25a5e4c2010-12-01 08:53:58 +00001034 UndefElts.clearBit(IdxNo);
Chris Lattner7e044912010-01-04 07:17:19 +00001035 break;
1036 }
1037 case Instruction::ShuffleVector: {
1038 ShuffleVectorInst *Shuffle = cast<ShuffleVectorInst>(I);
Craig Topper2e18bcf2016-12-29 04:24:32 +00001039 unsigned LHSVWidth =
1040 Shuffle->getOperand(0)->getType()->getVectorNumElements();
Chris Lattner7e044912010-01-04 07:17:19 +00001041 APInt LeftDemanded(LHSVWidth, 0), RightDemanded(LHSVWidth, 0);
1042 for (unsigned i = 0; i < VWidth; i++) {
1043 if (DemandedElts[i]) {
1044 unsigned MaskVal = Shuffle->getMaskValue(i);
1045 if (MaskVal != -1u) {
1046 assert(MaskVal < LHSVWidth * 2 &&
1047 "shufflevector mask index out of range!");
1048 if (MaskVal < LHSVWidth)
Jay Foad25a5e4c2010-12-01 08:53:58 +00001049 LeftDemanded.setBit(MaskVal);
Chris Lattner7e044912010-01-04 07:17:19 +00001050 else
Jay Foad25a5e4c2010-12-01 08:53:58 +00001051 RightDemanded.setBit(MaskVal - LHSVWidth);
Chris Lattner7e044912010-01-04 07:17:19 +00001052 }
1053 }
1054 }
1055
Alexey Bataevfee90782016-09-23 09:14:08 +00001056 APInt LHSUndefElts(LHSVWidth, 0);
Chris Lattner7e044912010-01-04 07:17:19 +00001057 TmpV = SimplifyDemandedVectorElts(I->getOperand(0), LeftDemanded,
Alexey Bataevfee90782016-09-23 09:14:08 +00001058 LHSUndefElts, Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001059 if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; }
1060
Alexey Bataevfee90782016-09-23 09:14:08 +00001061 APInt RHSUndefElts(LHSVWidth, 0);
Chris Lattner7e044912010-01-04 07:17:19 +00001062 TmpV = SimplifyDemandedVectorElts(I->getOperand(1), RightDemanded,
Alexey Bataevfee90782016-09-23 09:14:08 +00001063 RHSUndefElts, Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001064 if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; }
1065
1066 bool NewUndefElts = false;
Alexey Bataev793c9462016-09-26 13:18:59 +00001067 unsigned LHSIdx = -1u, LHSValIdx = -1u;
1068 unsigned RHSIdx = -1u, RHSValIdx = -1u;
Alexey Bataevfee90782016-09-23 09:14:08 +00001069 bool LHSUniform = true;
1070 bool RHSUniform = true;
Chris Lattner7e044912010-01-04 07:17:19 +00001071 for (unsigned i = 0; i < VWidth; i++) {
1072 unsigned MaskVal = Shuffle->getMaskValue(i);
1073 if (MaskVal == -1u) {
Jay Foad25a5e4c2010-12-01 08:53:58 +00001074 UndefElts.setBit(i);
Eli Friedman888bea02011-09-15 01:14:29 +00001075 } else if (!DemandedElts[i]) {
1076 NewUndefElts = true;
1077 UndefElts.setBit(i);
Chris Lattner7e044912010-01-04 07:17:19 +00001078 } else if (MaskVal < LHSVWidth) {
Alexey Bataevfee90782016-09-23 09:14:08 +00001079 if (LHSUndefElts[MaskVal]) {
Chris Lattner7e044912010-01-04 07:17:19 +00001080 NewUndefElts = true;
Jay Foad25a5e4c2010-12-01 08:53:58 +00001081 UndefElts.setBit(i);
Alexey Bataevfee90782016-09-23 09:14:08 +00001082 } else {
Alexey Bataev793c9462016-09-26 13:18:59 +00001083 LHSIdx = LHSIdx == -1u ? i : LHSVWidth;
1084 LHSValIdx = LHSValIdx == -1u ? MaskVal : LHSVWidth;
Alexey Bataevfee90782016-09-23 09:14:08 +00001085 LHSUniform = LHSUniform && (MaskVal == i);
Chris Lattner7e044912010-01-04 07:17:19 +00001086 }
1087 } else {
Alexey Bataevfee90782016-09-23 09:14:08 +00001088 if (RHSUndefElts[MaskVal - LHSVWidth]) {
Chris Lattner7e044912010-01-04 07:17:19 +00001089 NewUndefElts = true;
Jay Foad25a5e4c2010-12-01 08:53:58 +00001090 UndefElts.setBit(i);
Alexey Bataevfee90782016-09-23 09:14:08 +00001091 } else {
Alexey Bataev793c9462016-09-26 13:18:59 +00001092 RHSIdx = RHSIdx == -1u ? i : LHSVWidth;
1093 RHSValIdx = RHSValIdx == -1u ? MaskVal - LHSVWidth : LHSVWidth;
Alexey Bataevfee90782016-09-23 09:14:08 +00001094 RHSUniform = RHSUniform && (MaskVal - LHSVWidth == i);
Chris Lattner7e044912010-01-04 07:17:19 +00001095 }
1096 }
1097 }
1098
Alexey Bataevfee90782016-09-23 09:14:08 +00001099 // Try to transform shuffle with constant vector and single element from
1100 // this constant vector to single insertelement instruction.
1101 // shufflevector V, C, <v1, v2, .., ci, .., vm> ->
1102 // insertelement V, C[ci], ci-n
1103 if (LHSVWidth == Shuffle->getType()->getNumElements()) {
1104 Value *Op = nullptr;
1105 Constant *Value = nullptr;
1106 unsigned Idx = -1u;
1107
Craig Topper62f06e22016-12-29 05:38:31 +00001108 // Find constant vector with the single element in shuffle (LHS or RHS).
Alexey Bataevfee90782016-09-23 09:14:08 +00001109 if (LHSIdx < LHSVWidth && RHSUniform) {
1110 if (auto *CV = dyn_cast<ConstantVector>(Shuffle->getOperand(0))) {
1111 Op = Shuffle->getOperand(1);
Alexey Bataev793c9462016-09-26 13:18:59 +00001112 Value = CV->getOperand(LHSValIdx);
Alexey Bataevfee90782016-09-23 09:14:08 +00001113 Idx = LHSIdx;
1114 }
1115 }
1116 if (RHSIdx < LHSVWidth && LHSUniform) {
1117 if (auto *CV = dyn_cast<ConstantVector>(Shuffle->getOperand(1))) {
1118 Op = Shuffle->getOperand(0);
Alexey Bataev793c9462016-09-26 13:18:59 +00001119 Value = CV->getOperand(RHSValIdx);
Alexey Bataevfee90782016-09-23 09:14:08 +00001120 Idx = RHSIdx;
1121 }
1122 }
1123 // Found constant vector with single element - convert to insertelement.
1124 if (Op && Value) {
1125 Instruction *New = InsertElementInst::Create(
1126 Op, Value, ConstantInt::get(Type::getInt32Ty(I->getContext()), Idx),
1127 Shuffle->getName());
1128 InsertNewInstWith(New, *Shuffle);
1129 return New;
1130 }
1131 }
Chris Lattner7e044912010-01-04 07:17:19 +00001132 if (NewUndefElts) {
1133 // Add additional discovered undefs.
Chris Lattner0256be92012-01-27 03:08:05 +00001134 SmallVector<Constant*, 16> Elts;
Chris Lattner7e044912010-01-04 07:17:19 +00001135 for (unsigned i = 0; i < VWidth; ++i) {
1136 if (UndefElts[i])
1137 Elts.push_back(UndefValue::get(Type::getInt32Ty(I->getContext())));
1138 else
1139 Elts.push_back(ConstantInt::get(Type::getInt32Ty(I->getContext()),
1140 Shuffle->getMaskValue(i)));
1141 }
1142 I->setOperand(2, ConstantVector::get(Elts));
1143 MadeChange = true;
1144 }
1145 break;
1146 }
Pete Cooperabc13af2012-07-26 23:10:24 +00001147 case Instruction::Select: {
1148 APInt LeftDemanded(DemandedElts), RightDemanded(DemandedElts);
1149 if (ConstantVector* CV = dyn_cast<ConstantVector>(I->getOperand(0))) {
1150 for (unsigned i = 0; i < VWidth; i++) {
Andrea Di Biagio40f59e42015-10-06 10:34:53 +00001151 Constant *CElt = CV->getAggregateElement(i);
1152 // Method isNullValue always returns false when called on a
1153 // ConstantExpr. If CElt is a ConstantExpr then skip it in order to
1154 // to avoid propagating incorrect information.
1155 if (isa<ConstantExpr>(CElt))
1156 continue;
1157 if (CElt->isNullValue())
Pete Cooperabc13af2012-07-26 23:10:24 +00001158 LeftDemanded.clearBit(i);
1159 else
1160 RightDemanded.clearBit(i);
1161 }
1162 }
1163
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001164 TmpV = SimplifyDemandedVectorElts(I->getOperand(1), LeftDemanded, UndefElts,
1165 Depth + 1);
Pete Cooperabc13af2012-07-26 23:10:24 +00001166 if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; }
1167
1168 TmpV = SimplifyDemandedVectorElts(I->getOperand(2), RightDemanded,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001169 UndefElts2, Depth + 1);
Pete Cooperabc13af2012-07-26 23:10:24 +00001170 if (TmpV) { I->setOperand(2, TmpV); MadeChange = true; }
Craig Topper4c947752012-12-22 18:09:02 +00001171
Pete Cooperabc13af2012-07-26 23:10:24 +00001172 // Output elements are undefined if both are undefined.
1173 UndefElts &= UndefElts2;
1174 break;
1175 }
Chris Lattner7e044912010-01-04 07:17:19 +00001176 case Instruction::BitCast: {
1177 // Vector->vector casts only.
Chris Lattner229907c2011-07-18 04:54:35 +00001178 VectorType *VTy = dyn_cast<VectorType>(I->getOperand(0)->getType());
Chris Lattner7e044912010-01-04 07:17:19 +00001179 if (!VTy) break;
1180 unsigned InVWidth = VTy->getNumElements();
1181 APInt InputDemandedElts(InVWidth, 0);
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001182 UndefElts2 = APInt(InVWidth, 0);
Chris Lattner7e044912010-01-04 07:17:19 +00001183 unsigned Ratio;
1184
1185 if (VWidth == InVWidth) {
1186 // If we are converting from <4 x i32> -> <4 x f32>, we demand the same
1187 // elements as are demanded of us.
1188 Ratio = 1;
1189 InputDemandedElts = DemandedElts;
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001190 } else if ((VWidth % InVWidth) == 0) {
1191 // If the number of elements in the output is a multiple of the number of
1192 // elements in the input then an input element is live if any of the
1193 // corresponding output elements are live.
1194 Ratio = VWidth / InVWidth;
1195 for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx)
Chris Lattner7e044912010-01-04 07:17:19 +00001196 if (DemandedElts[OutIdx])
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001197 InputDemandedElts.setBit(OutIdx / Ratio);
1198 } else if ((InVWidth % VWidth) == 0) {
1199 // If the number of elements in the input is a multiple of the number of
1200 // elements in the output then an input element is live if the
1201 // corresponding output element is live.
1202 Ratio = InVWidth / VWidth;
Chris Lattner7e044912010-01-04 07:17:19 +00001203 for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx)
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001204 if (DemandedElts[InIdx / Ratio])
Jay Foad25a5e4c2010-12-01 08:53:58 +00001205 InputDemandedElts.setBit(InIdx);
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001206 } else {
1207 // Unsupported so far.
1208 break;
Chris Lattner7e044912010-01-04 07:17:19 +00001209 }
Craig Topper4c947752012-12-22 18:09:02 +00001210
Chris Lattner7e044912010-01-04 07:17:19 +00001211 // div/rem demand all inputs, because they don't want divide by zero.
1212 TmpV = SimplifyDemandedVectorElts(I->getOperand(0), InputDemandedElts,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001213 UndefElts2, Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001214 if (TmpV) {
1215 I->setOperand(0, TmpV);
1216 MadeChange = true;
1217 }
Craig Topper4c947752012-12-22 18:09:02 +00001218
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001219 if (VWidth == InVWidth) {
1220 UndefElts = UndefElts2;
1221 } else if ((VWidth % InVWidth) == 0) {
1222 // If the number of elements in the output is a multiple of the number of
1223 // elements in the input then an output element is undef if the
1224 // corresponding input element is undef.
Chris Lattner7e044912010-01-04 07:17:19 +00001225 for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx)
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001226 if (UndefElts2[OutIdx / Ratio])
Jay Foad25a5e4c2010-12-01 08:53:58 +00001227 UndefElts.setBit(OutIdx);
Simon Pilgrim43f5e082015-09-29 08:19:11 +00001228 } else if ((InVWidth % VWidth) == 0) {
1229 // If the number of elements in the input is a multiple of the number of
1230 // elements in the output then an output element is undef if all of the
1231 // corresponding input elements are undef.
1232 for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) {
1233 APInt SubUndef = UndefElts2.lshr(OutIdx * Ratio).zextOrTrunc(Ratio);
1234 if (SubUndef.countPopulation() == Ratio)
1235 UndefElts.setBit(OutIdx);
1236 }
1237 } else {
Chris Lattner7e044912010-01-04 07:17:19 +00001238 llvm_unreachable("Unimp");
Chris Lattner7e044912010-01-04 07:17:19 +00001239 }
1240 break;
1241 }
1242 case Instruction::And:
1243 case Instruction::Or:
1244 case Instruction::Xor:
1245 case Instruction::Add:
1246 case Instruction::Sub:
1247 case Instruction::Mul:
1248 // div/rem demand all inputs, because they don't want divide by zero.
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001249 TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, UndefElts,
1250 Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001251 if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; }
1252 TmpV = SimplifyDemandedVectorElts(I->getOperand(1), DemandedElts,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001253 UndefElts2, Depth + 1);
Chris Lattner7e044912010-01-04 07:17:19 +00001254 if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; }
Craig Topper4c947752012-12-22 18:09:02 +00001255
Chris Lattner7e044912010-01-04 07:17:19 +00001256 // Output elements are undefined if both are undefined. Consider things
1257 // like undef&0. The result is known zero, not undef.
1258 UndefElts &= UndefElts2;
1259 break;
Pete Coopere807e452012-07-26 22:37:04 +00001260 case Instruction::FPTrunc:
1261 case Instruction::FPExt:
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001262 TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, UndefElts,
1263 Depth + 1);
Pete Coopere807e452012-07-26 22:37:04 +00001264 if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; }
1265 break;
Craig Topper4c947752012-12-22 18:09:02 +00001266
Chris Lattner7e044912010-01-04 07:17:19 +00001267 case Instruction::Call: {
1268 IntrinsicInst *II = dyn_cast<IntrinsicInst>(I);
1269 if (!II) break;
1270 switch (II->getIntrinsicID()) {
1271 default: break;
Craig Topper4c947752012-12-22 18:09:02 +00001272
Craig Topper7fc6d342016-12-11 22:32:38 +00001273 case Intrinsic::x86_xop_vfrcz_ss:
1274 case Intrinsic::x86_xop_vfrcz_sd:
1275 // The instructions for these intrinsics are speced to zero upper bits not
1276 // pass them through like other scalar intrinsics. So we shouldn't just
1277 // use Arg0 if DemandedElts[0] is clear like we do for other intrinsics.
1278 // Instead we should return a zero vector.
Craig Topper1a8a3372016-12-29 03:30:17 +00001279 if (!DemandedElts[0]) {
1280 Worklist.Add(II);
Craig Topper7fc6d342016-12-11 22:32:38 +00001281 return ConstantAggregateZero::get(II->getType());
Craig Topper1a8a3372016-12-29 03:30:17 +00001282 }
Craig Topper7fc6d342016-12-11 22:32:38 +00001283
Craig Topperac75bca2016-12-13 07:45:45 +00001284 // Only the lower element is used.
1285 DemandedElts = 1;
Craig Topper7fc6d342016-12-11 22:32:38 +00001286 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), DemandedElts,
1287 UndefElts, Depth + 1);
1288 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
Craig Topperac75bca2016-12-13 07:45:45 +00001289
1290 // Only the lower element is undefined. The high elements are zero.
1291 UndefElts = UndefElts[0];
Craig Topper7fc6d342016-12-11 22:32:38 +00001292 break;
1293
Simon Pilgrim4c564ad2016-04-24 19:31:56 +00001294 // Unary scalar-as-vector operations that work column-wise.
Simon Pilgrim83020942016-04-24 18:23:14 +00001295 case Intrinsic::x86_sse_rcp_ss:
1296 case Intrinsic::x86_sse_rsqrt_ss:
1297 case Intrinsic::x86_sse_sqrt_ss:
1298 case Intrinsic::x86_sse2_sqrt_sd:
Simon Pilgrim83020942016-04-24 18:23:14 +00001299 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), DemandedElts,
1300 UndefElts, Depth + 1);
1301 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
1302
1303 // If lowest element of a scalar op isn't used then use Arg0.
Craig Topper1a8a3372016-12-29 03:30:17 +00001304 if (!DemandedElts[0]) {
1305 Worklist.Add(II);
Simon Pilgrim83020942016-04-24 18:23:14 +00001306 return II->getArgOperand(0);
Craig Topper1a8a3372016-12-29 03:30:17 +00001307 }
Simon Pilgrim4c564ad2016-04-24 19:31:56 +00001308 // TODO: If only low elt lower SQRT to FSQRT (with rounding/exceptions
1309 // checks).
Simon Pilgrim83020942016-04-24 18:23:14 +00001310 break;
1311
Craig Toppera0372de2016-12-14 03:17:27 +00001312 // Binary scalar-as-vector operations that work column-wise. The high
1313 // elements come from operand 0. The low element is a function of both
1314 // operands.
Chris Lattner7e044912010-01-04 07:17:19 +00001315 case Intrinsic::x86_sse_min_ss:
1316 case Intrinsic::x86_sse_max_ss:
Simon Pilgrim83020942016-04-24 18:23:14 +00001317 case Intrinsic::x86_sse_cmp_ss:
Chris Lattner7e044912010-01-04 07:17:19 +00001318 case Intrinsic::x86_sse2_min_sd:
1319 case Intrinsic::x86_sse2_max_sd:
Craig Toppera0372de2016-12-14 03:17:27 +00001320 case Intrinsic::x86_sse2_cmp_sd: {
1321 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), DemandedElts,
1322 UndefElts, Depth + 1);
1323 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
1324
1325 // If lowest element of a scalar op isn't used then use Arg0.
Craig Topper1a8a3372016-12-29 03:30:17 +00001326 if (!DemandedElts[0]) {
1327 Worklist.Add(II);
Craig Toppera0372de2016-12-14 03:17:27 +00001328 return II->getArgOperand(0);
Craig Topper1a8a3372016-12-29 03:30:17 +00001329 }
Craig Toppera0372de2016-12-14 03:17:27 +00001330
1331 // Only lower element is used for operand 1.
1332 DemandedElts = 1;
1333 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(1), DemandedElts,
1334 UndefElts2, Depth + 1);
1335 if (TmpV) { II->setArgOperand(1, TmpV); MadeChange = true; }
1336
1337 // Lower element is undefined if both lower elements are undefined.
1338 // Consider things like undef&0. The result is known zero, not undef.
1339 if (!UndefElts2[0])
1340 UndefElts.clearBit(0);
1341
1342 break;
1343 }
1344
Craig Toppereb6a20e2016-12-14 03:17:30 +00001345 // Binary scalar-as-vector operations that work column-wise. The high
1346 // elements come from operand 0 and the low element comes from operand 1.
Simon Pilgrim83020942016-04-24 18:23:14 +00001347 case Intrinsic::x86_sse41_round_ss:
Craig Toppereb6a20e2016-12-14 03:17:30 +00001348 case Intrinsic::x86_sse41_round_sd: {
1349 // Don't use the low element of operand 0.
1350 APInt DemandedElts2 = DemandedElts;
1351 DemandedElts2.clearBit(0);
1352 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), DemandedElts2,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001353 UndefElts, Depth + 1);
Gabor Greife23efee2010-06-28 16:45:00 +00001354 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
Craig Toppereb6a20e2016-12-14 03:17:30 +00001355
1356 // If lowest element of a scalar op isn't used then use Arg0.
Craig Topper1a8a3372016-12-29 03:30:17 +00001357 if (!DemandedElts[0]) {
1358 Worklist.Add(II);
Craig Toppereb6a20e2016-12-14 03:17:30 +00001359 return II->getArgOperand(0);
Craig Topper1a8a3372016-12-29 03:30:17 +00001360 }
Craig Toppereb6a20e2016-12-14 03:17:30 +00001361
1362 // Only lower element is used for operand 1.
1363 DemandedElts = 1;
Gabor Greife23efee2010-06-28 16:45:00 +00001364 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(1), DemandedElts,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001365 UndefElts2, Depth + 1);
Gabor Greife23efee2010-06-28 16:45:00 +00001366 if (TmpV) { II->setArgOperand(1, TmpV); MadeChange = true; }
Chris Lattner7e044912010-01-04 07:17:19 +00001367
Craig Toppereb6a20e2016-12-14 03:17:30 +00001368 // Take the high undef elements from operand 0 and take the lower element
1369 // from operand 1.
1370 UndefElts.clearBit(0);
1371 UndefElts |= UndefElts2[0];
Chris Lattner7e044912010-01-04 07:17:19 +00001372 break;
Craig Toppereb6a20e2016-12-14 03:17:30 +00001373 }
Simon Pilgrim61116dd2015-09-17 20:32:45 +00001374
Craig Topperdfd268d2016-12-14 05:43:05 +00001375 // Three input scalar-as-vector operations that work column-wise. The high
1376 // elements come from operand 0 and the low element is a function of all
1377 // three inputs.
Craig Topper268b3ab2016-12-14 06:06:58 +00001378 case Intrinsic::x86_avx512_mask_add_ss_round:
1379 case Intrinsic::x86_avx512_mask_div_ss_round:
1380 case Intrinsic::x86_avx512_mask_mul_ss_round:
1381 case Intrinsic::x86_avx512_mask_sub_ss_round:
1382 case Intrinsic::x86_avx512_mask_max_ss_round:
1383 case Intrinsic::x86_avx512_mask_min_ss_round:
1384 case Intrinsic::x86_avx512_mask_add_sd_round:
1385 case Intrinsic::x86_avx512_mask_div_sd_round:
1386 case Intrinsic::x86_avx512_mask_mul_sd_round:
1387 case Intrinsic::x86_avx512_mask_sub_sd_round:
1388 case Intrinsic::x86_avx512_mask_max_sd_round:
1389 case Intrinsic::x86_avx512_mask_min_sd_round:
Craig Topper23ebd952016-12-11 08:54:52 +00001390 case Intrinsic::x86_fma_vfmadd_ss:
1391 case Intrinsic::x86_fma_vfmsub_ss:
1392 case Intrinsic::x86_fma_vfnmadd_ss:
1393 case Intrinsic::x86_fma_vfnmsub_ss:
1394 case Intrinsic::x86_fma_vfmadd_sd:
1395 case Intrinsic::x86_fma_vfmsub_sd:
1396 case Intrinsic::x86_fma_vfnmadd_sd:
1397 case Intrinsic::x86_fma_vfnmsub_sd:
Craig Topperab5f3552016-12-15 03:49:45 +00001398 case Intrinsic::x86_avx512_mask_vfmadd_ss:
1399 case Intrinsic::x86_avx512_mask_vfmadd_sd:
1400 case Intrinsic::x86_avx512_maskz_vfmadd_ss:
1401 case Intrinsic::x86_avx512_maskz_vfmadd_sd:
Craig Topper23ebd952016-12-11 08:54:52 +00001402 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), DemandedElts,
1403 UndefElts, Depth + 1);
1404 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
Craig Topperdfd268d2016-12-14 05:43:05 +00001405
1406 // If lowest element of a scalar op isn't used then use Arg0.
Craig Topper1a8a3372016-12-29 03:30:17 +00001407 if (!DemandedElts[0]) {
1408 Worklist.Add(II);
Craig Topperdfd268d2016-12-14 05:43:05 +00001409 return II->getArgOperand(0);
Craig Topper1a8a3372016-12-29 03:30:17 +00001410 }
Craig Topperdfd268d2016-12-14 05:43:05 +00001411
1412 // Only lower element is used for operand 1 and 2.
1413 DemandedElts = 1;
Craig Topper23ebd952016-12-11 08:54:52 +00001414 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(1), DemandedElts,
1415 UndefElts2, Depth + 1);
1416 if (TmpV) { II->setArgOperand(1, TmpV); MadeChange = true; }
1417 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(2), DemandedElts,
1418 UndefElts3, Depth + 1);
1419 if (TmpV) { II->setArgOperand(2, TmpV); MadeChange = true; }
1420
Craig Topperdfd268d2016-12-14 05:43:05 +00001421 // Lower element is undefined if all three lower elements are undefined.
1422 // Consider things like undef&0. The result is known zero, not undef.
1423 if (!UndefElts2[0] || !UndefElts3[0])
1424 UndefElts.clearBit(0);
Craig Topper23ebd952016-12-11 08:54:52 +00001425
Craig Topper23ebd952016-12-11 08:54:52 +00001426 break;
1427
Craig Topperab5f3552016-12-15 03:49:45 +00001428 case Intrinsic::x86_avx512_mask3_vfmadd_ss:
1429 case Intrinsic::x86_avx512_mask3_vfmadd_sd:
1430 case Intrinsic::x86_avx512_mask3_vfmsub_ss:
1431 case Intrinsic::x86_avx512_mask3_vfmsub_sd:
1432 case Intrinsic::x86_avx512_mask3_vfnmsub_ss:
1433 case Intrinsic::x86_avx512_mask3_vfnmsub_sd:
1434 // These intrinsics get the passthru bits from operand 2.
1435 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(2), DemandedElts,
1436 UndefElts, Depth + 1);
1437 if (TmpV) { II->setArgOperand(2, TmpV); MadeChange = true; }
1438
1439 // If lowest element of a scalar op isn't used then use Arg2.
Craig Topper1a8a3372016-12-29 03:30:17 +00001440 if (!DemandedElts[0]) {
1441 Worklist.Add(II);
Craig Topperab5f3552016-12-15 03:49:45 +00001442 return II->getArgOperand(2);
Craig Topper1a8a3372016-12-29 03:30:17 +00001443 }
Craig Topperab5f3552016-12-15 03:49:45 +00001444
1445 // Only lower element is used for operand 0 and 1.
1446 DemandedElts = 1;
1447 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), DemandedElts,
1448 UndefElts2, Depth + 1);
1449 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
1450 TmpV = SimplifyDemandedVectorElts(II->getArgOperand(1), DemandedElts,
1451 UndefElts3, Depth + 1);
1452 if (TmpV) { II->setArgOperand(1, TmpV); MadeChange = true; }
1453
1454 // Lower element is undefined if all three lower elements are undefined.
1455 // Consider things like undef&0. The result is known zero, not undef.
1456 if (!UndefElts2[0] || !UndefElts3[0])
1457 UndefElts.clearBit(0);
1458
1459 break;
1460
Simon Pilgrimc9cf7fc2016-12-26 23:28:17 +00001461 case Intrinsic::x86_sse2_pmulu_dq:
1462 case Intrinsic::x86_sse41_pmuldq:
1463 case Intrinsic::x86_avx2_pmul_dq:
Craig Topper72f2d4e2016-12-27 05:30:09 +00001464 case Intrinsic::x86_avx2_pmulu_dq:
1465 case Intrinsic::x86_avx512_pmul_dq_512:
1466 case Intrinsic::x86_avx512_pmulu_dq_512: {
Simon Pilgrimc9cf7fc2016-12-26 23:28:17 +00001467 Value *Op0 = II->getArgOperand(0);
1468 Value *Op1 = II->getArgOperand(1);
1469 unsigned InnerVWidth = Op0->getType()->getVectorNumElements();
1470 assert((VWidth * 2) == InnerVWidth && "Unexpected input size");
1471
1472 APInt InnerDemandedElts(InnerVWidth, 0);
1473 for (unsigned i = 0; i != VWidth; ++i)
1474 if (DemandedElts[i])
1475 InnerDemandedElts.setBit(i * 2);
1476
1477 UndefElts2 = APInt(InnerVWidth, 0);
1478 TmpV = SimplifyDemandedVectorElts(Op0, InnerDemandedElts, UndefElts2,
1479 Depth + 1);
1480 if (TmpV) { II->setArgOperand(0, TmpV); MadeChange = true; }
1481
1482 UndefElts3 = APInt(InnerVWidth, 0);
1483 TmpV = SimplifyDemandedVectorElts(Op1, InnerDemandedElts, UndefElts3,
1484 Depth + 1);
1485 if (TmpV) { II->setArgOperand(1, TmpV); MadeChange = true; }
1486
1487 break;
1488 }
1489
Simon Pilgrim51b3b982017-01-20 09:28:21 +00001490 case Intrinsic::x86_sse2_packssdw_128:
1491 case Intrinsic::x86_sse2_packsswb_128:
1492 case Intrinsic::x86_sse2_packuswb_128:
1493 case Intrinsic::x86_sse41_packusdw:
1494 case Intrinsic::x86_avx2_packssdw:
1495 case Intrinsic::x86_avx2_packsswb:
1496 case Intrinsic::x86_avx2_packusdw:
Craig Topper3731f4d2017-02-16 07:35:23 +00001497 case Intrinsic::x86_avx2_packuswb:
1498 case Intrinsic::x86_avx512_packssdw_512:
1499 case Intrinsic::x86_avx512_packsswb_512:
1500 case Intrinsic::x86_avx512_packusdw_512:
1501 case Intrinsic::x86_avx512_packuswb_512: {
Simon Pilgrim51b3b982017-01-20 09:28:21 +00001502 auto *Ty0 = II->getArgOperand(0)->getType();
1503 unsigned InnerVWidth = Ty0->getVectorNumElements();
1504 assert(VWidth == (InnerVWidth * 2) && "Unexpected input size");
1505
1506 unsigned NumLanes = Ty0->getPrimitiveSizeInBits() / 128;
1507 unsigned VWidthPerLane = VWidth / NumLanes;
1508 unsigned InnerVWidthPerLane = InnerVWidth / NumLanes;
1509
1510 // Per lane, pack the elements of the first input and then the second.
1511 // e.g.
1512 // v8i16 PACK(v4i32 X, v4i32 Y) - (X[0..3],Y[0..3])
1513 // v32i8 PACK(v16i16 X, v16i16 Y) - (X[0..7],Y[0..7]),(X[8..15],Y[8..15])
1514 for (int OpNum = 0; OpNum != 2; ++OpNum) {
1515 APInt OpDemandedElts(InnerVWidth, 0);
1516 for (unsigned Lane = 0; Lane != NumLanes; ++Lane) {
1517 unsigned LaneIdx = Lane * VWidthPerLane;
1518 for (unsigned Elt = 0; Elt != InnerVWidthPerLane; ++Elt) {
1519 unsigned Idx = LaneIdx + Elt + InnerVWidthPerLane * OpNum;
1520 if (DemandedElts[Idx])
1521 OpDemandedElts.setBit((Lane * InnerVWidthPerLane) + Elt);
1522 }
1523 }
1524
1525 // Demand elements from the operand.
1526 auto *Op = II->getArgOperand(OpNum);
1527 APInt OpUndefElts(InnerVWidth, 0);
1528 TmpV = SimplifyDemandedVectorElts(Op, OpDemandedElts, OpUndefElts,
1529 Depth + 1);
1530 if (TmpV) {
1531 II->setArgOperand(OpNum, TmpV);
1532 MadeChange = true;
1533 }
1534
1535 // Pack the operand's UNDEF elements, one lane at a time.
1536 OpUndefElts = OpUndefElts.zext(VWidth);
1537 for (unsigned Lane = 0; Lane != NumLanes; ++Lane) {
1538 APInt LaneElts = OpUndefElts.lshr(InnerVWidthPerLane * Lane);
1539 LaneElts = LaneElts.getLoBits(InnerVWidthPerLane);
Craig Topper24e71012017-04-28 03:36:24 +00001540 LaneElts <<= InnerVWidthPerLane * (2 * Lane + OpNum);
Simon Pilgrim51b3b982017-01-20 09:28:21 +00001541 UndefElts |= LaneElts;
1542 }
1543 }
1544 break;
1545 }
1546
Simon Pilgrimd4eb8002017-01-17 11:35:03 +00001547 // PSHUFB
Simon Pilgrim73a68c22017-01-16 11:30:41 +00001548 case Intrinsic::x86_ssse3_pshuf_b_128:
1549 case Intrinsic::x86_avx2_pshuf_b:
Simon Pilgrimd4eb8002017-01-17 11:35:03 +00001550 case Intrinsic::x86_avx512_pshuf_b_512:
1551 // PERMILVAR
1552 case Intrinsic::x86_avx_vpermilvar_ps:
1553 case Intrinsic::x86_avx_vpermilvar_ps_256:
1554 case Intrinsic::x86_avx512_vpermilvar_ps_512:
1555 case Intrinsic::x86_avx_vpermilvar_pd:
1556 case Intrinsic::x86_avx_vpermilvar_pd_256:
Simon Pilgrimfe2c0ed2017-01-18 14:47:49 +00001557 case Intrinsic::x86_avx512_vpermilvar_pd_512:
1558 // PERMV
1559 case Intrinsic::x86_avx2_permd:
1560 case Intrinsic::x86_avx2_permps: {
Simon Pilgrim73a68c22017-01-16 11:30:41 +00001561 Value *Op1 = II->getArgOperand(1);
1562 TmpV = SimplifyDemandedVectorElts(Op1, DemandedElts, UndefElts,
1563 Depth + 1);
1564 if (TmpV) { II->setArgOperand(1, TmpV); MadeChange = true; }
1565 break;
1566 }
1567
Simon Pilgrim61116dd2015-09-17 20:32:45 +00001568 // SSE4A instructions leave the upper 64-bits of the 128-bit result
1569 // in an undefined state.
1570 case Intrinsic::x86_sse4a_extrq:
1571 case Intrinsic::x86_sse4a_extrqi:
1572 case Intrinsic::x86_sse4a_insertq:
1573 case Intrinsic::x86_sse4a_insertqi:
Craig Topper3a86a042017-03-19 05:49:16 +00001574 UndefElts.setHighBits(VWidth / 2);
Simon Pilgrim61116dd2015-09-17 20:32:45 +00001575 break;
Matt Arsenaultefe949c2017-03-09 20:34:27 +00001576 case Intrinsic::amdgcn_buffer_load:
Matt Arsenault7205f3c2017-04-17 15:12:44 +00001577 case Intrinsic::amdgcn_buffer_load_format:
1578 case Intrinsic::amdgcn_image_sample:
1579 case Intrinsic::amdgcn_image_sample_cl:
1580 case Intrinsic::amdgcn_image_sample_d:
1581 case Intrinsic::amdgcn_image_sample_d_cl:
1582 case Intrinsic::amdgcn_image_sample_l:
1583 case Intrinsic::amdgcn_image_sample_b:
1584 case Intrinsic::amdgcn_image_sample_b_cl:
1585 case Intrinsic::amdgcn_image_sample_lz:
1586 case Intrinsic::amdgcn_image_sample_cd:
1587 case Intrinsic::amdgcn_image_sample_cd_cl:
1588
1589 case Intrinsic::amdgcn_image_sample_c:
1590 case Intrinsic::amdgcn_image_sample_c_cl:
1591 case Intrinsic::amdgcn_image_sample_c_d:
1592 case Intrinsic::amdgcn_image_sample_c_d_cl:
1593 case Intrinsic::amdgcn_image_sample_c_l:
1594 case Intrinsic::amdgcn_image_sample_c_b:
1595 case Intrinsic::amdgcn_image_sample_c_b_cl:
1596 case Intrinsic::amdgcn_image_sample_c_lz:
1597 case Intrinsic::amdgcn_image_sample_c_cd:
1598 case Intrinsic::amdgcn_image_sample_c_cd_cl:
1599
1600 case Intrinsic::amdgcn_image_sample_o:
1601 case Intrinsic::amdgcn_image_sample_cl_o:
1602 case Intrinsic::amdgcn_image_sample_d_o:
1603 case Intrinsic::amdgcn_image_sample_d_cl_o:
1604 case Intrinsic::amdgcn_image_sample_l_o:
1605 case Intrinsic::amdgcn_image_sample_b_o:
1606 case Intrinsic::amdgcn_image_sample_b_cl_o:
1607 case Intrinsic::amdgcn_image_sample_lz_o:
1608 case Intrinsic::amdgcn_image_sample_cd_o:
1609 case Intrinsic::amdgcn_image_sample_cd_cl_o:
1610
1611 case Intrinsic::amdgcn_image_sample_c_o:
1612 case Intrinsic::amdgcn_image_sample_c_cl_o:
1613 case Intrinsic::amdgcn_image_sample_c_d_o:
1614 case Intrinsic::amdgcn_image_sample_c_d_cl_o:
1615 case Intrinsic::amdgcn_image_sample_c_l_o:
1616 case Intrinsic::amdgcn_image_sample_c_b_o:
1617 case Intrinsic::amdgcn_image_sample_c_b_cl_o:
1618 case Intrinsic::amdgcn_image_sample_c_lz_o:
1619 case Intrinsic::amdgcn_image_sample_c_cd_o:
1620 case Intrinsic::amdgcn_image_sample_c_cd_cl_o:
1621
1622 case Intrinsic::amdgcn_image_getlod: {
Craig Topperd33ee1b2017-04-03 16:34:59 +00001623 if (VWidth == 1 || !DemandedElts.isMask())
Matt Arsenaultefe949c2017-03-09 20:34:27 +00001624 return nullptr;
1625
1626 // TODO: Handle 3 vectors when supported in code gen.
1627 unsigned NewNumElts = PowerOf2Ceil(DemandedElts.countTrailingOnes());
1628 if (NewNumElts == VWidth)
1629 return nullptr;
1630
1631 Module *M = II->getParent()->getParent()->getParent();
1632 Type *EltTy = V->getType()->getVectorElementType();
1633
1634 Type *NewTy = (NewNumElts == 1) ? EltTy :
1635 VectorType::get(EltTy, NewNumElts);
1636
Matt Arsenault7205f3c2017-04-17 15:12:44 +00001637 auto IID = II->getIntrinsicID();
1638
1639 bool IsBuffer = IID == Intrinsic::amdgcn_buffer_load ||
1640 IID == Intrinsic::amdgcn_buffer_load_format;
1641
1642 Function *NewIntrin = IsBuffer ?
1643 Intrinsic::getDeclaration(M, IID, NewTy) :
1644 // Samplers have 3 mangled types.
1645 Intrinsic::getDeclaration(M, IID,
1646 { NewTy, II->getArgOperand(0)->getType(),
1647 II->getArgOperand(1)->getType()});
Matt Arsenaultefe949c2017-03-09 20:34:27 +00001648
1649 SmallVector<Value *, 5> Args;
1650 for (unsigned I = 0, E = II->getNumArgOperands(); I != E; ++I)
1651 Args.push_back(II->getArgOperand(I));
1652
Matt Arsenaulta3bdd8f2017-03-10 05:25:49 +00001653 IRBuilderBase::InsertPointGuard Guard(*Builder);
1654 Builder->SetInsertPoint(II);
1655
Matt Arsenaultefe949c2017-03-09 20:34:27 +00001656 CallInst *NewCall = Builder->CreateCall(NewIntrin, Args);
1657 NewCall->takeName(II);
1658 NewCall->copyMetadata(*II);
Matt Arsenault7205f3c2017-04-17 15:12:44 +00001659
1660 if (!IsBuffer) {
1661 ConstantInt *DMask = dyn_cast<ConstantInt>(NewCall->getArgOperand(3));
1662 if (DMask) {
1663 unsigned DMaskVal = DMask->getZExtValue() & 0xf;
1664
1665 unsigned PopCnt = 0;
1666 unsigned NewDMask = 0;
1667 for (unsigned I = 0; I < 4; ++I) {
1668 const unsigned Bit = 1 << I;
1669 if (!!(DMaskVal & Bit)) {
1670 if (++PopCnt > NewNumElts)
1671 break;
1672
1673 NewDMask |= Bit;
1674 }
1675 }
1676
1677 NewCall->setArgOperand(3, ConstantInt::get(DMask->getType(), NewDMask));
1678 }
1679 }
1680
1681
Matt Arsenaultefe949c2017-03-09 20:34:27 +00001682 if (NewNumElts == 1) {
1683 return Builder->CreateInsertElement(UndefValue::get(V->getType()),
1684 NewCall, static_cast<uint64_t>(0));
1685 }
1686
1687 SmallVector<uint32_t, 8> EltMask;
1688 for (unsigned I = 0; I < VWidth; ++I)
1689 EltMask.push_back(I);
1690
1691 Value *Shuffle = Builder->CreateShuffleVector(
1692 NewCall, UndefValue::get(NewTy), EltMask);
1693
1694 MadeChange = true;
1695 return Shuffle;
1696 }
Chris Lattner7e044912010-01-04 07:17:19 +00001697 }
1698 break;
1699 }
1700 }
Craig Topperf40110f2014-04-25 05:29:35 +00001701 return MadeChange ? I : nullptr;
Chris Lattner7e044912010-01-04 07:17:19 +00001702}