core/IReduceMemoryImpl.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2025 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* IReduceMemoryImpl.h                                         (C) 2000-2025 */
/*                                                                           */
/* Interface de la gestion mémoire pour les réductions.                      */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_ACCELERATOR_CORE_IREDUCEMEMORYIMPL_H
#define ARCANE_ACCELERATOR_CORE_IREDUCEMEMORYIMPL_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/accelerator/core/AcceleratorCoreGlobal.h"
 
#include "arcane/utils/MemoryView.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Accelerator::impl
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_ACCELERATOR_CORE_EXPORT IReduceMemoryImpl
{
 public:

  struct GridMemoryInfo
  {
    MutableMemoryView m_grid_memory_values;
    unsigned int* m_grid_device_count = nullptr;
    void* m_host_memory_for_reduced_value = nullptr;
    Int32 m_warp_size = 64;
  };
 
 public:
 
  virtual ~IReduceMemoryImpl() = default;
 
 public:

  virtual void* allocateReduceDataMemory(ConstMemoryView identity_view) = 0;

  virtual void setGridSizeAndAllocate(Int32 grid_size) = 0;

  virtual Int32 gridSize() const = 0;

  virtual GridMemoryInfo gridMemoryInfo() = 0;

  virtual void copyReduceValueFromDevice() =0;

  virtual void release() = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template<typename T> T*
allocateReduceDataMemory(IReduceMemoryImpl* p,T identity)
{
  T* ptr = reinterpret_cast<T*>(p->allocateReduceDataMemory(makeMemoryView(&identity)));
  return ptr;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Accelerator::impl
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif