RunQueuePool.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2024 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* RunQueuePool.h                                              (C) 2000-2024 */
/*                                                                           */
/* Collection de RunQueue.                                                   */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_ACCELERATOR_CORE_RUNQUEUE_POOL_H
#define ARCANE_ACCELERATOR_CORE_RUNQUEUE_POOL_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/UniqueArray.h"
 
#include "arcane/accelerator/core/AcceleratorCoreGlobal.h"
#include "arcane/accelerator/core/Runner.h"
#include "arcane/accelerator/core/RunQueue.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Accelerator
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_ACCELERATOR_CORE_EXPORT RunQueuePool
{
 public:
 
  RunQueuePool();
 
 public:
 
  RunQueuePool(const RunQueuePool&) = delete;
  RunQueuePool(RunQueuePool&&) = delete;
  RunQueuePool& operator=(const RunQueuePool&) = delete;
  RunQueuePool& operator=(RunQueuePool&&) = delete;
 
 public:
 
  void initialize(Runner& runner, Int32 pool_size);
  void initialize(Runner& runner, Int32 pool_size, const RunQueueBuildInfo& bi);
 
 public:
 
  const RunQueue& operator[](Int32 i) const
  {
    return m_pool[i % m_pool_size];
  }
 
  RunQueue& operator[](Int32 i)
  {
    return m_pool[i % m_pool_size];
  }
 
  void barrier() const;
 
  Int32 size() const { return m_pool_size; }
 
  void setAsync(bool v) const;
 
 private:
 
  UniqueArray<RunQueue> m_pool;
  Runner m_runner;
  Int32 m_pool_size = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Accelerator
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif