ActsTrk::CUDAClusterizationAlgProviderTool Class Reference

#include <CUDAClusterizationAlgProviderTool.h>

Inheritance diagram for ActsTrk::CUDAClusterizationAlgProviderTool:

Collaboration diagram for ActsTrk::CUDAClusterizationAlgProviderTool:

Public Member Functions
virtual StatusCode	initialize () override
virtual StatusCode	finalize () override
virtual std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::clusterization_algorithm > >	getClusterizationAlgorithm (const EventContext &ctx) const override
virtual std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const IDeviceClusterizationAlgProviderTool::sorting_algorithm_type > >	getSortingAlgorithm (const EventContext &ctx) const override

Private Attributes
traccc::clustering_config	m_clusteringConfig {}
ToolHandle< AthDevice::IMemoryResourceTool >	m_hostMR
ToolHandle< AthDevice::IMemoryResourceTool >	m_deviceMR
ToolHandle< AthDevice::ICopyTool >	m_copy
ToolHandle< AthCUDA::IStreamTool >	m_streamTool

Detailed Description

Definition at line 20 of file CUDAClusterizationAlgProviderTool.h.

Member Function Documentation

finalize()

StatusCode ActsTrk::CUDAClusterizationAlgProviderTool::finalize ( )

overridevirtual

Definition at line 27 of file CUDAClusterizationAlgProviderTool.cxx.

{
  ATH_MSG_DEBUG("Finalizing.");
  return StatusCode::SUCCESS;
}

getClusterizationAlgorithm()

std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::clusterization_algorithm > > ActsTrk::CUDAClusterizationAlgProviderTool::getClusterizationAlgorithm ( const EventContext & ctx ) const

overridevirtual

Definition at line 34 of file CUDAClusterizationAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc clusterization algorithm");
  traccc::memory_resource mr{m_deviceMR->mr(), &m_hostMR->mr()};
  auto copy = m_copy->copy(ctx);
 
  return std::make_pair(copy, std::make_shared<traccc::cuda::clusterization_algorithm>(
      mr,
      *copy,
      traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
      m_clusteringConfig,
      makeActsAthenaLogger(this, "TracccClusterizationCUDA")));
 
}

getSortingAlgorithm()

std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const IDeviceClusterizationAlgProviderTool::sorting_algorithm_type > > ActsTrk::CUDAClusterizationAlgProviderTool::getSortingAlgorithm ( const EventContext & ctx ) const

overridevirtual

Definition at line 51 of file CUDAClusterizationAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc measurement sorting algorithm");
  traccc::memory_resource mr{m_deviceMR->mr(), &m_hostMR->mr()};
  auto copy = m_copy->copy(ctx);
 
  return std::make_pair(copy, std::make_shared<traccc::cuda::measurement_sorting_algorithm>(
      mr,
      *copy,
      traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
      makeActsAthenaLogger(this, "TracccMeasurementSortingCUDA")));
 
}

initialize()

StatusCode ActsTrk::CUDAClusterizationAlgProviderTool::initialize ( )

overridevirtual

Definition at line 14 of file CUDAClusterizationAlgProviderTool.cxx.

{
  ATH_MSG_DEBUG("Initializing.");
 
  ATH_CHECK(m_hostMR.retrieve());
  ATH_CHECK(m_deviceMR.retrieve());
  ATH_CHECK(m_copy.retrieve());
  ATH_CHECK(m_streamTool.retrieve());
 
  ATH_MSG_DEBUG("Successfully initialized");
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_clusteringConfig

traccc::clustering_config ActsTrk::CUDAClusterizationAlgProviderTool::m_clusteringConfig {}

private

Definition at line 36 of file CUDAClusterizationAlgProviderTool.h.

{};

m_copy

ToolHandle<AthDevice::ICopyTool> ActsTrk::CUDAClusterizationAlgProviderTool::m_copy

private

Initial value:

{
      this, "CopyProviderTool", "", "Vecmem copy provider tool"}

Definition at line 42 of file CUDAClusterizationAlgProviderTool.h.

                                       {
      this, "CopyProviderTool", "", "Vecmem copy provider tool"};

m_deviceMR

ToolHandle<AthDevice::IMemoryResourceTool> ActsTrk::CUDAClusterizationAlgProviderTool::m_deviceMR

private

Initial value:

{
    this, "DeviceMR", "", "Device memory resource tool"}

Definition at line 40 of file CUDAClusterizationAlgProviderTool.h.

                                                     {
    this, "DeviceMR", "", "Device memory resource tool"};

m_hostMR

ToolHandle<AthDevice::IMemoryResourceTool> ActsTrk::CUDAClusterizationAlgProviderTool::m_hostMR

private

Initial value:

{
    this, "HostMR", "", "Host memory resource tool"}

Definition at line 38 of file CUDAClusterizationAlgProviderTool.h.

                                                   {
    this, "HostMR", "", "Host memory resource tool"};

m_streamTool

ToolHandle<AthCUDA::IStreamTool> ActsTrk::CUDAClusterizationAlgProviderTool::m_streamTool

private

Initial value:

{
      this, "StreamTool", "", "CUDA stream provider tool"}

Definition at line 44 of file CUDAClusterizationAlgProviderTool.h.

                                             {
      this, "StreamTool", "", "CUDA stream provider tool"};

---

The documentation for this class was generated from the following files:

• CUDAClusterizationAlgProviderTool.h
• CUDAClusterizationAlgProviderTool.cxx