Merge pull request #2573 from trentxintong/MemLeak

Fix a memory leak in DSE

Author: trentxintong

lib/SILOptimizer/Transforms/DeadStoreElimination.cpp

@@ -323,7 +323,7 @@ enum class ProcessKind {
   TypeExpansionAnalysis *TE;
 
   /// The allocator we are using.
-  llvm::BumpPtrAllocator &BPA;
+  llvm::SpecificBumpPtrAllocator<BlockState> &BPA;
 
   /// The epilogue release matcher we are using.
   ConsumedArgToEpilogueReleaseMatcher& ERM;
@@ -429,7 +429,7 @@ enum class ProcessKind {
   /// Constructor.
   DSEContext(SILFunction *F, SILModule *M, SILPassManager *PM,
              AliasAnalysis *AA, TypeExpansionAnalysis *TE,
-             llvm::BumpPtrAllocator &BPA,
+             llvm::SpecificBumpPtrAllocator<BlockState> &BPA,
              ConsumedArgToEpilogueReleaseMatcher &ERM)
     : Mod(M), F(F), PM(PM), AA(AA), TE(TE), BPA(BPA), ERM(ERM) {}
 
@@ -1154,7 +1154,7 @@ bool DSEContext::run() {
   // Initialize the BBToLocState mapping.
   unsigned LocationNum = this->getLocationVault().size();
   for (auto &B : *F) {
-    auto *State = new (BPA) BlockState(&B, LocationNum, Optimistic);
+    auto *State = new (BPA.Allocate()) BlockState(&B, LocationNum, Optimistic);
     BBToLocState[&B] = State;
     State->initStoreSetAtEndOfBlock(*this);
   }
@@ -1234,7 +1234,7 @@ class DeadStoreElimination : public SILFunctionTransform {
     auto *RCFI = PM->getAnalysis<RCIdentityAnalysis>()->get(F);
 
     // The allocator we are using.
-    llvm::BumpPtrAllocator BPA;
+    llvm::SpecificBumpPtrAllocator<BlockState> BPA;
 
     // The epilogue release matcher we are using.
     ConsumedArgToEpilogueReleaseMatcher ERM(RCFI, F);