ActsTrk::CUDATrkParamAlgProviderTool Class Reference

Tool providing CUDA-based traccc track parameter estimation algorithm. More...

#include <CUDATrkParamAlgProviderTool.h>

Inheritance diagram for ActsTrk::CUDATrkParamAlgProviderTool:

Collaboration diagram for ActsTrk::CUDATrkParamAlgProviderTool:

Public Member Functions
virtual StatusCode	initialize () override
	Function initializing the algorithm.
virtual DeviceAlgorithmT< traccc::device::seed_parameter_estimation_algorithm >	getAlgorithm (const EventContext &ctx, const traccc::track_params_estimation_config &trkparam_config) const override
	Function constructing the traccc cuda track parameter estimation 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 parameter estimation algorithm.

This tool constructs the traccc device track parameter estimation 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, so strip seeding and param est. should be appropriately configured.

Author

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

Definition at line 34 of file CUDATrkParamAlgProviderTool.h.

Member Function Documentation

getAlgorithm()

DeviceAlgorithmT< traccc::device::seed_parameter_estimation_algorithm > ActsTrk::CUDATrkParamAlgProviderTool::getAlgorithm ( const EventContext & ctx, const traccc::track_params_estimation_config & trkparam_config ) const

overridevirtual

Function constructing the traccc cuda track parameter estimation algorithm.

Returns

cuda track parameter estimation algorithm and vecmem copy object

Definition at line 29 of file CUDATrkParamAlgProviderTool.cxx.

{
 
  ATH_MSG_VERBOSE("Constructing CUDA traccc track parameter estimation algorithm");
  auto copy = m_copy->copy(ctx);
 
  return {copy, std::make_shared<traccc::cuda::seed_parameter_estimation_algorithm>(
    trkparam_config,
    traccc::memory_resource{m_MRs->mainMR(), m_MRs->hostMR()},
    *copy,
    traccc::cuda::stream_wrapper{m_streamTool->stream(ctx)},
    makeActsAthenaLogger(this, "TracccTrkParamEstimationCUDA"))};
 
}

initialize()

StatusCode ActsTrk::CUDATrkParamAlgProviderTool::initialize ( )

overridevirtual

Function initializing the algorithm.

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

private

Initial value:

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

Definition at line 56 of file CUDATrkParamAlgProviderTool.h.

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

m_MRs

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

private

Initial value:

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

Definition at line 51 of file CUDATrkParamAlgProviderTool.h.

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

m_streamTool

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

private

Initial value:

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

Definition at line 59 of file CUDATrkParamAlgProviderTool.h.

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

---

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

• CUDATrkParamAlgProviderTool.h
• CUDATrkParamAlgProviderTool.cxx