ActsTrk::CUDAClusterizationAlgProviderTool Class Reference

Tool providing CUDA-based traccc clusterization and measurement sorting algorithms. More...

#include <CUDAClusterizationAlgProviderTool.h>

Inheritance diagram for ActsTrk::CUDAClusterizationAlgProviderTool:

Collaboration diagram for ActsTrk::CUDAClusterizationAlgProviderTool:

Public Member Functions
virtual StatusCode	initialize () override
	Function initializing the algorithm.
virtual std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::clusterization_algorithm > >	getClusterizationAlgorithm (const EventContext &ctx) const override
	Function constructing the traccc cuda clusterization algorithm.
virtual std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const IDeviceClusterizationAlgProviderTool::sorting_algorithm_type > >	getSortingAlgorithm (const EventContext &ctx) const override
	Function constructing the traccc cuda measurement sorting algorithm Neccesary because the CKF requires measurements to be sorted by module ID.

Private Attributes
traccc::clustering_config	m_clusteringConfig {}
Whether to sort traccc cells on GPU prior to clusterization
Gaudi::Property< bool >	m_sortCells
The host and device memory resources tool to use for memory allocations
ToolHandle< AthDevice::IMemoryResourcesTool >	m_MRs
The device copy tool to use
ToolHandle< AthDevice::ICopyTool >	m_copy
The cuda stream provider tool
ToolHandle< AthCUDA::IStreamTool >	m_streamTool

Detailed Description

Tool providing CUDA-based traccc clusterization and measurement sorting algorithms.

This tool constructs the traccc device clusterization algorithm, configured to run on with CUDA backend. It relies on separate tools to provide the host/device memory resources, the vecmem copy object, and the CUDA stream used for asynchronous execution.

Author

Neža Ribarič neza..nosp@m.riba.nosp@m.ric@c.nosp@m.ern..nosp@m.ch

Definition at line 35 of file CUDAClusterizationAlgProviderTool.h.

Member Function Documentation

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

Function constructing the traccc cuda clusterization algorithm.

Returns

cuda clusterization algorithm and vecmem copy object

Definition at line 29 of file CUDAClusterizationAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc clusterization algorithm");
  traccc::memory_resource mr{m_MRs->mainMR(), m_MRs->hostMR()};
  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

Function constructing the traccc cuda measurement sorting algorithm Neccesary because the CKF requires measurements to be sorted by module ID.

Returns

cuda measurement sorting algorithm and vecmem copy object

Definition at line 46 of file CUDAClusterizationAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc measurement sorting algorithm");
  traccc::memory_resource mr{m_MRs->mainMR(), m_MRs->hostMR()};
  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

Function initializing the algorithm.

Definition at line 14 of file CUDAClusterizationAlgProviderTool.cxx.

{
  ATH_MSG_DEBUG("Initializing.");
 
  ATH_CHECK(m_MRs.retrieve());
  ATH_CHECK(m_copy.retrieve());
  ATH_CHECK(m_streamTool.retrieve());
 
  m_clusteringConfig.sort_cells = m_sortCells;
 
  ATH_MSG_DEBUG("Successfully initialized");
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_clusteringConfig

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

private

Definition at line 56 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 67 of file CUDAClusterizationAlgProviderTool.h.

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

m_MRs

ToolHandle<AthDevice::IMemoryResourcesTool> ActsTrk::CUDAClusterizationAlgProviderTool::m_MRs

private

Initial value:

{
        this, "MemoryResourcesTool", "",
        "The memory resources tool to use for allocating memory on the device"}

Definition at line 62 of file CUDAClusterizationAlgProviderTool.h.

                                                 {
        this, "MemoryResourcesTool", "",
        "The memory resources tool to use for allocating memory on the device"};

m_sortCells

Gaudi::Property<bool> ActsTrk::CUDAClusterizationAlgProviderTool::m_sortCells

private

Initial value:

{this, "CellSorting", true,
    "Whether to sort traccc cells on GPU prior to clusterization"}

Definition at line 59 of file CUDAClusterizationAlgProviderTool.h.

                                 {this, "CellSorting", true,
    "Whether to sort traccc cells on GPU prior to clusterization"};

m_streamTool

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

private

Initial value:

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

Definition at line 70 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