ReduceMemoryImpl.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2023 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* ReduceMemoryImpl.h                                          (C) 2000-2023 */
/*                                                                           */
/* Gestion de la mémoire pour les réductions.                                */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_ACCELERATOR_CORE_INTERNAL_REDUCEMEMORYIMPL_H
#define ARCANE_ACCELERATOR_CORE_INTERNAL_REDUCEMEMORYIMPL_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/accelerator/core/IReduceMemoryImpl.h"
 
#include "arcane/utils/Array.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Accelerator::impl
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
class ReduceMemoryImpl
: public IReduceMemoryImpl
{
 public:
 
  explicit ReduceMemoryImpl(RunCommandImpl* p);
 
 public:
 
  void* allocateReduceDataMemory(ConstMemoryView identity_view) override;
  void setGridSizeAndAllocate(Int32 grid_size) override
  {
    m_grid_size = grid_size;
    _setReducePolicy();
    _allocateGridDataMemory();
  }
  Int32 gridSize() const override { return m_grid_size; }
 
  GridMemoryInfo gridMemoryInfo() override
  {
    return m_grid_memory_info;
  }
  void copyReduceValueFromDevice() override;
  void release() override;
 
 private:
 
  RunCommandImpl* m_command = nullptr;
 
  std::byte* m_device_memory = nullptr;
 
  UniqueArray<std::byte> m_device_memory_bytes;
 
  UniqueArray<std::byte> m_host_memory_bytes;
 
  Int64 m_size = 0;
 
  Int32 m_grid_size = 0;
 
  Int64 m_data_type_size = 0;
 
  GridMemoryInfo m_grid_memory_info;
 
  UniqueArray<Byte> m_grid_buffer;
 
  UniqueArray<std::byte> m_identity_buffer;
 
  UniqueArray<unsigned int> m_grid_device_count;
 
 private:
 
  void _allocateGridDataMemory();
  void _allocateMemoryForGridDeviceCount();
  void _setReducePolicy();
  void _allocateMemoryForReduceData(Int32 new_size)
  {
    m_device_memory_bytes.resize(new_size);
    m_device_memory = m_device_memory_bytes.data();
 
    m_host_memory_bytes.resize(new_size);
    m_grid_memory_info.m_host_memory_for_reduced_value = m_host_memory_bytes.data();
 
    m_size = new_size;
  }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane::Accelerator::impl
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif