ActsTrk::CUDASeedingAlgProviderTool Class Reference

Tool providing CUDA-based traccc seeding algorithm. More...

#include <CUDASeedingAlgProviderTool.h>

Inheritance diagram for ActsTrk::CUDASeedingAlgProviderTool:

Collaboration diagram for ActsTrk::CUDASeedingAlgProviderTool:

Public Member Functions
virtual StatusCode	initialize () override
	Function initializing the algorithm.
virtual DeviceAlgorithmT< traccc::device::triplet_seeding_algorithm >	getTripletSeedingAlgorithm (const EventContext &ctx, const traccc::seedfinder_config &seedfinder, const traccc::spacepoint_grid_config &grid, const traccc::seedfilter_config &seedfilter) const override
	Function constructing the traccc cuda seeding algorithm.
virtual DeviceAlgorithmT< traccc::device::gbts_seeding_algorithm >	getGBTSAlgorithm (const EventContext &ctx, const traccc::gbts_seedfinder_config &gbts_config) const override
	Get the device specific GBTS seeding algorithm.

Private Attributes
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 seeding algorithm.

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

Currently only pixel seeding is implemented in Traccc

Author

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

Definition at line 33 of file CUDASeedingAlgProviderTool.h.

Member Function Documentation

getGBTSAlgorithm()

DeviceAlgorithmT< traccc::device::gbts_seeding_algorithm > ActsTrk::CUDASeedingAlgProviderTool::getGBTSAlgorithm ( const EventContext & ctx, const traccc::gbts_seedfinder_config & gbts_config ) const

overridevirtual

Get the device specific GBTS seeding algorithm.

Definition at line 48 of file CUDASeedingAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc pixel GBTS seeding algorithm");
  auto copy = m_copy->copy(ctx);
 
  return {copy, std::make_shared<traccc::cuda::gbts_seeding_algorithm>(
    gbts_config,
    traccc::memory_resource{m_MRs->mainMR(), m_MRs->hostMR()},
    *copy,
    traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
    makeActsAthenaLogger(this, "TracccGBTSSeedingCUDA"))};
 
}

getTripletSeedingAlgorithm()

DeviceAlgorithmT< traccc::device::triplet_seeding_algorithm > ActsTrk::CUDASeedingAlgProviderTool::getTripletSeedingAlgorithm ( const EventContext & ctx, const traccc::seedfinder_config & seedfinder, const traccc::spacepoint_grid_config & grid, const traccc::seedfilter_config & seedfilter ) const

overridevirtual

Function constructing the traccc cuda seeding algorithm.

Returns

cuda triplet seeding algorithm and vecmem copy object

Definition at line 30 of file CUDASeedingAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc pixel triplet seeding algorithm");
  auto copy = m_copy->copy(ctx);
 
  return {copy, std::make_shared<traccc::cuda::triplet_seeding_algorithm>(
    seedfinder,
    grid,
    seedfilter,
    traccc::memory_resource{m_MRs->mainMR(), m_MRs->hostMR()},
    *copy,
    traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
    makeActsAthenaLogger(this, "TracccTripletSeedingCUDA"))};
 
}

initialize()

StatusCode ActsTrk::CUDASeedingAlgProviderTool::initialize ( )

overridevirtual

Function initializing the algorithm.

Definition at line 17 of file CUDASeedingAlgProviderTool.cxx.

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

Member Data Documentation

m_copy

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

private

Initial value:

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

Definition at line 61 of file CUDASeedingAlgProviderTool.h.

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

m_MRs

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

private

Initial value:

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

Definition at line 56 of file CUDASeedingAlgProviderTool.h.

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

m_streamTool

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

private

Initial value:

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

Definition at line 64 of file CUDASeedingAlgProviderTool.h.

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

---

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

• CUDASeedingAlgProviderTool.h
• CUDASeedingAlgProviderTool.cxx