Program Listing for File AllocationAdvisor.cpp

↰ Return to documentation for file (umpire/strategy/AllocationAdvisor.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/strategy/AllocationAdvisor.hpp"

#include "umpire/ResourceManager.hpp"
#include "umpire/op/MemoryOperationRegistry.hpp"

#if defined(UMPIRE_ENABLE_CUDA)
#include <cuda_runtime_api.h>
#endif

namespace umpire {
namespace strategy {

AllocationAdvisor::AllocationAdvisor(const std::string& name, int id,
                                     Allocator allocator,
                                     const std::string& advice_operation,
                                     int device_id)
    : AllocationAdvisor(name, id, allocator, advice_operation, allocator,
                        device_id)
{
}

AllocationAdvisor::AllocationAdvisor(const std::string& name, int id,
                                     Allocator allocator,
                                     const std::string& advice_operation,
                                     Allocator accessing_allocator,
                                     int device_id)
    : AllocationStrategy(name, id),
      m_allocator{allocator.getAllocationStrategy()},
      m_device{device_id}
{
  auto& op_registry = op::MemoryOperationRegistry::getInstance();

  m_advice_operation =
      op_registry.find(advice_operation, m_allocator, m_allocator);

#if defined(UMPIRE_ENABLE_CUDA)
  if (accessing_allocator.getPlatform() == Platform::host) {
    m_device = cudaCpuDeviceId;
  }
#else
  UMPIRE_USE_VAR(accessing_allocator);
#endif
}

void* AllocationAdvisor::allocate(std::size_t bytes)
{
  void* ptr = m_allocator->allocate(bytes);
  m_advice_operation->apply(ptr, nullptr, m_device, bytes);

  return ptr;
}

void AllocationAdvisor::deallocate(void* ptr)
{
  m_allocator->deallocate(ptr);
}

Platform AllocationAdvisor::getPlatform() noexcept
{
  return m_allocator->getPlatform();
}

MemoryResourceTraits AllocationAdvisor::getTraits() const noexcept
{
  return m_allocator->getTraits();
}

} // end of namespace strategy
} // end of namespace umpire