internal/RunCommandImpl.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* RunCommandImpl.h                                            (C) 2000-2025 */
/*                                                                           */
/* Implémentation de la gestion d'une commande sur accélérateur.             */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_ACCELERATOR_CORE_INTERNAL_RUNCOMMANDIMPL_H
#define ARCANE_ACCELERATOR_CORE_INTERNAL_RUNCOMMANDIMPL_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arccore/base/TraceInfo.h"
#include "arccore/base/Profiling.h"
#include "arccore/base/String.h"
#include "arccore/base/ParallelLoopOptions.h"
 
#include "arcane/accelerator/core/AcceleratorCoreGlobal.h"
 
#include <set>
#include <stack>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Accelerator::impl
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class RunCommandImpl
{
  friend RunCommand;
  friend RunQueueImpl;
 
 public:
 
  explicit RunCommandImpl(RunQueueImpl* queue);
  ~RunCommandImpl();
  RunCommandImpl(const RunCommandImpl&) = delete;
  RunCommandImpl& operator=(const RunCommandImpl&) = delete;
 
 public:
 
  static RunCommandImpl* create(RunQueueImpl* r);
 
 public:
 
  const TraceInfo& traceInfo() const { return m_trace_info; }
  const String& kernelName() const { return m_kernel_name; }
 
 public:
 
  void notifyBeginLaunchKernel();
  void notifyEndLaunchKernel();
  void notifyEndExecuteKernel();
  impl::IReduceMemoryImpl* getOrCreateReduceMemoryImpl();
  void releaseReduceMemoryImpl(ReduceMemoryImpl* p);
  IRunQueueStream* internalStream() const;
  RunnerImpl* runner() const;
 
 public:
 
  void notifyLaunchKernelSyclEvent(void* sycl_event_ptr);
 
 private:
 
  ReduceMemoryImpl* _getOrCreateReduceMemoryImpl();
 
 private:
 
  RunQueueImpl* m_queue;
  TraceInfo m_trace_info;
  String m_kernel_name;
  Int32 m_nb_thread_per_block = 0;
  ParallelLoopOptions m_parallel_loop_options;
 
  // NOTE: cette pile gère la mémoire associé à un seul runtime
  // Si on souhaite un jour supporté plusieurs runtimes il faudra une pile
  // par runtime. On peut éventuellement limiter cela si on est sur
  // qu'une commande est associée à un seul type (au sens runtime) de RunQueue.
  std::stack<ReduceMemoryImpl*> m_reduce_memory_pool;

  std::set<ReduceMemoryImpl*> m_active_reduce_memory_list;

  bool m_has_been_launched = false;

  bool m_use_profiling = false;

  bool m_use_sequential_timer_event = false;

  IRunQueueEventImpl* m_start_event = nullptr;
  IRunQueueEventImpl* m_stop_event = nullptr;

  Int64 m_begin_time = 0;
 
  ForLoopOneExecStat m_loop_one_exec_stat;
  ForLoopOneExecStat* m_loop_one_exec_stat_ptr = nullptr;

  const bool m_use_accelerator = false;

  bool m_is_allow_reuse_command = false;

  bool m_has_living_run_command = false;

  bool m_may_be_put_in_pool = false;

  Int32 m_shared_memory_size = 0;
 
 private:
 
  void _freePools();
  void _reset();
  void _init();
  IRunQueueEventImpl* _createEvent();
  void _notifyDestroyRunCommand();
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane::Accelerator::impl
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif