getLocationForDest should work for memset as well.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@120380 91177308-0d34-0410-b5e6-96231b3b80d8

2 files changed, 3 insertions, 2 deletions

diff --git a/include/llvm/Analysis/AliasAnalysis.h b/include/llvm/Analysis/AliasAnalysis.h
index 8d7f3a1c40..d4f006fd11 100644
--- a/include/llvm/Analysis/AliasAnalysis.h
+++ b/include/llvm/Analysis/AliasAnalysis.h
@@ -49,6 +49,7 @@ class TargetData;
 class Pass;
 class AnalysisUsage;
 class MemTransferInst;
+class MemIntrinsic;
 
 class AliasAnalysis {
 protected:
@@ -137,7 +138,7 @@ public:
   Location getLocation(const StoreInst *SI);
   Location getLocation(const VAArgInst *VI);
   static Location getLocationForSource(const MemTransferInst *MTI);
-  static Location getLocationForDest(const MemTransferInst *MTI);
+  static Location getLocationForDest(const MemIntrinsic *MI);
 
   /// Alias analysis result - Either we know for sure that it does not alias, we
   /// know for sure it must alias, or we don't know anything: The two pointers
diff --git a/lib/Analysis/AliasAnalysis.cpp b/lib/Analysis/AliasAnalysis.cpp
index f452c9e676..588d68da33 100644
--- a/lib/Analysis/AliasAnalysis.cpp
+++ b/lib/Analysis/AliasAnalysis.cpp
@@ -224,7 +224,7 @@ AliasAnalysis::getLocationForSource(const MemTransferInst *MTI) {
 }
 
 AliasAnalysis::Location 
-AliasAnalysis::getLocationForDest(const MemTransferInst *MTI) {
+AliasAnalysis::getLocationForDest(const MemIntrinsic *MTI) {
   uint64_t Size = UnknownSize;
   if (ConstantInt *C = dyn_cast<ConstantInt>(MTI->getLength()))
     Size = C->getValue().getZExtValue();