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 std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::triplet_seeding_algorithm > >	getTripletSeedingAlgorithm (const EventContext &ctx, const traccc::seedfinder_config &seedfinder, const traccc::seedfilter_config &seedfilter) const override
	Function constructing the traccc cuda seeding algorithm.
virtual std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::gbts_seeding_algorithm > >	getGBTSAlgorithm (const EventContext &ctx, const traccc::gbts_seedfinder_config &gbts_config) const override

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 35 of file CUDASeedingAlgProviderTool.h.

Member Function Documentation

getGBTSAlgorithm()

std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::gbts_seeding_algorithm > > ActsTrk::CUDASeedingAlgProviderTool::getGBTSAlgorithm ( const EventContext & ctx, const traccc::gbts_seedfinder_config & gbts_config ) const

overridevirtual

Definition at line 45 of file CUDASeedingAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc pixel GBTS seeding 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::gbts_seeding_algorithm>(
    gbts_config,
    mr,
    *copy,
    traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
    makeActsAthenaLogger(this, "TracccGBTSSeedingCUDA")));
 
}

getTripletSeedingAlgorithm()

std::pair< std::shared_ptr< const vecmem::copy >, std::shared_ptr< const traccc::device::triplet_seeding_algorithm > > ActsTrk::CUDASeedingAlgProviderTool::getTripletSeedingAlgorithm ( const EventContext & ctx, const traccc::seedfinder_config & seedfinder, const traccc::seedfilter_config & seedfilter ) const

overridevirtual

Function constructing the traccc cuda seeding algorithm.

Returns

cuda seeding algorithm and vecmem copy object

Definition at line 26 of file CUDASeedingAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc pixel triplet seeding 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::triplet_seeding_algorithm>(
    seedfinder,
    seedfinder,
    seedfilter,
    mr,
    *copy,
    traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
    makeActsAthenaLogger(this, "TracccSPFormationCUDA")));
 
}

initialize()

StatusCode ActsTrk::CUDASeedingAlgProviderTool::initialize ( )

overridevirtual

Function initializing the algorithm.

Definition at line 13 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 58 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 53 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 61 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