Program Listing for File HostReallocateOperation.cpp¶
↰ Return to documentation for file (umpire/op/HostReallocateOperation.cpp
)
//////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2016-20, Lawrence Livermore National Security, LLC and Umpire
// project contributors. See the COPYRIGHT file for details.
//
// SPDX-License-Identifier: (MIT)
//////////////////////////////////////////////////////////////////////////////
#include "umpire/op/HostReallocateOperation.hpp"
#include <cstdlib>
#include "umpire/ResourceManager.hpp"
#include "umpire/strategy/mixins/Inspector.hpp"
#include "umpire/util/Macros.hpp"
namespace umpire {
namespace op {
void HostReallocateOperation::transform(
void* current_ptr, void** new_ptr,
util::AllocationRecord* current_allocation,
util::AllocationRecord* new_allocation, std::size_t new_size)
{
auto allocator = new_allocation->strategy;
const std::size_t old_size = current_allocation->size;
//
// Since Umpire implements its own semantics for zero-length allocations, we
// cannot simply call ::realloc() with a pointer to a zero-length allocation.
//
if (old_size == 0) {
*new_ptr = allocator->allocate(new_size);
const std::size_t copy_size = (old_size > new_size) ? new_size : old_size;
ResourceManager::getInstance().copy(*new_ptr, current_ptr, copy_size);
allocator->deallocate(current_ptr);
} else {
auto old_record =
ResourceManager::getInstance().deregisterAllocation(current_ptr);
*new_ptr = ::realloc(current_ptr, new_size);
if (!*new_ptr) {
UMPIRE_ERROR("::realloc(current_ptr="
<< current_ptr << ", old_size=" << old_record.size
<< ", new_size=" << new_size << ") failed");
}
ResourceManager::getInstance().registerAllocation(
*new_ptr, {*new_ptr, new_size, allocator});
}
UMPIRE_RECORD_STATISTIC("HostReallocate", "current_ptr",
reinterpret_cast<uintptr_t>(current_ptr), "new_ptr",
reinterpret_cast<uintptr_t>(*new_ptr), "size",
new_size, "event", "reallocate");
}
} // end of namespace op
} // end of namespace umpire