Profiling.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2022 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* Profiling.h                                                 (C) 2000-2022 */
/*                                                                           */
/* Classes pour gérer le profilage.                                          */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_UTILS_PROFILING_H
#define ARCANE_UTILS_PROFILING_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/UtilsTypes.h"
 
#include "arcane/utils/String.h"
 
#include <atomic>
#include <functional>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::impl
{
class AcceleratorStatInfoList;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_UTILS_EXPORT ScopedStatLoop
{
 public:
 
  explicit ScopedStatLoop(ForLoopOneExecStat* s);
  ~ScopedStatLoop();
 
 public:
 
  Int64 m_begin_time = 0.0;
  ForLoopOneExecStat* m_stat_info = nullptr;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_UTILS_EXPORT ForLoopStatInfoList
{
 public:
 
  ForLoopStatInfoList();
  ~ForLoopStatInfoList();
 
 public:
 
  void merge(const ForLoopOneExecStat& loop_stat_info, const ForLoopTraceInfo& loop_trace_info);
 
 public:

  ForLoopStatInfoListImpl* _internalImpl() const { return m_p; }
 
 private:
 
  ForLoopStatInfoListImpl* m_p = nullptr;
};
 
} // namespace Arcane::impl
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_UTILS_EXPORT ForLoopOneExecStat
{
 public:

  void incrementNbChunk() { ++m_nb_chunk; }

  void setBeginTime(Int64 v) { m_begin_time = v; }

  void setEndTime(Int64 v) { m_end_time = v; }

  Int64 nbChunk() const { return m_nb_chunk; }

  Int64 execTime() const { return m_end_time - m_begin_time; }
 
  void reset()
  {
    m_nb_chunk = 0;
    m_begin_time = 0;
    m_end_time = 0;
  }
 
 private:

  std::atomic<Int64> m_nb_chunk = 0;
 
  // Temps de début d'exécution
  Int64 m_begin_time = 0;
 
  // Temps de fin d'exécution
  Int64 m_end_time = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_UTILS_EXPORT ProfilingRegistry
{
 public:

  ARCANE_DEPRECATED_REASON("Y2023: Use _threadLocalForLoopInstance() instead")
  static impl::ForLoopStatInfoList* threadLocalInstance();

  static void setProfilingLevel(Int32 level);

  static Int32 profilingLevel() { return m_profiling_level; }

  static bool hasProfiling() { return m_profiling_level > 0; }

  static void visitLoopStat(const std::function<void(const impl::ForLoopStatInfoList&)>& f);

  static void visitAcceleratorStat(const std::function<void(const impl::AcceleratorStatInfoList&)>& f);
 
  static const impl::ForLoopCumulativeStat& globalLoopStat();
 
 public:
 
  // API publique mais réservée à Arcane.

  static impl::ForLoopStatInfoList* _threadLocalForLoopInstance();

  static impl::AcceleratorStatInfoList* _threadLocalAcceleratorInstance();
 
 private:
 
  static Int32 m_profiling_level;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif