ImutAVLTree now allocates tree nodes from the BumpPtrAllocator using

the new type-aligned Allocate() method.

llvm-svn: 43100

1 files changed, 3 insertions, 4 deletions

diff --git a/llvm/include/llvm/ADT/ImmutableSet.h b/llvm/include/llvm/ADT/ImmutableSet.h
index 3ff6571481a..0210f27b895 100644
--- a/llvm/include/llvm/ADT/ImmutableSet.h
+++ b/llvm/include/llvm/ADT/ImmutableSet.h
@@ -373,12 +373,11 @@ private:
     
     assert (InsertPos != NULL);
     
-    // FIXME: more intelligent calculation of alignment.
-    TreeTy* T = (TreeTy*) Allocator.Allocate(sizeof(*T),16);
-    
+    // Allocate the new tree node and insert it into the cache.
+    TreeTy* T = Allocator.Allocate<TreeTy>();    
     new (T) TreeTy(L,R,V,IncrementHeight(L,R));
-    
     Cache.InsertNode(T,InsertPos);
+
     return T;      
   }