GPUClusterInfoAndMomentsCalculator.cxx

Go to the documentation of this file.

//
// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
//
// Dear emacs, this is -*- c++ -*-
//
 
#include "GPUClusterInfoAndMomentsCalculator.h"
#include "GPUClusterInfoAndMomentsCalculatorImpl.h"
 
 
using namespace CaloRecGPU;
using namespace ClusterMomentsCalculator;
 
GPUClusterInfoAndMomentsCalculator::GPUClusterInfoAndMomentsCalculator(const std::string & type, const std::string & name, const IInterface * parent):
  base_class(type, name, parent),
  CaloGPUTimed(this)
{
}
 
StatusCode GPUClusterInfoAndMomentsCalculator::initialize_non_CUDA()
{
  m_options.allocate();
 
  m_options.m_options->use_abs_energy         = m_absOpt;
  m_options.m_options->use_two_gaussian_noise = m_twoGaussianNoise;
  m_options.m_options->skip_invalid_clusters  = m_skipInvalidClusters;
  m_options.m_options->min_LAr_quality        = m_minBadLArQuality;
  m_options.m_options->max_axis_angle         = m_maxAxisAngle;
  m_options.m_options->eta_inner_wheel        = m_etaInnerWheel;
  m_options.m_options->min_l_longitudinal     = m_minLLongitudinal;
  m_options.m_options->min_r_lateral          = m_minRLateral;
 
 
  ATH_CHECK( m_kernelSizeOptimizer.retrieve() );
  ATH_CHECK( m_streamTool.retrieve() );
 
  return StatusCode::SUCCESS;
}
 
StatusCode GPUClusterInfoAndMomentsCalculator::initialize_CUDA()
{
  m_options.sendToGPU();
  register_kernels( *(m_kernelSizeOptimizer.get()) );
 
  return StatusCode::SUCCESS;
}
 
StatusCode GPUClusterInfoAndMomentsCalculator::execute(const EventContext & ctx, const ConstantDataHolder & constant_data,
                                                       EventDataHolder & event_data, void * /*temporary_buffer*/) const
{
  size_t times[num_time_measurements];
 
  calculateClusterPropertiesAndMoments(event_data, constant_data, m_options, *(m_kernelSizeOptimizer.get()), m_streamTool->stream(ctx), times, m_measureTimes);
 
  if (m_measureTimes)
    {
      record_times(ctx.evt(),
                   times[ 0],
                   times[ 1],
                   times[ 2],
                   times[ 3],
                   times[ 4],
                   times[ 5],
                   times[ 6],
                   times[ 7],
                   times[ 8],
                   times[ 9],
                   times[10]
                   );
    }
 
  return StatusCode::SUCCESS;
}
 
StatusCode GPUClusterInfoAndMomentsCalculator::finalize()
{
  if (m_measureTimes)
    {
      print_times("Isolation_Clusters Isolation_Cells Zeroth_Clusters "
                  "First_Cells First_Clusters Second_Cells Shower_Axis Second_Clusters "
                  "Third_Cells Third_Clusters Finalize_Clusters", num_time_measurements);
    }
  return StatusCode::SUCCESS;
}