2008-04-09 16:23:16 +08:00
|
|
|
//===- MemCpyOptimizer.cpp - Optimize use of memcpy and friends -----------===//
|
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// This pass performs various transformations related to eliminating memcpy
|
|
|
|
// calls, or transforming sets of stores into memset's.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "llvm/Transforms/Scalar.h"
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
|
|
#include "llvm/ADT/Statistic.h"
|
|
|
|
#include "llvm/Analysis/AliasAnalysis.h"
|
2015-01-04 20:03:27 +08:00
|
|
|
#include "llvm/Analysis/AssumptionCache.h"
|
2008-04-09 16:23:16 +08:00
|
|
|
#include "llvm/Analysis/MemoryDependenceAnalysis.h"
|
2010-12-27 04:15:01 +08:00
|
|
|
#include "llvm/Analysis/ValueTracking.h"
|
2013-01-02 19:36:10 +08:00
|
|
|
#include "llvm/IR/DataLayout.h"
|
2014-01-13 17:26:24 +08:00
|
|
|
#include "llvm/IR/Dominators.h"
|
2014-03-04 18:40:04 +08:00
|
|
|
#include "llvm/IR/GetElementPtrTypeIterator.h"
|
2013-01-02 19:36:10 +08:00
|
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
|
|
#include "llvm/IR/IRBuilder.h"
|
|
|
|
#include "llvm/IR/Instructions.h"
|
|
|
|
#include "llvm/IR/IntrinsicInst.h"
|
2008-04-09 16:23:16 +08:00
|
|
|
#include "llvm/Support/Debug.h"
|
2009-08-23 12:37:46 +08:00
|
|
|
#include "llvm/Support/raw_ostream.h"
|
2015-01-15 10:16:27 +08:00
|
|
|
#include "llvm/Analysis/TargetLibraryInfo.h"
|
2012-06-29 20:38:19 +08:00
|
|
|
#include "llvm/Transforms/Utils/Local.h"
|
2008-04-09 16:23:16 +08:00
|
|
|
#include <list>
|
|
|
|
using namespace llvm;
|
|
|
|
|
2014-04-22 10:55:47 +08:00
|
|
|
#define DEBUG_TYPE "memcpyopt"
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
STATISTIC(NumMemCpyInstr, "Number of memcpy instructions deleted");
|
|
|
|
STATISTIC(NumMemSetInfer, "Number of memsets inferred");
|
2009-09-03 21:37:16 +08:00
|
|
|
STATISTIC(NumMoveToCpy, "Number of memmoves converted to memcpy");
|
2010-12-25 05:17:12 +08:00
|
|
|
STATISTIC(NumCpyToSet, "Number of memcpys converted to memset");
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2012-09-14 00:29:49 +08:00
|
|
|
static int64_t GetOffsetFromIndex(const GEPOperator *GEP, unsigned Idx,
|
2012-10-09 00:38:25 +08:00
|
|
|
bool &VariableIdxFound, const DataLayout &TD){
|
2008-04-09 16:23:16 +08:00
|
|
|
// Skip over the first indices.
|
|
|
|
gep_type_iterator GTI = gep_type_begin(GEP);
|
|
|
|
for (unsigned i = 1; i != Idx; ++i, ++GTI)
|
|
|
|
/*skip along*/;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Compute the offset implied by the rest of the indices.
|
|
|
|
int64_t Offset = 0;
|
|
|
|
for (unsigned i = Idx, e = GEP->getNumOperands(); i != e; ++i, ++GTI) {
|
|
|
|
ConstantInt *OpC = dyn_cast<ConstantInt>(GEP->getOperand(i));
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!OpC)
|
2008-04-09 16:23:16 +08:00
|
|
|
return VariableIdxFound = true;
|
|
|
|
if (OpC->isZero()) continue; // No offset.
|
|
|
|
|
|
|
|
// Handle struct indices, which add their field offset to the pointer.
|
2011-07-18 12:54:35 +08:00
|
|
|
if (StructType *STy = dyn_cast<StructType>(*GTI)) {
|
2008-04-09 16:23:16 +08:00
|
|
|
Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
|
|
|
|
continue;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Otherwise, we have a sequential type like an array or vector. Multiply
|
|
|
|
// the index by the ElementSize.
|
2009-05-09 15:06:46 +08:00
|
|
|
uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
|
2008-04-09 16:23:16 +08:00
|
|
|
Offset += Size*OpC->getSExtValue();
|
|
|
|
}
|
|
|
|
|
|
|
|
return Offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// IsPointerOffset - Return true if Ptr1 is provably equal to Ptr2 plus a
|
|
|
|
/// constant offset, and return that constant offset. For example, Ptr1 might
|
|
|
|
/// be &A[42], and Ptr2 might be &A[40]. In this case offset would be -8.
|
|
|
|
static bool IsPointerOffset(Value *Ptr1, Value *Ptr2, int64_t &Offset,
|
2012-10-09 00:38:25 +08:00
|
|
|
const DataLayout &TD) {
|
2011-01-12 09:43:46 +08:00
|
|
|
Ptr1 = Ptr1->stripPointerCasts();
|
|
|
|
Ptr2 = Ptr2->stripPointerCasts();
|
2014-03-11 05:05:13 +08:00
|
|
|
|
|
|
|
// Handle the trivial case first.
|
|
|
|
if (Ptr1 == Ptr2) {
|
|
|
|
Offset = 0;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2012-09-14 00:29:49 +08:00
|
|
|
GEPOperator *GEP1 = dyn_cast<GEPOperator>(Ptr1);
|
|
|
|
GEPOperator *GEP2 = dyn_cast<GEPOperator>(Ptr2);
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 05:07:56 +08:00
|
|
|
bool VariableIdxFound = false;
|
|
|
|
|
|
|
|
// If one pointer is a GEP and the other isn't, then see if the GEP is a
|
|
|
|
// constant offset from the base, as in "P" and "gep P, 1".
|
2014-04-25 13:29:35 +08:00
|
|
|
if (GEP1 && !GEP2 && GEP1->getOperand(0)->stripPointerCasts() == Ptr2) {
|
2011-01-09 05:07:56 +08:00
|
|
|
Offset = -GetOffsetFromIndex(GEP1, 1, VariableIdxFound, TD);
|
|
|
|
return !VariableIdxFound;
|
|
|
|
}
|
|
|
|
|
2014-04-25 13:29:35 +08:00
|
|
|
if (GEP2 && !GEP1 && GEP2->getOperand(0)->stripPointerCasts() == Ptr1) {
|
2011-01-09 05:07:56 +08:00
|
|
|
Offset = GetOffsetFromIndex(GEP2, 1, VariableIdxFound, TD);
|
|
|
|
return !VariableIdxFound;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Right now we handle the case when Ptr1/Ptr2 are both GEPs with an identical
|
|
|
|
// base. After that base, they may have some number of common (and
|
|
|
|
// potentially variable) indices. After that they handle some constant
|
|
|
|
// offset, which determines their offset from each other. At this point, we
|
|
|
|
// handle no other case.
|
|
|
|
if (!GEP1 || !GEP2 || GEP1->getOperand(0) != GEP2->getOperand(0))
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Skip any common indices and track the GEP types.
|
|
|
|
unsigned Idx = 1;
|
|
|
|
for (; Idx != GEP1->getNumOperands() && Idx != GEP2->getNumOperands(); ++Idx)
|
|
|
|
if (GEP1->getOperand(Idx) != GEP2->getOperand(Idx))
|
|
|
|
break;
|
|
|
|
|
|
|
|
int64_t Offset1 = GetOffsetFromIndex(GEP1, Idx, VariableIdxFound, TD);
|
|
|
|
int64_t Offset2 = GetOffsetFromIndex(GEP2, Idx, VariableIdxFound, TD);
|
|
|
|
if (VariableIdxFound) return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
Offset = Offset2-Offset1;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/// MemsetRange - Represents a range of memset'd bytes with the ByteVal value.
|
|
|
|
/// This allows us to analyze stores like:
|
|
|
|
/// store 0 -> P+1
|
|
|
|
/// store 0 -> P+0
|
|
|
|
/// store 0 -> P+3
|
|
|
|
/// store 0 -> P+2
|
|
|
|
/// which sometimes happens with stores to arrays of structs etc. When we see
|
|
|
|
/// the first store, we make a range [1, 2). The second store extends the range
|
|
|
|
/// to [0, 2). The third makes a new range [2, 3). The fourth store joins the
|
|
|
|
/// two ranges into [0, 3) which is memset'able.
|
|
|
|
namespace {
|
|
|
|
struct MemsetRange {
|
|
|
|
// Start/End - A semi range that describes the span that this range covers.
|
2012-07-24 18:51:42 +08:00
|
|
|
// The range is closed at the start and open at the end: [Start, End).
|
2008-04-09 16:23:16 +08:00
|
|
|
int64_t Start, End;
|
|
|
|
|
|
|
|
/// StartPtr - The getelementptr instruction that points to the start of the
|
|
|
|
/// range.
|
|
|
|
Value *StartPtr;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
/// Alignment - The known alignment of the first store.
|
|
|
|
unsigned Alignment;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
/// TheStores - The actual stores that make up this range.
|
2011-01-09 04:54:51 +08:00
|
|
|
SmallVector<Instruction*, 16> TheStores;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2012-10-09 00:38:25 +08:00
|
|
|
bool isProfitableToUseMemset(const DataLayout &TD) const;
|
2008-04-09 16:23:16 +08:00
|
|
|
|
|
|
|
};
|
|
|
|
} // end anon namespace
|
|
|
|
|
2012-10-09 00:38:25 +08:00
|
|
|
bool MemsetRange::isProfitableToUseMemset(const DataLayout &TD) const {
|
2011-12-06 06:53:09 +08:00
|
|
|
// If we found more than 4 stores to merge or 16 bytes, use memset.
|
2011-12-06 06:37:00 +08:00
|
|
|
if (TheStores.size() >= 4 || End-Start >= 16) return true;
|
2011-01-09 04:54:51 +08:00
|
|
|
|
|
|
|
// If there is nothing to merge, don't do anything.
|
|
|
|
if (TheStores.size() < 2) return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
// If any of the stores are a memset, then it is always good to extend the
|
|
|
|
// memset.
|
|
|
|
for (unsigned i = 0, e = TheStores.size(); i != e; ++i)
|
|
|
|
if (!isa<StoreInst>(TheStores[i]))
|
|
|
|
return true;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Assume that the code generator is capable of merging pairs of stores
|
|
|
|
// together if it wants to.
|
2011-01-09 04:54:51 +08:00
|
|
|
if (TheStores.size() == 2) return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// If we have fewer than 8 stores, it can still be worthwhile to do this.
|
|
|
|
// For example, merging 4 i8 stores into an i32 store is useful almost always.
|
|
|
|
// However, merging 2 32-bit stores isn't useful on a 32-bit architecture (the
|
|
|
|
// memset will be split into 2 32-bit stores anyway) and doing so can
|
|
|
|
// pessimize the llvm optimizer.
|
|
|
|
//
|
|
|
|
// Since we don't have perfect knowledge here, make some assumptions: assume
|
2013-09-17 06:43:16 +08:00
|
|
|
// the maximum GPR width is the same size as the largest legal integer
|
|
|
|
// size. If so, check to see whether we will end up actually reducing the
|
|
|
|
// number of stores used.
|
2008-04-09 16:23:16 +08:00
|
|
|
unsigned Bytes = unsigned(End-Start);
|
2013-09-17 06:43:16 +08:00
|
|
|
unsigned MaxIntSize = TD.getLargestLegalIntTypeSize();
|
|
|
|
if (MaxIntSize == 0)
|
|
|
|
MaxIntSize = 1;
|
|
|
|
unsigned NumPointerStores = Bytes / MaxIntSize;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Assume the remaining bytes if any are done a byte at a time.
|
2013-09-17 06:43:16 +08:00
|
|
|
unsigned NumByteStores = Bytes - NumPointerStores * MaxIntSize;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// If we will reduce the # stores (according to this heuristic), do the
|
|
|
|
// transformation. This encourages merging 4 x i8 -> i32 and 2 x i16 -> i32
|
|
|
|
// etc.
|
|
|
|
return TheStores.size() > NumPointerStores+NumByteStores;
|
2012-07-24 18:51:42 +08:00
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
class MemsetRanges {
|
|
|
|
/// Ranges - A sorted list of the memset ranges. We use std::list here
|
|
|
|
/// because each element is relatively large and expensive to copy.
|
|
|
|
std::list<MemsetRange> Ranges;
|
|
|
|
typedef std::list<MemsetRange>::iterator range_iterator;
|
2014-02-21 08:06:31 +08:00
|
|
|
const DataLayout &DL;
|
2008-04-09 16:23:16 +08:00
|
|
|
public:
|
2014-02-21 08:06:31 +08:00
|
|
|
MemsetRanges(const DataLayout &DL) : DL(DL) {}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
typedef std::list<MemsetRange>::const_iterator const_iterator;
|
|
|
|
const_iterator begin() const { return Ranges.begin(); }
|
|
|
|
const_iterator end() const { return Ranges.end(); }
|
|
|
|
bool empty() const { return Ranges.empty(); }
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
void addInst(int64_t OffsetFromFirst, Instruction *Inst) {
|
2011-01-09 04:54:51 +08:00
|
|
|
if (StoreInst *SI = dyn_cast<StoreInst>(Inst))
|
|
|
|
addStore(OffsetFromFirst, SI);
|
|
|
|
else
|
|
|
|
addMemSet(OffsetFromFirst, cast<MemSetInst>(Inst));
|
|
|
|
}
|
|
|
|
|
|
|
|
void addStore(int64_t OffsetFromFirst, StoreInst *SI) {
|
2014-02-21 08:06:31 +08:00
|
|
|
int64_t StoreSize = DL.getTypeStoreSize(SI->getOperand(0)->getType());
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
addRange(OffsetFromFirst, StoreSize,
|
|
|
|
SI->getPointerOperand(), SI->getAlignment(), SI);
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
void addMemSet(int64_t OffsetFromFirst, MemSetInst *MSI) {
|
|
|
|
int64_t Size = cast<ConstantInt>(MSI->getLength())->getZExtValue();
|
|
|
|
addRange(OffsetFromFirst, Size, MSI->getDest(), MSI->getAlignment(), MSI);
|
2011-01-09 04:24:01 +08:00
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
void addRange(int64_t Start, int64_t Size, Value *Ptr,
|
|
|
|
unsigned Alignment, Instruction *Inst);
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
};
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
} // end anon namespace
|
|
|
|
|
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
/// addRange - Add a new store to the MemsetRanges data structure. This adds a
|
2008-04-09 16:23:16 +08:00
|
|
|
/// new range for the specified store at the specified offset, merging into
|
|
|
|
/// existing ranges as appropriate.
|
2011-01-09 04:54:51 +08:00
|
|
|
///
|
|
|
|
/// Do a linear search of the ranges to see if this can be joined and/or to
|
|
|
|
/// find the insertion point in the list. We keep the ranges sorted for
|
|
|
|
/// simplicity here. This is a linear search of a linked list, which is ugly,
|
|
|
|
/// however the number of ranges is limited, so this won't get crazy slow.
|
|
|
|
void MemsetRanges::addRange(int64_t Start, int64_t Size, Value *Ptr,
|
|
|
|
unsigned Alignment, Instruction *Inst) {
|
|
|
|
int64_t End = Start+Size;
|
2008-04-09 16:23:16 +08:00
|
|
|
range_iterator I = Ranges.begin(), E = Ranges.end();
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
while (I != E && Start > I->End)
|
|
|
|
++I;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// We now know that I == E, in which case we didn't find anything to merge
|
|
|
|
// with, or that Start <= I->End. If End < I->Start or I == E, then we need
|
|
|
|
// to insert a new range. Handle this now.
|
|
|
|
if (I == E || End < I->Start) {
|
|
|
|
MemsetRange &R = *Ranges.insert(I, MemsetRange());
|
|
|
|
R.Start = Start;
|
|
|
|
R.End = End;
|
2011-01-09 04:54:51 +08:00
|
|
|
R.StartPtr = Ptr;
|
|
|
|
R.Alignment = Alignment;
|
|
|
|
R.TheStores.push_back(Inst);
|
2008-04-09 16:23:16 +08:00
|
|
|
return;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// This store overlaps with I, add it.
|
2011-01-09 04:54:51 +08:00
|
|
|
I->TheStores.push_back(Inst);
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// At this point, we may have an interval that completely contains our store.
|
|
|
|
// If so, just add it to the interval and return.
|
|
|
|
if (I->Start <= Start && I->End >= End)
|
|
|
|
return;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Now we know that Start <= I->End and End >= I->Start so the range overlaps
|
|
|
|
// but is not entirely contained within the range.
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// See if the range extends the start of the range. In this case, it couldn't
|
|
|
|
// possibly cause it to join the prior range, because otherwise we would have
|
|
|
|
// stopped on *it*.
|
|
|
|
if (Start < I->Start) {
|
|
|
|
I->Start = Start;
|
2011-01-09 04:54:51 +08:00
|
|
|
I->StartPtr = Ptr;
|
|
|
|
I->Alignment = Alignment;
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Now we know that Start <= I->End and Start >= I->Start (so the startpoint
|
|
|
|
// is in or right at the end of I), and that End >= I->Start. Extend I out to
|
|
|
|
// End.
|
|
|
|
if (End > I->End) {
|
|
|
|
I->End = End;
|
2009-03-19 13:51:39 +08:00
|
|
|
range_iterator NextI = I;
|
2008-04-09 16:23:16 +08:00
|
|
|
while (++NextI != E && End >= NextI->Start) {
|
|
|
|
// Merge the range in.
|
|
|
|
I->TheStores.append(NextI->TheStores.begin(), NextI->TheStores.end());
|
|
|
|
if (NextI->End > I->End)
|
|
|
|
I->End = NextI->End;
|
|
|
|
Ranges.erase(NextI);
|
|
|
|
NextI = I;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// MemCpyOpt Pass
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
namespace {
|
2009-09-02 14:11:42 +08:00
|
|
|
class MemCpyOpt : public FunctionPass {
|
2010-11-21 08:28:59 +08:00
|
|
|
MemoryDependenceAnalysis *MD;
|
2011-05-02 02:27:11 +08:00
|
|
|
TargetLibraryInfo *TLI;
|
2014-02-21 08:06:31 +08:00
|
|
|
const DataLayout *DL;
|
2008-04-09 16:23:16 +08:00
|
|
|
public:
|
|
|
|
static char ID; // Pass identification, replacement for typeid
|
2010-10-20 01:21:58 +08:00
|
|
|
MemCpyOpt() : FunctionPass(ID) {
|
|
|
|
initializeMemCpyOptPass(*PassRegistry::getPassRegistry());
|
2014-04-25 13:29:35 +08:00
|
|
|
MD = nullptr;
|
|
|
|
TLI = nullptr;
|
|
|
|
DL = nullptr;
|
2010-10-20 01:21:58 +08:00
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2014-03-05 17:10:37 +08:00
|
|
|
bool runOnFunction(Function &F) override;
|
2011-01-09 04:24:01 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
private:
|
|
|
|
// This transformation requires dominator postdominator info
|
2014-03-05 17:10:37 +08:00
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {
|
2008-04-09 16:23:16 +08:00
|
|
|
AU.setPreservesCFG();
|
2015-01-04 20:03:27 +08:00
|
|
|
AU.addRequired<AssumptionCacheTracker>();
|
2014-01-13 21:07:17 +08:00
|
|
|
AU.addRequired<DominatorTreeWrapperPass>();
|
2008-04-09 16:23:16 +08:00
|
|
|
AU.addRequired<MemoryDependenceAnalysis>();
|
|
|
|
AU.addRequired<AliasAnalysis>();
|
2011-05-02 02:27:11 +08:00
|
|
|
AU.addRequired<TargetLibraryInfo>();
|
2008-04-09 16:23:16 +08:00
|
|
|
AU.addPreserved<AliasAnalysis>();
|
|
|
|
AU.addPreserved<MemoryDependenceAnalysis>();
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Helper fuctions
|
2009-09-02 01:09:55 +08:00
|
|
|
bool processStore(StoreInst *SI, BasicBlock::iterator &BBI);
|
2011-01-09 05:19:19 +08:00
|
|
|
bool processMemSet(MemSetInst *SI, BasicBlock::iterator &BBI);
|
2009-09-02 01:09:55 +08:00
|
|
|
bool processMemCpy(MemCpyInst *M);
|
2009-09-02 01:56:32 +08:00
|
|
|
bool processMemMove(MemMoveInst *M);
|
2010-10-16 06:52:12 +08:00
|
|
|
bool performCallSlotOptzn(Instruction *cpy, Value *cpyDst, Value *cpySrc,
|
2012-10-04 18:54:40 +08:00
|
|
|
uint64_t cpyLen, unsigned cpyAlign, CallInst *C);
|
2010-11-18 15:02:37 +08:00
|
|
|
bool processMemCpyMemCpyDependence(MemCpyInst *M, MemCpyInst *MDep,
|
|
|
|
uint64_t MSize);
|
2010-11-21 08:28:59 +08:00
|
|
|
bool processByValArgument(CallSite CS, unsigned ArgNo);
|
2011-01-09 04:24:01 +08:00
|
|
|
Instruction *tryMergingIntoMemset(Instruction *I, Value *StartPtr,
|
|
|
|
Value *ByteVal);
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
bool iterateOnFunction(Function &F);
|
|
|
|
};
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
char MemCpyOpt::ID = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// createMemCpyOptPass - The public interface to this file...
|
|
|
|
FunctionPass *llvm::createMemCpyOptPass() { return new MemCpyOpt(); }
|
|
|
|
|
2010-10-13 03:48:12 +08:00
|
|
|
INITIALIZE_PASS_BEGIN(MemCpyOpt, "memcpyopt", "MemCpy Optimization",
|
|
|
|
false, false)
|
2015-01-04 20:03:27 +08:00
|
|
|
INITIALIZE_PASS_DEPENDENCY(AssumptionCacheTracker)
|
2014-01-13 21:07:17 +08:00
|
|
|
INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass)
|
2010-10-13 03:48:12 +08:00
|
|
|
INITIALIZE_PASS_DEPENDENCY(MemoryDependenceAnalysis)
|
2011-05-02 02:27:11 +08:00
|
|
|
INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfo)
|
2010-10-13 03:48:12 +08:00
|
|
|
INITIALIZE_AG_DEPENDENCY(AliasAnalysis)
|
|
|
|
INITIALIZE_PASS_END(MemCpyOpt, "memcpyopt", "MemCpy Optimization",
|
|
|
|
false, false)
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
/// tryMergingIntoMemset - When scanning forward over instructions, we look for
|
2008-04-09 16:23:16 +08:00
|
|
|
/// some other patterns to fold away. In particular, this looks for stores to
|
2011-02-15 17:23:02 +08:00
|
|
|
/// neighboring locations of memory. If it sees enough consecutive ones, it
|
2011-01-09 04:24:01 +08:00
|
|
|
/// attempts to merge them together into a memcpy/memset.
|
2012-07-24 18:51:42 +08:00
|
|
|
Instruction *MemCpyOpt::tryMergingIntoMemset(Instruction *StartInst,
|
2011-01-09 04:24:01 +08:00
|
|
|
Value *StartPtr, Value *ByteVal) {
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!DL) return nullptr;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Okay, so we now have a single store that can be splatable. Scan to find
|
|
|
|
// all subsequent stores of the same value to offset from the same pointer.
|
|
|
|
// Join these together into ranges, so we can decide whether contiguous blocks
|
|
|
|
// are stored.
|
2014-02-21 08:06:31 +08:00
|
|
|
MemsetRanges Ranges(*DL);
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
BasicBlock::iterator BI = StartInst;
|
2008-04-09 16:23:16 +08:00
|
|
|
for (++BI; !isa<TerminatorInst>(BI); ++BI) {
|
2011-01-09 04:54:51 +08:00
|
|
|
if (!isa<StoreInst>(BI) && !isa<MemSetInst>(BI)) {
|
|
|
|
// If the instruction is readnone, ignore it, otherwise bail out. We
|
|
|
|
// don't even allow readonly here because we don't want something like:
|
2008-04-09 16:23:16 +08:00
|
|
|
// A[1] = 2; strlen(A); A[2] = 2; -> memcpy(A, ...); strlen(A).
|
2011-01-09 04:54:51 +08:00
|
|
|
if (BI->mayWriteToMemory() || BI->mayReadFromMemory())
|
|
|
|
break;
|
|
|
|
continue;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
if (StoreInst *NextStore = dyn_cast<StoreInst>(BI)) {
|
|
|
|
// If this is a store, see if we can merge it in.
|
2011-08-18 06:22:24 +08:00
|
|
|
if (!NextStore->isSimple()) break;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
// Check to see if this stored value is of the same byte-splattable value.
|
|
|
|
if (ByteVal != isBytewiseValue(NextStore->getOperand(0)))
|
|
|
|
break;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
// Check to see if this store is to a constant offset from the start ptr.
|
|
|
|
int64_t Offset;
|
2011-01-10 03:26:10 +08:00
|
|
|
if (!IsPointerOffset(StartPtr, NextStore->getPointerOperand(),
|
2014-02-21 08:06:31 +08:00
|
|
|
Offset, *DL))
|
2011-01-09 04:54:51 +08:00
|
|
|
break;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
Ranges.addStore(Offset, NextStore);
|
|
|
|
} else {
|
|
|
|
MemSetInst *MSI = cast<MemSetInst>(BI);
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
if (MSI->isVolatile() || ByteVal != MSI->getValue() ||
|
|
|
|
!isa<ConstantInt>(MSI->getLength()))
|
|
|
|
break;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
// Check to see if this store is to a constant offset from the start ptr.
|
|
|
|
int64_t Offset;
|
2014-02-21 08:06:31 +08:00
|
|
|
if (!IsPointerOffset(StartPtr, MSI->getDest(), Offset, *DL))
|
2011-01-09 04:54:51 +08:00
|
|
|
break;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:54:51 +08:00
|
|
|
Ranges.addMemSet(Offset, MSI);
|
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// If we have no ranges, then we just had a single store with nothing that
|
|
|
|
// could be merged in. This is a very common case of course.
|
|
|
|
if (Ranges.empty())
|
2014-04-25 13:29:35 +08:00
|
|
|
return nullptr;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// If we had at least one store that could be merged in, add the starting
|
|
|
|
// store as well. We try to avoid this unless there is at least something
|
|
|
|
// interesting as a small compile-time optimization.
|
2011-01-09 04:24:01 +08:00
|
|
|
Ranges.addInst(0, StartInst);
|
|
|
|
|
|
|
|
// If we create any memsets, we put it right before the first instruction that
|
|
|
|
// isn't part of the memset block. This ensure that the memset is dominated
|
|
|
|
// by any addressing instruction needed by the start of the block.
|
|
|
|
IRBuilder<> Builder(BI);
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Now that we have full information about ranges, loop over the ranges and
|
|
|
|
// emit memset's for anything big enough to be worthwhile.
|
2014-04-25 13:29:35 +08:00
|
|
|
Instruction *AMemSet = nullptr;
|
2008-04-09 16:23:16 +08:00
|
|
|
for (MemsetRanges::const_iterator I = Ranges.begin(), E = Ranges.end();
|
|
|
|
I != E; ++I) {
|
|
|
|
const MemsetRange &Range = *I;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
if (Range.TheStores.size() == 1) continue;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// If it is profitable to lower this range to memset, do so now.
|
2014-02-21 08:06:31 +08:00
|
|
|
if (!Range.isProfitableToUseMemset(*DL))
|
2008-04-09 16:23:16 +08:00
|
|
|
continue;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
// Otherwise, we do want to transform this! Create a new memset.
|
2008-04-09 16:23:16 +08:00
|
|
|
// Get the starting pointer of the block.
|
|
|
|
StartPtr = Range.StartPtr;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
Reapply address space patch after fixing an issue in MemCopyOptimizer.
Added support for address spaces and added a isVolatile field to memcpy, memmove, and memset,
e.g., llvm.memcpy.i32(i8*, i8*, i32, i32) -> llvm.memcpy.p0i8.p0i8.i32(i8*, i8*, i32, i32, i1)
llvm-svn: 100304
2010-04-04 11:10:48 +08:00
|
|
|
// Determine alignment
|
|
|
|
unsigned Alignment = Range.Alignment;
|
|
|
|
if (Alignment == 0) {
|
2012-07-24 18:51:42 +08:00
|
|
|
Type *EltType =
|
2011-01-09 04:24:01 +08:00
|
|
|
cast<PointerType>(StartPtr->getType())->getElementType();
|
2014-02-21 08:06:31 +08:00
|
|
|
Alignment = DL->getABITypeAlignment(EltType);
|
Reapply address space patch after fixing an issue in MemCopyOptimizer.
Added support for address spaces and added a isVolatile field to memcpy, memmove, and memset,
e.g., llvm.memcpy.i32(i8*, i8*, i32, i32) -> llvm.memcpy.p0i8.p0i8.i32(i8*, i8*, i32, i32, i1)
llvm-svn: 100304
2010-04-04 11:10:48 +08:00
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
|
|
|
AMemSet =
|
2010-12-27 06:57:41 +08:00
|
|
|
Builder.CreateMemSet(StartPtr, ByteVal, Range.End-Range.Start, Alignment);
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-01-05 09:27:47 +08:00
|
|
|
DEBUG(dbgs() << "Replace stores:\n";
|
2008-04-09 16:23:16 +08:00
|
|
|
for (unsigned i = 0, e = Range.TheStores.size(); i != e; ++i)
|
2010-11-21 08:28:59 +08:00
|
|
|
dbgs() << *Range.TheStores[i] << '\n';
|
2011-01-09 04:24:01 +08:00
|
|
|
dbgs() << "With: " << *AMemSet << '\n');
|
2011-05-05 05:58:58 +08:00
|
|
|
|
|
|
|
if (!Range.TheStores.empty())
|
|
|
|
AMemSet->setDebugLoc(Range.TheStores[0]->getDebugLoc());
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Zap all the stores.
|
2013-07-03 23:07:05 +08:00
|
|
|
for (SmallVectorImpl<Instruction *>::const_iterator
|
2009-09-08 08:27:14 +08:00
|
|
|
SI = Range.TheStores.begin(),
|
2011-01-09 06:19:21 +08:00
|
|
|
SE = Range.TheStores.end(); SI != SE; ++SI) {
|
|
|
|
MD->removeInstruction(*SI);
|
2008-04-21 15:45:10 +08:00
|
|
|
(*SI)->eraseFromParent();
|
2011-01-09 06:19:21 +08:00
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
++NumMemSetInfer;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
return AMemSet;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool MemCpyOpt::processStore(StoreInst *SI, BasicBlock::iterator &BBI) {
|
2011-08-18 06:22:24 +08:00
|
|
|
if (!SI->isSimple()) return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!DL) return false;
|
2011-01-09 04:24:01 +08:00
|
|
|
|
|
|
|
// Detect cases where we're performing call slot forwarding, but
|
|
|
|
// happen to be using a load-store pair to implement it, rather than
|
|
|
|
// a memcpy.
|
|
|
|
if (LoadInst *LI = dyn_cast<LoadInst>(SI->getOperand(0))) {
|
2011-08-18 06:22:24 +08:00
|
|
|
if (LI->isSimple() && LI->hasOneUse() &&
|
2011-06-15 09:25:56 +08:00
|
|
|
LI->getParent() == SI->getParent()) {
|
2011-06-03 05:24:42 +08:00
|
|
|
MemDepResult ldep = MD->getDependency(LI);
|
2014-04-25 13:29:35 +08:00
|
|
|
CallInst *C = nullptr;
|
2011-06-03 05:24:42 +08:00
|
|
|
if (ldep.isClobber() && !isa<MemCpyInst>(ldep.getInst()))
|
|
|
|
C = dyn_cast<CallInst>(ldep.getInst());
|
|
|
|
|
|
|
|
if (C) {
|
|
|
|
// Check that nothing touches the dest of the "copy" between
|
|
|
|
// the call and the store.
|
2011-06-15 09:25:56 +08:00
|
|
|
AliasAnalysis &AA = getAnalysis<AliasAnalysis>();
|
|
|
|
AliasAnalysis::Location StoreLoc = AA.getLocation(SI);
|
|
|
|
for (BasicBlock::iterator I = --BasicBlock::iterator(SI),
|
|
|
|
E = C; I != E; --I) {
|
|
|
|
if (AA.getModRefInfo(&*I, StoreLoc) != AliasAnalysis::NoModRef) {
|
2014-04-25 13:29:35 +08:00
|
|
|
C = nullptr;
|
2011-06-15 09:25:56 +08:00
|
|
|
break;
|
|
|
|
}
|
2011-06-03 05:24:42 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
if (C) {
|
2012-10-04 18:54:40 +08:00
|
|
|
unsigned storeAlign = SI->getAlignment();
|
|
|
|
if (!storeAlign)
|
2014-02-21 08:06:31 +08:00
|
|
|
storeAlign = DL->getABITypeAlignment(SI->getOperand(0)->getType());
|
2012-10-04 18:54:40 +08:00
|
|
|
unsigned loadAlign = LI->getAlignment();
|
|
|
|
if (!loadAlign)
|
2014-02-21 08:06:31 +08:00
|
|
|
loadAlign = DL->getABITypeAlignment(LI->getType());
|
2012-10-04 18:54:40 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
bool changed = performCallSlotOptzn(LI,
|
2012-07-24 18:51:42 +08:00
|
|
|
SI->getPointerOperand()->stripPointerCasts(),
|
2011-01-09 04:24:01 +08:00
|
|
|
LI->getPointerOperand()->stripPointerCasts(),
|
2014-02-21 08:06:31 +08:00
|
|
|
DL->getTypeStoreSize(SI->getOperand(0)->getType()),
|
2012-10-04 18:54:40 +08:00
|
|
|
std::min(storeAlign, loadAlign), C);
|
2011-01-09 04:24:01 +08:00
|
|
|
if (changed) {
|
|
|
|
MD->removeInstruction(SI);
|
|
|
|
SI->eraseFromParent();
|
2011-01-10 03:26:10 +08:00
|
|
|
MD->removeInstruction(LI);
|
2011-01-09 04:24:01 +08:00
|
|
|
LI->eraseFromParent();
|
|
|
|
++NumMemCpyInstr;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
// There are two cases that are interesting for this code to handle: memcpy
|
|
|
|
// and memset. Right now we only handle memset.
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
// Ensure that the value being stored is something that can be memset'able a
|
|
|
|
// byte at a time like "0" or "-1" or any width, as well as things like
|
|
|
|
// 0xA0A0A0A0 and 0.0.
|
|
|
|
if (Value *ByteVal = isBytewiseValue(SI->getOperand(0)))
|
|
|
|
if (Instruction *I = tryMergingIntoMemset(SI, SI->getPointerOperand(),
|
|
|
|
ByteVal)) {
|
|
|
|
BBI = I; // Don't invalidate iterator.
|
|
|
|
return true;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-01-09 04:24:01 +08:00
|
|
|
return false;
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
|
|
|
|
2011-01-09 05:19:19 +08:00
|
|
|
bool MemCpyOpt::processMemSet(MemSetInst *MSI, BasicBlock::iterator &BBI) {
|
|
|
|
// See if there is another memset or store neighboring this memset which
|
|
|
|
// allows us to widen out the memset to do a single larger store.
|
2011-01-09 06:11:56 +08:00
|
|
|
if (isa<ConstantInt>(MSI->getLength()) && !MSI->isVolatile())
|
|
|
|
if (Instruction *I = tryMergingIntoMemset(MSI, MSI->getDest(),
|
|
|
|
MSI->getValue())) {
|
|
|
|
BBI = I; // Don't invalidate iterator.
|
|
|
|
return true;
|
|
|
|
}
|
2011-01-09 05:19:19 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
|
|
|
|
/// performCallSlotOptzn - takes a memcpy and a call that it depends on,
|
|
|
|
/// and checks for the possibility of a call slot optimization by having
|
|
|
|
/// the call write its result directly into the destination of the memcpy.
|
2010-10-16 06:52:12 +08:00
|
|
|
bool MemCpyOpt::performCallSlotOptzn(Instruction *cpy,
|
|
|
|
Value *cpyDest, Value *cpySrc,
|
2012-10-04 18:54:40 +08:00
|
|
|
uint64_t cpyLen, unsigned cpyAlign,
|
|
|
|
CallInst *C) {
|
2008-04-09 16:23:16 +08:00
|
|
|
// The general transformation to keep in mind is
|
|
|
|
//
|
|
|
|
// call @func(..., src, ...)
|
|
|
|
// memcpy(dest, src, ...)
|
|
|
|
//
|
|
|
|
// ->
|
|
|
|
//
|
|
|
|
// memcpy(dest, src, ...)
|
|
|
|
// call @func(..., dest, ...)
|
|
|
|
//
|
|
|
|
// Since moving the memcpy is technically awkward, we additionally check that
|
|
|
|
// src only holds uninitialized values at the moment of the call, meaning that
|
|
|
|
// the memcpy can be discarded rather than moved.
|
|
|
|
|
|
|
|
// Deliberately get the source and destination with bitcasts stripped away,
|
|
|
|
// because we'll need to do type comparisons based on the underlying type.
|
2010-07-29 06:50:26 +08:00
|
|
|
CallSite CS(C);
|
2008-04-09 16:23:16 +08:00
|
|
|
|
|
|
|
// Require that src be an alloca. This simplifies the reasoning considerably.
|
2009-09-02 01:09:55 +08:00
|
|
|
AllocaInst *srcAlloca = dyn_cast<AllocaInst>(cpySrc);
|
2008-04-09 16:23:16 +08:00
|
|
|
if (!srcAlloca)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
// Check that all of src is copied to dest.
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!DL) return false;
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2009-09-02 01:09:55 +08:00
|
|
|
ConstantInt *srcArraySize = dyn_cast<ConstantInt>(srcAlloca->getArraySize());
|
2008-04-09 16:23:16 +08:00
|
|
|
if (!srcArraySize)
|
|
|
|
return false;
|
|
|
|
|
2014-02-21 08:06:31 +08:00
|
|
|
uint64_t srcSize = DL->getTypeAllocSize(srcAlloca->getAllocatedType()) *
|
2008-04-09 16:23:16 +08:00
|
|
|
srcArraySize->getZExtValue();
|
|
|
|
|
2010-10-16 06:52:12 +08:00
|
|
|
if (cpyLen < srcSize)
|
2008-04-09 16:23:16 +08:00
|
|
|
return false;
|
|
|
|
|
|
|
|
// Check that accessing the first srcSize bytes of dest will not cause a
|
|
|
|
// trap. Otherwise the transform is invalid since it might cause a trap
|
|
|
|
// to occur earlier than it otherwise would.
|
2009-09-02 01:09:55 +08:00
|
|
|
if (AllocaInst *A = dyn_cast<AllocaInst>(cpyDest)) {
|
2008-04-09 16:23:16 +08:00
|
|
|
// The destination is an alloca. Check it is larger than srcSize.
|
2009-09-02 01:09:55 +08:00
|
|
|
ConstantInt *destArraySize = dyn_cast<ConstantInt>(A->getArraySize());
|
2008-04-09 16:23:16 +08:00
|
|
|
if (!destArraySize)
|
|
|
|
return false;
|
|
|
|
|
2014-02-21 08:06:31 +08:00
|
|
|
uint64_t destSize = DL->getTypeAllocSize(A->getAllocatedType()) *
|
2008-04-09 16:23:16 +08:00
|
|
|
destArraySize->getZExtValue();
|
|
|
|
|
|
|
|
if (destSize < srcSize)
|
|
|
|
return false;
|
2009-09-02 01:09:55 +08:00
|
|
|
} else if (Argument *A = dyn_cast<Argument>(cpyDest)) {
|
2014-10-17 03:43:08 +08:00
|
|
|
if (A->getDereferenceableBytes() < srcSize) {
|
|
|
|
// If the destination is an sret parameter then only accesses that are
|
|
|
|
// outside of the returned struct type can trap.
|
|
|
|
if (!A->hasStructRetAttr())
|
|
|
|
return false;
|
|
|
|
|
|
|
|
Type *StructTy = cast<PointerType>(A->getType())->getElementType();
|
|
|
|
if (!StructTy->isSized()) {
|
|
|
|
// The call may never return and hence the copy-instruction may never
|
|
|
|
// be executed, and therefore it's not safe to say "the destination
|
|
|
|
// has at least <cpyLen> bytes, as implied by the copy-instruction",
|
|
|
|
return false;
|
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2014-10-17 03:43:08 +08:00
|
|
|
uint64_t destSize = DL->getTypeAllocSize(StructTy);
|
|
|
|
if (destSize < srcSize)
|
|
|
|
return false;
|
2013-06-08 12:56:05 +08:00
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
} else {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2012-10-05 15:29:46 +08:00
|
|
|
// Check that dest points to memory that is at least as aligned as src.
|
|
|
|
unsigned srcAlign = srcAlloca->getAlignment();
|
|
|
|
if (!srcAlign)
|
2014-02-21 08:06:31 +08:00
|
|
|
srcAlign = DL->getABITypeAlignment(srcAlloca->getAllocatedType());
|
2012-10-05 15:29:46 +08:00
|
|
|
bool isDestSufficientlyAligned = srcAlign <= cpyAlign;
|
|
|
|
// If dest is not aligned enough and we can't increase its alignment then
|
|
|
|
// bail out.
|
|
|
|
if (!isDestSufficientlyAligned && !isa<AllocaInst>(cpyDest))
|
|
|
|
return false;
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Check that src is not accessed except via the call and the memcpy. This
|
|
|
|
// guarantees that it holds only undefined values when passed in (so the final
|
|
|
|
// memcpy can be dropped), that it is not read or written between the call and
|
|
|
|
// the memcpy, and that writing beyond the end of it is undefined.
|
2014-03-09 11:16:01 +08:00
|
|
|
SmallVector<User*, 8> srcUseList(srcAlloca->user_begin(),
|
|
|
|
srcAlloca->user_end());
|
2008-04-09 16:23:16 +08:00
|
|
|
while (!srcUseList.empty()) {
|
2014-03-09 11:16:01 +08:00
|
|
|
User *U = srcUseList.pop_back_val();
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2014-03-09 11:16:01 +08:00
|
|
|
if (isa<BitCastInst>(U) || isa<AddrSpaceCastInst>(U)) {
|
|
|
|
for (User *UU : U->users())
|
|
|
|
srcUseList.push_back(UU);
|
2014-09-01 18:09:18 +08:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (GetElementPtrInst *G = dyn_cast<GetElementPtrInst>(U)) {
|
|
|
|
if (!G->hasAllZeroIndices())
|
2008-06-02 06:26:26 +08:00
|
|
|
return false;
|
2014-09-01 18:09:18 +08:00
|
|
|
|
|
|
|
for (User *UU : U->users())
|
|
|
|
srcUseList.push_back(UU);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (const IntrinsicInst *IT = dyn_cast<IntrinsicInst>(U))
|
|
|
|
if (IT->getIntrinsicID() == Intrinsic::lifetime_start ||
|
|
|
|
IT->getIntrinsicID() == Intrinsic::lifetime_end)
|
2014-09-01 14:03:11 +08:00
|
|
|
continue;
|
2014-09-01 18:09:18 +08:00
|
|
|
|
|
|
|
if (U != C && U != cpy)
|
2008-04-09 16:23:16 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2014-07-15 02:52:02 +08:00
|
|
|
// Check that src isn't captured by the called function since the
|
|
|
|
// transformation can cause aliasing issues in that case.
|
|
|
|
for (unsigned i = 0, e = CS.arg_size(); i != e; ++i)
|
|
|
|
if (CS.getArgument(i) == cpySrc && !CS.doesNotCapture(i))
|
|
|
|
return false;
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Since we're changing the parameter to the callsite, we need to make sure
|
|
|
|
// that what would be the new parameter dominates the callsite.
|
2014-01-13 21:07:17 +08:00
|
|
|
DominatorTree &DT = getAnalysis<DominatorTreeWrapperPass>().getDomTree();
|
2009-09-02 01:09:55 +08:00
|
|
|
if (Instruction *cpyDestInst = dyn_cast<Instruction>(cpyDest))
|
2008-04-09 16:23:16 +08:00
|
|
|
if (!DT.dominates(cpyDestInst, C))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
// In addition to knowing that the call does not access src in some
|
|
|
|
// unexpected manner, for example via a global, which we deduce from
|
|
|
|
// the use analysis, we also need to know that it does not sneakily
|
|
|
|
// access dest. We rely on AA to figure this out for us.
|
2009-09-02 01:09:55 +08:00
|
|
|
AliasAnalysis &AA = getAnalysis<AliasAnalysis>();
|
2012-05-15 04:35:04 +08:00
|
|
|
AliasAnalysis::ModRefResult MR = AA.getModRefInfo(C, cpyDest, srcSize);
|
|
|
|
// If necessary, perform additional analysis.
|
|
|
|
if (MR != AliasAnalysis::NoModRef)
|
|
|
|
MR = AA.callCapturesBefore(C, cpyDest, srcSize, &DT);
|
|
|
|
if (MR != AliasAnalysis::NoModRef)
|
2008-04-09 16:23:16 +08:00
|
|
|
return false;
|
|
|
|
|
|
|
|
// All the checks have passed, so do the transformation.
|
2008-06-02 05:52:16 +08:00
|
|
|
bool changedArgument = false;
|
2008-04-09 16:23:16 +08:00
|
|
|
for (unsigned i = 0; i < CS.arg_size(); ++i)
|
2008-06-02 06:26:26 +08:00
|
|
|
if (CS.getArgument(i)->stripPointerCasts() == cpySrc) {
|
2012-10-04 21:53:21 +08:00
|
|
|
Value *Dest = cpySrc->getType() == cpyDest->getType() ? cpyDest
|
|
|
|
: CastInst::CreatePointerCast(cpyDest, cpySrc->getType(),
|
|
|
|
cpyDest->getName(), C);
|
2008-06-02 05:52:16 +08:00
|
|
|
changedArgument = true;
|
2012-10-04 21:53:21 +08:00
|
|
|
if (CS.getArgument(i)->getType() == Dest->getType())
|
|
|
|
CS.setArgument(i, Dest);
|
2009-09-02 01:09:55 +08:00
|
|
|
else
|
2012-10-04 21:53:21 +08:00
|
|
|
CS.setArgument(i, CastInst::CreatePointerCast(Dest,
|
|
|
|
CS.getArgument(i)->getType(), Dest->getName(), C));
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
|
|
|
|
2008-06-02 05:52:16 +08:00
|
|
|
if (!changedArgument)
|
|
|
|
return false;
|
|
|
|
|
2012-10-04 18:54:40 +08:00
|
|
|
// If the destination wasn't sufficiently aligned then increase its alignment.
|
|
|
|
if (!isDestSufficientlyAligned) {
|
|
|
|
assert(isa<AllocaInst>(cpyDest) && "Can only increase alloca alignment!");
|
|
|
|
cast<AllocaInst>(cpyDest)->setAlignment(srcAlign);
|
|
|
|
}
|
|
|
|
|
2008-04-09 16:23:16 +08:00
|
|
|
// Drop any cached information about the call, because we may have changed
|
|
|
|
// its dependence information by changing its parameter.
|
2010-11-21 08:28:59 +08:00
|
|
|
MD->removeInstruction(C);
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
// Remove the memcpy.
|
|
|
|
MD->removeInstruction(cpy);
|
2010-06-22 23:08:57 +08:00
|
|
|
++NumMemCpyInstr;
|
2008-04-09 16:23:16 +08:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2010-11-18 15:02:37 +08:00
|
|
|
/// processMemCpyMemCpyDependence - We've found that the (upward scanning)
|
|
|
|
/// memory dependence of memcpy 'M' is the memcpy 'MDep'. Try to simplify M to
|
|
|
|
/// copy from MDep's input if we can. MSize is the size of M's copy.
|
2012-07-24 18:51:42 +08:00
|
|
|
///
|
2010-11-18 15:02:37 +08:00
|
|
|
bool MemCpyOpt::processMemCpyMemCpyDependence(MemCpyInst *M, MemCpyInst *MDep,
|
|
|
|
uint64_t MSize) {
|
2008-04-09 16:23:16 +08:00
|
|
|
// We can only transforms memcpy's where the dest of one is the source of the
|
2010-11-18 15:02:37 +08:00
|
|
|
// other.
|
2010-11-21 08:28:59 +08:00
|
|
|
if (M->getSource() != MDep->getDest() || MDep->isVolatile())
|
2008-04-09 16:23:16 +08:00
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-12-09 15:39:50 +08:00
|
|
|
// If dep instruction is reading from our current input, then it is a noop
|
|
|
|
// transfer and substituting the input won't change this instruction. Just
|
|
|
|
// ignore the input and let someone else zap MDep. This handles cases like:
|
|
|
|
// memcpy(a <- a)
|
|
|
|
// memcpy(b <- a)
|
|
|
|
if (M->getSource() == MDep->getSource())
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-04-15 13:18:47 +08:00
|
|
|
// Second, the length of the memcpy's must be the same, or the preceding one
|
2008-04-09 16:23:16 +08:00
|
|
|
// must be larger than the following one.
|
2011-01-22 06:07:57 +08:00
|
|
|
ConstantInt *MDepLen = dyn_cast<ConstantInt>(MDep->getLength());
|
|
|
|
ConstantInt *MLen = dyn_cast<ConstantInt>(M->getLength());
|
|
|
|
if (!MDepLen || !MLen || MDepLen->getZExtValue() < MLen->getZExtValue())
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
AliasAnalysis &AA = getAnalysis<AliasAnalysis>();
|
2010-11-21 16:06:10 +08:00
|
|
|
|
|
|
|
// Verify that the copied-from memory doesn't change in between the two
|
|
|
|
// transfers. For example, in:
|
|
|
|
// memcpy(a <- b)
|
|
|
|
// *b = 42;
|
|
|
|
// memcpy(c <- a)
|
|
|
|
// It would be invalid to transform the second memcpy into memcpy(c <- b).
|
|
|
|
//
|
|
|
|
// TODO: If the code between M and MDep is transparent to the destination "c",
|
|
|
|
// then we could still perform the xform by moving M up to the first memcpy.
|
|
|
|
//
|
|
|
|
// NOTE: This is conservative, it will stop on any read from the source loc,
|
|
|
|
// not just the defining memcpy.
|
|
|
|
MemDepResult SourceDep =
|
|
|
|
MD->getPointerDependencyFrom(AA.getLocationForSource(MDep),
|
|
|
|
false, M, M->getParent());
|
|
|
|
if (!SourceDep.isClobber() || SourceDep.getInst() != MDep)
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-18 16:00:57 +08:00
|
|
|
// If the dest of the second might alias the source of the first, then the
|
|
|
|
// source and dest might overlap. We still want to eliminate the intermediate
|
|
|
|
// value, but we have to generate a memmove instead of memcpy.
|
2010-12-27 06:57:41 +08:00
|
|
|
bool UseMemMove = false;
|
|
|
|
if (!AA.isNoAlias(AA.getLocationForDest(M), AA.getLocationForSource(MDep)))
|
|
|
|
UseMemMove = true;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
// If all checks passed, then we can transform M.
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-10-01 17:02:05 +08:00
|
|
|
// Make sure to use the lesser of the alignment of the source and the dest
|
|
|
|
// since we're changing where we're reading from, but don't want to increase
|
|
|
|
// the alignment past what can be read from or written to.
|
2010-09-25 08:57:26 +08:00
|
|
|
// TODO: Is this worth it if we're creating a less aligned memcpy? For
|
|
|
|
// example we could be moving from movaps -> movq on x86.
|
2010-11-18 16:07:09 +08:00
|
|
|
unsigned Align = std::min(MDep->getAlignment(), M->getAlignment());
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-12-27 06:57:41 +08:00
|
|
|
IRBuilder<> Builder(M);
|
|
|
|
if (UseMemMove)
|
|
|
|
Builder.CreateMemMove(M->getRawDest(), MDep->getRawSource(), M->getLength(),
|
|
|
|
Align, M->isVolatile());
|
|
|
|
else
|
|
|
|
Builder.CreateMemCpy(M->getRawDest(), MDep->getRawSource(), M->getLength(),
|
|
|
|
Align, M->isVolatile());
|
2010-11-18 16:07:09 +08:00
|
|
|
|
2010-11-21 16:06:10 +08:00
|
|
|
// Remove the instruction we're replacing.
|
2010-11-21 08:28:59 +08:00
|
|
|
MD->removeInstruction(M);
|
2010-11-18 16:07:09 +08:00
|
|
|
M->eraseFromParent();
|
|
|
|
++NumMemCpyInstr;
|
|
|
|
return true;
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
|
|
|
|
2010-11-18 15:02:37 +08:00
|
|
|
|
|
|
|
/// processMemCpy - perform simplification of memcpy's. If we have memcpy A
|
|
|
|
/// which copies X to Y, and memcpy B which copies Y to Z, then we can rewrite
|
|
|
|
/// B to be a memcpy from X to Z (or potentially a memmove, depending on
|
|
|
|
/// circumstances). This allows later passes to remove the first memcpy
|
|
|
|
/// altogether.
|
|
|
|
bool MemCpyOpt::processMemCpy(MemCpyInst *M) {
|
2014-02-04 08:18:54 +08:00
|
|
|
// We can only optimize non-volatile memcpy's.
|
|
|
|
if (M->isVolatile()) return false;
|
2010-11-18 15:02:37 +08:00
|
|
|
|
2010-12-09 15:45:45 +08:00
|
|
|
// If the source and destination of the memcpy are the same, then zap it.
|
|
|
|
if (M->getSource() == M->getDest()) {
|
|
|
|
MD->removeInstruction(M);
|
|
|
|
M->eraseFromParent();
|
|
|
|
return false;
|
|
|
|
}
|
2010-12-25 05:17:12 +08:00
|
|
|
|
|
|
|
// If copying from a constant, try to turn the memcpy into a memset.
|
2010-12-25 06:23:59 +08:00
|
|
|
if (GlobalVariable *GV = dyn_cast<GlobalVariable>(M->getSource()))
|
2010-12-26 23:23:45 +08:00
|
|
|
if (GV->isConstant() && GV->hasDefinitiveInitializer())
|
2010-12-25 06:23:59 +08:00
|
|
|
if (Value *ByteVal = isBytewiseValue(GV->getInitializer())) {
|
2010-12-27 06:57:41 +08:00
|
|
|
IRBuilder<> Builder(M);
|
2014-02-04 08:18:54 +08:00
|
|
|
Builder.CreateMemSet(M->getRawDest(), ByteVal, M->getLength(),
|
2010-12-27 06:57:41 +08:00
|
|
|
M->getAlignment(), false);
|
2010-12-25 06:23:59 +08:00
|
|
|
MD->removeInstruction(M);
|
|
|
|
M->eraseFromParent();
|
|
|
|
++NumCpyToSet;
|
|
|
|
return true;
|
|
|
|
}
|
2010-12-25 05:17:12 +08:00
|
|
|
|
2014-02-04 08:18:54 +08:00
|
|
|
// The optimizations after this point require the memcpy size.
|
|
|
|
ConstantInt *CopySize = dyn_cast<ConstantInt>(M->getLength());
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!CopySize) return false;
|
2014-02-04 08:18:54 +08:00
|
|
|
|
2014-02-06 14:29:19 +08:00
|
|
|
// The are three possible optimizations we can do for memcpy:
|
2010-11-18 15:02:37 +08:00
|
|
|
// a) memcpy-memcpy xform which exposes redundance for DSE.
|
|
|
|
// b) call-memcpy xform for return slot optimization.
|
2014-03-27 07:45:15 +08:00
|
|
|
// c) memcpy from freshly alloca'd space or space that has just started its
|
|
|
|
// lifetime copies undefined data, and we can therefore eliminate the
|
|
|
|
// memcpy in favor of the data that was already at the destination.
|
2010-11-21 08:28:59 +08:00
|
|
|
MemDepResult DepInfo = MD->getDependency(M);
|
2011-10-17 04:13:32 +08:00
|
|
|
if (DepInfo.isClobber()) {
|
|
|
|
if (CallInst *C = dyn_cast<CallInst>(DepInfo.getInst())) {
|
|
|
|
if (performCallSlotOptzn(M, M->getDest(), M->getSource(),
|
2012-10-04 18:54:40 +08:00
|
|
|
CopySize->getZExtValue(), M->getAlignment(),
|
|
|
|
C)) {
|
2011-10-17 04:13:32 +08:00
|
|
|
MD->removeInstruction(M);
|
|
|
|
M->eraseFromParent();
|
|
|
|
return true;
|
|
|
|
}
|
2010-12-09 15:45:45 +08:00
|
|
|
}
|
2010-11-18 15:02:37 +08:00
|
|
|
}
|
2012-02-13 14:30:56 +08:00
|
|
|
|
|
|
|
AliasAnalysis::Location SrcLoc = AliasAnalysis::getLocationForSource(M);
|
2011-10-17 04:13:32 +08:00
|
|
|
MemDepResult SrcDepInfo = MD->getPointerDependencyFrom(SrcLoc, true,
|
|
|
|
M, M->getParent());
|
|
|
|
if (SrcDepInfo.isClobber()) {
|
|
|
|
if (MemCpyInst *MDep = dyn_cast<MemCpyInst>(SrcDepInfo.getInst()))
|
|
|
|
return processMemCpyMemCpyDependence(M, MDep, CopySize->getZExtValue());
|
2014-02-06 14:29:19 +08:00
|
|
|
} else if (SrcDepInfo.isDef()) {
|
2014-03-27 07:45:15 +08:00
|
|
|
Instruction *I = SrcDepInfo.getInst();
|
|
|
|
bool hasUndefContents = false;
|
|
|
|
|
|
|
|
if (isa<AllocaInst>(I)) {
|
|
|
|
hasUndefContents = true;
|
|
|
|
} else if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) {
|
|
|
|
if (II->getIntrinsicID() == Intrinsic::lifetime_start)
|
|
|
|
if (ConstantInt *LTSize = dyn_cast<ConstantInt>(II->getArgOperand(0)))
|
|
|
|
if (LTSize->getZExtValue() >= CopySize->getZExtValue())
|
|
|
|
hasUndefContents = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (hasUndefContents) {
|
2014-02-06 14:29:19 +08:00
|
|
|
MD->removeInstruction(M);
|
|
|
|
M->eraseFromParent();
|
|
|
|
++NumMemCpyInstr;
|
|
|
|
return true;
|
|
|
|
}
|
2011-10-17 04:13:32 +08:00
|
|
|
}
|
|
|
|
|
2010-11-18 15:02:37 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2009-09-02 01:56:32 +08:00
|
|
|
/// processMemMove - Transforms memmove calls to memcpy calls when the src/dst
|
|
|
|
/// are guaranteed not to alias.
|
|
|
|
bool MemCpyOpt::processMemMove(MemMoveInst *M) {
|
|
|
|
AliasAnalysis &AA = getAnalysis<AliasAnalysis>();
|
|
|
|
|
2011-05-02 02:27:11 +08:00
|
|
|
if (!TLI->has(LibFunc::memmove))
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2009-09-02 01:56:32 +08:00
|
|
|
// See if the pointers alias.
|
2010-12-27 06:57:41 +08:00
|
|
|
if (!AA.isNoAlias(AA.getLocationForDest(M), AA.getLocationForSource(M)))
|
2009-09-02 01:56:32 +08:00
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-01-05 09:27:47 +08:00
|
|
|
DEBUG(dbgs() << "MemCpyOpt: Optimizing memmove -> memcpy: " << *M << "\n");
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2009-09-02 01:56:32 +08:00
|
|
|
// If not, then we know we can transform this.
|
|
|
|
Module *Mod = M->getParent()->getParent()->getParent();
|
2011-07-12 22:06:48 +08:00
|
|
|
Type *ArgTys[3] = { M->getRawDest()->getType(),
|
|
|
|
M->getRawSource()->getType(),
|
|
|
|
M->getLength()->getType() };
|
2010-07-22 18:37:47 +08:00
|
|
|
M->setCalledFunction(Intrinsic::getDeclaration(Mod, Intrinsic::memcpy,
|
2011-07-15 01:45:39 +08:00
|
|
|
ArgTys));
|
2009-09-03 21:37:16 +08:00
|
|
|
|
2009-09-02 01:56:32 +08:00
|
|
|
// MemDep may have over conservative information about this instruction, just
|
|
|
|
// conservatively flush it from the cache.
|
2010-11-21 08:28:59 +08:00
|
|
|
MD->removeInstruction(M);
|
2009-09-03 21:37:16 +08:00
|
|
|
|
|
|
|
++NumMoveToCpy;
|
2009-09-02 01:56:32 +08:00
|
|
|
return true;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
/// processByValArgument - This is called on every byval argument in call sites.
|
|
|
|
bool MemCpyOpt::processByValArgument(CallSite CS, unsigned ArgNo) {
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!DL) return false;
|
2009-09-02 01:56:32 +08:00
|
|
|
|
2010-11-21 16:06:10 +08:00
|
|
|
// Find out what feeds this byval argument.
|
2010-11-21 08:28:59 +08:00
|
|
|
Value *ByValArg = CS.getArgument(ArgNo);
|
2011-10-12 08:14:31 +08:00
|
|
|
Type *ByValTy = cast<PointerType>(ByValArg->getType())->getElementType();
|
2014-02-21 08:06:31 +08:00
|
|
|
uint64_t ByValSize = DL->getTypeAllocSize(ByValTy);
|
2010-11-21 16:06:10 +08:00
|
|
|
MemDepResult DepInfo =
|
|
|
|
MD->getPointerDependencyFrom(AliasAnalysis::Location(ByValArg, ByValSize),
|
|
|
|
true, CS.getInstruction(),
|
|
|
|
CS.getInstruction()->getParent());
|
2010-11-21 08:28:59 +08:00
|
|
|
if (!DepInfo.isClobber())
|
|
|
|
return false;
|
|
|
|
|
|
|
|
// If the byval argument isn't fed by a memcpy, ignore it. If it is fed by
|
|
|
|
// a memcpy, see if we can byval from the source of the memcpy instead of the
|
|
|
|
// result.
|
|
|
|
MemCpyInst *MDep = dyn_cast<MemCpyInst>(DepInfo.getInst());
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!MDep || MDep->isVolatile() ||
|
2010-11-21 08:28:59 +08:00
|
|
|
ByValArg->stripPointerCasts() != MDep->getDest())
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
// The length of the memcpy must be larger or equal to the size of the byval.
|
|
|
|
ConstantInt *C1 = dyn_cast<ConstantInt>(MDep->getLength());
|
2014-04-25 13:29:35 +08:00
|
|
|
if (!C1 || C1->getValue().getZExtValue() < ByValSize)
|
2010-11-21 08:28:59 +08:00
|
|
|
return false;
|
|
|
|
|
2011-05-23 08:03:39 +08:00
|
|
|
// Get the alignment of the byval. If the call doesn't specify the alignment,
|
|
|
|
// then it is some target specific value that we can't know.
|
2010-11-21 08:28:59 +08:00
|
|
|
unsigned ByValAlign = CS.getParamAlignment(ArgNo+1);
|
2011-05-23 08:03:39 +08:00
|
|
|
if (ByValAlign == 0) return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-05-23 08:03:39 +08:00
|
|
|
// If it is greater than the memcpy, then we check to see if we can force the
|
|
|
|
// source of the memcpy to the alignment we need. If we fail, we bail out.
|
2015-01-04 20:03:27 +08:00
|
|
|
AssumptionCache &AC =
|
|
|
|
getAnalysis<AssumptionCacheTracker>().getAssumptionCache(
|
|
|
|
*CS->getParent()->getParent());
|
Make use of @llvm.assume in ValueTracking (computeKnownBits, etc.)
This change, which allows @llvm.assume to be used from within computeKnownBits
(and other associated functions in ValueTracking), adds some (optional)
parameters to computeKnownBits and friends. These functions now (optionally)
take a "context" instruction pointer, an AssumptionTracker pointer, and also a
DomTree pointer, and most of the changes are just to pass this new information
when it is easily available from InstSimplify, InstCombine, etc.
As explained below, the significant conceptual change is that known properties
of a value might depend on the control-flow location of the use (because we
care that the @llvm.assume dominates the use because assumptions have
control-flow dependencies). This means that, when we ask if bits are known in a
value, we might get different answers for different uses.
The significant changes are all in ValueTracking. Two main changes: First, as
with the rest of the code, new parameters need to be passed around. To make
this easier, I grouped them into a structure, and I made internal static
versions of the relevant functions that take this structure as a parameter. The
new code does as you might expect, it looks for @llvm.assume calls that make
use of the value we're trying to learn something about (often indirectly),
attempts to pattern match that expression, and uses the result if successful.
By making use of the AssumptionTracker, the process of finding @llvm.assume
calls is not expensive.
Part of the structure being passed around inside ValueTracking is a set of
already-considered @llvm.assume calls. This is to prevent a query using, for
example, the assume(a == b), to recurse on itself. The context and DT params
are used to find applicable assumptions. An assumption needs to dominate the
context instruction, or come after it deterministically. In this latter case we
only handle the specific case where both the assumption and the context
instruction are in the same block, and we need to exclude assumptions from
being used to simplify their own ephemeral values (those which contribute only
to the assumption) because otherwise the assumption would prove its feeding
comparison trivial and would be removed.
This commit adds the plumbing and the logic for a simple masked-bit propagation
(just enough to write a regression test). Future commits add more patterns
(and, correspondingly, more regression tests).
llvm-svn: 217342
2014-09-08 02:57:58 +08:00
|
|
|
DominatorTree &DT = getAnalysis<DominatorTreeWrapperPass>().getDomTree();
|
2011-05-23 08:03:39 +08:00
|
|
|
if (MDep->getAlignment() < ByValAlign &&
|
2015-01-04 20:03:27 +08:00
|
|
|
getOrEnforceKnownAlignment(MDep->getSource(), ByValAlign, DL, &AC,
|
|
|
|
CS.getInstruction(), &DT) < ByValAlign)
|
2011-05-23 08:03:39 +08:00
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
// Verify that the copied-from memory doesn't change in between the memcpy and
|
|
|
|
// the byval call.
|
|
|
|
// memcpy(a <- b)
|
|
|
|
// *b = 42;
|
|
|
|
// foo(*a)
|
|
|
|
// It would be invalid to transform the second memcpy into foo(*b).
|
2010-11-21 16:06:10 +08:00
|
|
|
//
|
|
|
|
// NOTE: This is conservative, it will stop on any read from the source loc,
|
|
|
|
// not just the defining memcpy.
|
|
|
|
MemDepResult SourceDep =
|
|
|
|
MD->getPointerDependencyFrom(AliasAnalysis::getLocationForSource(MDep),
|
|
|
|
false, CS.getInstruction(), MDep->getParent());
|
|
|
|
if (!SourceDep.isClobber() || SourceDep.getInst() != MDep)
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
Value *TmpCast = MDep->getSource();
|
|
|
|
if (MDep->getSource()->getType() != ByValArg->getType())
|
|
|
|
TmpCast = new BitCastInst(MDep->getSource(), ByValArg->getType(),
|
|
|
|
"tmpcast", CS.getInstruction());
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
DEBUG(dbgs() << "MemCpyOpt: Forwarding memcpy to byval:\n"
|
|
|
|
<< " " << *MDep << "\n"
|
|
|
|
<< " " << *CS.getInstruction() << "\n");
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
// Otherwise we're good! Update the byval argument.
|
|
|
|
CS.setArgument(ArgNo, TmpCast);
|
|
|
|
++NumMemCpyInstr;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// iterateOnFunction - Executes one iteration of MemCpyOpt.
|
2008-04-09 16:23:16 +08:00
|
|
|
bool MemCpyOpt::iterateOnFunction(Function &F) {
|
2009-09-02 01:09:55 +08:00
|
|
|
bool MadeChange = false;
|
2008-04-09 16:23:16 +08:00
|
|
|
|
2009-09-02 01:09:55 +08:00
|
|
|
// Walk all instruction in the function.
|
2008-04-21 15:45:10 +08:00
|
|
|
for (Function::iterator BB = F.begin(), BBE = F.end(); BB != BBE; ++BB) {
|
2010-11-21 08:28:59 +08:00
|
|
|
for (BasicBlock::iterator BI = BB->begin(), BE = BB->end(); BI != BE;) {
|
2009-09-02 01:09:55 +08:00
|
|
|
// Avoid invalidating the iterator.
|
|
|
|
Instruction *I = BI++;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2010-11-21 08:28:59 +08:00
|
|
|
bool RepeatInstruction = false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2008-04-21 15:45:10 +08:00
|
|
|
if (StoreInst *SI = dyn_cast<StoreInst>(I))
|
2009-09-02 01:09:55 +08:00
|
|
|
MadeChange |= processStore(SI, BI);
|
2011-01-09 05:19:19 +08:00
|
|
|
else if (MemSetInst *M = dyn_cast<MemSetInst>(I))
|
|
|
|
RepeatInstruction = processMemSet(M, BI);
|
|
|
|
else if (MemCpyInst *M = dyn_cast<MemCpyInst>(I))
|
2010-11-21 08:28:59 +08:00
|
|
|
RepeatInstruction = processMemCpy(M);
|
2011-01-09 05:19:19 +08:00
|
|
|
else if (MemMoveInst *M = dyn_cast<MemMoveInst>(I))
|
2010-11-21 08:28:59 +08:00
|
|
|
RepeatInstruction = processMemMove(M);
|
2011-01-09 05:19:19 +08:00
|
|
|
else if (CallSite CS = (Value*)I) {
|
2010-11-21 08:28:59 +08:00
|
|
|
for (unsigned i = 0, e = CS.arg_size(); i != e; ++i)
|
2011-11-21 03:09:04 +08:00
|
|
|
if (CS.isByValArgument(i))
|
2010-11-21 08:28:59 +08:00
|
|
|
MadeChange |= processByValArgument(CS, i);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Reprocess the instruction if desired.
|
|
|
|
if (RepeatInstruction) {
|
2011-01-09 06:19:21 +08:00
|
|
|
if (BI != BB->begin()) --BI;
|
2010-11-21 08:28:59 +08:00
|
|
|
MadeChange = true;
|
2009-09-02 01:56:32 +08:00
|
|
|
}
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2009-09-02 01:09:55 +08:00
|
|
|
return MadeChange;
|
2008-04-09 16:23:16 +08:00
|
|
|
}
|
2009-09-02 01:09:55 +08:00
|
|
|
|
|
|
|
// MemCpyOpt::runOnFunction - This is the main transformation entry point for a
|
|
|
|
// function.
|
|
|
|
//
|
|
|
|
bool MemCpyOpt::runOnFunction(Function &F) {
|
2014-02-06 08:07:05 +08:00
|
|
|
if (skipOptnoneFunction(F))
|
|
|
|
return false;
|
|
|
|
|
2009-09-02 01:09:55 +08:00
|
|
|
bool MadeChange = false;
|
2010-11-21 08:28:59 +08:00
|
|
|
MD = &getAnalysis<MemoryDependenceAnalysis>();
|
2014-02-26 01:30:31 +08:00
|
|
|
DataLayoutPass *DLP = getAnalysisIfAvailable<DataLayoutPass>();
|
2014-04-25 13:29:35 +08:00
|
|
|
DL = DLP ? &DLP->getDataLayout() : nullptr;
|
2011-05-02 02:27:11 +08:00
|
|
|
TLI = &getAnalysis<TargetLibraryInfo>();
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2011-05-02 02:27:11 +08:00
|
|
|
// If we don't have at least memset and memcpy, there is little point of doing
|
|
|
|
// anything here. These are required by a freestanding implementation, so if
|
|
|
|
// even they are disabled, there is no point in trying hard.
|
|
|
|
if (!TLI->has(LibFunc::memset) || !TLI->has(LibFunc::memcpy))
|
|
|
|
return false;
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2009-09-02 01:09:55 +08:00
|
|
|
while (1) {
|
|
|
|
if (!iterateOnFunction(F))
|
|
|
|
break;
|
|
|
|
MadeChange = true;
|
|
|
|
}
|
2012-07-24 18:51:42 +08:00
|
|
|
|
2014-04-25 13:29:35 +08:00
|
|
|
MD = nullptr;
|
2009-09-02 01:09:55 +08:00
|
|
|
return MadeChange;
|
|
|
|
}
|