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/util/Macros.hpp"
#include "umpire/strategy/mixins/Inspector.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