arena_allocator.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_CORE_ARENA_ALLOCATOR_HPP
#define STAN_MATH_REV_CORE_ARENA_ALLOCATOR_HPP
 
#include <stan/math/rev/core/chainablestack.hpp>
 
namespace stan {
namespace math {
 
template <typename T>
struct arena_allocator {
  using value_type = T;
 
  arena_allocator() = default;
 
  arena_allocator(const arena_allocator& rhs) = default;
 
  template <class U>
  arena_allocator(const arena_allocator<U>& rhs) {}
 
  T* allocate(std::size_t n) {
    return ChainableStack::instance_->memalloc_.alloc_array<T>(n);
  }
 
  void deallocate(T* /*p*/, std::size_t /*n*/) noexcept {}
 
  constexpr bool operator==(const arena_allocator&) const noexcept {
    return true;
  }
  constexpr bool operator!=(const arena_allocator&) const noexcept {
    return false;
  }
};
 
}  // namespace math
}  // namespace stan
 
#endif