|
Packit Service |
7f3b24 |
// SPDX-License-Identifier: BSD-2-Clause
|
|
Packit Service |
7f3b24 |
/* Copyright (C) 2014 - 2020 Intel Corporation. */
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
#pragma once
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
#include <hbwmalloc.h>
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
#include <stddef.h>
|
|
Packit Service |
7f3b24 |
#include <new>
|
|
Packit Service |
7f3b24 |
/*
|
|
Packit Service |
7f3b24 |
* Header file for the C++ allocator compatible with the C++ standard library allocator concepts.
|
|
Packit Service |
7f3b24 |
* More details in hbwallocator(3) man page.
|
|
Packit Service |
7f3b24 |
* Note: memory heap management is based on hbwmalloc, refer to the hbwmalloc man page for more information.
|
|
Packit Service |
7f3b24 |
*
|
|
Packit Service |
7f3b24 |
* Functionality defined in this header is considered as EXPERIMENTAL API.
|
|
Packit Service |
7f3b24 |
* API standards are described in memkind(3) man page.
|
|
Packit Service |
7f3b24 |
*/
|
|
Packit Service |
7f3b24 |
namespace hbw
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
template <class T>
|
|
Packit Service |
7f3b24 |
class allocator
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
public:
|
|
Packit Service |
7f3b24 |
/*
|
|
Packit Service |
7f3b24 |
* Public member types required and defined by the standard library allocator concepts.
|
|
Packit Service |
7f3b24 |
*/
|
|
Packit Service |
7f3b24 |
typedef size_t size_type;
|
|
Packit Service |
7f3b24 |
typedef ptrdiff_t difference_type;
|
|
Packit Service |
7f3b24 |
typedef T *pointer;
|
|
Packit Service |
7f3b24 |
typedef const T *const_pointer;
|
|
Packit Service |
7f3b24 |
typedef T &reference;
|
|
Packit Service |
7f3b24 |
typedef const T &const_reference;
|
|
Packit Service |
7f3b24 |
typedef T value_type;
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
template <class U>
|
|
Packit Service |
7f3b24 |
struct rebind {
|
|
Packit Service |
7f3b24 |
typedef hbw::allocator<U> other;
|
|
Packit Service |
7f3b24 |
};
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
/*
|
|
Packit Service |
7f3b24 |
* Public member functions required and defined by the standard library allocator concepts.
|
|
Packit Service |
7f3b24 |
*/
|
|
Packit Service |
7f3b24 |
allocator() throw() { }
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
template <class U>
|
|
Packit Service |
7f3b24 |
allocator(const allocator<U> &) throw() { }
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
~allocator() throw() { }
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
pointer address(reference x) const
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
return &x;
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
const_pointer address(const_reference x) const
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
return &x;
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
/*
|
|
Packit Service |
7f3b24 |
* Allocates n*sizeof(T) bytes of high bandwidth memory using hbw_malloc().
|
|
Packit Service |
7f3b24 |
* Throws std::bad_alloc when cannot allocate memory.
|
|
Packit Service |
7f3b24 |
*/
|
|
Packit Service |
7f3b24 |
pointer allocate(size_type n, const void * = 0)
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
if (n > this->max_size()) {
|
|
Packit Service |
7f3b24 |
throw std::bad_alloc();
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
pointer result = static_cast<pointer>(hbw_malloc(n * sizeof(T)));
|
|
Packit Service |
7f3b24 |
if (!result) {
|
|
Packit Service |
7f3b24 |
throw std::bad_alloc();
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
return result;
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
/*
|
|
Packit Service |
7f3b24 |
* Deallocates memory associated with pointer returned by allocate() using hbw_free().
|
|
Packit Service |
7f3b24 |
*/
|
|
Packit Service |
7f3b24 |
void deallocate(pointer p, size_type n)
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
hbw_free(static_cast<void *>(p));
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
size_type max_size() const throw()
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
return size_t(-1) / sizeof(T);
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
void construct(pointer p, const_reference val)
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
::new(p) value_type(val);
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
void destroy(pointer p)
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
p->~T();
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
};
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
template <class T, class U>
|
|
Packit Service |
7f3b24 |
bool operator==(const allocator<T> &, const allocator<U> &)
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
return true;
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
template <class T, class U>
|
|
Packit Service |
7f3b24 |
bool operator!=(const allocator<T> &, const allocator<U> &)
|
|
Packit Service |
7f3b24 |
{
|
|
Packit Service |
7f3b24 |
return false;
|
|
Packit Service |
7f3b24 |
}
|
|
Packit Service |
7f3b24 |
|
|
Packit Service |
7f3b24 |
}
|