Added Reset() to free all allocated memory regions and reset state to be the same as right after ctor.

llvm-svn: 41728

1 files changed, 6 insertions, 0 deletions

diff --git a/llvm/lib/Support/Allocator.cpp b/llvm/lib/Support/Allocator.cpp
index 234fd41b738..e68c3e26c7d 100644
--- a/llvm/lib/Support/Allocator.cpp
+++ b/llvm/lib/Support/Allocator.cpp
@@ -92,6 +92,12 @@ BumpPtrAllocator::~BumpPtrAllocator() {
   ((MemRegion*)TheMemory)->Deallocate();
 }
 
+void BumpPtrAllocator::Reset() {
+  ((MemRegion*)TheMemory)->Deallocate();
+  TheMemory = malloc(4096);
+  ((MemRegion*)TheMemory)->Init(4096, 1, 0);
+}
+
 void *BumpPtrAllocator::Allocate(unsigned Size, unsigned Align) {
   MemRegion *MRP = (MemRegion*)TheMemory;
   void *Ptr = MRP->Allocate(Size, Align, &MRP);