blob: 92b0555913758ff32837ba385c612686ef96f1f2 [file] [log] [blame]
Dan Gohman91d598d2009-09-10 23:07:18 +00001//===-- ConstantFolding.cpp - Fold instructions into constants ------------===//
John Criswell970af112005-10-27 16:00:10 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
John Criswell970af112005-10-27 16:00:10 +00007//
8//===----------------------------------------------------------------------===//
9//
Dan Gohman91d598d2009-09-10 23:07:18 +000010// This file defines routines for folding instructions into constants.
11//
Chandler Carruthef860a22013-01-02 09:10:48 +000012// Also, to supplement the basic IR ConstantExpr simplifications,
Dan Gohman91d598d2009-09-10 23:07:18 +000013// this file defines some additional folding routines that can make use of
Chandler Carruthef860a22013-01-02 09:10:48 +000014// DataLayout information. These functions cannot go in IR due to library
Dan Gohman91d598d2009-09-10 23:07:18 +000015// dependency issues.
John Criswell970af112005-10-27 16:00:10 +000016//
17//===----------------------------------------------------------------------===//
18
19#include "llvm/Analysis/ConstantFolding.h"
Eugene Zelenko1804a772016-08-25 00:45:04 +000020#include "llvm/ADT/APFloat.h"
21#include "llvm/ADT/APInt.h"
22#include "llvm/ADT/ArrayRef.h"
23#include "llvm/ADT/DenseMap.h"
David Majnemere61e4bf2016-06-21 05:10:24 +000024#include "llvm/ADT/STLExtras.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000025#include "llvm/ADT/SmallVector.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +000026#include "llvm/ADT/StringRef.h"
Chandler Carruth62d42152015-01-15 02:16:27 +000027#include "llvm/Analysis/TargetLibraryInfo.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000028#include "llvm/Analysis/ValueTracking.h"
Alp Tokerc817d6a2014-06-09 18:28:53 +000029#include "llvm/Config/config.h"
Eugene Zelenko1804a772016-08-25 00:45:04 +000030#include "llvm/IR/Constant.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000031#include "llvm/IR/Constants.h"
32#include "llvm/IR/DataLayout.h"
33#include "llvm/IR/DerivedTypes.h"
34#include "llvm/IR/Function.h"
Eugene Zelenko1804a772016-08-25 00:45:04 +000035#include "llvm/IR/GlobalValue.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000036#include "llvm/IR/GlobalVariable.h"
Eugene Zelenko1804a772016-08-25 00:45:04 +000037#include "llvm/IR/InstrTypes.h"
38#include "llvm/IR/Instruction.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000039#include "llvm/IR/Instructions.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000040#include "llvm/IR/Operator.h"
Eugene Zelenko1804a772016-08-25 00:45:04 +000041#include "llvm/IR/Type.h"
42#include "llvm/IR/Value.h"
43#include "llvm/Support/Casting.h"
Torok Edwin56d06592009-07-11 20:10:48 +000044#include "llvm/Support/ErrorHandling.h"
Craig Topperb45eabc2017-04-26 16:39:58 +000045#include "llvm/Support/KnownBits.h"
John Criswell970af112005-10-27 16:00:10 +000046#include "llvm/Support/MathExtras.h"
Eugene Zelenko35623fb2016-03-28 17:40:08 +000047#include <cassert>
John Criswell970af112005-10-27 16:00:10 +000048#include <cerrno>
Eugene Zelenko35623fb2016-03-28 17:40:08 +000049#include <cfenv>
Jeff Cohencc08c832006-12-02 02:22:01 +000050#include <cmath>
Eugene Zelenko1804a772016-08-25 00:45:04 +000051#include <cstddef>
52#include <cstdint>
Alp Tokerc817d6a2014-06-09 18:28:53 +000053
John Criswell970af112005-10-27 16:00:10 +000054using namespace llvm;
55
Eugene Zelenko35623fb2016-03-28 17:40:08 +000056namespace {
57
Chris Lattner44d68b92007-01-31 00:51:48 +000058//===----------------------------------------------------------------------===//
59// Constant Folding internal helper functions
60//===----------------------------------------------------------------------===//
61
Matt Arsenault47a4b392016-12-02 02:26:02 +000062static Constant *foldConstVectorToAPInt(APInt &Result, Type *DestTy,
63 Constant *C, Type *SrcEltTy,
64 unsigned NumSrcElts,
65 const DataLayout &DL) {
66 // Now that we know that the input value is a vector of integers, just shift
67 // and insert them into our result.
68 unsigned BitShift = DL.getTypeSizeInBits(SrcEltTy);
69 for (unsigned i = 0; i != NumSrcElts; ++i) {
70 Constant *Element;
71 if (DL.isLittleEndian())
72 Element = C->getAggregateElement(NumSrcElts - i - 1);
73 else
74 Element = C->getAggregateElement(i);
75
76 if (Element && isa<UndefValue>(Element)) {
77 Result <<= BitShift;
78 continue;
79 }
80
81 auto *ElementCI = dyn_cast_or_null<ConstantInt>(Element);
82 if (!ElementCI)
83 return ConstantExpr::getBitCast(C, DestTy);
84
85 Result <<= BitShift;
86 Result |= ElementCI->getValue().zextOrSelf(Result.getBitWidth());
87 }
88
89 return nullptr;
90}
91
Matt Arsenault624e1b32016-12-07 20:56:11 +000092/// Constant fold bitcast, symbolically evaluating it with DataLayout.
Sanjay Patel0d7dee62014-10-02 15:13:22 +000093/// This always returns a non-null constant, but it may be a
Chris Lattner9d051242009-10-25 06:08:26 +000094/// ConstantExpr if unfoldable.
Eugene Zelenko35623fb2016-03-28 17:40:08 +000095Constant *FoldBitCast(Constant *C, Type *DestTy, const DataLayout &DL) {
Nadav Rotem365af6f2011-08-24 20:18:38 +000096 // Catch the obvious splat cases.
97 if (C->isNullValue() && !DestTy->isX86_MMXTy())
98 return Constant::getNullValue(DestTy);
Bruno Cardoso Lopesc29520c2014-10-22 12:18:48 +000099 if (C->isAllOnesValue() && !DestTy->isX86_MMXTy() &&
100 !DestTy->isPtrOrPtrVectorTy()) // Don't get ones for ptr types!
Nadav Rotem365af6f2011-08-24 20:18:38 +0000101 return Constant::getAllOnesValue(DestTy);
102
Matt Arsenault624e1b32016-12-07 20:56:11 +0000103 if (auto *VTy = dyn_cast<VectorType>(C->getType())) {
104 // Handle a vector->scalar integer/fp cast.
105 if (isa<IntegerType>(DestTy) || DestTy->isFloatingPointTy()) {
106 unsigned NumSrcElts = VTy->getNumElements();
107 Type *SrcEltTy = VTy->getElementType();
Rafael Espindolabb893fe2012-01-27 23:33:07 +0000108
Matt Arsenault624e1b32016-12-07 20:56:11 +0000109 // If the vector is a vector of floating point, convert it to vector of int
110 // to simplify things.
111 if (SrcEltTy->isFloatingPointTy()) {
112 unsigned FPWidth = SrcEltTy->getPrimitiveSizeInBits();
113 Type *SrcIVTy =
114 VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumSrcElts);
115 // Ask IR to do the conversion now that #elts line up.
116 C = ConstantExpr::getBitCast(C, SrcIVTy);
117 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000118
Matt Arsenault624e1b32016-12-07 20:56:11 +0000119 APInt Result(DL.getTypeSizeInBits(DestTy), 0);
120 if (Constant *CE = foldConstVectorToAPInt(Result, DestTy, C,
121 SrcEltTy, NumSrcElts, DL))
122 return CE;
123
124 if (isa<IntegerType>(DestTy))
125 return ConstantInt::get(DestTy, Result);
126
127 APFloat FP(DestTy->getFltSemantics(), Result);
128 return ConstantFP::get(DestTy->getContext(), FP);
Rafael Espindolabb893fe2012-01-27 23:33:07 +0000129 }
Rafael Espindolabb893fe2012-01-27 23:33:07 +0000130 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000131
Nadav Rotemad4a70a2011-08-20 14:02:29 +0000132 // The code below only handles casts to vectors currently.
David Majnemer90a97042016-07-13 04:22:12 +0000133 auto *DestVTy = dyn_cast<VectorType>(DestTy);
Craig Topper9f008862014-04-15 04:59:12 +0000134 if (!DestVTy)
Chris Lattnerd8e8fb42009-10-25 06:15:37 +0000135 return ConstantExpr::getBitCast(C, DestTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000136
Chris Lattnerd8e8fb42009-10-25 06:15:37 +0000137 // If this is a scalar -> vector cast, convert the input into a <1 x scalar>
138 // vector so the code below can handle it uniformly.
139 if (isa<ConstantFP>(C) || isa<ConstantInt>(C)) {
140 Constant *Ops = C; // don't take the address of C!
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000141 return FoldBitCast(ConstantVector::get(Ops), DestTy, DL);
Chris Lattnerd8e8fb42009-10-25 06:15:37 +0000142 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000143
Chris Lattner9d051242009-10-25 06:08:26 +0000144 // If this is a bitcast from constant vector -> vector, fold it.
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000145 if (!isa<ConstantDataVector>(C) && !isa<ConstantVector>(C))
Chris Lattner9d051242009-10-25 06:08:26 +0000146 return ConstantExpr::getBitCast(C, DestTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000147
Chandler Carruthef860a22013-01-02 09:10:48 +0000148 // If the element types match, IR can fold it.
Chris Lattner9d051242009-10-25 06:08:26 +0000149 unsigned NumDstElt = DestVTy->getNumElements();
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000150 unsigned NumSrcElt = C->getType()->getVectorNumElements();
Chris Lattner9d051242009-10-25 06:08:26 +0000151 if (NumDstElt == NumSrcElt)
152 return ConstantExpr::getBitCast(C, DestTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000153
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000154 Type *SrcEltTy = C->getType()->getVectorElementType();
Chris Lattner229907c2011-07-18 04:54:35 +0000155 Type *DstEltTy = DestVTy->getElementType();
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000156
157 // Otherwise, we're changing the number of elements in a vector, which
Chris Lattner9d051242009-10-25 06:08:26 +0000158 // requires endianness information to do the right thing. For example,
159 // bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>)
160 // folds to (little endian):
161 // <4 x i32> <i32 0, i32 0, i32 1, i32 0>
162 // and to (big endian):
163 // <4 x i32> <i32 0, i32 0, i32 0, i32 1>
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000164
Chris Lattner9d051242009-10-25 06:08:26 +0000165 // First thing is first. We only want to think about integer here, so if
166 // we have something in FP form, recast it as integer.
Duncan Sands9dff9be2010-02-15 16:12:20 +0000167 if (DstEltTy->isFloatingPointTy()) {
Chris Lattner9d051242009-10-25 06:08:26 +0000168 // Fold to an vector of integers with same size as our FP type.
169 unsigned FPWidth = DstEltTy->getPrimitiveSizeInBits();
Chris Lattner229907c2011-07-18 04:54:35 +0000170 Type *DestIVTy =
Chris Lattner9d051242009-10-25 06:08:26 +0000171 VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumDstElt);
172 // Recursively handle this integer conversion, if possible.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000173 C = FoldBitCast(C, DestIVTy, DL);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000174
Chandler Carruthef860a22013-01-02 09:10:48 +0000175 // Finally, IR can handle this now that #elts line up.
Chris Lattner9d051242009-10-25 06:08:26 +0000176 return ConstantExpr::getBitCast(C, DestTy);
177 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000178
Chris Lattner9d051242009-10-25 06:08:26 +0000179 // Okay, we know the destination is integer, if the input is FP, convert
180 // it to integer first.
Duncan Sands9dff9be2010-02-15 16:12:20 +0000181 if (SrcEltTy->isFloatingPointTy()) {
Chris Lattner9d051242009-10-25 06:08:26 +0000182 unsigned FPWidth = SrcEltTy->getPrimitiveSizeInBits();
Chris Lattner229907c2011-07-18 04:54:35 +0000183 Type *SrcIVTy =
Chris Lattner9d051242009-10-25 06:08:26 +0000184 VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumSrcElt);
Chandler Carruthef860a22013-01-02 09:10:48 +0000185 // Ask IR to do the conversion now that #elts line up.
Chris Lattner9d051242009-10-25 06:08:26 +0000186 C = ConstantExpr::getBitCast(C, SrcIVTy);
Chandler Carruthef860a22013-01-02 09:10:48 +0000187 // If IR wasn't able to fold it, bail out.
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000188 if (!isa<ConstantVector>(C) && // FIXME: Remove ConstantVector.
189 !isa<ConstantDataVector>(C))
Chris Lattner9d051242009-10-25 06:08:26 +0000190 return C;
191 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000192
Chris Lattner9d051242009-10-25 06:08:26 +0000193 // Now we know that the input and output vectors are both integer vectors
194 // of the same size, and that their #elements is not the same. Do the
195 // conversion here, which depends on whether the input or output has
196 // more elements.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000197 bool isLittleEndian = DL.isLittleEndian();
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000198
Chris Lattner9d051242009-10-25 06:08:26 +0000199 SmallVector<Constant*, 32> Result;
200 if (NumDstElt < NumSrcElt) {
201 // Handle: bitcast (<4 x i32> <i32 0, i32 1, i32 2, i32 3> to <2 x i64>)
202 Constant *Zero = Constant::getNullValue(DstEltTy);
203 unsigned Ratio = NumSrcElt/NumDstElt;
204 unsigned SrcBitSize = SrcEltTy->getPrimitiveSizeInBits();
205 unsigned SrcElt = 0;
206 for (unsigned i = 0; i != NumDstElt; ++i) {
207 // Build each element of the result.
208 Constant *Elt = Zero;
209 unsigned ShiftAmt = isLittleEndian ? 0 : SrcBitSize*(Ratio-1);
210 for (unsigned j = 0; j != Ratio; ++j) {
David Majnemere4218cf2016-07-29 04:06:09 +0000211 Constant *Src = C->getAggregateElement(SrcElt++);
212 if (Src && isa<UndefValue>(Src))
David Majnemer718da3d2016-07-29 18:48:27 +0000213 Src = Constant::getNullValue(C->getType()->getVectorElementType());
David Majnemere4218cf2016-07-29 04:06:09 +0000214 else
215 Src = dyn_cast_or_null<ConstantInt>(Src);
Chris Lattner9d051242009-10-25 06:08:26 +0000216 if (!Src) // Reject constantexpr elements.
217 return ConstantExpr::getBitCast(C, DestTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000218
Chris Lattner9d051242009-10-25 06:08:26 +0000219 // Zero extend the element to the right size.
220 Src = ConstantExpr::getZExt(Src, Elt->getType());
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000221
Chris Lattner9d051242009-10-25 06:08:26 +0000222 // Shift it to the right place, depending on endianness.
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000223 Src = ConstantExpr::getShl(Src,
Chris Lattner9d051242009-10-25 06:08:26 +0000224 ConstantInt::get(Src->getType(), ShiftAmt));
225 ShiftAmt += isLittleEndian ? SrcBitSize : -SrcBitSize;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000226
Chris Lattner9d051242009-10-25 06:08:26 +0000227 // Mix it in.
228 Elt = ConstantExpr::getOr(Elt, Src);
229 }
230 Result.push_back(Elt);
231 }
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000232 return ConstantVector::get(Result);
233 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000234
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000235 // Handle: bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>)
236 unsigned Ratio = NumDstElt/NumSrcElt;
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000237 unsigned DstBitSize = DL.getTypeSizeInBits(DstEltTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000238
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000239 // Loop over each source value, expanding into multiple results.
240 for (unsigned i = 0; i != NumSrcElt; ++i) {
Andrea Di Biagio7277afe2016-09-13 14:50:47 +0000241 auto *Element = C->getAggregateElement(i);
242
243 if (!Element) // Reject constantexpr elements.
244 return ConstantExpr::getBitCast(C, DestTy);
245
246 if (isa<UndefValue>(Element)) {
247 // Correctly Propagate undef values.
248 Result.append(Ratio, UndefValue::get(DstEltTy));
249 continue;
250 }
251
252 auto *Src = dyn_cast<ConstantInt>(Element);
253 if (!Src)
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000254 return ConstantExpr::getBitCast(C, DestTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000255
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000256 unsigned ShiftAmt = isLittleEndian ? 0 : DstBitSize*(Ratio-1);
257 for (unsigned j = 0; j != Ratio; ++j) {
258 // Shift the piece of the value into the right place, depending on
259 // endianness.
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000260 Constant *Elt = ConstantExpr::getLShr(Src,
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000261 ConstantInt::get(Src->getType(), ShiftAmt));
262 ShiftAmt += isLittleEndian ? DstBitSize : -DstBitSize;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000263
Bruno Cardoso Lopesc29520c2014-10-22 12:18:48 +0000264 // Truncate the element to an integer with the same pointer size and
265 // convert the element back to a pointer using a inttoptr.
266 if (DstEltTy->isPointerTy()) {
267 IntegerType *DstIntTy = Type::getIntNTy(C->getContext(), DstBitSize);
268 Constant *CE = ConstantExpr::getTrunc(Elt, DstIntTy);
269 Result.push_back(ConstantExpr::getIntToPtr(CE, DstEltTy));
270 continue;
271 }
272
Chris Lattner61a1d6c2012-01-26 21:37:55 +0000273 // Truncate and remember this piece.
274 Result.push_back(ConstantExpr::getTrunc(Elt, DstEltTy));
Chris Lattner9d051242009-10-25 06:08:26 +0000275 }
276 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000277
Chris Lattner69229312011-02-15 00:14:00 +0000278 return ConstantVector::get(Result);
Chris Lattner9d051242009-10-25 06:08:26 +0000279}
280
Peter Collingbourne265ebd72016-04-22 20:40:10 +0000281} // end anonymous namespace
282
Sanjay Patel0d7dee62014-10-02 15:13:22 +0000283/// If this constant is a constant offset from a global, return the global and
284/// the constant. Because of constantexprs, this function is recursive.
Peter Collingbourne265ebd72016-04-22 20:40:10 +0000285bool llvm::IsConstantOffsetFromGlobal(Constant *C, GlobalValue *&GV,
286 APInt &Offset, const DataLayout &DL) {
Chris Lattner44d68b92007-01-31 00:51:48 +0000287 // Trivial case, constant is the global.
288 if ((GV = dyn_cast<GlobalValue>(C))) {
Elena Demikhovsky945b7e52018-02-14 06:58:08 +0000289 unsigned BitWidth = DL.getIndexTypeSizeInBits(GV->getType());
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000290 Offset = APInt(BitWidth, 0);
Chris Lattner44d68b92007-01-31 00:51:48 +0000291 return true;
292 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000293
Chris Lattner44d68b92007-01-31 00:51:48 +0000294 // Otherwise, if this isn't a constant expr, bail out.
David Majnemer90a97042016-07-13 04:22:12 +0000295 auto *CE = dyn_cast<ConstantExpr>(C);
Chris Lattner44d68b92007-01-31 00:51:48 +0000296 if (!CE) return false;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000297
Chris Lattner44d68b92007-01-31 00:51:48 +0000298 // Look through ptr->int and ptr->ptr casts.
299 if (CE->getOpcode() == Instruction::PtrToInt ||
Matt Arsenault95365ca2015-07-27 18:31:03 +0000300 CE->getOpcode() == Instruction::BitCast)
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000301 return IsConstantOffsetFromGlobal(CE->getOperand(0), GV, Offset, DL);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000302
303 // i32* getelementptr ([5 x i32]* @a, i32 0, i32 5)
David Majnemer90a97042016-07-13 04:22:12 +0000304 auto *GEP = dyn_cast<GEPOperator>(CE);
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000305 if (!GEP)
306 return false;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000307
Elena Demikhovsky945b7e52018-02-14 06:58:08 +0000308 unsigned BitWidth = DL.getIndexTypeSizeInBits(GEP->getType());
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000309 APInt TmpOffset(BitWidth, 0);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000310
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000311 // If the base isn't a global+constant, we aren't either.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000312 if (!IsConstantOffsetFromGlobal(CE->getOperand(0), GV, TmpOffset, DL))
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000313 return false;
314
315 // Otherwise, add any offset that our operands provide.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000316 if (!GEP->accumulateConstantOffset(DL, TmpOffset))
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000317 return false;
318
319 Offset = TmpOffset;
320 return true;
Chris Lattner44d68b92007-01-31 00:51:48 +0000321}
322
Eugene Leviant6f42a2c2018-03-13 10:19:50 +0000323Constant *llvm::ConstantFoldLoadThroughBitcast(Constant *C, Type *DestTy,
324 const DataLayout &DL) {
325 do {
326 Type *SrcTy = C->getType();
327
328 // If the type sizes are the same and a cast is legal, just directly
329 // cast the constant.
330 if (DL.getTypeSizeInBits(DestTy) == DL.getTypeSizeInBits(SrcTy)) {
331 Instruction::CastOps Cast = Instruction::BitCast;
332 // If we are going from a pointer to int or vice versa, we spell the cast
333 // differently.
334 if (SrcTy->isIntegerTy() && DestTy->isPointerTy())
335 Cast = Instruction::IntToPtr;
336 else if (SrcTy->isPointerTy() && DestTy->isIntegerTy())
337 Cast = Instruction::PtrToInt;
338
339 if (CastInst::castIsValid(Cast, C, DestTy))
340 return ConstantExpr::getCast(Cast, C, DestTy);
341 }
342
343 // If this isn't an aggregate type, there is nothing we can do to drill down
344 // and find a bitcastable constant.
345 if (!SrcTy->isAggregateType())
346 return nullptr;
347
348 // We're simulating a load through a pointer that was bitcast to point to
349 // a different type, so we can try to walk down through the initial
350 // elements of an aggregate to see if some part of th e aggregate is
351 // castable to implement the "load" semantic model.
352 C = C->getAggregateElement(0u);
353 } while (C);
354
355 return nullptr;
356}
357
Peter Collingbourne265ebd72016-04-22 20:40:10 +0000358namespace {
359
Sanjay Patel0d7dee62014-10-02 15:13:22 +0000360/// Recursive helper to read bits out of global. C is the constant being copied
361/// out of. ByteOffset is an offset into C. CurPtr is the pointer to copy
362/// results into and BytesLeft is the number of bytes left in
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000363/// the CurPtr buffer. DL is the DataLayout.
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000364bool ReadDataFromGlobal(Constant *C, uint64_t ByteOffset, unsigned char *CurPtr,
365 unsigned BytesLeft, const DataLayout &DL) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000366 assert(ByteOffset <= DL.getTypeAllocSize(C->getType()) &&
Chris Lattnered00b802009-10-23 06:23:49 +0000367 "Out of range access");
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000368
Chris Lattner3db7bd22009-10-24 05:27:19 +0000369 // If this element is zero or undefined, we can just return since *CurPtr is
370 // zero initialized.
Chris Lattnered00b802009-10-23 06:23:49 +0000371 if (isa<ConstantAggregateZero>(C) || isa<UndefValue>(C))
372 return true;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000373
David Majnemer90a97042016-07-13 04:22:12 +0000374 if (auto *CI = dyn_cast<ConstantInt>(C)) {
Chris Lattnered00b802009-10-23 06:23:49 +0000375 if (CI->getBitWidth() > 64 ||
376 (CI->getBitWidth() & 7) != 0)
377 return false;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000378
Chris Lattnered00b802009-10-23 06:23:49 +0000379 uint64_t Val = CI->getZExtValue();
380 unsigned IntBytes = unsigned(CI->getBitWidth()/8);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000381
Chris Lattnered00b802009-10-23 06:23:49 +0000382 for (unsigned i = 0; i != BytesLeft && ByteOffset != IntBytes; ++i) {
NAKAMURA Takumi43ab4ef2012-11-08 20:34:25 +0000383 int n = ByteOffset;
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000384 if (!DL.isLittleEndian())
NAKAMURA Takumi43ab4ef2012-11-08 20:34:25 +0000385 n = IntBytes - n - 1;
386 CurPtr[i] = (unsigned char)(Val >> (n * 8));
Chris Lattnered00b802009-10-23 06:23:49 +0000387 ++ByteOffset;
388 }
389 return true;
390 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000391
David Majnemer90a97042016-07-13 04:22:12 +0000392 if (auto *CFP = dyn_cast<ConstantFP>(C)) {
Chris Lattnered00b802009-10-23 06:23:49 +0000393 if (CFP->getType()->isDoubleTy()) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000394 C = FoldBitCast(C, Type::getInt64Ty(C->getContext()), DL);
395 return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, DL);
Chris Lattnered00b802009-10-23 06:23:49 +0000396 }
397 if (CFP->getType()->isFloatTy()){
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000398 C = FoldBitCast(C, Type::getInt32Ty(C->getContext()), DL);
399 return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, DL);
Chris Lattnered00b802009-10-23 06:23:49 +0000400 }
Owen Andersond4ebfd82013-02-06 22:43:31 +0000401 if (CFP->getType()->isHalfTy()){
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000402 C = FoldBitCast(C, Type::getInt16Ty(C->getContext()), DL);
403 return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, DL);
Owen Andersond4ebfd82013-02-06 22:43:31 +0000404 }
Chris Lattner3db7bd22009-10-24 05:27:19 +0000405 return false;
Chris Lattnered00b802009-10-23 06:23:49 +0000406 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000407
David Majnemer90a97042016-07-13 04:22:12 +0000408 if (auto *CS = dyn_cast<ConstantStruct>(C)) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000409 const StructLayout *SL = DL.getStructLayout(CS->getType());
Chris Lattnered00b802009-10-23 06:23:49 +0000410 unsigned Index = SL->getElementContainingOffset(ByteOffset);
411 uint64_t CurEltOffset = SL->getElementOffset(Index);
412 ByteOffset -= CurEltOffset;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000413
Eugene Zelenko1804a772016-08-25 00:45:04 +0000414 while (true) {
Chris Lattnered00b802009-10-23 06:23:49 +0000415 // If the element access is to the element itself and not to tail padding,
416 // read the bytes from the element.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000417 uint64_t EltSize = DL.getTypeAllocSize(CS->getOperand(Index)->getType());
Chris Lattnered00b802009-10-23 06:23:49 +0000418
419 if (ByteOffset < EltSize &&
420 !ReadDataFromGlobal(CS->getOperand(Index), ByteOffset, CurPtr,
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000421 BytesLeft, DL))
Chris Lattnered00b802009-10-23 06:23:49 +0000422 return false;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000423
Chris Lattnered00b802009-10-23 06:23:49 +0000424 ++Index;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000425
Chris Lattnered00b802009-10-23 06:23:49 +0000426 // Check to see if we read from the last struct element, if so we're done.
427 if (Index == CS->getType()->getNumElements())
428 return true;
429
430 // If we read all of the bytes we needed from this element we're done.
431 uint64_t NextEltOffset = SL->getElementOffset(Index);
432
Matt Arsenault8c789092013-08-12 23:15:58 +0000433 if (BytesLeft <= NextEltOffset - CurEltOffset - ByteOffset)
Chris Lattnered00b802009-10-23 06:23:49 +0000434 return true;
435
436 // Move to the next element of the struct.
Matt Arsenault8c789092013-08-12 23:15:58 +0000437 CurPtr += NextEltOffset - CurEltOffset - ByteOffset;
438 BytesLeft -= NextEltOffset - CurEltOffset - ByteOffset;
Chris Lattnered00b802009-10-23 06:23:49 +0000439 ByteOffset = 0;
440 CurEltOffset = NextEltOffset;
441 }
442 // not reached.
443 }
444
Chris Lattner67058832012-01-25 06:48:06 +0000445 if (isa<ConstantArray>(C) || isa<ConstantVector>(C) ||
446 isa<ConstantDataSequential>(C)) {
Matt Arsenault8c789092013-08-12 23:15:58 +0000447 Type *EltTy = C->getType()->getSequentialElementType();
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000448 uint64_t EltSize = DL.getTypeAllocSize(EltTy);
Chris Lattnered00b802009-10-23 06:23:49 +0000449 uint64_t Index = ByteOffset / EltSize;
450 uint64_t Offset = ByteOffset - Index * EltSize;
Chris Lattner67058832012-01-25 06:48:06 +0000451 uint64_t NumElts;
David Majnemer90a97042016-07-13 04:22:12 +0000452 if (auto *AT = dyn_cast<ArrayType>(C->getType()))
Chris Lattner67058832012-01-25 06:48:06 +0000453 NumElts = AT->getNumElements();
454 else
Matt Arsenault8c789092013-08-12 23:15:58 +0000455 NumElts = C->getType()->getVectorNumElements();
Duncan Sands0b875a02012-07-25 09:14:54 +0000456
Chris Lattner67058832012-01-25 06:48:06 +0000457 for (; Index != NumElts; ++Index) {
458 if (!ReadDataFromGlobal(C->getAggregateElement(Index), Offset, CurPtr,
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000459 BytesLeft, DL))
Chris Lattner67058832012-01-25 06:48:06 +0000460 return false;
Duncan Sands0b875a02012-07-25 09:14:54 +0000461
462 uint64_t BytesWritten = EltSize - Offset;
463 assert(BytesWritten <= EltSize && "Not indexing into this element?");
464 if (BytesWritten >= BytesLeft)
Chris Lattner67058832012-01-25 06:48:06 +0000465 return true;
Duncan Sands0b875a02012-07-25 09:14:54 +0000466
Chris Lattner67058832012-01-25 06:48:06 +0000467 Offset = 0;
Duncan Sands0b875a02012-07-25 09:14:54 +0000468 BytesLeft -= BytesWritten;
469 CurPtr += BytesWritten;
Chris Lattner67058832012-01-25 06:48:06 +0000470 }
471 return true;
472 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000473
David Majnemer90a97042016-07-13 04:22:12 +0000474 if (auto *CE = dyn_cast<ConstantExpr>(C)) {
Anders Carlssonecf8e152011-02-06 20:22:49 +0000475 if (CE->getOpcode() == Instruction::IntToPtr &&
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000476 CE->getOperand(0)->getType() == DL.getIntPtrType(CE->getType())) {
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000477 return ReadDataFromGlobal(CE->getOperand(0), ByteOffset, CurPtr,
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000478 BytesLeft, DL);
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000479 }
Anders Carlssond21b06a2011-02-06 20:11:56 +0000480 }
481
Chris Lattnered00b802009-10-23 06:23:49 +0000482 // Otherwise, unknown initializer type.
483 return false;
484}
485
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000486Constant *FoldReinterpretLoadFromConstPtr(Constant *C, Type *LoadTy,
487 const DataLayout &DL) {
David Majnemer90a97042016-07-13 04:22:12 +0000488 auto *PTy = cast<PointerType>(C->getType());
489 auto *IntType = dyn_cast<IntegerType>(LoadTy);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000490
Chris Lattnered00b802009-10-23 06:23:49 +0000491 // If this isn't an integer load we can't fold it directly.
492 if (!IntType) {
Matt Arsenault7a960a82013-08-20 21:20:04 +0000493 unsigned AS = PTy->getAddressSpace();
494
Chris Lattnered00b802009-10-23 06:23:49 +0000495 // If this is a float/double load, we can try folding it as an int32/64 load
Chris Lattnerccf1e842009-10-23 06:57:37 +0000496 // and then bitcast the result. This can be useful for union cases. Note
497 // that address spaces don't matter here since we're not going to result in
498 // an actual new load.
Chris Lattner229907c2011-07-18 04:54:35 +0000499 Type *MapTy;
Owen Andersond4ebfd82013-02-06 22:43:31 +0000500 if (LoadTy->isHalfTy())
Eduard Burtescu14239212016-01-22 01:17:26 +0000501 MapTy = Type::getInt16Ty(C->getContext());
Owen Andersond4ebfd82013-02-06 22:43:31 +0000502 else if (LoadTy->isFloatTy())
Eduard Burtescu14239212016-01-22 01:17:26 +0000503 MapTy = Type::getInt32Ty(C->getContext());
Chris Lattnerccf1e842009-10-23 06:57:37 +0000504 else if (LoadTy->isDoubleTy())
Eduard Burtescu14239212016-01-22 01:17:26 +0000505 MapTy = Type::getInt64Ty(C->getContext());
Duncan Sands19d0b472010-02-16 11:11:14 +0000506 else if (LoadTy->isVectorTy()) {
Eduard Burtescu14239212016-01-22 01:17:26 +0000507 MapTy = PointerType::getIntNTy(C->getContext(),
508 DL.getTypeAllocSizeInBits(LoadTy));
Chris Lattnerccf1e842009-10-23 06:57:37 +0000509 } else
Craig Topper9f008862014-04-15 04:59:12 +0000510 return nullptr;
Chris Lattnered00b802009-10-23 06:23:49 +0000511
Eduard Burtescu14239212016-01-22 01:17:26 +0000512 C = FoldBitCast(C, MapTy->getPointerTo(AS), DL);
513 if (Constant *Res = FoldReinterpretLoadFromConstPtr(C, MapTy, DL))
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000514 return FoldBitCast(Res, LoadTy, DL);
Craig Topper9f008862014-04-15 04:59:12 +0000515 return nullptr;
Chris Lattnered00b802009-10-23 06:23:49 +0000516 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000517
Chris Lattnered00b802009-10-23 06:23:49 +0000518 unsigned BytesLoaded = (IntType->getBitWidth() + 7) / 8;
Matt Arsenault8c789092013-08-12 23:15:58 +0000519 if (BytesLoaded > 32 || BytesLoaded == 0)
Craig Topper9f008862014-04-15 04:59:12 +0000520 return nullptr;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000521
Chris Lattnered00b802009-10-23 06:23:49 +0000522 GlobalValue *GVal;
David Majnemerf89660a2016-07-13 23:33:07 +0000523 APInt OffsetAI;
524 if (!IsConstantOffsetFromGlobal(C, GVal, OffsetAI, DL))
Craig Topper9f008862014-04-15 04:59:12 +0000525 return nullptr;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000526
David Majnemer90a97042016-07-13 04:22:12 +0000527 auto *GV = dyn_cast<GlobalVariable>(GVal);
Chris Lattner3db7bd22009-10-24 05:27:19 +0000528 if (!GV || !GV->isConstant() || !GV->hasDefinitiveInitializer() ||
Chris Lattnered00b802009-10-23 06:23:49 +0000529 !GV->getInitializer()->getType()->isSized())
Craig Topper9f008862014-04-15 04:59:12 +0000530 return nullptr;
Chris Lattnered00b802009-10-23 06:23:49 +0000531
David Majnemerf89660a2016-07-13 23:33:07 +0000532 int64_t Offset = OffsetAI.getSExtValue();
533 int64_t InitializerSize = DL.getTypeAllocSize(GV->getInitializer()->getType());
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000534
Chris Lattnered00b802009-10-23 06:23:49 +0000535 // If we're not accessing anything in this constant, the result is undefined.
David Majnemerf89660a2016-07-13 23:33:07 +0000536 if (Offset + BytesLoaded <= 0)
537 return UndefValue::get(IntType);
538
539 // If we're not accessing anything in this constant, the result is undefined.
540 if (Offset >= InitializerSize)
Chris Lattnered00b802009-10-23 06:23:49 +0000541 return UndefValue::get(IntType);
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000542
Chris Lattner59f94c02009-10-23 06:50:36 +0000543 unsigned char RawBytes[32] = {0};
David Majnemerf89660a2016-07-13 23:33:07 +0000544 unsigned char *CurPtr = RawBytes;
545 unsigned BytesLeft = BytesLoaded;
546
547 // If we're loading off the beginning of the global, some bytes may be valid.
548 if (Offset < 0) {
549 CurPtr += -Offset;
550 BytesLeft += Offset;
551 Offset = 0;
552 }
553
554 if (!ReadDataFromGlobal(GV->getInitializer(), Offset, CurPtr, BytesLeft, DL))
Craig Topper9f008862014-04-15 04:59:12 +0000555 return nullptr;
Chris Lattnered00b802009-10-23 06:23:49 +0000556
NAKAMURA Takumi43ab4ef2012-11-08 20:34:25 +0000557 APInt ResultVal = APInt(IntType->getBitWidth(), 0);
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000558 if (DL.isLittleEndian()) {
NAKAMURA Takumi43ab4ef2012-11-08 20:34:25 +0000559 ResultVal = RawBytes[BytesLoaded - 1];
560 for (unsigned i = 1; i != BytesLoaded; ++i) {
561 ResultVal <<= 8;
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000562 ResultVal |= RawBytes[BytesLoaded - 1 - i];
NAKAMURA Takumi43ab4ef2012-11-08 20:34:25 +0000563 }
564 } else {
565 ResultVal = RawBytes[0];
566 for (unsigned i = 1; i != BytesLoaded; ++i) {
567 ResultVal <<= 8;
568 ResultVal |= RawBytes[i];
569 }
Chris Lattner59f94c02009-10-23 06:50:36 +0000570 }
Chris Lattnered00b802009-10-23 06:23:49 +0000571
Chris Lattner59f94c02009-10-23 06:50:36 +0000572 return ConstantInt::get(IntType->getContext(), ResultVal);
Chris Lattnered00b802009-10-23 06:23:49 +0000573}
574
Eugene Leviant6f42a2c2018-03-13 10:19:50 +0000575Constant *ConstantFoldLoadThroughBitcastExpr(ConstantExpr *CE, Type *DestTy,
576 const DataLayout &DL) {
Eduard Burtescu14239212016-01-22 01:17:26 +0000577 auto *SrcPtr = CE->getOperand(0);
578 auto *SrcPtrTy = dyn_cast<PointerType>(SrcPtr->getType());
579 if (!SrcPtrTy)
Chandler Carrutha0e56952014-05-15 09:56:28 +0000580 return nullptr;
Eduard Burtescu14239212016-01-22 01:17:26 +0000581 Type *SrcTy = SrcPtrTy->getPointerElementType();
Chandler Carrutha0e56952014-05-15 09:56:28 +0000582
Eduard Burtescu14239212016-01-22 01:17:26 +0000583 Constant *C = ConstantFoldLoadFromConstPtr(SrcPtr, SrcTy, DL);
Chandler Carrutha0e56952014-05-15 09:56:28 +0000584 if (!C)
585 return nullptr;
586
Eugene Leviant6f42a2c2018-03-13 10:19:50 +0000587 return llvm::ConstantFoldLoadThroughBitcast(C, DestTy, DL);
Chandler Carrutha0e56952014-05-15 09:56:28 +0000588}
589
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000590} // end anonymous namespace
591
Eduard Burtescu14239212016-01-22 01:17:26 +0000592Constant *llvm::ConstantFoldLoadFromConstPtr(Constant *C, Type *Ty,
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000593 const DataLayout &DL) {
Chris Lattner1664a4f2009-10-22 06:25:11 +0000594 // First, try the easy cases:
David Majnemer90a97042016-07-13 04:22:12 +0000595 if (auto *GV = dyn_cast<GlobalVariable>(C))
Chris Lattner1664a4f2009-10-22 06:25:11 +0000596 if (GV->isConstant() && GV->hasDefinitiveInitializer())
597 return GV->getInitializer();
598
David Majnemered9abe12015-07-22 22:29:30 +0000599 if (auto *GA = dyn_cast<GlobalAlias>(C))
Sanjoy Das5ce32722016-04-08 00:48:30 +0000600 if (GA->getAliasee() && !GA->isInterposable())
Eduard Burtescu14239212016-01-22 01:17:26 +0000601 return ConstantFoldLoadFromConstPtr(GA->getAliasee(), Ty, DL);
David Majnemered9abe12015-07-22 22:29:30 +0000602
Chris Lattnercf7e8942009-10-22 06:44:07 +0000603 // If the loaded value isn't a constant expr, we can't handle it.
David Majnemer90a97042016-07-13 04:22:12 +0000604 auto *CE = dyn_cast<ConstantExpr>(C);
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000605 if (!CE)
Craig Topper9f008862014-04-15 04:59:12 +0000606 return nullptr;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000607
Chris Lattnercf7e8942009-10-22 06:44:07 +0000608 if (CE->getOpcode() == Instruction::GetElementPtr) {
David Majnemer90a97042016-07-13 04:22:12 +0000609 if (auto *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) {
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000610 if (GV->isConstant() && GV->hasDefinitiveInitializer()) {
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000611 if (Constant *V =
Chris Lattnercf7e8942009-10-22 06:44:07 +0000612 ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE))
613 return V;
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000614 }
615 }
Chris Lattnercf7e8942009-10-22 06:44:07 +0000616 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000617
Chandler Carrutha0e56952014-05-15 09:56:28 +0000618 if (CE->getOpcode() == Instruction::BitCast)
Eugene Leviant6f42a2c2018-03-13 10:19:50 +0000619 if (Constant *LoadedC = ConstantFoldLoadThroughBitcastExpr(CE, Ty, DL))
Chandler Carrutha0e56952014-05-15 09:56:28 +0000620 return LoadedC;
621
Chris Lattnercf7e8942009-10-22 06:44:07 +0000622 // Instead of loading constant c string, use corresponding integer value
623 // directly if string length is small enough.
Chris Lattnercf9e8f62012-02-05 02:29:43 +0000624 StringRef Str;
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000625 if (getConstantStringInfo(CE, Str) && !Str.empty()) {
David Majnemere61e4bf2016-06-21 05:10:24 +0000626 size_t StrLen = Str.size();
Chris Lattnered00b802009-10-23 06:23:49 +0000627 unsigned NumBits = Ty->getPrimitiveSizeInBits();
Chris Lattnerfd4a09f2010-07-12 00:22:51 +0000628 // Replace load with immediate integer if the result is an integer or fp
629 // value.
630 if ((NumBits >> 3) == StrLen + 1 && (NumBits & 7) == 0 &&
Chandler Carruth57041d82010-07-12 06:47:05 +0000631 (isa<IntegerType>(Ty) || Ty->isFloatingPointTy())) {
Chris Lattnered00b802009-10-23 06:23:49 +0000632 APInt StrVal(NumBits, 0);
633 APInt SingleChar(NumBits, 0);
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000634 if (DL.isLittleEndian()) {
David Majnemere61e4bf2016-06-21 05:10:24 +0000635 for (unsigned char C : reverse(Str.bytes())) {
636 SingleChar = static_cast<uint64_t>(C);
Chris Lattner51d2f702009-10-22 06:38:35 +0000637 StrVal = (StrVal << 8) | SingleChar;
638 }
Chris Lattnercf7e8942009-10-22 06:44:07 +0000639 } else {
David Majnemere61e4bf2016-06-21 05:10:24 +0000640 for (unsigned char C : Str.bytes()) {
641 SingleChar = static_cast<uint64_t>(C);
Chris Lattnercf7e8942009-10-22 06:44:07 +0000642 StrVal = (StrVal << 8) | SingleChar;
643 }
644 // Append NULL at the end.
645 SingleChar = 0;
646 StrVal = (StrVal << 8) | SingleChar;
Chris Lattner51d2f702009-10-22 06:38:35 +0000647 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000648
Chris Lattnerfd4a09f2010-07-12 00:22:51 +0000649 Constant *Res = ConstantInt::get(CE->getContext(), StrVal);
650 if (Ty->isFloatingPointTy())
651 Res = ConstantExpr::getBitCast(Res, Ty);
652 return Res;
Chris Lattner51d2f702009-10-22 06:38:35 +0000653 }
Chris Lattner1664a4f2009-10-22 06:25:11 +0000654 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000655
Chris Lattnercf7e8942009-10-22 06:44:07 +0000656 // If this load comes from anywhere in a constant global, and if the global
657 // is all undef or zero, we know what it loads.
David Majnemer90a97042016-07-13 04:22:12 +0000658 if (auto *GV = dyn_cast<GlobalVariable>(GetUnderlyingObject(CE, DL))) {
Chris Lattnercf7e8942009-10-22 06:44:07 +0000659 if (GV->isConstant() && GV->hasDefinitiveInitializer()) {
Chris Lattnercf7e8942009-10-22 06:44:07 +0000660 if (GV->getInitializer()->isNullValue())
Eduard Burtescu14239212016-01-22 01:17:26 +0000661 return Constant::getNullValue(Ty);
Chris Lattnercf7e8942009-10-22 06:44:07 +0000662 if (isa<UndefValue>(GV->getInitializer()))
Eduard Burtescu14239212016-01-22 01:17:26 +0000663 return UndefValue::get(Ty);
Chris Lattnercf7e8942009-10-22 06:44:07 +0000664 }
665 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000666
NAKAMURA Takumi43ab4ef2012-11-08 20:34:25 +0000667 // Try hard to fold loads from bitcasted strange and non-type-safe things.
Eduard Burtescu14239212016-01-22 01:17:26 +0000668 return FoldReinterpretLoadFromConstPtr(CE, Ty, DL);
Chris Lattner1664a4f2009-10-22 06:25:11 +0000669}
670
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000671namespace {
672
673Constant *ConstantFoldLoadInst(const LoadInst *LI, const DataLayout &DL) {
Craig Topper9f008862014-04-15 04:59:12 +0000674 if (LI->isVolatile()) return nullptr;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000675
David Majnemer90a97042016-07-13 04:22:12 +0000676 if (auto *C = dyn_cast<Constant>(LI->getOperand(0)))
Eduard Burtescu14239212016-01-22 01:17:26 +0000677 return ConstantFoldLoadFromConstPtr(C, LI->getType(), DL);
Chris Lattnered00b802009-10-23 06:23:49 +0000678
Craig Topper9f008862014-04-15 04:59:12 +0000679 return nullptr;
Chris Lattner1664a4f2009-10-22 06:25:11 +0000680}
Chris Lattner44d68b92007-01-31 00:51:48 +0000681
Sanjay Patel0d7dee62014-10-02 15:13:22 +0000682/// One of Op0/Op1 is a constant expression.
Nick Lewyckye88d3882008-12-15 01:35:36 +0000683/// Attempt to symbolically evaluate the result of a binary operator merging
Nick Lewycky06417742013-02-14 03:23:37 +0000684/// these together. If target data info is available, it is provided as DL,
685/// otherwise DL is null.
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000686Constant *SymbolicallyEvaluateBinop(unsigned Opc, Constant *Op0, Constant *Op1,
687 const DataLayout &DL) {
Chris Lattner44d68b92007-01-31 00:51:48 +0000688 // SROA
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000689
Chris Lattner44d68b92007-01-31 00:51:48 +0000690 // Fold (and 0xffffffff00000000, (shl x, 32)) -> shl.
691 // Fold (lshr (or X, Y), 32) -> (lshr [X/Y], 32) if one doesn't contribute
692 // bits.
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000693
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000694 if (Opc == Instruction::And) {
Craig Topper8205a1a2017-05-24 16:53:07 +0000695 KnownBits Known0 = computeKnownBits(Op0, DL);
696 KnownBits Known1 = computeKnownBits(Op1, DL);
Craig Topperb45eabc2017-04-26 16:39:58 +0000697 if ((Known1.One | Known0.Zero).isAllOnesValue()) {
Nick Lewycky06417742013-02-14 03:23:37 +0000698 // All the bits of Op0 that the 'and' could be masking are already zero.
699 return Op0;
700 }
Craig Topperb45eabc2017-04-26 16:39:58 +0000701 if ((Known0.One | Known1.Zero).isAllOnesValue()) {
Nick Lewycky06417742013-02-14 03:23:37 +0000702 // All the bits of Op1 that the 'and' could be masking are already zero.
703 return Op1;
704 }
705
Craig Topper8189a872017-05-03 23:12:29 +0000706 Known0.Zero |= Known1.Zero;
707 Known0.One &= Known1.One;
708 if (Known0.isConstant())
709 return ConstantInt::get(Op0->getType(), Known0.getConstant());
Nick Lewycky06417742013-02-14 03:23:37 +0000710 }
711
Chris Lattner44d68b92007-01-31 00:51:48 +0000712 // If the constant expr is something like &A[123] - &A[4].f, fold this into a
713 // constant. This happens frequently when iterating over a global array.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000714 if (Opc == Instruction::Sub) {
Chris Lattner44d68b92007-01-31 00:51:48 +0000715 GlobalValue *GV1, *GV2;
Matt Arsenaulta8e89442013-11-04 20:46:52 +0000716 APInt Offs1, Offs2;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000717
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000718 if (IsConstantOffsetFromGlobal(Op0, GV1, Offs1, DL))
719 if (IsConstantOffsetFromGlobal(Op1, GV2, Offs2, DL) && GV1 == GV2) {
720 unsigned OpSize = DL.getTypeSizeInBits(Op0->getType());
Matt Arsenaultbed5bf22013-09-12 01:07:58 +0000721
Chris Lattner44d68b92007-01-31 00:51:48 +0000722 // (&GV+C1) - (&GV+C2) -> C1-C2, pointer arithmetic cannot overflow.
Benjamin Kramera5a9ec52013-02-05 19:04:36 +0000723 // PtrToInt may change the bitwidth so we have convert to the right size
724 // first.
725 return ConstantInt::get(Op0->getType(), Offs1.zextOrTrunc(OpSize) -
726 Offs2.zextOrTrunc(OpSize));
Chris Lattner44d68b92007-01-31 00:51:48 +0000727 }
728 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000729
Craig Topper9f008862014-04-15 04:59:12 +0000730 return nullptr;
Chris Lattner44d68b92007-01-31 00:51:48 +0000731}
732
Sanjay Patel0d7dee62014-10-02 15:13:22 +0000733/// If array indices are not pointer-sized integers, explicitly cast them so
734/// that they aren't implicitly casted by the getelementptr.
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000735Constant *CastGEPIndices(Type *SrcElemTy, ArrayRef<Constant *> Ops,
Peter Collingbourned93620b2016-11-10 22:34:55 +0000736 Type *ResultTy, Optional<unsigned> InRangeIndex,
737 const DataLayout &DL, const TargetLibraryInfo *TLI) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000738 Type *IntPtrTy = DL.getIntPtrType(ResultTy);
Keno Fischerdc091192016-12-08 17:22:35 +0000739 Type *IntPtrScalarTy = IntPtrTy->getScalarType();
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000740
741 bool Any = false;
742 SmallVector<Constant*, 32> NewIdxs;
Jay Foadf4b14a22011-07-19 13:32:40 +0000743 for (unsigned i = 1, e = Ops.size(); i != e; ++i) {
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000744 if ((i == 1 ||
Keno Fischerdc091192016-12-08 17:22:35 +0000745 !isa<StructType>(GetElementPtrInst::getIndexedType(
746 SrcElemTy, Ops.slice(1, i - 1)))) &&
Michael Kupersteindd92c782016-12-21 17:34:21 +0000747 Ops[i]->getType()->getScalarType() != IntPtrScalarTy) {
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000748 Any = true;
Michael Kupersteindd92c782016-12-21 17:34:21 +0000749 Type *NewType = Ops[i]->getType()->isVectorTy()
750 ? IntPtrTy
751 : IntPtrTy->getScalarType();
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000752 NewIdxs.push_back(ConstantExpr::getCast(CastInst::getCastOpcode(Ops[i],
753 true,
Michael Kupersteindd92c782016-12-21 17:34:21 +0000754 NewType,
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000755 true),
Michael Kupersteindd92c782016-12-21 17:34:21 +0000756 Ops[i], NewType));
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000757 } else
758 NewIdxs.push_back(Ops[i]);
759 }
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000760
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000761 if (!Any)
Craig Topper9f008862014-04-15 04:59:12 +0000762 return nullptr;
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000763
Peter Collingbourned93620b2016-11-10 22:34:55 +0000764 Constant *C = ConstantExpr::getGetElementPtr(
765 SrcElemTy, Ops[0], NewIdxs, /*InBounds=*/false, InRangeIndex);
David Majnemerd536f232016-07-29 03:27:26 +0000766 if (Constant *Folded = ConstantFoldConstant(C, DL, TLI))
767 C = Folded;
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000768
Dan Gohmane5e1b7b2010-02-01 18:27:38 +0000769 return C;
770}
771
Nadav Rotem77f1b9c2012-07-30 07:25:20 +0000772/// Strip the pointer casts, but preserve the address space information.
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000773Constant* StripPtrCastKeepAS(Constant* Ptr, Type *&ElemTy) {
Nadav Rotem77f1b9c2012-07-30 07:25:20 +0000774 assert(Ptr->getType()->isPointerTy() && "Not a pointer type");
David Majnemer90a97042016-07-13 04:22:12 +0000775 auto *OldPtrTy = cast<PointerType>(Ptr->getType());
Rafael Espindola78598d92014-06-04 19:01:48 +0000776 Ptr = Ptr->stripPointerCasts();
David Majnemer90a97042016-07-13 04:22:12 +0000777 auto *NewPtrTy = cast<PointerType>(Ptr->getType());
Nadav Rotem77f1b9c2012-07-30 07:25:20 +0000778
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000779 ElemTy = NewPtrTy->getPointerElementType();
780
Nadav Rotem77f1b9c2012-07-30 07:25:20 +0000781 // Preserve the address space number of the pointer.
782 if (NewPtrTy->getAddressSpace() != OldPtrTy->getAddressSpace()) {
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000783 NewPtrTy = ElemTy->getPointerTo(OldPtrTy->getAddressSpace());
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +0000784 Ptr = ConstantExpr::getPointerCast(Ptr, NewPtrTy);
Nadav Rotem77f1b9c2012-07-30 07:25:20 +0000785 }
786 return Ptr;
787}
788
Sanjay Patel0d7dee62014-10-02 15:13:22 +0000789/// If we can symbolically evaluate the GEP constant expression, do so.
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000790Constant *SymbolicallyEvaluateGEP(const GEPOperator *GEP,
791 ArrayRef<Constant *> Ops,
792 const DataLayout &DL,
793 const TargetLibraryInfo *TLI) {
Peter Collingbourned93620b2016-11-10 22:34:55 +0000794 const GEPOperator *InnermostGEP = GEP;
Peter Collingbourne0a4fc462016-11-22 01:03:40 +0000795 bool InBounds = GEP->isInBounds();
Peter Collingbourned93620b2016-11-10 22:34:55 +0000796
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000797 Type *SrcElemTy = GEP->getSourceElementType();
798 Type *ResElemTy = GEP->getResultElementType();
799 Type *ResTy = GEP->getType();
800 if (!SrcElemTy->isSized())
801 return nullptr;
802
Peter Collingbourned93620b2016-11-10 22:34:55 +0000803 if (Constant *C = CastGEPIndices(SrcElemTy, Ops, ResTy,
804 GEP->getInRangeIndex(), DL, TLI))
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000805 return C;
806
Chris Lattner44d68b92007-01-31 00:51:48 +0000807 Constant *Ptr = Ops[0];
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000808 if (!Ptr->getType()->isPointerTy())
Craig Topper9f008862014-04-15 04:59:12 +0000809 return nullptr;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000810
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000811 Type *IntPtrTy = DL.getIntPtrType(Ptr->getType());
Chris Lattnerf4b42f52008-05-08 04:54:43 +0000812
813 // If this is a constant expr gep that is effectively computing an
814 // "offsetof", fold it into 'cast int Size to T*' instead of 'gep 0, 0, 12'
Jay Foadf4b14a22011-07-19 13:32:40 +0000815 for (unsigned i = 1, e = Ops.size(); i != e; ++i)
Elena Demikhovsky945b7e52018-02-14 06:58:08 +0000816 if (!isa<ConstantInt>(Ops[i])) {
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000817
Elena Demikhovsky945b7e52018-02-14 06:58:08 +0000818 // If this is "gep i8* Ptr, (sub 0, V)", fold this as:
819 // "inttoptr (sub (ptrtoint Ptr), V)"
820 if (Ops.size() == 2 && ResElemTy->isIntegerTy(8)) {
821 auto *CE = dyn_cast<ConstantExpr>(Ops[1]);
822 assert((!CE || CE->getType() == IntPtrTy) &&
823 "CastGEPIndices didn't canonicalize index types!");
824 if (CE && CE->getOpcode() == Instruction::Sub &&
825 CE->getOperand(0)->isNullValue()) {
826 Constant *Res = ConstantExpr::getPtrToInt(Ptr, CE->getType());
827 Res = ConstantExpr::getSub(Res, CE->getOperand(1));
828 Res = ConstantExpr::getIntToPtr(Res, ResTy);
829 if (auto *FoldedRes = ConstantFoldConstant(Res, DL, TLI))
830 Res = FoldedRes;
831 return Res;
832 }
Chris Lattner5858e092011-01-06 06:19:46 +0000833 }
Elena Demikhovsky945b7e52018-02-14 06:58:08 +0000834 return nullptr;
Chris Lattner5858e092011-01-06 06:19:46 +0000835 }
Nadav Rotem77f1b9c2012-07-30 07:25:20 +0000836
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000837 unsigned BitWidth = DL.getTypeSizeInBits(IntPtrTy);
Jay Foadbf904772011-07-19 14:01:37 +0000838 APInt Offset =
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000839 APInt(BitWidth,
Eduard Burtescu68e7f492016-01-22 03:08:27 +0000840 DL.getIndexedOffsetInType(
841 SrcElemTy,
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000842 makeArrayRef((Value * const *)Ops.data() + 1, Ops.size() - 1)));
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000843 Ptr = StripPtrCastKeepAS(Ptr, SrcElemTy);
Dan Gohman474e488c2010-03-10 19:31:51 +0000844
845 // If this is a GEP of a GEP, fold it all into a single GEP.
David Majnemer90a97042016-07-13 04:22:12 +0000846 while (auto *GEP = dyn_cast<GEPOperator>(Ptr)) {
Peter Collingbourned93620b2016-11-10 22:34:55 +0000847 InnermostGEP = GEP;
Peter Collingbourne0a4fc462016-11-22 01:03:40 +0000848 InBounds &= GEP->isInBounds();
Peter Collingbourned93620b2016-11-10 22:34:55 +0000849
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000850 SmallVector<Value *, 4> NestedOps(GEP->op_begin() + 1, GEP->op_end());
Duncan Sands8c355062010-03-12 17:55:20 +0000851
852 // Do not try the incorporate the sub-GEP if some index is not a number.
853 bool AllConstantInt = true;
David Majnemer90a97042016-07-13 04:22:12 +0000854 for (Value *NestedOp : NestedOps)
855 if (!isa<ConstantInt>(NestedOp)) {
Duncan Sands8c355062010-03-12 17:55:20 +0000856 AllConstantInt = false;
857 break;
858 }
859 if (!AllConstantInt)
860 break;
861
Dan Gohman474e488c2010-03-10 19:31:51 +0000862 Ptr = cast<Constant>(GEP->getOperand(0));
Eduard Burtescu68e7f492016-01-22 03:08:27 +0000863 SrcElemTy = GEP->getSourceElementType();
864 Offset += APInt(BitWidth, DL.getIndexedOffsetInType(SrcElemTy, NestedOps));
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000865 Ptr = StripPtrCastKeepAS(Ptr, SrcElemTy);
Dan Gohman474e488c2010-03-10 19:31:51 +0000866 }
867
Dan Gohman81ce8422009-08-19 18:18:36 +0000868 // If the base value for this address is a literal integer value, fold the
869 // getelementptr to the resulting integer value casted to the pointer type.
Dan Gohmana5ca5782010-03-18 19:34:33 +0000870 APInt BasePtr(BitWidth, 0);
David Majnemer90a97042016-07-13 04:22:12 +0000871 if (auto *CE = dyn_cast<ConstantExpr>(Ptr)) {
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000872 if (CE->getOpcode() == Instruction::IntToPtr) {
David Majnemer90a97042016-07-13 04:22:12 +0000873 if (auto *Base = dyn_cast<ConstantInt>(CE->getOperand(0)))
Jay Foad583abbc2010-12-07 08:25:19 +0000874 BasePtr = Base->getValue().zextOrTrunc(BitWidth);
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000875 }
876 }
877
Sanjoy Das6fa08aa2016-08-05 19:23:29 +0000878 auto *PTy = cast<PointerType>(Ptr->getType());
879 if ((Ptr->isNullValue() || BasePtr != 0) &&
880 !DL.isNonIntegralPointerType(PTy)) {
Matt Arsenaulta5e56982013-08-12 22:56:15 +0000881 Constant *C = ConstantInt::get(Ptr->getContext(), Offset + BasePtr);
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000882 return ConstantExpr::getIntToPtr(C, ResTy);
Dan Gohman81ce8422009-08-19 18:18:36 +0000883 }
884
885 // Otherwise form a regular getelementptr. Recompute the indices so that
886 // we eliminate over-indexing of the notional static type array bounds.
887 // This makes it easy to determine if the getelementptr is "inbounds".
888 // Also, this helps GlobalOpt do SROA on GlobalVariables.
Sanjoy Das6fa08aa2016-08-05 19:23:29 +0000889 Type *Ty = PTy;
Matt Arsenault7a960a82013-08-20 21:20:04 +0000890 SmallVector<Constant *, 32> NewIdxs;
891
Dan Gohmanc59ba422009-08-19 22:46:59 +0000892 do {
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000893 if (!Ty->isStructTy()) {
894 if (Ty->isPointerTy()) {
Chris Lattner77c36d62009-12-03 01:05:45 +0000895 // The only pointer indexing we'll do is on the first index of the GEP.
896 if (!NewIdxs.empty())
897 break;
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000898
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000899 Ty = SrcElemTy;
900
Chris Lattner77c36d62009-12-03 01:05:45 +0000901 // Only handle pointers to sized types, not pointers to functions.
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000902 if (!Ty->isSized())
Craig Topper9f008862014-04-15 04:59:12 +0000903 return nullptr;
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000904 } else if (auto *ATy = dyn_cast<SequentialType>(Ty)) {
905 Ty = ATy->getElementType();
906 } else {
907 // We've reached some non-indexable type.
908 break;
Chris Lattner77c36d62009-12-03 01:05:45 +0000909 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +0000910
Dan Gohman81ce8422009-08-19 18:18:36 +0000911 // Determine which element of the array the offset points into.
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000912 APInt ElemSize(BitWidth, DL.getTypeAllocSize(Ty));
David Majnemer1b3db332016-07-13 05:16:16 +0000913 if (ElemSize == 0) {
Chris Lattner6ce03802010-11-21 08:39:01 +0000914 // The element size is 0. This may be [0 x Ty]*, so just use a zero
Duncan Sands1f86be92010-11-21 12:43:13 +0000915 // index for this level and proceed to the next level to see if it can
916 // accommodate the offset.
Chris Lattner6ce03802010-11-21 08:39:01 +0000917 NewIdxs.push_back(ConstantInt::get(IntPtrTy, 0));
David Majnemer1b3db332016-07-13 05:16:16 +0000918 } else {
Chris Lattner6ce03802010-11-21 08:39:01 +0000919 // The element size is non-zero divide the offset by the element
920 // size (rounding down), to compute the index at this level.
David Majnemer4cff2f82016-07-13 15:53:46 +0000921 bool Overflow;
922 APInt NewIdx = Offset.sdiv_ov(ElemSize, Overflow);
923 if (Overflow)
924 break;
Chris Lattner6ce03802010-11-21 08:39:01 +0000925 Offset -= NewIdx * ElemSize;
926 NewIdxs.push_back(ConstantInt::get(IntPtrTy, NewIdx));
927 }
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000928 } else {
David Majnemer90a97042016-07-13 04:22:12 +0000929 auto *STy = cast<StructType>(Ty);
Chandler Carruthaacb8a52012-04-24 18:42:47 +0000930 // If we end up with an offset that isn't valid for this struct type, we
931 // can't re-form this GEP in a regular form, so bail out. The pointer
932 // operand likely went through casts that are necessary to make the GEP
933 // sensible.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000934 const StructLayout &SL = *DL.getStructLayout(STy);
David Majnemer1b3db332016-07-13 05:16:16 +0000935 if (Offset.isNegative() || Offset.uge(SL.getSizeInBytes()))
Chandler Carruthaacb8a52012-04-24 18:42:47 +0000936 break;
937
938 // Determine which field of the struct the offset points into. The
939 // getZExtValue is fine as we've already ensured that the offset is
940 // within the range representable by the StructLayout API.
Dan Gohman23e62c52009-08-21 16:52:54 +0000941 unsigned ElIdx = SL.getElementContainingOffset(Offset.getZExtValue());
Chris Lattner46b5c642009-11-06 04:27:31 +0000942 NewIdxs.push_back(ConstantInt::get(Type::getInt32Ty(Ty->getContext()),
943 ElIdx));
Dan Gohman23e62c52009-08-21 16:52:54 +0000944 Offset -= APInt(BitWidth, SL.getElementOffset(ElIdx));
Dan Gohman81ce8422009-08-19 18:18:36 +0000945 Ty = STy->getTypeAtIndex(ElIdx);
Dan Gohman81ce8422009-08-19 18:18:36 +0000946 }
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000947 } while (Ty != ResElemTy);
Dan Gohmanc59ba422009-08-19 22:46:59 +0000948
949 // If we haven't used up the entire offset by descending the static
950 // type, then the offset is pointing into the middle of an indivisible
951 // member, so we can't simplify it.
952 if (Offset != 0)
Craig Topper9f008862014-04-15 04:59:12 +0000953 return nullptr;
Dan Gohman81ce8422009-08-19 18:18:36 +0000954
Peter Collingbourned93620b2016-11-10 22:34:55 +0000955 // Preserve the inrange index from the innermost GEP if possible. We must
956 // have calculated the same indices up to and including the inrange index.
957 Optional<unsigned> InRangeIndex;
958 if (Optional<unsigned> LastIRIndex = InnermostGEP->getInRangeIndex())
959 if (SrcElemTy == InnermostGEP->getSourceElementType() &&
960 NewIdxs.size() > *LastIRIndex) {
961 InRangeIndex = LastIRIndex;
962 for (unsigned I = 0; I <= *LastIRIndex; ++I)
Peter Collingbournec7d28192018-09-11 01:53:36 +0000963 if (NewIdxs[I] != InnermostGEP->getOperand(I + 1))
964 return nullptr;
Peter Collingbourned93620b2016-11-10 22:34:55 +0000965 }
966
Dan Gohman21c62162009-09-11 00:04:14 +0000967 // Create a GEP.
Peter Collingbourne0a4fc462016-11-22 01:03:40 +0000968 Constant *C = ConstantExpr::getGetElementPtr(SrcElemTy, Ptr, NewIdxs,
969 InBounds, InRangeIndex);
Matt Arsenault8c789092013-08-12 23:15:58 +0000970 assert(C->getType()->getPointerElementType() == Ty &&
Dan Gohmane4ca02d2009-09-03 23:34:49 +0000971 "Computed GetElementPtr has unexpected type!");
Dan Gohman81ce8422009-08-19 18:18:36 +0000972
Dan Gohmanc59ba422009-08-19 22:46:59 +0000973 // If we ended up indexing a member with a type that doesn't match
Dan Gohman8a8ad7d2009-08-20 16:42:55 +0000974 // the type of what the original indices indexed, add a cast.
Eduard Burtescu2f4758b2016-01-21 23:42:06 +0000975 if (Ty != ResElemTy)
976 C = FoldBitCast(C, ResTy, DL);
Dan Gohmanc59ba422009-08-19 22:46:59 +0000977
978 return C;
Chris Lattner44d68b92007-01-31 00:51:48 +0000979}
980
Manuel Jacobe9024592016-01-21 06:33:22 +0000981/// Attempt to constant fold an instruction with the
982/// specified opcode and operands. If successful, the constant result is
983/// returned, if not, null is returned. Note that this function can fail when
984/// attempting to fold instructions like loads and stores, which have no
985/// constant expression form.
David Majnemera926b3e2016-07-29 03:27:33 +0000986Constant *ConstantFoldInstOperandsImpl(const Value *InstOrCE, unsigned Opcode,
Eugene Zelenko35623fb2016-03-28 17:40:08 +0000987 ArrayRef<Constant *> Ops,
988 const DataLayout &DL,
989 const TargetLibraryInfo *TLI) {
David Majnemera926b3e2016-07-29 03:27:33 +0000990 Type *DestTy = InstOrCE->getType();
991
Manuel Jacobe9024592016-01-21 06:33:22 +0000992 // Handle easy binops first.
993 if (Instruction::isBinaryOp(Opcode))
994 return ConstantFoldBinaryOpOperands(Opcode, Ops[0], Ops[1], DL);
995
996 if (Instruction::isCast(Opcode))
997 return ConstantFoldCastOperand(Opcode, Ops[0], DestTy, DL);
998
David Majnemer17bdf442016-07-13 03:42:38 +0000999 if (auto *GEP = dyn_cast<GEPOperator>(InstOrCE)) {
Eduard Burtescu2f4758b2016-01-21 23:42:06 +00001000 if (Constant *C = SymbolicallyEvaluateGEP(GEP, Ops, DL, TLI))
1001 return C;
1002
Peter Collingbourned93620b2016-11-10 22:34:55 +00001003 return ConstantExpr::getGetElementPtr(GEP->getSourceElementType(), Ops[0],
1004 Ops.slice(1), GEP->isInBounds(),
1005 GEP->getInRangeIndex());
Eduard Burtescu2f4758b2016-01-21 23:42:06 +00001006 }
1007
David Majnemer57b94c82016-07-29 03:27:31 +00001008 if (auto *CE = dyn_cast<ConstantExpr>(InstOrCE))
1009 return CE->getWithOperands(Ops);
1010
Manuel Jacobe9024592016-01-21 06:33:22 +00001011 switch (Opcode) {
1012 default: return nullptr;
1013 case Instruction::ICmp:
1014 case Instruction::FCmp: llvm_unreachable("Invalid for compares");
1015 case Instruction::Call:
Andrew Kaylor647025f2017-06-09 23:18:11 +00001016 if (auto *F = dyn_cast<Function>(Ops.back())) {
1017 ImmutableCallSite CS(cast<CallInst>(InstOrCE));
1018 if (canConstantFoldCallTo(CS, F))
1019 return ConstantFoldCall(CS, F, Ops.slice(0, Ops.size() - 1), TLI);
1020 }
Manuel Jacobe9024592016-01-21 06:33:22 +00001021 return nullptr;
1022 case Instruction::Select:
1023 return ConstantExpr::getSelect(Ops[0], Ops[1], Ops[2]);
1024 case Instruction::ExtractElement:
1025 return ConstantExpr::getExtractElement(Ops[0], Ops[1]);
1026 case Instruction::InsertElement:
1027 return ConstantExpr::getInsertElement(Ops[0], Ops[1], Ops[2]);
1028 case Instruction::ShuffleVector:
1029 return ConstantExpr::getShuffleVector(Ops[0], Ops[1], Ops[2]);
Manuel Jacobe9024592016-01-21 06:33:22 +00001030 }
1031}
1032
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001033} // end anonymous namespace
Chris Lattner44d68b92007-01-31 00:51:48 +00001034
1035//===----------------------------------------------------------------------===//
1036// Constant Folding public APIs
1037//===----------------------------------------------------------------------===//
1038
David Majnemerd536f232016-07-29 03:27:26 +00001039namespace {
1040
1041Constant *
1042ConstantFoldConstantImpl(const Constant *C, const DataLayout &DL,
1043 const TargetLibraryInfo *TLI,
1044 SmallDenseMap<Constant *, Constant *> &FoldedOps) {
1045 if (!isa<ConstantVector>(C) && !isa<ConstantExpr>(C))
1046 return nullptr;
1047
1048 SmallVector<Constant *, 8> Ops;
1049 for (const Use &NewU : C->operands()) {
1050 auto *NewC = cast<Constant>(&NewU);
1051 // Recursively fold the ConstantExpr's operands. If we have already folded
1052 // a ConstantExpr, we don't have to process it again.
1053 if (isa<ConstantVector>(NewC) || isa<ConstantExpr>(NewC)) {
1054 auto It = FoldedOps.find(NewC);
1055 if (It == FoldedOps.end()) {
1056 if (auto *FoldedC =
1057 ConstantFoldConstantImpl(NewC, DL, TLI, FoldedOps)) {
David Majnemerd536f232016-07-29 03:27:26 +00001058 FoldedOps.insert({NewC, FoldedC});
David Greenda211702017-03-21 10:17:39 +00001059 NewC = FoldedC;
David Majnemerd536f232016-07-29 03:27:26 +00001060 } else {
1061 FoldedOps.insert({NewC, NewC});
1062 }
1063 } else {
1064 NewC = It->second;
1065 }
1066 }
1067 Ops.push_back(NewC);
1068 }
1069
1070 if (auto *CE = dyn_cast<ConstantExpr>(C)) {
1071 if (CE->isCompare())
1072 return ConstantFoldCompareInstOperands(CE->getPredicate(), Ops[0], Ops[1],
1073 DL, TLI);
1074
David Majnemera926b3e2016-07-29 03:27:33 +00001075 return ConstantFoldInstOperandsImpl(CE, CE->getOpcode(), Ops, DL, TLI);
David Majnemerd536f232016-07-29 03:27:26 +00001076 }
1077
1078 assert(isa<ConstantVector>(C));
1079 return ConstantVector::get(Ops);
1080}
1081
1082} // end anonymous namespace
1083
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001084Constant *llvm::ConstantFoldInstruction(Instruction *I, const DataLayout &DL,
Chad Rosierc24b86f2011-12-01 03:08:23 +00001085 const TargetLibraryInfo *TLI) {
Duncan Sands763dec02010-11-23 10:16:18 +00001086 // Handle PHI nodes quickly here...
David Majnemer90a97042016-07-13 04:22:12 +00001087 if (auto *PN = dyn_cast<PHINode>(I)) {
Craig Topper9f008862014-04-15 04:59:12 +00001088 Constant *CommonValue = nullptr;
John Criswell970af112005-10-27 16:00:10 +00001089
David Majnemerd536f232016-07-29 03:27:26 +00001090 SmallDenseMap<Constant *, Constant *> FoldedOps;
Pete Cooper833f34d2015-05-12 20:05:31 +00001091 for (Value *Incoming : PN->incoming_values()) {
Duncan Sands763dec02010-11-23 10:16:18 +00001092 // If the incoming value is undef then skip it. Note that while we could
1093 // skip the value if it is equal to the phi node itself we choose not to
1094 // because that would break the rule that constant folding only applies if
1095 // all operands are constants.
1096 if (isa<UndefValue>(Incoming))
Duncan Sands1d27f0122010-11-14 12:53:18 +00001097 continue;
Dan Gohman1ccecdb2012-04-27 17:50:22 +00001098 // If the incoming value is not a constant, then give up.
David Majnemer90a97042016-07-13 04:22:12 +00001099 auto *C = dyn_cast<Constant>(Incoming);
Dan Gohman1ccecdb2012-04-27 17:50:22 +00001100 if (!C)
Craig Topper9f008862014-04-15 04:59:12 +00001101 return nullptr;
Dan Gohman1ccecdb2012-04-27 17:50:22 +00001102 // Fold the PHI's operands.
David Majnemerd536f232016-07-29 03:27:26 +00001103 if (auto *FoldedC = ConstantFoldConstantImpl(C, DL, TLI, FoldedOps))
1104 C = FoldedC;
Dan Gohman1ccecdb2012-04-27 17:50:22 +00001105 // If the incoming value is a different constant to
1106 // the one we saw previously, then give up.
1107 if (CommonValue && C != CommonValue)
Craig Topper9f008862014-04-15 04:59:12 +00001108 return nullptr;
Duncan Sands1d27f0122010-11-14 12:53:18 +00001109 CommonValue = C;
1110 }
Chris Lattner2ae054a2007-01-30 23:45:45 +00001111
Duncan Sands1d27f0122010-11-14 12:53:18 +00001112 // If we reach here, all incoming values are the same constant or undef.
1113 return CommonValue ? CommonValue : UndefValue::get(PN->getType());
Chris Lattner2ae054a2007-01-30 23:45:45 +00001114 }
1115
1116 // Scan the operand list, checking to see if they are all constants, if so,
Manuel Jacobe9024592016-01-21 06:33:22 +00001117 // hand off to ConstantFoldInstOperandsImpl.
Fiona Glaser2e5c0c22016-03-13 05:36:15 +00001118 if (!all_of(I->operands(), [](Use &U) { return isa<Constant>(U); }))
1119 return nullptr;
Chris Lattner2ae054a2007-01-30 23:45:45 +00001120
David Majnemerd536f232016-07-29 03:27:26 +00001121 SmallDenseMap<Constant *, Constant *> FoldedOps;
Fiona Glaser2e5c0c22016-03-13 05:36:15 +00001122 SmallVector<Constant *, 8> Ops;
David Majnemer90a97042016-07-13 04:22:12 +00001123 for (const Use &OpU : I->operands()) {
1124 auto *Op = cast<Constant>(&OpU);
Dan Gohman1ccecdb2012-04-27 17:50:22 +00001125 // Fold the Instruction's operands.
David Majnemerd536f232016-07-29 03:27:26 +00001126 if (auto *FoldedOp = ConstantFoldConstantImpl(Op, DL, TLI, FoldedOps))
1127 Op = FoldedOp;
Dan Gohman1ccecdb2012-04-27 17:50:22 +00001128
1129 Ops.push_back(Op);
1130 }
1131
David Majnemer90a97042016-07-13 04:22:12 +00001132 if (const auto *CI = dyn_cast<CmpInst>(I))
Chris Lattnercdfb80d2009-11-09 23:06:58 +00001133 return ConstantFoldCompareInstOperands(CI->getPredicate(), Ops[0], Ops[1],
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001134 DL, TLI);
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001135
David Majnemer90a97042016-07-13 04:22:12 +00001136 if (const auto *LI = dyn_cast<LoadInst>(I))
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001137 return ConstantFoldLoadInst(LI, DL);
Frits van Bommela98214d2010-11-29 20:36:52 +00001138
David Majnemer90a97042016-07-13 04:22:12 +00001139 if (auto *IVI = dyn_cast<InsertValueInst>(I)) {
Frits van Bommela98214d2010-11-29 20:36:52 +00001140 return ConstantExpr::getInsertValue(
1141 cast<Constant>(IVI->getAggregateOperand()),
1142 cast<Constant>(IVI->getInsertedValueOperand()),
Jay Foad57aa6362011-07-13 10:26:04 +00001143 IVI->getIndices());
Matt Arsenaulta5e56982013-08-12 22:56:15 +00001144 }
Frits van Bommela98214d2010-11-29 20:36:52 +00001145
David Majnemer90a97042016-07-13 04:22:12 +00001146 if (auto *EVI = dyn_cast<ExtractValueInst>(I)) {
Frits van Bommela98214d2010-11-29 20:36:52 +00001147 return ConstantExpr::getExtractValue(
1148 cast<Constant>(EVI->getAggregateOperand()),
Jay Foad57aa6362011-07-13 10:26:04 +00001149 EVI->getIndices());
Matt Arsenaulta5e56982013-08-12 22:56:15 +00001150 }
Frits van Bommela98214d2010-11-29 20:36:52 +00001151
Eduard Burtescu2f4758b2016-01-21 23:42:06 +00001152 return ConstantFoldInstOperands(I, Ops, DL, TLI);
Chris Lattner2ae054a2007-01-30 23:45:45 +00001153}
1154
David Majnemerd536f232016-07-29 03:27:26 +00001155Constant *llvm::ConstantFoldConstant(const Constant *C, const DataLayout &DL,
1156 const TargetLibraryInfo *TLI) {
1157 SmallDenseMap<Constant *, Constant *> FoldedOps;
1158 return ConstantFoldConstantImpl(C, DL, TLI, FoldedOps);
Benjamin Kramer89ca4bc2013-04-13 12:53:18 +00001159}
1160
Manuel Jacobe9024592016-01-21 06:33:22 +00001161Constant *llvm::ConstantFoldInstOperands(Instruction *I,
Jay Foadf4b14a22011-07-19 13:32:40 +00001162 ArrayRef<Constant *> Ops,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001163 const DataLayout &DL,
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001164 const TargetLibraryInfo *TLI) {
David Majnemera926b3e2016-07-29 03:27:33 +00001165 return ConstantFoldInstOperandsImpl(I, I->getOpcode(), Ops, DL, TLI);
Chris Lattner2ae054a2007-01-30 23:45:45 +00001166}
1167
Chris Lattnerd2265b42007-12-10 22:53:04 +00001168Constant *llvm::ConstantFoldCompareInstOperands(unsigned Predicate,
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001169 Constant *Ops0, Constant *Ops1,
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001170 const DataLayout &DL,
Chad Rosierc24b86f2011-12-01 03:08:23 +00001171 const TargetLibraryInfo *TLI) {
Chris Lattnerd2265b42007-12-10 22:53:04 +00001172 // fold: icmp (inttoptr x), null -> icmp x, 0
Craig Topperb23e7c72017-06-02 16:17:32 +00001173 // fold: icmp null, (inttoptr x) -> icmp 0, x
Chris Lattnerd2265b42007-12-10 22:53:04 +00001174 // fold: icmp (ptrtoint x), 0 -> icmp x, null
Craig Topperb23e7c72017-06-02 16:17:32 +00001175 // fold: icmp 0, (ptrtoint x) -> icmp null, x
Nick Lewyckyf6ccd252008-05-25 20:56:15 +00001176 // fold: icmp (inttoptr x), (inttoptr y) -> icmp trunc/zext x, trunc/zext y
Chris Lattnerd2265b42007-12-10 22:53:04 +00001177 // fold: icmp (ptrtoint x), (ptrtoint y) -> icmp x, y
1178 //
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001179 // FIXME: The following comment is out of data and the DataLayout is here now.
1180 // ConstantExpr::getCompare cannot do this, because it doesn't have DL
Chris Lattnerd2265b42007-12-10 22:53:04 +00001181 // around to know if bit truncation is happening.
David Majnemer90a97042016-07-13 04:22:12 +00001182 if (auto *CE0 = dyn_cast<ConstantExpr>(Ops0)) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001183 if (Ops1->isNullValue()) {
Chris Lattnerd2265b42007-12-10 22:53:04 +00001184 if (CE0->getOpcode() == Instruction::IntToPtr) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001185 Type *IntPtrTy = DL.getIntPtrType(CE0->getType());
Chris Lattnerd2265b42007-12-10 22:53:04 +00001186 // Convert the integer value to the right size to ensure we get the
1187 // proper extension or truncation.
Owen Anderson487375e2009-07-29 18:55:55 +00001188 Constant *C = ConstantExpr::getIntegerCast(CE0->getOperand(0),
Chris Lattnerd2265b42007-12-10 22:53:04 +00001189 IntPtrTy, false);
Chris Lattnercdfb80d2009-11-09 23:06:58 +00001190 Constant *Null = Constant::getNullValue(C->getType());
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001191 return ConstantFoldCompareInstOperands(Predicate, C, Null, DL, TLI);
Chris Lattnerd2265b42007-12-10 22:53:04 +00001192 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001193
Chris Lattnerd2265b42007-12-10 22:53:04 +00001194 // Only do this transformation if the int is intptrty in size, otherwise
1195 // there is a truncation or extension that we aren't modeling.
Matt Arsenault7a960a82013-08-20 21:20:04 +00001196 if (CE0->getOpcode() == Instruction::PtrToInt) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001197 Type *IntPtrTy = DL.getIntPtrType(CE0->getOperand(0)->getType());
Matt Arsenault7a960a82013-08-20 21:20:04 +00001198 if (CE0->getType() == IntPtrTy) {
1199 Constant *C = CE0->getOperand(0);
1200 Constant *Null = Constant::getNullValue(C->getType());
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001201 return ConstantFoldCompareInstOperands(Predicate, C, Null, DL, TLI);
Matt Arsenault7a960a82013-08-20 21:20:04 +00001202 }
Chris Lattnerd2265b42007-12-10 22:53:04 +00001203 }
1204 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001205
David Majnemer90a97042016-07-13 04:22:12 +00001206 if (auto *CE1 = dyn_cast<ConstantExpr>(Ops1)) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001207 if (CE0->getOpcode() == CE1->getOpcode()) {
Nick Lewyckyf6ccd252008-05-25 20:56:15 +00001208 if (CE0->getOpcode() == Instruction::IntToPtr) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001209 Type *IntPtrTy = DL.getIntPtrType(CE0->getType());
Matt Arsenault7a960a82013-08-20 21:20:04 +00001210
Nick Lewyckyf6ccd252008-05-25 20:56:15 +00001211 // Convert the integer value to the right size to ensure we get the
1212 // proper extension or truncation.
Owen Anderson487375e2009-07-29 18:55:55 +00001213 Constant *C0 = ConstantExpr::getIntegerCast(CE0->getOperand(0),
Nick Lewyckyf6ccd252008-05-25 20:56:15 +00001214 IntPtrTy, false);
Owen Anderson487375e2009-07-29 18:55:55 +00001215 Constant *C1 = ConstantExpr::getIntegerCast(CE1->getOperand(0),
Nick Lewyckyf6ccd252008-05-25 20:56:15 +00001216 IntPtrTy, false);
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001217 return ConstantFoldCompareInstOperands(Predicate, C0, C1, DL, TLI);
Nick Lewyckyf6ccd252008-05-25 20:56:15 +00001218 }
1219
Chandler Carruth7ec50852012-11-01 08:07:29 +00001220 // Only do this transformation if the int is intptrty in size, otherwise
1221 // there is a truncation or extension that we aren't modeling.
Matt Arsenault7a960a82013-08-20 21:20:04 +00001222 if (CE0->getOpcode() == Instruction::PtrToInt) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001223 Type *IntPtrTy = DL.getIntPtrType(CE0->getOperand(0)->getType());
Matt Arsenault7a960a82013-08-20 21:20:04 +00001224 if (CE0->getType() == IntPtrTy &&
1225 CE0->getOperand(0)->getType() == CE1->getOperand(0)->getType()) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001226 return ConstantFoldCompareInstOperands(
1227 Predicate, CE0->getOperand(0), CE1->getOperand(0), DL, TLI);
Matt Arsenault7a960a82013-08-20 21:20:04 +00001228 }
1229 }
Chris Lattnerd2265b42007-12-10 22:53:04 +00001230 }
1231 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001232
Chris Lattner8fb74c62010-01-02 01:22:23 +00001233 // icmp eq (or x, y), 0 -> (icmp eq x, 0) & (icmp eq y, 0)
1234 // icmp ne (or x, y), 0 -> (icmp ne x, 0) | (icmp ne y, 0)
1235 if ((Predicate == ICmpInst::ICMP_EQ || Predicate == ICmpInst::ICMP_NE) &&
1236 CE0->getOpcode() == Instruction::Or && Ops1->isNullValue()) {
Mehdi Aminia28d91d2015-03-10 02:37:25 +00001237 Constant *LHS = ConstantFoldCompareInstOperands(
1238 Predicate, CE0->getOperand(0), Ops1, DL, TLI);
1239 Constant *RHS = ConstantFoldCompareInstOperands(
1240 Predicate, CE0->getOperand(1), Ops1, DL, TLI);
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001241 unsigned OpC =
Chris Lattner8fb74c62010-01-02 01:22:23 +00001242 Predicate == ICmpInst::ICMP_EQ ? Instruction::And : Instruction::Or;
Manuel Jacoba61ca372016-01-21 06:26:35 +00001243 return ConstantFoldBinaryOpOperands(OpC, LHS, RHS, DL);
Chris Lattner8fb74c62010-01-02 01:22:23 +00001244 }
Craig Topperb23e7c72017-06-02 16:17:32 +00001245 } else if (isa<ConstantExpr>(Ops1)) {
1246 // If RHS is a constant expression, but the left side isn't, swap the
1247 // operands and try again.
1248 Predicate = ICmpInst::getSwappedPredicate((ICmpInst::Predicate)Predicate);
1249 return ConstantFoldCompareInstOperands(Predicate, Ops1, Ops0, DL, TLI);
Chris Lattnerd2265b42007-12-10 22:53:04 +00001250 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001251
Chris Lattnercdfb80d2009-11-09 23:06:58 +00001252 return ConstantExpr::getCompare(Predicate, Ops0, Ops1);
Chris Lattnerd2265b42007-12-10 22:53:04 +00001253}
1254
Manuel Jacoba61ca372016-01-21 06:26:35 +00001255Constant *llvm::ConstantFoldBinaryOpOperands(unsigned Opcode, Constant *LHS,
1256 Constant *RHS,
1257 const DataLayout &DL) {
1258 assert(Instruction::isBinaryOp(Opcode));
1259 if (isa<ConstantExpr>(LHS) || isa<ConstantExpr>(RHS))
1260 if (Constant *C = SymbolicallyEvaluateBinop(Opcode, LHS, RHS, DL))
1261 return C;
1262
1263 return ConstantExpr::get(Opcode, LHS, RHS);
1264}
Chris Lattnerd2265b42007-12-10 22:53:04 +00001265
Manuel Jacob925d0292016-01-21 06:31:08 +00001266Constant *llvm::ConstantFoldCastOperand(unsigned Opcode, Constant *C,
1267 Type *DestTy, const DataLayout &DL) {
1268 assert(Instruction::isCast(Opcode));
1269 switch (Opcode) {
1270 default:
1271 llvm_unreachable("Missing case");
1272 case Instruction::PtrToInt:
1273 // If the input is a inttoptr, eliminate the pair. This requires knowing
1274 // the width of a pointer, so it can't be done in ConstantExpr::getCast.
David Majnemer90a97042016-07-13 04:22:12 +00001275 if (auto *CE = dyn_cast<ConstantExpr>(C)) {
Manuel Jacob925d0292016-01-21 06:31:08 +00001276 if (CE->getOpcode() == Instruction::IntToPtr) {
1277 Constant *Input = CE->getOperand(0);
1278 unsigned InWidth = Input->getType()->getScalarSizeInBits();
1279 unsigned PtrWidth = DL.getPointerTypeSizeInBits(CE->getType());
1280 if (PtrWidth < InWidth) {
1281 Constant *Mask =
1282 ConstantInt::get(CE->getContext(),
1283 APInt::getLowBitsSet(InWidth, PtrWidth));
1284 Input = ConstantExpr::getAnd(Input, Mask);
1285 }
1286 // Do a zext or trunc to get to the dest size.
1287 return ConstantExpr::getIntegerCast(Input, DestTy, false);
1288 }
1289 }
1290 return ConstantExpr::getCast(Opcode, C, DestTy);
1291 case Instruction::IntToPtr:
1292 // If the input is a ptrtoint, turn the pair into a ptr to ptr bitcast if
1293 // the int size is >= the ptr size and the address spaces are the same.
1294 // This requires knowing the width of a pointer, so it can't be done in
1295 // ConstantExpr::getCast.
David Majnemer90a97042016-07-13 04:22:12 +00001296 if (auto *CE = dyn_cast<ConstantExpr>(C)) {
Manuel Jacob925d0292016-01-21 06:31:08 +00001297 if (CE->getOpcode() == Instruction::PtrToInt) {
1298 Constant *SrcPtr = CE->getOperand(0);
1299 unsigned SrcPtrSize = DL.getPointerTypeSizeInBits(SrcPtr->getType());
1300 unsigned MidIntSize = CE->getType()->getScalarSizeInBits();
1301
1302 if (MidIntSize >= SrcPtrSize) {
1303 unsigned SrcAS = SrcPtr->getType()->getPointerAddressSpace();
1304 if (SrcAS == DestTy->getPointerAddressSpace())
1305 return FoldBitCast(CE->getOperand(0), DestTy, DL);
1306 }
1307 }
1308 }
1309
1310 return ConstantExpr::getCast(Opcode, C, DestTy);
1311 case Instruction::Trunc:
1312 case Instruction::ZExt:
1313 case Instruction::SExt:
1314 case Instruction::FPTrunc:
1315 case Instruction::FPExt:
1316 case Instruction::UIToFP:
1317 case Instruction::SIToFP:
1318 case Instruction::FPToUI:
1319 case Instruction::FPToSI:
1320 case Instruction::AddrSpaceCast:
1321 return ConstantExpr::getCast(Opcode, C, DestTy);
1322 case Instruction::BitCast:
1323 return FoldBitCast(C, DestTy, DL);
1324 }
1325}
1326
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001327Constant *llvm::ConstantFoldLoadThroughGEPConstantExpr(Constant *C,
Dan Gohmane525d9d2009-10-05 16:36:26 +00001328 ConstantExpr *CE) {
Chris Lattnerf488b352012-01-24 05:43:50 +00001329 if (!CE->getOperand(1)->isNullValue())
Craig Topper9f008862014-04-15 04:59:12 +00001330 return nullptr; // Do not allow stepping over the value!
Chris Lattner67058832012-01-25 06:48:06 +00001331
1332 // Loop over all of the operands, tracking down which value we are
1333 // addressing.
1334 for (unsigned i = 2, e = CE->getNumOperands(); i != e; ++i) {
1335 C = C->getAggregateElement(CE->getOperand(i));
Craig Topper9f008862014-04-15 04:59:12 +00001336 if (!C)
1337 return nullptr;
Chris Lattner67058832012-01-25 06:48:06 +00001338 }
1339 return C;
Chris Lattnerf488b352012-01-24 05:43:50 +00001340}
1341
David Majnemer90a97042016-07-13 04:22:12 +00001342Constant *
1343llvm::ConstantFoldLoadThroughGEPIndices(Constant *C,
1344 ArrayRef<Constant *> Indices) {
Chris Lattner2ae054a2007-01-30 23:45:45 +00001345 // Loop over all of the operands, tracking down which value we are
Chris Lattnerf488b352012-01-24 05:43:50 +00001346 // addressing.
David Majnemer90a97042016-07-13 04:22:12 +00001347 for (Constant *Index : Indices) {
1348 C = C->getAggregateElement(Index);
Craig Topper9f008862014-04-15 04:59:12 +00001349 if (!C)
1350 return nullptr;
Chris Lattnerf488b352012-01-24 05:43:50 +00001351 }
Chris Lattner2ae054a2007-01-30 23:45:45 +00001352 return C;
1353}
1354
Chris Lattner2ae054a2007-01-30 23:45:45 +00001355//===----------------------------------------------------------------------===//
1356// Constant Folding for Calls
1357//
John Criswell970af112005-10-27 16:00:10 +00001358
Andrew Kaylor647025f2017-06-09 23:18:11 +00001359bool llvm::canConstantFoldCallTo(ImmutableCallSite CS, const Function *F) {
Andrew Kaylor53a5fbb2017-08-14 21:15:13 +00001360 if (CS.isNoBuiltin() || CS.isStrictFP())
Andrew Kaylor647025f2017-06-09 23:18:11 +00001361 return false;
John Criswell970af112005-10-27 16:00:10 +00001362 switch (F->getIntrinsicID()) {
Owen Andersond4ebfd82013-02-06 22:43:31 +00001363 case Intrinsic::fabs:
Matt Arsenaultd6511b42014-10-21 23:00:20 +00001364 case Intrinsic::minnum:
1365 case Intrinsic::maxnum:
Thomas Livelyfa54e562018-10-19 18:15:32 +00001366 case Intrinsic::minimum:
1367 case Intrinsic::maximum:
Owen Andersond4ebfd82013-02-06 22:43:31 +00001368 case Intrinsic::log:
1369 case Intrinsic::log2:
1370 case Intrinsic::log10:
1371 case Intrinsic::exp:
1372 case Intrinsic::exp2:
1373 case Intrinsic::floor:
Karthik Bhat195e9dd2014-03-24 04:36:06 +00001374 case Intrinsic::ceil:
Dale Johannesen4d4e77a2007-10-02 17:43:59 +00001375 case Intrinsic::sqrt:
Karthik Bhatd2bc0d82015-07-08 03:55:47 +00001376 case Intrinsic::sin:
1377 case Intrinsic::cos:
Karthik Bhatd818e382015-07-21 08:52:23 +00001378 case Intrinsic::trunc:
1379 case Intrinsic::rint:
1380 case Intrinsic::nearbyint:
Chad Rosier0155a632011-12-03 00:00:03 +00001381 case Intrinsic::pow:
Dale Johannesen4d4e77a2007-10-02 17:43:59 +00001382 case Intrinsic::powi:
Reid Spencer6bba6c82007-04-01 07:35:23 +00001383 case Intrinsic::bswap:
1384 case Intrinsic::ctpop:
1385 case Intrinsic::ctlz:
1386 case Intrinsic::cttz:
Sanjay Patel411b8602018-08-17 13:23:44 +00001387 case Intrinsic::fshl:
1388 case Intrinsic::fshr:
Matt Arsenault83778582014-03-05 00:02:00 +00001389 case Intrinsic::fma:
1390 case Intrinsic::fmuladd:
Karthik Bhatdaa8cd12014-03-06 05:32:52 +00001391 case Intrinsic::copysign:
Piotr Padlewskia26a08c2018-05-18 23:52:57 +00001392 case Intrinsic::launder_invariant_group:
Piotr Padlewski5b3db452018-07-02 04:49:30 +00001393 case Intrinsic::strip_invariant_group:
Karthik Bhatb67688a2014-03-07 04:36:21 +00001394 case Intrinsic::round:
David Majnemer7f781ab2016-07-14 00:29:50 +00001395 case Intrinsic::masked_load:
Evan Phoenix44e5dbc2009-10-05 22:53:52 +00001396 case Intrinsic::sadd_with_overflow:
Frits van Bommel0bb2ad22011-03-27 14:26:13 +00001397 case Intrinsic::uadd_with_overflow:
Evan Phoenix44e5dbc2009-10-05 22:53:52 +00001398 case Intrinsic::ssub_with_overflow:
Frits van Bommel0bb2ad22011-03-27 14:26:13 +00001399 case Intrinsic::usub_with_overflow:
Chris Lattner698661c2010-10-14 00:05:07 +00001400 case Intrinsic::smul_with_overflow:
Frits van Bommel0bb2ad22011-03-27 14:26:13 +00001401 case Intrinsic::umul_with_overflow:
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001402 case Intrinsic::convert_from_fp16:
1403 case Intrinsic::convert_to_fp16:
Matt Arsenault155dda92016-03-21 15:00:35 +00001404 case Intrinsic::bitreverse:
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001405 case Intrinsic::x86_sse_cvtss2si:
1406 case Intrinsic::x86_sse_cvtss2si64:
1407 case Intrinsic::x86_sse_cvttss2si:
1408 case Intrinsic::x86_sse_cvttss2si64:
1409 case Intrinsic::x86_sse2_cvtsd2si:
1410 case Intrinsic::x86_sse2_cvtsd2si64:
1411 case Intrinsic::x86_sse2_cvttsd2si:
1412 case Intrinsic::x86_sse2_cvttsd2si64:
Craig Topper484b3422018-08-12 22:09:54 +00001413 case Intrinsic::x86_avx512_vcvtss2si32:
1414 case Intrinsic::x86_avx512_vcvtss2si64:
1415 case Intrinsic::x86_avx512_cvttss2si:
1416 case Intrinsic::x86_avx512_cvttss2si64:
1417 case Intrinsic::x86_avx512_vcvtsd2si32:
1418 case Intrinsic::x86_avx512_vcvtsd2si64:
1419 case Intrinsic::x86_avx512_cvttsd2si:
1420 case Intrinsic::x86_avx512_cvttsd2si64:
1421 case Intrinsic::x86_avx512_vcvtss2usi32:
1422 case Intrinsic::x86_avx512_vcvtss2usi64:
1423 case Intrinsic::x86_avx512_cvttss2usi:
1424 case Intrinsic::x86_avx512_cvttss2usi64:
1425 case Intrinsic::x86_avx512_vcvtsd2usi32:
1426 case Intrinsic::x86_avx512_vcvtsd2usi64:
1427 case Intrinsic::x86_avx512_cvttsd2usi:
1428 case Intrinsic::x86_avx512_cvttsd2usi64:
James Y Knight72f76bf2018-11-07 15:24:12 +00001429 case Intrinsic::is_constant:
John Criswell970af112005-10-27 16:00:10 +00001430 return true;
Chris Lattner9ca7c092009-10-05 05:00:35 +00001431 default:
1432 return false;
Craig Topper492db482017-04-07 21:36:32 +00001433 case Intrinsic::not_intrinsic: break;
John Criswell970af112005-10-27 16:00:10 +00001434 }
1435
Matt Arsenaulta5e56982013-08-12 22:56:15 +00001436 if (!F->hasName())
1437 return false;
Daniel Dunbarca414c72009-07-26 08:34:35 +00001438 StringRef Name = F->getName();
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001439
Chris Lattner785f9982007-08-08 06:55:43 +00001440 // In these cases, the check of the length is required. We don't want to
1441 // return true for a name like "cos\0blah" which strcmp would return equal to
1442 // "cos", but has length 8.
Daniel Dunbarca414c72009-07-26 08:34:35 +00001443 switch (Name[0]) {
Erik Schnetter5e93e282015-08-27 19:56:57 +00001444 default:
1445 return false;
Chris Lattner785f9982007-08-08 06:55:43 +00001446 case 'a':
Erik Schnetter5e93e282015-08-27 19:56:57 +00001447 return Name == "acos" || Name == "asin" || Name == "atan" ||
1448 Name == "atan2" || Name == "acosf" || Name == "asinf" ||
1449 Name == "atanf" || Name == "atan2f";
Chris Lattner785f9982007-08-08 06:55:43 +00001450 case 'c':
Erik Schnetter5e93e282015-08-27 19:56:57 +00001451 return Name == "ceil" || Name == "cos" || Name == "cosh" ||
1452 Name == "ceilf" || Name == "cosf" || Name == "coshf";
Chris Lattner785f9982007-08-08 06:55:43 +00001453 case 'e':
Erik Schnetter5e93e282015-08-27 19:56:57 +00001454 return Name == "exp" || Name == "exp2" || Name == "expf" || Name == "exp2f";
Chris Lattner785f9982007-08-08 06:55:43 +00001455 case 'f':
Erik Schnetter5e93e282015-08-27 19:56:57 +00001456 return Name == "fabs" || Name == "floor" || Name == "fmod" ||
1457 Name == "fabsf" || Name == "floorf" || Name == "fmodf";
Chris Lattner785f9982007-08-08 06:55:43 +00001458 case 'l':
Xin Tongc063c3f2017-10-01 00:09:53 +00001459 return Name == "log" || Name == "log10" || Name == "logf" ||
Erik Schnetter5e93e282015-08-27 19:56:57 +00001460 Name == "log10f";
Chris Lattner785f9982007-08-08 06:55:43 +00001461 case 'p':
Erik Schnetter5e93e282015-08-27 19:56:57 +00001462 return Name == "pow" || Name == "powf";
Bryant Wongb5e03b62016-12-26 14:29:29 +00001463 case 'r':
1464 return Name == "round" || Name == "roundf";
Chris Lattner785f9982007-08-08 06:55:43 +00001465 case 's':
Daniel Dunbarca414c72009-07-26 08:34:35 +00001466 return Name == "sin" || Name == "sinh" || Name == "sqrt" ||
Erik Schnetter5e93e282015-08-27 19:56:57 +00001467 Name == "sinf" || Name == "sinhf" || Name == "sqrtf";
Chris Lattner785f9982007-08-08 06:55:43 +00001468 case 't':
Erik Schnetter5e93e282015-08-27 19:56:57 +00001469 return Name == "tan" || Name == "tanh" || Name == "tanf" || Name == "tanhf";
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001470 case '_':
1471
1472 // Check for various function names that get used for the math functions
1473 // when the header files are preprocessed with the macro
1474 // __FINITE_MATH_ONLY__ enabled.
1475 // The '12' here is the length of the shortest name that can match.
1476 // We need to check the size before looking at Name[1] and Name[2]
1477 // so we may as well check a limit that will eliminate mismatches.
1478 if (Name.size() < 12 || Name[1] != '_')
1479 return false;
1480 switch (Name[2]) {
1481 default:
1482 return false;
1483 case 'a':
1484 return Name == "__acos_finite" || Name == "__acosf_finite" ||
1485 Name == "__asin_finite" || Name == "__asinf_finite" ||
1486 Name == "__atan2_finite" || Name == "__atan2f_finite";
1487 case 'c':
1488 return Name == "__cosh_finite" || Name == "__coshf_finite";
1489 case 'e':
1490 return Name == "__exp_finite" || Name == "__expf_finite" ||
1491 Name == "__exp2_finite" || Name == "__exp2f_finite";
1492 case 'l':
1493 return Name == "__log_finite" || Name == "__logf_finite" ||
1494 Name == "__log10_finite" || Name == "__log10f_finite";
1495 case 'p':
1496 return Name == "__pow_finite" || Name == "__powf_finite";
1497 case 's':
1498 return Name == "__sinh_finite" || Name == "__sinhf_finite";
1499 }
John Criswell970af112005-10-27 16:00:10 +00001500 }
1501}
1502
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001503namespace {
1504
1505Constant *GetConstantFoldFPValue(double V, Type *Ty) {
Owen Andersond4ebfd82013-02-06 22:43:31 +00001506 if (Ty->isHalfTy()) {
1507 APFloat APF(V);
1508 bool unused;
Stephan Bergmann17c7f702016-12-14 11:57:17 +00001509 APF.convert(APFloat::IEEEhalf(), APFloat::rmNearestTiesToEven, &unused);
Owen Andersond4ebfd82013-02-06 22:43:31 +00001510 return ConstantFP::get(Ty->getContext(), APF);
1511 }
Chris Lattner351534f2009-10-05 05:06:24 +00001512 if (Ty->isFloatTy())
Chris Lattner46b5c642009-11-06 04:27:31 +00001513 return ConstantFP::get(Ty->getContext(), APFloat((float)V));
Chris Lattner351534f2009-10-05 05:06:24 +00001514 if (Ty->isDoubleTy())
Chris Lattner46b5c642009-11-06 04:27:31 +00001515 return ConstantFP::get(Ty->getContext(), APFloat(V));
Xin Tongc063c3f2017-10-01 00:09:53 +00001516 llvm_unreachable("Can only constant fold half/float/double");
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001517}
1518
Sanjay Patel0d7dee62014-10-02 15:13:22 +00001519/// Clear the floating-point exception state.
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001520inline void llvm_fenv_clearexcept() {
Alp Toker51420a82014-06-09 19:00:52 +00001521#if defined(HAVE_FENV_H) && HAVE_DECL_FE_ALL_EXCEPT
Alp Tokerc817d6a2014-06-09 18:28:53 +00001522 feclearexcept(FE_ALL_EXCEPT);
1523#endif
1524 errno = 0;
1525}
1526
Sanjay Patel0d7dee62014-10-02 15:13:22 +00001527/// Test if a floating-point exception was raised.
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001528inline bool llvm_fenv_testexcept() {
Alp Tokerc817d6a2014-06-09 18:28:53 +00001529 int errno_val = errno;
1530 if (errno_val == ERANGE || errno_val == EDOM)
1531 return true;
Alp Toker51420a82014-06-09 19:00:52 +00001532#if defined(HAVE_FENV_H) && HAVE_DECL_FE_ALL_EXCEPT && HAVE_DECL_FE_INEXACT
Alp Tokerc817d6a2014-06-09 18:28:53 +00001533 if (fetestexcept(FE_ALL_EXCEPT & ~FE_INEXACT))
1534 return true;
1535#endif
1536 return false;
1537}
Alp Tokerc817d6a2014-06-09 18:28:53 +00001538
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001539Constant *ConstantFoldFP(double (*NativeFP)(double), double V, Type *Ty) {
Alp Tokerc817d6a2014-06-09 18:28:53 +00001540 llvm_fenv_clearexcept();
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001541 V = NativeFP(V);
Alp Tokerc817d6a2014-06-09 18:28:53 +00001542 if (llvm_fenv_testexcept()) {
1543 llvm_fenv_clearexcept();
Craig Topper9f008862014-04-15 04:59:12 +00001544 return nullptr;
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001545 }
1546
1547 return GetConstantFoldFPValue(V, Ty);
John Criswell970af112005-10-27 16:00:10 +00001548}
1549
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001550Constant *ConstantFoldBinaryFP(double (*NativeFP)(double, double), double V,
1551 double W, Type *Ty) {
Alp Tokerc817d6a2014-06-09 18:28:53 +00001552 llvm_fenv_clearexcept();
Dan Gohman72efc042007-07-16 15:26:22 +00001553 V = NativeFP(V, W);
Alp Tokerc817d6a2014-06-09 18:28:53 +00001554 if (llvm_fenv_testexcept()) {
1555 llvm_fenv_clearexcept();
Craig Topper9f008862014-04-15 04:59:12 +00001556 return nullptr;
Dale Johannesenbed9dc42007-09-06 18:13:44 +00001557 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001558
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001559 return GetConstantFoldFPValue(V, Ty);
Dan Gohman72efc042007-07-16 15:26:22 +00001560}
1561
Sanjay Patel0d7dee62014-10-02 15:13:22 +00001562/// Attempt to fold an SSE floating point to integer conversion of a constant
1563/// floating point. If roundTowardZero is false, the default IEEE rounding is
1564/// used (toward nearest, ties to even). This matches the behavior of the
1565/// non-truncating SSE instructions in the default rounding mode. The desired
1566/// integer type Ty is used to select how many bits are available for the
1567/// result. Returns null if the conversion cannot be performed, otherwise
1568/// returns the Constant value resulting from the conversion.
Simon Pilgrim0ea8d272016-07-19 15:07:43 +00001569Constant *ConstantFoldSSEConvertToInt(const APFloat &Val, bool roundTowardZero,
Craig Topper484b3422018-08-12 22:09:54 +00001570 Type *Ty, bool IsSigned) {
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001571 // All of these conversion intrinsics form an integer of at most 64bits.
Matt Arsenault8c789092013-08-12 23:15:58 +00001572 unsigned ResultWidth = Ty->getIntegerBitWidth();
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001573 assert(ResultWidth <= 64 &&
1574 "Can only constant fold conversions to 64 and 32 bit ints");
1575
1576 uint64_t UIntVal;
1577 bool isExact = false;
1578 APFloat::roundingMode mode = roundTowardZero? APFloat::rmTowardZero
1579 : APFloat::rmNearestTiesToEven;
Simon Pilgrim00b34992017-03-20 14:40:12 +00001580 APFloat::opStatus status =
1581 Val.convertToInteger(makeMutableArrayRef(UIntVal), ResultWidth,
Craig Topper484b3422018-08-12 22:09:54 +00001582 IsSigned, mode, &isExact);
Simon Pilgrim0ea8d272016-07-19 15:07:43 +00001583 if (status != APFloat::opOK &&
1584 (!roundTowardZero || status != APFloat::opInexact))
Craig Topper9f008862014-04-15 04:59:12 +00001585 return nullptr;
Craig Topper484b3422018-08-12 22:09:54 +00001586 return ConstantInt::get(Ty, UIntVal, IsSigned);
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001587}
1588
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001589double getValueAsDouble(ConstantFP *Op) {
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001590 Type *Ty = Op->getType();
1591
1592 if (Ty->isFloatTy())
1593 return Op->getValueAPF().convertToFloat();
1594
1595 if (Ty->isDoubleTy())
1596 return Op->getValueAPF().convertToDouble();
1597
1598 bool unused;
1599 APFloat APF = Op->getValueAPF();
Stephan Bergmann17c7f702016-12-14 11:57:17 +00001600 APF.convert(APFloat::IEEEdouble(), APFloat::rmNearestTiesToEven, &unused);
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001601 return APF.convertToDouble();
1602}
1603
James Y Knight72f76bf2018-11-07 15:24:12 +00001604static bool isManifestConstant(const Constant *c) {
1605 if (isa<ConstantData>(c)) {
1606 return true;
1607 } else if (isa<ConstantAggregate>(c) || isa<ConstantExpr>(c)) {
1608 for (const Value *subc : c->operand_values()) {
1609 if (!isManifestConstant(cast<Constant>(subc)))
1610 return false;
1611 }
1612 return true;
1613 }
1614 return false;
1615}
1616
Eugene Zelenko35623fb2016-03-28 17:40:08 +00001617Constant *ConstantFoldScalarCall(StringRef Name, unsigned IntrinsicID, Type *Ty,
1618 ArrayRef<Constant *> Operands,
Manoj Gupta77eeac32018-07-09 22:27:23 +00001619 const TargetLibraryInfo *TLI,
1620 ImmutableCallSite CS) {
Jay Foadf4b14a22011-07-19 13:32:40 +00001621 if (Operands.size() == 1) {
James Y Knight72f76bf2018-11-07 15:24:12 +00001622 if (IntrinsicID == Intrinsic::is_constant) {
1623 // We know we have a "Constant" argument. But we want to only
1624 // return true for manifest constants, not those that depend on
1625 // constants with unknowable values, e.g. GlobalValue or BlockAddress.
1626 if (isManifestConstant(Operands[0]))
1627 return ConstantInt::getTrue(Ty->getContext());
1628 return nullptr;
1629 }
Sanjoy Das87b9e1b2016-04-08 18:21:11 +00001630 if (isa<UndefValue>(Operands[0])) {
1631 // cosine(arg) is between -1 and 1. cosine(invalid arg) is NaN
1632 if (IntrinsicID == Intrinsic::cos)
1633 return Constant::getNullValue(Ty);
Craig Topperd470d732017-06-04 08:21:53 +00001634 if (IntrinsicID == Intrinsic::bswap ||
Piotr Padlewskia26a08c2018-05-18 23:52:57 +00001635 IntrinsicID == Intrinsic::bitreverse ||
Piotr Padlewski5b3db452018-07-02 04:49:30 +00001636 IntrinsicID == Intrinsic::launder_invariant_group ||
1637 IntrinsicID == Intrinsic::strip_invariant_group)
Craig Topperd470d732017-06-04 08:21:53 +00001638 return Operands[0];
Sanjoy Das87b9e1b2016-04-08 18:21:11 +00001639 }
Piotr Padlewskia26a08c2018-05-18 23:52:57 +00001640
Manoj Guptaf9f50f62018-07-23 21:20:00 +00001641 if (isa<ConstantPointerNull>(Operands[0])) {
Piotr Padlewski5b3db452018-07-02 04:49:30 +00001642 // launder(null) == null == strip(null) iff in addrspace 0
1643 if (IntrinsicID == Intrinsic::launder_invariant_group ||
Manoj Guptaf9f50f62018-07-23 21:20:00 +00001644 IntrinsicID == Intrinsic::strip_invariant_group) {
1645 // If instruction is not yet put in a basic block (e.g. when cloning
1646 // a function during inlining), CS caller may not be available.
1647 // So check CS's BB first before querying CS.getCaller.
1648 const Function *Caller = CS.getParent() ? CS.getCaller() : nullptr;
1649 if (Caller &&
1650 !NullPointerIsDefined(
1651 Caller, Operands[0]->getType()->getPointerAddressSpace())) {
1652 return Operands[0];
1653 }
1654 return nullptr;
1655 }
Piotr Padlewskia26a08c2018-05-18 23:52:57 +00001656 }
1657
David Majnemer90a97042016-07-13 04:22:12 +00001658 if (auto *Op = dyn_cast<ConstantFP>(Operands[0])) {
Benjamin Kramer061d1472014-03-05 19:41:48 +00001659 if (IntrinsicID == Intrinsic::convert_to_fp16) {
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001660 APFloat Val(Op->getValueAPF());
1661
1662 bool lost = false;
Stephan Bergmann17c7f702016-12-14 11:57:17 +00001663 Val.convert(APFloat::IEEEhalf(), APFloat::rmNearestTiesToEven, &lost);
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001664
Benjamin Kramer061d1472014-03-05 19:41:48 +00001665 return ConstantInt::get(Ty->getContext(), Val.bitcastToAPInt());
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001666 }
1667
Xin Tongc063c3f2017-10-01 00:09:53 +00001668 if (!Ty->isHalfTy() && !Ty->isFloatTy() && !Ty->isDoubleTy())
Craig Topper9f008862014-04-15 04:59:12 +00001669 return nullptr;
Jakob Stoklund Olesen10835732010-09-27 21:29:20 +00001670
Karthik Bhatb67688a2014-03-07 04:36:21 +00001671 if (IntrinsicID == Intrinsic::round) {
1672 APFloat V = Op->getValueAPF();
1673 V.roundToIntegral(APFloat::rmNearestTiesToAway);
1674 return ConstantFP::get(Ty->getContext(), V);
1675 }
1676
Karthik Bhatd818e382015-07-21 08:52:23 +00001677 if (IntrinsicID == Intrinsic::floor) {
1678 APFloat V = Op->getValueAPF();
1679 V.roundToIntegral(APFloat::rmTowardNegative);
1680 return ConstantFP::get(Ty->getContext(), V);
1681 }
1682
1683 if (IntrinsicID == Intrinsic::ceil) {
1684 APFloat V = Op->getValueAPF();
1685 V.roundToIntegral(APFloat::rmTowardPositive);
1686 return ConstantFP::get(Ty->getContext(), V);
1687 }
1688
1689 if (IntrinsicID == Intrinsic::trunc) {
1690 APFloat V = Op->getValueAPF();
1691 V.roundToIntegral(APFloat::rmTowardZero);
1692 return ConstantFP::get(Ty->getContext(), V);
1693 }
1694
1695 if (IntrinsicID == Intrinsic::rint) {
1696 APFloat V = Op->getValueAPF();
1697 V.roundToIntegral(APFloat::rmNearestTiesToEven);
1698 return ConstantFP::get(Ty->getContext(), V);
1699 }
1700
1701 if (IntrinsicID == Intrinsic::nearbyint) {
1702 APFloat V = Op->getValueAPF();
1703 V.roundToIntegral(APFloat::rmNearestTiesToEven);
1704 return ConstantFP::get(Ty->getContext(), V);
1705 }
1706
Jakob Stoklund Olesen10835732010-09-27 21:29:20 +00001707 /// We only fold functions with finite arguments. Folding NaN and inf is
1708 /// likely to be aborted with an exception anyway, and some host libms
1709 /// have known errors raising exceptions.
1710 if (Op->getValueAPF().isNaN() || Op->getValueAPF().isInfinity())
Craig Topper9f008862014-04-15 04:59:12 +00001711 return nullptr;
Jakob Stoklund Olesen10835732010-09-27 21:29:20 +00001712
Dale Johannesenbed9dc42007-09-06 18:13:44 +00001713 /// Currently APFloat versions of these functions do not exist, so we use
1714 /// the host native double versions. Float versions are not called
1715 /// directly but for all these it is true (float)(f((double)arg)) ==
1716 /// f(arg). Long double not supported yet.
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001717 double V = getValueAsDouble(Op);
Owen Andersond4ebfd82013-02-06 22:43:31 +00001718
Benjamin Kramer061d1472014-03-05 19:41:48 +00001719 switch (IntrinsicID) {
Owen Andersond4ebfd82013-02-06 22:43:31 +00001720 default: break;
1721 case Intrinsic::fabs:
1722 return ConstantFoldFP(fabs, V, Ty);
1723 case Intrinsic::log2:
Vince Harrond5281122015-05-07 00:05:26 +00001724 return ConstantFoldFP(Log2, V, Ty);
Owen Andersond4ebfd82013-02-06 22:43:31 +00001725 case Intrinsic::log:
1726 return ConstantFoldFP(log, V, Ty);
1727 case Intrinsic::log10:
1728 return ConstantFoldFP(log10, V, Ty);
1729 case Intrinsic::exp:
1730 return ConstantFoldFP(exp, V, Ty);
1731 case Intrinsic::exp2:
1732 return ConstantFoldFP(exp2, V, Ty);
Karthik Bhatd2bc0d82015-07-08 03:55:47 +00001733 case Intrinsic::sin:
1734 return ConstantFoldFP(sin, V, Ty);
1735 case Intrinsic::cos:
1736 return ConstantFoldFP(cos, V, Ty);
Justin Lebar8b18a342017-01-21 00:59:57 +00001737 case Intrinsic::sqrt:
1738 return ConstantFoldFP(sqrt, V, Ty);
Owen Andersond4ebfd82013-02-06 22:43:31 +00001739 }
1740
Benjamin Kramer061d1472014-03-05 19:41:48 +00001741 if (!TLI)
Craig Topper9f008862014-04-15 04:59:12 +00001742 return nullptr;
Benjamin Kramer061d1472014-03-05 19:41:48 +00001743
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001744 char NameKeyChar = Name[0];
1745 if (Name[0] == '_' && Name.size() > 2 && Name[1] == '_')
1746 NameKeyChar = Name[2];
1747
1748 switch (NameKeyChar) {
Chris Lattner785f9982007-08-08 06:55:43 +00001749 case 'a':
David L. Jonesd21529f2017-01-23 23:16:46 +00001750 if ((Name == "acos" && TLI->has(LibFunc_acos)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001751 (Name == "acosf" && TLI->has(LibFunc_acosf)) ||
1752 (Name == "__acos_finite" && TLI->has(LibFunc_acos_finite)) ||
1753 (Name == "__acosf_finite" && TLI->has(LibFunc_acosf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001754 return ConstantFoldFP(acos, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001755 else if ((Name == "asin" && TLI->has(LibFunc_asin)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001756 (Name == "asinf" && TLI->has(LibFunc_asinf)) ||
1757 (Name == "__asin_finite" && TLI->has(LibFunc_asin_finite)) ||
1758 (Name == "__asinf_finite" && TLI->has(LibFunc_asinf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001759 return ConstantFoldFP(asin, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001760 else if ((Name == "atan" && TLI->has(LibFunc_atan)) ||
1761 (Name == "atanf" && TLI->has(LibFunc_atanf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001762 return ConstantFoldFP(atan, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001763 break;
1764 case 'c':
David L. Jonesd21529f2017-01-23 23:16:46 +00001765 if ((Name == "ceil" && TLI->has(LibFunc_ceil)) ||
1766 (Name == "ceilf" && TLI->has(LibFunc_ceilf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001767 return ConstantFoldFP(ceil, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001768 else if ((Name == "cos" && TLI->has(LibFunc_cos)) ||
1769 (Name == "cosf" && TLI->has(LibFunc_cosf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001770 return ConstantFoldFP(cos, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001771 else if ((Name == "cosh" && TLI->has(LibFunc_cosh)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001772 (Name == "coshf" && TLI->has(LibFunc_coshf)) ||
1773 (Name == "__cosh_finite" && TLI->has(LibFunc_cosh_finite)) ||
1774 (Name == "__coshf_finite" && TLI->has(LibFunc_coshf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001775 return ConstantFoldFP(cosh, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001776 break;
1777 case 'e':
David L. Jonesd21529f2017-01-23 23:16:46 +00001778 if ((Name == "exp" && TLI->has(LibFunc_exp)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001779 (Name == "expf" && TLI->has(LibFunc_expf)) ||
1780 (Name == "__exp_finite" && TLI->has(LibFunc_exp_finite)) ||
1781 (Name == "__expf_finite" && TLI->has(LibFunc_expf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001782 return ConstantFoldFP(exp, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001783 if ((Name == "exp2" && TLI->has(LibFunc_exp2)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001784 (Name == "exp2f" && TLI->has(LibFunc_exp2f)) ||
1785 (Name == "__exp2_finite" && TLI->has(LibFunc_exp2_finite)) ||
1786 (Name == "__exp2f_finite" && TLI->has(LibFunc_exp2f_finite)))
Chris Lattner713d5232011-05-22 22:22:35 +00001787 // Constant fold exp2(x) as pow(2,x) in case the host doesn't have a
1788 // C99 library.
1789 return ConstantFoldBinaryFP(pow, 2.0, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001790 break;
1791 case 'f':
David L. Jonesd21529f2017-01-23 23:16:46 +00001792 if ((Name == "fabs" && TLI->has(LibFunc_fabs)) ||
1793 (Name == "fabsf" && TLI->has(LibFunc_fabsf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001794 return ConstantFoldFP(fabs, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001795 else if ((Name == "floor" && TLI->has(LibFunc_floor)) ||
1796 (Name == "floorf" && TLI->has(LibFunc_floorf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001797 return ConstantFoldFP(floor, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001798 break;
1799 case 'l':
David L. Jonesd21529f2017-01-23 23:16:46 +00001800 if ((Name == "log" && V > 0 && TLI->has(LibFunc_log)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001801 (Name == "logf" && V > 0 && TLI->has(LibFunc_logf)) ||
1802 (Name == "__log_finite" && V > 0 &&
1803 TLI->has(LibFunc_log_finite)) ||
1804 (Name == "__logf_finite" && V > 0 &&
1805 TLI->has(LibFunc_logf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001806 return ConstantFoldFP(log, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001807 else if ((Name == "log10" && V > 0 && TLI->has(LibFunc_log10)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001808 (Name == "log10f" && V > 0 && TLI->has(LibFunc_log10f)) ||
1809 (Name == "__log10_finite" && V > 0 &&
1810 TLI->has(LibFunc_log10_finite)) ||
1811 (Name == "__log10f_finite" && V > 0 &&
1812 TLI->has(LibFunc_log10f_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001813 return ConstantFoldFP(log10, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001814 break;
Bryant Wongb5e03b62016-12-26 14:29:29 +00001815 case 'r':
David L. Jonesd21529f2017-01-23 23:16:46 +00001816 if ((Name == "round" && TLI->has(LibFunc_round)) ||
1817 (Name == "roundf" && TLI->has(LibFunc_roundf)))
Bryant Wongb5e03b62016-12-26 14:29:29 +00001818 return ConstantFoldFP(round, V, Ty);
Galina Kistanovac2b642d2017-05-31 20:25:13 +00001819 break;
Chris Lattner785f9982007-08-08 06:55:43 +00001820 case 's':
David L. Jonesd21529f2017-01-23 23:16:46 +00001821 if ((Name == "sin" && TLI->has(LibFunc_sin)) ||
1822 (Name == "sinf" && TLI->has(LibFunc_sinf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001823 return ConstantFoldFP(sin, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001824 else if ((Name == "sinh" && TLI->has(LibFunc_sinh)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001825 (Name == "sinhf" && TLI->has(LibFunc_sinhf)) ||
1826 (Name == "__sinh_finite" && TLI->has(LibFunc_sinh_finite)) ||
1827 (Name == "__sinhf_finite" && TLI->has(LibFunc_sinhf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001828 return ConstantFoldFP(sinh, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001829 else if ((Name == "sqrt" && V >= 0 && TLI->has(LibFunc_sqrt)) ||
1830 (Name == "sqrtf" && V >= 0 && TLI->has(LibFunc_sqrtf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001831 return ConstantFoldFP(sqrt, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001832 break;
1833 case 't':
David L. Jonesd21529f2017-01-23 23:16:46 +00001834 if ((Name == "tan" && TLI->has(LibFunc_tan)) ||
1835 (Name == "tanf" && TLI->has(LibFunc_tanf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001836 return ConstantFoldFP(tan, V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001837 else if ((Name == "tanh" && TLI->has(LibFunc_tanh)) ||
1838 (Name == "tanhf" && TLI->has(LibFunc_tanhf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001839 return ConstantFoldFP(tanh, V, Ty);
Chris Lattner785f9982007-08-08 06:55:43 +00001840 break;
1841 default:
1842 break;
John Criswell970af112005-10-27 16:00:10 +00001843 }
Craig Topper9f008862014-04-15 04:59:12 +00001844 return nullptr;
Chris Lattner9ca7c092009-10-05 05:00:35 +00001845 }
Chandler Carruth352d9b12011-01-10 09:02:58 +00001846
David Majnemer90a97042016-07-13 04:22:12 +00001847 if (auto *Op = dyn_cast<ConstantInt>(Operands[0])) {
Benjamin Kramer061d1472014-03-05 19:41:48 +00001848 switch (IntrinsicID) {
Chandler Carruth352d9b12011-01-10 09:02:58 +00001849 case Intrinsic::bswap:
Benjamin Kramer061d1472014-03-05 19:41:48 +00001850 return ConstantInt::get(Ty->getContext(), Op->getValue().byteSwap());
Chandler Carruth352d9b12011-01-10 09:02:58 +00001851 case Intrinsic::ctpop:
Owen Andersonedb4a702009-07-24 23:12:02 +00001852 return ConstantInt::get(Ty, Op->getValue().countPopulation());
Matt Arsenault155dda92016-03-21 15:00:35 +00001853 case Intrinsic::bitreverse:
1854 return ConstantInt::get(Ty->getContext(), Op->getValue().reverseBits());
Chandler Carruth352d9b12011-01-10 09:02:58 +00001855 case Intrinsic::convert_from_fp16: {
Stephan Bergmann17c7f702016-12-14 11:57:17 +00001856 APFloat Val(APFloat::IEEEhalf(), Op->getValue());
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001857
1858 bool lost = false;
Andrea Di Biagio29059992015-05-14 18:01:48 +00001859 APFloat::opStatus status = Val.convert(
1860 Ty->getFltSemantics(), APFloat::rmNearestTiesToEven, &lost);
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001861
1862 // Conversion is always precise.
Jeffrey Yasskin9b43f332010-12-23 00:58:24 +00001863 (void)status;
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001864 assert(status == APFloat::opOK && !lost &&
1865 "Precision lost during fp16 constfolding");
1866
Benjamin Kramer061d1472014-03-05 19:41:48 +00001867 return ConstantFP::get(Ty->getContext(), Val);
Anton Korobeynikov065232f2010-03-19 00:36:35 +00001868 }
Chandler Carruth352d9b12011-01-10 09:02:58 +00001869 default:
Craig Topper9f008862014-04-15 04:59:12 +00001870 return nullptr;
Chandler Carruth352d9b12011-01-10 09:02:58 +00001871 }
John Criswell970af112005-10-27 16:00:10 +00001872 }
Chandler Carruth352d9b12011-01-10 09:02:58 +00001873
Chris Lattner61a1d6c2012-01-26 21:37:55 +00001874 // Support ConstantVector in case we have an Undef in the top.
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001875 if (isa<ConstantVector>(Operands[0]) ||
Chris Lattner61a1d6c2012-01-26 21:37:55 +00001876 isa<ConstantDataVector>(Operands[0])) {
David Majnemer90a97042016-07-13 04:22:12 +00001877 auto *Op = cast<Constant>(Operands[0]);
Benjamin Kramer061d1472014-03-05 19:41:48 +00001878 switch (IntrinsicID) {
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001879 default: break;
1880 case Intrinsic::x86_sse_cvtss2si:
1881 case Intrinsic::x86_sse_cvtss2si64:
1882 case Intrinsic::x86_sse2_cvtsd2si:
1883 case Intrinsic::x86_sse2_cvtsd2si64:
Chris Lattner61a1d6c2012-01-26 21:37:55 +00001884 if (ConstantFP *FPOp =
Simon Pilgrim0ea8d272016-07-19 15:07:43 +00001885 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U)))
1886 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(),
Craig Topper484b3422018-08-12 22:09:54 +00001887 /*roundTowardZero=*/false, Ty,
1888 /*IsSigned*/true);
Craig Topper0dd29e22017-06-04 08:21:51 +00001889 break;
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001890 case Intrinsic::x86_sse_cvttss2si:
1891 case Intrinsic::x86_sse_cvttss2si64:
1892 case Intrinsic::x86_sse2_cvttsd2si:
1893 case Intrinsic::x86_sse2_cvttsd2si64:
Chris Lattner61a1d6c2012-01-26 21:37:55 +00001894 if (ConstantFP *FPOp =
Simon Pilgrim0ea8d272016-07-19 15:07:43 +00001895 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U)))
1896 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(),
Craig Topper484b3422018-08-12 22:09:54 +00001897 /*roundTowardZero=*/true, Ty,
1898 /*IsSigned*/true);
Craig Topper0dd29e22017-06-04 08:21:51 +00001899 break;
Chandler Carruthb1e7f552011-01-11 01:07:24 +00001900 }
1901 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001902
Craig Topper9f008862014-04-15 04:59:12 +00001903 return nullptr;
Chris Lattner9ca7c092009-10-05 05:00:35 +00001904 }
Chandler Carruth352d9b12011-01-10 09:02:58 +00001905
Jay Foadf4b14a22011-07-19 13:32:40 +00001906 if (Operands.size() == 2) {
David Majnemer90a97042016-07-13 04:22:12 +00001907 if (auto *Op1 = dyn_cast<ConstantFP>(Operands[0])) {
Owen Andersond4ebfd82013-02-06 22:43:31 +00001908 if (!Ty->isHalfTy() && !Ty->isFloatTy() && !Ty->isDoubleTy())
Craig Topper9f008862014-04-15 04:59:12 +00001909 return nullptr;
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001910 double Op1V = getValueAsDouble(Op1);
Owen Andersond4ebfd82013-02-06 22:43:31 +00001911
David Majnemer90a97042016-07-13 04:22:12 +00001912 if (auto *Op2 = dyn_cast<ConstantFP>(Operands[1])) {
Chris Lattner351534f2009-10-05 05:06:24 +00001913 if (Op2->getType() != Op1->getType())
Craig Topper9f008862014-04-15 04:59:12 +00001914 return nullptr;
Chad Rosier576c0f82011-12-01 23:16:03 +00001915
Matt Arsenaultf8ecf9b2014-03-05 00:01:58 +00001916 double Op2V = getValueAsDouble(Op2);
Benjamin Kramer061d1472014-03-05 19:41:48 +00001917 if (IntrinsicID == Intrinsic::pow) {
Chad Rosier0155a632011-12-03 00:00:03 +00001918 return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty);
1919 }
Karthik Bhatdaa8cd12014-03-06 05:32:52 +00001920 if (IntrinsicID == Intrinsic::copysign) {
1921 APFloat V1 = Op1->getValueAPF();
Benjamin Kramer8f59adb2016-02-13 16:54:14 +00001922 const APFloat &V2 = Op2->getValueAPF();
Karthik Bhatdaa8cd12014-03-06 05:32:52 +00001923 V1.copySign(V2);
1924 return ConstantFP::get(Ty->getContext(), V1);
1925 }
Matt Arsenaultd6511b42014-10-21 23:00:20 +00001926
1927 if (IntrinsicID == Intrinsic::minnum) {
1928 const APFloat &C1 = Op1->getValueAPF();
1929 const APFloat &C2 = Op2->getValueAPF();
1930 return ConstantFP::get(Ty->getContext(), minnum(C1, C2));
1931 }
1932
1933 if (IntrinsicID == Intrinsic::maxnum) {
1934 const APFloat &C1 = Op1->getValueAPF();
1935 const APFloat &C2 = Op2->getValueAPF();
1936 return ConstantFP::get(Ty->getContext(), maxnum(C1, C2));
1937 }
1938
Thomas Livelyfa54e562018-10-19 18:15:32 +00001939 if (IntrinsicID == Intrinsic::minimum) {
1940 const APFloat &C1 = Op1->getValueAPF();
1941 const APFloat &C2 = Op2->getValueAPF();
1942 return ConstantFP::get(Ty->getContext(), minimum(C1, C2));
1943 }
1944
1945 if (IntrinsicID == Intrinsic::maximum) {
1946 const APFloat &C1 = Op1->getValueAPF();
1947 const APFloat &C2 = Op2->getValueAPF();
1948 return ConstantFP::get(Ty->getContext(), maximum(C1, C2));
1949 }
1950
Chad Rosier0155a632011-12-03 00:00:03 +00001951 if (!TLI)
Craig Topper9f008862014-04-15 04:59:12 +00001952 return nullptr;
David L. Jonesd21529f2017-01-23 23:16:46 +00001953 if ((Name == "pow" && TLI->has(LibFunc_pow)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001954 (Name == "powf" && TLI->has(LibFunc_powf)) ||
1955 (Name == "__pow_finite" && TLI->has(LibFunc_pow_finite)) ||
1956 (Name == "__powf_finite" && TLI->has(LibFunc_powf_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001957 return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001958 if ((Name == "fmod" && TLI->has(LibFunc_fmod)) ||
1959 (Name == "fmodf" && TLI->has(LibFunc_fmodf)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001960 return ConstantFoldBinaryFP(fmod, Op1V, Op2V, Ty);
David L. Jonesd21529f2017-01-23 23:16:46 +00001961 if ((Name == "atan2" && TLI->has(LibFunc_atan2)) ||
Andrew Kaylorf7c864f2017-05-12 22:11:20 +00001962 (Name == "atan2f" && TLI->has(LibFunc_atan2f)) ||
1963 (Name == "__atan2_finite" && TLI->has(LibFunc_atan2_finite)) ||
1964 (Name == "__atan2f_finite" && TLI->has(LibFunc_atan2f_finite)))
Chris Lattner46b5c642009-11-06 04:27:31 +00001965 return ConstantFoldBinaryFP(atan2, Op1V, Op2V, Ty);
David Majnemer90a97042016-07-13 04:22:12 +00001966 } else if (auto *Op2C = dyn_cast<ConstantInt>(Operands[1])) {
Benjamin Kramer061d1472014-03-05 19:41:48 +00001967 if (IntrinsicID == Intrinsic::powi && Ty->isHalfTy())
1968 return ConstantFP::get(Ty->getContext(),
Owen Andersond4ebfd82013-02-06 22:43:31 +00001969 APFloat((float)std::pow((float)Op1V,
1970 (int)Op2C->getZExtValue())));
Benjamin Kramer061d1472014-03-05 19:41:48 +00001971 if (IntrinsicID == Intrinsic::powi && Ty->isFloatTy())
1972 return ConstantFP::get(Ty->getContext(),
Chris Lattner46b5c642009-11-06 04:27:31 +00001973 APFloat((float)std::pow((float)Op1V,
Chris Lattner3b187622008-04-20 00:41:09 +00001974 (int)Op2C->getZExtValue())));
Benjamin Kramer061d1472014-03-05 19:41:48 +00001975 if (IntrinsicID == Intrinsic::powi && Ty->isDoubleTy())
1976 return ConstantFP::get(Ty->getContext(),
Chris Lattner46b5c642009-11-06 04:27:31 +00001977 APFloat((double)std::pow((double)Op1V,
1978 (int)Op2C->getZExtValue())));
John Criswell970af112005-10-27 16:00:10 +00001979 }
Craig Topper9f008862014-04-15 04:59:12 +00001980 return nullptr;
John Criswell970af112005-10-27 16:00:10 +00001981 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00001982
David Majnemer90a97042016-07-13 04:22:12 +00001983 if (auto *Op1 = dyn_cast<ConstantInt>(Operands[0])) {
1984 if (auto *Op2 = dyn_cast<ConstantInt>(Operands[1])) {
Benjamin Kramer061d1472014-03-05 19:41:48 +00001985 switch (IntrinsicID) {
Chris Lattner59d93982009-10-05 05:26:04 +00001986 default: break;
Chris Lattner698661c2010-10-14 00:05:07 +00001987 case Intrinsic::sadd_with_overflow:
1988 case Intrinsic::uadd_with_overflow:
1989 case Intrinsic::ssub_with_overflow:
1990 case Intrinsic::usub_with_overflow:
Frits van Bommel0bb2ad22011-03-27 14:26:13 +00001991 case Intrinsic::smul_with_overflow:
1992 case Intrinsic::umul_with_overflow: {
Chris Lattner698661c2010-10-14 00:05:07 +00001993 APInt Res;
1994 bool Overflow;
Benjamin Kramer061d1472014-03-05 19:41:48 +00001995 switch (IntrinsicID) {
Craig Toppera2886c22012-02-07 05:05:23 +00001996 default: llvm_unreachable("Invalid case");
Chris Lattner698661c2010-10-14 00:05:07 +00001997 case Intrinsic::sadd_with_overflow:
1998 Res = Op1->getValue().sadd_ov(Op2->getValue(), Overflow);
1999 break;
2000 case Intrinsic::uadd_with_overflow:
2001 Res = Op1->getValue().uadd_ov(Op2->getValue(), Overflow);
2002 break;
2003 case Intrinsic::ssub_with_overflow:
2004 Res = Op1->getValue().ssub_ov(Op2->getValue(), Overflow);
2005 break;
2006 case Intrinsic::usub_with_overflow:
2007 Res = Op1->getValue().usub_ov(Op2->getValue(), Overflow);
2008 break;
2009 case Intrinsic::smul_with_overflow:
2010 Res = Op1->getValue().smul_ov(Op2->getValue(), Overflow);
2011 break;
Frits van Bommel0bb2ad22011-03-27 14:26:13 +00002012 case Intrinsic::umul_with_overflow:
2013 Res = Op1->getValue().umul_ov(Op2->getValue(), Overflow);
2014 break;
Chris Lattner698661c2010-10-14 00:05:07 +00002015 }
Chris Lattner59d93982009-10-05 05:26:04 +00002016 Constant *Ops[] = {
Benjamin Kramer061d1472014-03-05 19:41:48 +00002017 ConstantInt::get(Ty->getContext(), Res),
2018 ConstantInt::get(Type::getInt1Ty(Ty->getContext()), Overflow)
Chris Lattner59d93982009-10-05 05:26:04 +00002019 };
Benjamin Kramer061d1472014-03-05 19:41:48 +00002020 return ConstantStruct::get(cast<StructType>(Ty), Ops);
Chris Lattner59d93982009-10-05 05:26:04 +00002021 }
Chandler Carruth58a71ed2011-12-12 04:26:04 +00002022 case Intrinsic::cttz:
Benjamin Kramer435eba02013-01-24 16:28:28 +00002023 if (Op2->isOne() && Op1->isZero()) // cttz(0, 1) is undef.
2024 return UndefValue::get(Ty);
Chandler Carruth58a71ed2011-12-12 04:26:04 +00002025 return ConstantInt::get(Ty, Op1->getValue().countTrailingZeros());
2026 case Intrinsic::ctlz:
Benjamin Kramer435eba02013-01-24 16:28:28 +00002027 if (Op2->isOne() && Op1->isZero()) // ctlz(0, 1) is undef.
2028 return UndefValue::get(Ty);
Chandler Carruth58a71ed2011-12-12 04:26:04 +00002029 return ConstantInt::get(Ty, Op1->getValue().countLeadingZeros());
Chris Lattner59d93982009-10-05 05:26:04 +00002030 }
2031 }
NAKAMURA Takumidce89992012-11-05 00:11:11 +00002032
Craig Topper9f008862014-04-15 04:59:12 +00002033 return nullptr;
Chris Lattner59d93982009-10-05 05:26:04 +00002034 }
Craig Topper484b3422018-08-12 22:09:54 +00002035
2036 // Support ConstantVector in case we have an Undef in the top.
2037 if ((isa<ConstantVector>(Operands[0]) ||
2038 isa<ConstantDataVector>(Operands[0])) &&
2039 // Check for default rounding mode.
2040 // FIXME: Support other rounding modes?
2041 isa<ConstantInt>(Operands[1]) &&
2042 cast<ConstantInt>(Operands[1])->getValue() == 4) {
2043 auto *Op = cast<Constant>(Operands[0]);
2044 switch (IntrinsicID) {
2045 default: break;
2046 case Intrinsic::x86_avx512_vcvtss2si32:
2047 case Intrinsic::x86_avx512_vcvtss2si64:
2048 case Intrinsic::x86_avx512_vcvtsd2si32:
2049 case Intrinsic::x86_avx512_vcvtsd2si64:
2050 if (ConstantFP *FPOp =
2051 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U)))
2052 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(),
2053 /*roundTowardZero=*/false, Ty,
2054 /*IsSigned*/true);
2055 break;
2056 case Intrinsic::x86_avx512_vcvtss2usi32:
2057 case Intrinsic::x86_avx512_vcvtss2usi64:
2058 case Intrinsic::x86_avx512_vcvtsd2usi32:
2059 case Intrinsic::x86_avx512_vcvtsd2usi64:
2060 if (ConstantFP *FPOp =
2061 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U)))
2062 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(),
2063 /*roundTowardZero=*/false, Ty,
2064 /*IsSigned*/false);
2065 break;
2066 case Intrinsic::x86_avx512_cvttss2si:
2067 case Intrinsic::x86_avx512_cvttss2si64:
2068 case Intrinsic::x86_avx512_cvttsd2si:
2069 case Intrinsic::x86_avx512_cvttsd2si64:
2070 if (ConstantFP *FPOp =
2071 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U)))
2072 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(),
2073 /*roundTowardZero=*/true, Ty,
2074 /*IsSigned*/true);
2075 break;
2076 case Intrinsic::x86_avx512_cvttss2usi:
2077 case Intrinsic::x86_avx512_cvttss2usi64:
2078 case Intrinsic::x86_avx512_cvttsd2usi:
2079 case Intrinsic::x86_avx512_cvttsd2usi64:
2080 if (ConstantFP *FPOp =
2081 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U)))
2082 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(),
2083 /*roundTowardZero=*/true, Ty,
2084 /*IsSigned*/false);
2085 break;
2086 }
2087 }
Craig Topper9f008862014-04-15 04:59:12 +00002088 return nullptr;
John Criswell970af112005-10-27 16:00:10 +00002089 }
Matt Arsenault83778582014-03-05 00:02:00 +00002090
2091 if (Operands.size() != 3)
Craig Topper9f008862014-04-15 04:59:12 +00002092 return nullptr;
Matt Arsenault83778582014-03-05 00:02:00 +00002093
David Majnemer90a97042016-07-13 04:22:12 +00002094 if (const auto *Op1 = dyn_cast<ConstantFP>(Operands[0])) {
2095 if (const auto *Op2 = dyn_cast<ConstantFP>(Operands[1])) {
2096 if (const auto *Op3 = dyn_cast<ConstantFP>(Operands[2])) {
Benjamin Kramer061d1472014-03-05 19:41:48 +00002097 switch (IntrinsicID) {
Matt Arsenault83778582014-03-05 00:02:00 +00002098 default: break;
2099 case Intrinsic::fma:
2100 case Intrinsic::fmuladd: {
2101 APFloat V = Op1->getValueAPF();
2102 APFloat::opStatus s = V.fusedMultiplyAdd(Op2->getValueAPF(),
2103 Op3->getValueAPF(),
2104 APFloat::rmNearestTiesToEven);
2105 if (s != APFloat::opInvalidOp)
2106 return ConstantFP::get(Ty->getContext(), V);
2107
Craig Topper9f008862014-04-15 04:59:12 +00002108 return nullptr;
Matt Arsenault83778582014-03-05 00:02:00 +00002109 }
2110 }
2111 }
2112 }
2113 }
2114
Sanjay Patel411b8602018-08-17 13:23:44 +00002115 if (IntrinsicID == Intrinsic::fshl || IntrinsicID == Intrinsic::fshr) {
2116 auto *C0 = dyn_cast<ConstantInt>(Operands[0]);
2117 auto *C1 = dyn_cast<ConstantInt>(Operands[1]);
2118 auto *C2 = dyn_cast<ConstantInt>(Operands[2]);
2119 if (!(C0 && C1 && C2))
2120 return nullptr;
2121
2122 // The shift amount is interpreted as modulo the bitwidth. If the shift
2123 // amount is effectively 0, avoid UB due to oversized inverse shift below.
2124 unsigned BitWidth = C0->getBitWidth();
2125 unsigned ShAmt = C2->getValue().urem(BitWidth);
2126 bool IsRight = IntrinsicID == Intrinsic::fshr;
2127 if (!ShAmt)
2128 return IsRight ? C1 : C0;
2129
2130 // (X << ShlAmt) | (Y >> LshrAmt)
2131 const APInt &X = C0->getValue();
2132 const APInt &Y = C1->getValue();
2133 unsigned LshrAmt = IsRight ? ShAmt : BitWidth - ShAmt;
2134 unsigned ShlAmt = !IsRight ? ShAmt : BitWidth - ShAmt;
2135 return ConstantInt::get(Ty->getContext(), X.shl(ShlAmt) | Y.lshr(LshrAmt));
2136 }
2137
Craig Topper9f008862014-04-15 04:59:12 +00002138 return nullptr;
John Criswell970af112005-10-27 16:00:10 +00002139}
Benjamin Kramer061d1472014-03-05 19:41:48 +00002140
Eugene Zelenko35623fb2016-03-28 17:40:08 +00002141Constant *ConstantFoldVectorCall(StringRef Name, unsigned IntrinsicID,
2142 VectorType *VTy, ArrayRef<Constant *> Operands,
David Majnemer7f781ab2016-07-14 00:29:50 +00002143 const DataLayout &DL,
Manoj Gupta77eeac32018-07-09 22:27:23 +00002144 const TargetLibraryInfo *TLI,
2145 ImmutableCallSite CS) {
Benjamin Kramer061d1472014-03-05 19:41:48 +00002146 SmallVector<Constant *, 4> Result(VTy->getNumElements());
2147 SmallVector<Constant *, 4> Lane(Operands.size());
2148 Type *Ty = VTy->getElementType();
2149
David Majnemer7f781ab2016-07-14 00:29:50 +00002150 if (IntrinsicID == Intrinsic::masked_load) {
2151 auto *SrcPtr = Operands[0];
2152 auto *Mask = Operands[2];
2153 auto *Passthru = Operands[3];
David Majnemer17a95aa2016-07-14 06:58:37 +00002154
David Majnemer7f781ab2016-07-14 00:29:50 +00002155 Constant *VecData = ConstantFoldLoadFromConstPtr(SrcPtr, VTy, DL);
David Majnemer7f781ab2016-07-14 00:29:50 +00002156
2157 SmallVector<Constant *, 32> NewElements;
2158 for (unsigned I = 0, E = VTy->getNumElements(); I != E; ++I) {
David Majnemer17a95aa2016-07-14 06:58:37 +00002159 auto *MaskElt = Mask->getAggregateElement(I);
David Majnemer7f781ab2016-07-14 00:29:50 +00002160 if (!MaskElt)
2161 break;
David Majnemer17a95aa2016-07-14 06:58:37 +00002162 auto *PassthruElt = Passthru->getAggregateElement(I);
2163 auto *VecElt = VecData ? VecData->getAggregateElement(I) : nullptr;
2164 if (isa<UndefValue>(MaskElt)) {
2165 if (PassthruElt)
2166 NewElements.push_back(PassthruElt);
2167 else if (VecElt)
2168 NewElements.push_back(VecElt);
2169 else
2170 return nullptr;
2171 }
2172 if (MaskElt->isNullValue()) {
David Majnemer7f781ab2016-07-14 00:29:50 +00002173 if (!PassthruElt)
David Majnemer17a95aa2016-07-14 06:58:37 +00002174 return nullptr;
David Majnemer7f781ab2016-07-14 00:29:50 +00002175 NewElements.push_back(PassthruElt);
David Majnemer17a95aa2016-07-14 06:58:37 +00002176 } else if (MaskElt->isOneValue()) {
David Majnemer7f781ab2016-07-14 00:29:50 +00002177 if (!VecElt)
David Majnemer17a95aa2016-07-14 06:58:37 +00002178 return nullptr;
David Majnemer7f781ab2016-07-14 00:29:50 +00002179 NewElements.push_back(VecElt);
David Majnemer17a95aa2016-07-14 06:58:37 +00002180 } else {
2181 return nullptr;
David Majnemer7f781ab2016-07-14 00:29:50 +00002182 }
2183 }
David Majnemer17a95aa2016-07-14 06:58:37 +00002184 if (NewElements.size() != VTy->getNumElements())
2185 return nullptr;
2186 return ConstantVector::get(NewElements);
David Majnemer7f781ab2016-07-14 00:29:50 +00002187 }
2188
Benjamin Kramer061d1472014-03-05 19:41:48 +00002189 for (unsigned I = 0, E = VTy->getNumElements(); I != E; ++I) {
2190 // Gather a column of constants.
2191 for (unsigned J = 0, JE = Operands.size(); J != JE; ++J) {
Craig Topper7c553ed2017-06-03 18:50:29 +00002192 // These intrinsics use a scalar type for their second argument.
2193 if (J == 1 &&
Craig Topperfe9ad822017-06-04 07:30:28 +00002194 (IntrinsicID == Intrinsic::cttz || IntrinsicID == Intrinsic::ctlz ||
2195 IntrinsicID == Intrinsic::powi)) {
Craig Topper7c553ed2017-06-03 18:50:29 +00002196 Lane[J] = Operands[J];
2197 continue;
2198 }
2199
Benjamin Kramer061d1472014-03-05 19:41:48 +00002200 Constant *Agg = Operands[J]->getAggregateElement(I);
2201 if (!Agg)
2202 return nullptr;
2203
2204 Lane[J] = Agg;
2205 }
2206
2207 // Use the regular scalar folding to simplify this column.
Manoj Gupta77eeac32018-07-09 22:27:23 +00002208 Constant *Folded = ConstantFoldScalarCall(Name, IntrinsicID, Ty, Lane, TLI, CS);
Benjamin Kramer061d1472014-03-05 19:41:48 +00002209 if (!Folded)
2210 return nullptr;
2211 Result[I] = Folded;
2212 }
2213
2214 return ConstantVector::get(Result);
2215}
2216
Eugene Zelenko35623fb2016-03-28 17:40:08 +00002217} // end anonymous namespace
2218
Benjamin Kramer061d1472014-03-05 19:41:48 +00002219Constant *
Andrew Kaylor647025f2017-06-09 23:18:11 +00002220llvm::ConstantFoldCall(ImmutableCallSite CS, Function *F,
2221 ArrayRef<Constant *> Operands,
Benjamin Kramer061d1472014-03-05 19:41:48 +00002222 const TargetLibraryInfo *TLI) {
Andrew Kaylor53a5fbb2017-08-14 21:15:13 +00002223 if (CS.isNoBuiltin() || CS.isStrictFP())
Andrew Kaylor647025f2017-06-09 23:18:11 +00002224 return nullptr;
Benjamin Kramer061d1472014-03-05 19:41:48 +00002225 if (!F->hasName())
Craig Topper9f008862014-04-15 04:59:12 +00002226 return nullptr;
Benjamin Kramer061d1472014-03-05 19:41:48 +00002227 StringRef Name = F->getName();
2228
2229 Type *Ty = F->getReturnType();
2230
David Majnemer90a97042016-07-13 04:22:12 +00002231 if (auto *VTy = dyn_cast<VectorType>(Ty))
David Majnemer7f781ab2016-07-14 00:29:50 +00002232 return ConstantFoldVectorCall(Name, F->getIntrinsicID(), VTy, Operands,
Manoj Gupta77eeac32018-07-09 22:27:23 +00002233 F->getParent()->getDataLayout(), TLI, CS);
Benjamin Kramer061d1472014-03-05 19:41:48 +00002234
Manoj Gupta77eeac32018-07-09 22:27:23 +00002235 return ConstantFoldScalarCall(Name, F->getIntrinsicID(), Ty, Operands, TLI, CS);
Benjamin Kramer061d1472014-03-05 19:41:48 +00002236}
Eli Friedmanb6befc32016-11-02 20:48:11 +00002237
2238bool llvm::isMathLibCallNoop(CallSite CS, const TargetLibraryInfo *TLI) {
2239 // FIXME: Refactor this code; this duplicates logic in LibCallsShrinkWrap
2240 // (and to some extent ConstantFoldScalarCall).
Andrew Kaylor53a5fbb2017-08-14 21:15:13 +00002241 if (CS.isNoBuiltin() || CS.isStrictFP())
Andrew Kaylor647025f2017-06-09 23:18:11 +00002242 return false;
Eli Friedmanb6befc32016-11-02 20:48:11 +00002243 Function *F = CS.getCalledFunction();
2244 if (!F)
2245 return false;
2246
David L. Jonesd21529f2017-01-23 23:16:46 +00002247 LibFunc Func;
Eli Friedmanb6befc32016-11-02 20:48:11 +00002248 if (!TLI || !TLI->getLibFunc(*F, Func))
2249 return false;
2250
2251 if (CS.getNumArgOperands() == 1) {
2252 if (ConstantFP *OpC = dyn_cast<ConstantFP>(CS.getArgOperand(0))) {
2253 const APFloat &Op = OpC->getValueAPF();
2254 switch (Func) {
David L. Jonesd21529f2017-01-23 23:16:46 +00002255 case LibFunc_logl:
2256 case LibFunc_log:
2257 case LibFunc_logf:
2258 case LibFunc_log2l:
2259 case LibFunc_log2:
2260 case LibFunc_log2f:
2261 case LibFunc_log10l:
2262 case LibFunc_log10:
2263 case LibFunc_log10f:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002264 return Op.isNaN() || (!Op.isZero() && !Op.isNegative());
2265
David L. Jonesd21529f2017-01-23 23:16:46 +00002266 case LibFunc_expl:
2267 case LibFunc_exp:
2268 case LibFunc_expf:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002269 // FIXME: These boundaries are slightly conservative.
2270 if (OpC->getType()->isDoubleTy())
2271 return Op.compare(APFloat(-745.0)) != APFloat::cmpLessThan &&
2272 Op.compare(APFloat(709.0)) != APFloat::cmpGreaterThan;
2273 if (OpC->getType()->isFloatTy())
2274 return Op.compare(APFloat(-103.0f)) != APFloat::cmpLessThan &&
2275 Op.compare(APFloat(88.0f)) != APFloat::cmpGreaterThan;
2276 break;
2277
David L. Jonesd21529f2017-01-23 23:16:46 +00002278 case LibFunc_exp2l:
2279 case LibFunc_exp2:
2280 case LibFunc_exp2f:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002281 // FIXME: These boundaries are slightly conservative.
2282 if (OpC->getType()->isDoubleTy())
2283 return Op.compare(APFloat(-1074.0)) != APFloat::cmpLessThan &&
2284 Op.compare(APFloat(1023.0)) != APFloat::cmpGreaterThan;
2285 if (OpC->getType()->isFloatTy())
2286 return Op.compare(APFloat(-149.0f)) != APFloat::cmpLessThan &&
2287 Op.compare(APFloat(127.0f)) != APFloat::cmpGreaterThan;
2288 break;
2289
David L. Jonesd21529f2017-01-23 23:16:46 +00002290 case LibFunc_sinl:
2291 case LibFunc_sin:
2292 case LibFunc_sinf:
2293 case LibFunc_cosl:
2294 case LibFunc_cos:
2295 case LibFunc_cosf:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002296 return !Op.isInfinity();
2297
David L. Jonesd21529f2017-01-23 23:16:46 +00002298 case LibFunc_tanl:
2299 case LibFunc_tan:
2300 case LibFunc_tanf: {
Eli Friedmanb6befc32016-11-02 20:48:11 +00002301 // FIXME: Stop using the host math library.
2302 // FIXME: The computation isn't done in the right precision.
2303 Type *Ty = OpC->getType();
2304 if (Ty->isDoubleTy() || Ty->isFloatTy() || Ty->isHalfTy()) {
2305 double OpV = getValueAsDouble(OpC);
2306 return ConstantFoldFP(tan, OpV, Ty) != nullptr;
2307 }
2308 break;
2309 }
2310
David L. Jonesd21529f2017-01-23 23:16:46 +00002311 case LibFunc_asinl:
2312 case LibFunc_asin:
2313 case LibFunc_asinf:
2314 case LibFunc_acosl:
2315 case LibFunc_acos:
2316 case LibFunc_acosf:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002317 return Op.compare(APFloat(Op.getSemantics(), "-1")) !=
2318 APFloat::cmpLessThan &&
2319 Op.compare(APFloat(Op.getSemantics(), "1")) !=
2320 APFloat::cmpGreaterThan;
2321
David L. Jonesd21529f2017-01-23 23:16:46 +00002322 case LibFunc_sinh:
2323 case LibFunc_cosh:
2324 case LibFunc_sinhf:
2325 case LibFunc_coshf:
2326 case LibFunc_sinhl:
2327 case LibFunc_coshl:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002328 // FIXME: These boundaries are slightly conservative.
2329 if (OpC->getType()->isDoubleTy())
2330 return Op.compare(APFloat(-710.0)) != APFloat::cmpLessThan &&
2331 Op.compare(APFloat(710.0)) != APFloat::cmpGreaterThan;
2332 if (OpC->getType()->isFloatTy())
2333 return Op.compare(APFloat(-89.0f)) != APFloat::cmpLessThan &&
2334 Op.compare(APFloat(89.0f)) != APFloat::cmpGreaterThan;
2335 break;
2336
David L. Jonesd21529f2017-01-23 23:16:46 +00002337 case LibFunc_sqrtl:
2338 case LibFunc_sqrt:
2339 case LibFunc_sqrtf:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002340 return Op.isNaN() || Op.isZero() || !Op.isNegative();
2341
2342 // FIXME: Add more functions: sqrt_finite, atanh, expm1, log1p,
2343 // maybe others?
2344 default:
2345 break;
2346 }
2347 }
2348 }
2349
2350 if (CS.getNumArgOperands() == 2) {
2351 ConstantFP *Op0C = dyn_cast<ConstantFP>(CS.getArgOperand(0));
2352 ConstantFP *Op1C = dyn_cast<ConstantFP>(CS.getArgOperand(1));
2353 if (Op0C && Op1C) {
2354 const APFloat &Op0 = Op0C->getValueAPF();
2355 const APFloat &Op1 = Op1C->getValueAPF();
2356
2357 switch (Func) {
David L. Jonesd21529f2017-01-23 23:16:46 +00002358 case LibFunc_powl:
2359 case LibFunc_pow:
2360 case LibFunc_powf: {
Eli Friedmanb6befc32016-11-02 20:48:11 +00002361 // FIXME: Stop using the host math library.
2362 // FIXME: The computation isn't done in the right precision.
2363 Type *Ty = Op0C->getType();
2364 if (Ty->isDoubleTy() || Ty->isFloatTy() || Ty->isHalfTy()) {
2365 if (Ty == Op1C->getType()) {
2366 double Op0V = getValueAsDouble(Op0C);
2367 double Op1V = getValueAsDouble(Op1C);
2368 return ConstantFoldBinaryFP(pow, Op0V, Op1V, Ty) != nullptr;
2369 }
2370 }
2371 break;
2372 }
2373
David L. Jonesd21529f2017-01-23 23:16:46 +00002374 case LibFunc_fmodl:
2375 case LibFunc_fmod:
2376 case LibFunc_fmodf:
Eli Friedmanb6befc32016-11-02 20:48:11 +00002377 return Op0.isNaN() || Op1.isNaN() ||
2378 (!Op0.isInfinity() && !Op1.isZero());
2379
2380 default:
2381 break;
2382 }
2383 }
2384 }
2385
2386 return false;
2387}