ActsTrk::CUDATrackFindingAlgProviderTool Class Reference

Tool providing CUDA-based traccc track findng algorithm. More...

#include <CUDATrackFindingAlgProviderTool.h>

Inheritance diagram for ActsTrk::CUDATrackFindingAlgProviderTool:

Collaboration diagram for ActsTrk::CUDATrackFindingAlgProviderTool:

Public Member Functions
virtual StatusCode	initialize () override
	Function initializing the algorithm.
virtual DeviceAlgorithmT< traccc::device::combinatorial_kalman_filter_algorithm >	getAlgorithm (const EventContext &ctx, const traccc::finding_config &trkfinding_config) const override
	Function constructing the traccc cuda track finding 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 track findng algorithm.

This tool constructs the traccc device track reconstruction 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. It can accept a traccc device fitter as well, but currently MBF is emplyed ie we are fitting through finding

Author

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

Definition at line 33 of file CUDATrackFindingAlgProviderTool.h.

Member Function Documentation

getAlgorithm()

DeviceAlgorithmT< traccc::device::combinatorial_kalman_filter_algorithm > ActsTrk::CUDATrackFindingAlgProviderTool::getAlgorithm ( const EventContext & ctx, const traccc::finding_config & trkfinding_config ) const

overridevirtual

Function constructing the traccc cuda track finding algorithm.

Returns

cuda track finding algorithm and vecmem copy object

Definition at line 25 of file CUDATrackFindingAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc track finding algorithm");
  
  auto copy = m_copy->copy(ctx);
 
  return {copy, std::make_shared<traccc::cuda::combinatorial_kalman_filter_algorithm>(
    trkfinding_config,
    traccc::memory_resource{m_MRs->mainMR(), m_MRs->hostMR()},
    *copy,
    traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
    makeActsAthenaLogger(this, "TracccTrackFindingCUDA"),
    nullptr)};
 
}

initialize()

StatusCode ActsTrk::CUDATrackFindingAlgProviderTool::initialize ( )

overridevirtual

Function initializing the algorithm.

Definition at line 12 of file CUDATrackFindingAlgProviderTool.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::CUDATrackFindingAlgProviderTool::m_copy

private

Initial value:

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

Definition at line 55 of file CUDATrackFindingAlgProviderTool.h.

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

m_MRs

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

private

Initial value:

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

Definition at line 50 of file CUDATrackFindingAlgProviderTool.h.

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

m_streamTool

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

private

Initial value:

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

Definition at line 58 of file CUDATrackFindingAlgProviderTool.h.

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

---

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

• CUDATrackFindingAlgProviderTool.h
• CUDATrackFindingAlgProviderTool.cxx