Add allocator builder and use it in rake tasks
Add missing require for allocator builder