Program Listing for File AlignedAllocation.inl

↰ Return to documentation for file (umpire/strategy/mixins/AlignedAllocation.inl)

//////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2016-20, Lawrence Livermore National Security, LLC and Umpire
// project contributors. See the COPYRIGHT file for details.
//
// SPDX-License-Identifier: (MIT)
//////////////////////////////////////////////////////////////////////////////
#ifndef UMPIRE_aligned_allocation_INL
#define UMPIRE_aligned_allocation_INL

#include "umpire/util/allocation_statistics.hpp"
#include "umpire/util/memory_sanitizers.hpp"
#include "umpire/util/Macros.hpp"

namespace umpire {
namespace strategy {
namespace mixins {

inline std::size_t AlignedAllocation::aligned_round_up(std::size_t size)
{
  return size + (m_alignment - 1) - (size - 1) % m_alignment;
}

inline void* AlignedAllocation::aligned_allocate(std::size_t size)
{
  std::size_t total_bytes{ size + m_alignment };
  uintptr_t ptr{ reinterpret_cast<uintptr_t>(m_allocator->allocate(total_bytes)) };

  UMPIRE_POISON_MEMORY_REGION(m_allocator, reinterpret_cast<void*>(ptr), total_bytes);

  uintptr_t alignment{ m_alignment - 1 };
  uintptr_t mask{ m_mask };
  void *aligned_ptr{ reinterpret_cast<void*>( (ptr + alignment) & mask ) };

  base_pointer_map[aligned_ptr] = reinterpret_cast<void*>(ptr);

  return aligned_ptr;
}

inline void AlignedAllocation::aligned_deallocate(void* ptr)
{
  void* buffer{ base_pointer_map[ptr] };
  base_pointer_map.erase(ptr);
  m_allocator->deallocate(buffer);
}

} // namespace mixins
} // namespace strategy
} // namespace umpire
#endif // UMPIRE_aligned_allocation_INL