[llvm] r359627 - [MemorySSA] Invalidate MemorySSA if AA or DT are invalidated.

[Alina Sbirlea via llvm-commits]

Author: asbirlea
Date: Tue Apr 30 15:43:55 2019
New Revision: 359627

URL: http://llvm.org/viewvc/llvm-project?rev=359627&view=rev
Log:
[MemorySSA] Invalidate MemorySSA if AA or DT are invalidated.

Summary:
MemorySSA keeps internal pointers of AA and DT.
If these get invalidated, so should MemorySSA.

Reviewers: george.burgess.iv, chandlerc

Subscribers: jlebar, Prazek, llvm-commits

Tags: LLVM

Differential Revision: https://reviews.llvm.org/D61043

Added:
    llvm/trunk/test/Analysis/MemorySSA/invalidate.ll
Modified:
    llvm/trunk/include/llvm/Analysis/MemorySSA.h
    llvm/trunk/lib/Analysis/MemorySSA.cpp

Modified: llvm/trunk/include/llvm/Analysis/MemorySSA.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/Analysis/MemorySSA.h?rev=359627&r1=359626&r2=359627&view=diff
==============================================================================
--- llvm/trunk/include/llvm/Analysis/MemorySSA.h (original)
+++ llvm/trunk/include/llvm/Analysis/MemorySSA.h Tue Apr 30 15:43:55 2019
@@ -936,6 +936,9 @@ public:
     MemorySSA &getMSSA() { return *MSSA.get(); }
 
     std::unique_ptr<MemorySSA> MSSA;
+
+    bool invalidate(Function &F, const PreservedAnalyses &PA,
+                    FunctionAnalysisManager::Invalidator &Inv);
   };
 
   Result run(Function &F, FunctionAnalysisManager &AM);

Modified: llvm/trunk/lib/Analysis/MemorySSA.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Analysis/MemorySSA.cpp?rev=359627&r1=359626&r2=359627&view=diff
==============================================================================
--- llvm/trunk/lib/Analysis/MemorySSA.cpp (original)
+++ llvm/trunk/lib/Analysis/MemorySSA.cpp Tue Apr 30 15:43:55 2019
@@ -2213,6 +2213,15 @@ MemorySSAAnalysis::Result MemorySSAAnaly
   return MemorySSAAnalysis::Result(llvm::make_unique<MemorySSA>(F, &AA, &DT));
 }
 
+bool MemorySSAAnalysis::Result::invalidate(
+    Function &F, const PreservedAnalyses &PA,
+    FunctionAnalysisManager::Invalidator &Inv) {
+  auto PAC = PA.getChecker<MemorySSAAnalysis>();
+  return !(PAC.preserved() || PAC.preservedSet<AllAnalysesOn<Function>>()) ||
+         Inv.invalidate<AAManager>(F, PA) ||
+         Inv.invalidate<DominatorTreeAnalysis>(F, PA);
+}
+
 PreservedAnalyses MemorySSAPrinterPass::run(Function &F,
                                             FunctionAnalysisManager &AM) {
   OS << "MemorySSA for function: " << F.getName() << "\n";

Added: llvm/trunk/test/Analysis/MemorySSA/invalidate.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Analysis/MemorySSA/invalidate.ll?rev=359627&view=auto
==============================================================================
--- llvm/trunk/test/Analysis/MemorySSA/invalidate.ll (added)
+++ llvm/trunk/test/Analysis/MemorySSA/invalidate.ll Tue Apr 30 15:43:55 2019
@@ -0,0 +1,50 @@
+; RUN: opt -aa-pipeline=basic-aa -passes='require<memoryssa>,invalidate<aa>,early-cse-memssa' \
+; RUN:     -debug-pass-manager -disable-output %s 2>&1 \
+; RUN:     | FileCheck %s --check-prefix=CHECK-AA-INVALIDATE
+; RUN: opt -aa-pipeline=basic-aa -passes='require<memoryssa>,invalidate<domtree>,early-cse-memssa' \
+; RUN:     -debug-pass-manager -disable-output %s 2>&1 \
+; RUN:     | FileCheck %s --check-prefix=CHECK-DT-INVALIDATE
+
+; CHECK-AA-INVALIDATE: Running analysis: MemorySSAAnalysis
+; CHECK-AA-INVALIDATE: Running analysis: DominatorTreeAnalysis
+; CHECK-AA-INVALIDATE: Running analysis: AAManager
+; CHECK-AA-INVALIDATE: Running analysis: BasicAA
+; CHECK-NOT-AA-INVALIDATE: Invalidating analysis: MemorySSAAnalysis
+; CHECK-AA-INVALIDATE: Running pass: EarlyCSEPass
+; CHECK-NOT-AA-INVALIDATE: Running analysis: MemorySSAAnalysis
+
+; CHECK-DT-INVALIDATE: Running analysis: MemorySSAAnalysis
+; CHECK-DT-INVALIDATE: Running analysis: DominatorTreeAnalysis
+; CHECK-DT-INVALIDATE: Running analysis: AAManager
+; CHECK-DT-INVALIDATE: Running analysis: BasicAA
+; CHECK-DT-INVALIDATE: Running pass: InvalidateAnalysisPass<{{.*}}DominatorTreeAnalysis
+; CHECK-DT-INVALIDATE: Invalidating analysis: DominatorTreeAnalysis
+; CHECK-DT-INVALIDATE: Invalidating analysis: BasicAA
+; CHECK-DT-INVALIDATE: Invalidating analysis: AAManager
+; CHECK-DT-INVALIDATE: Invalidating analysis: MemorySSAAnalysis
+; CHECK-DT-INVALIDATE: Running pass: EarlyCSEPass
+; CHECK-DT-INVALIDATE: Running analysis: DominatorTreeAnalysis
+; CHECK-DT-INVALIDATE: Running analysis: MemorySSAAnalysis
+; CHECK-DT-INVALIDATE: Running analysis: AAManager
+; CHECK-DT-INVALIDATE: Running analysis: BasicAA
+
+
+; Function Attrs: ssp uwtable
+define i32 @main() {
+entry:
+  %call = call noalias i8* @_Znwm(i64 4)
+  %0 = bitcast i8* %call to i32*
+  %call1 = call noalias i8* @_Znwm(i64 4)
+  %1 = bitcast i8* %call1 to i32*
+  store i32 5, i32* %0, align 4
+  store i32 7, i32* %1, align 4
+  %2 = load i32, i32* %0, align 4
+  %3 = load i32, i32* %1, align 4
+  %4 = load i32, i32* %0, align 4
+  %5 = load i32, i32* %1, align 4
+  %add = add nsw i32 %3, %5
+  ret i32 %add
+}
+
+declare noalias i8* @_Znwm(i64)
+