0588de76bb
Reviewers: buda Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D526
44 lines
899 B
C++
44 lines
899 B
C++
#pragma once
|
|
|
|
#include <cstdlib>
|
|
#include <new>
|
|
|
|
template <class Tp>
|
|
struct fast_allocator {
|
|
typedef Tp value_type;
|
|
|
|
fast_allocator() = default;
|
|
|
|
template <class T>
|
|
fast_allocator(const fast_allocator<T>&) {}
|
|
|
|
Tp* allocate(std::size_t n);
|
|
void deallocate(Tp* p, std::size_t n);
|
|
};
|
|
|
|
template <class Tp>
|
|
Tp* fast_allocator<Tp>::allocate(std::size_t n) {
|
|
// hopefully we're using jemalloc here!
|
|
Tp* mem = static_cast<Tp*>(malloc(n * sizeof(Tp)));
|
|
|
|
if (mem != nullptr) return mem;
|
|
|
|
throw std::bad_alloc();
|
|
}
|
|
|
|
template <class Tp>
|
|
void fast_allocator<Tp>::deallocate(Tp* p, std::size_t) {
|
|
// hopefully we're using jemalloc here!
|
|
free(p);
|
|
}
|
|
|
|
template <class T, class U>
|
|
bool operator==(const fast_allocator<T>&, const fast_allocator<U>&) {
|
|
return true;
|
|
}
|
|
|
|
template <class T, class U>
|
|
bool operator!=(const fast_allocator<T>& a, const fast_allocator<U>& b) {
|
|
return !(a == b);
|
|
}
|