ACTS
Experiment-independent tracking
Loading...
Searching...
No Matches
ActsPlugins::CudaTrackBuilding Class Referencefinal

#include <ActsPlugins/Gnn/CudaTrackBuilding.hpp>

Inheritance diagram for ActsPlugins::CudaTrackBuilding:
[legend]
Collaboration diagram for ActsPlugins::CudaTrackBuilding:
[legend]

Classes

struct  Config

Public Member Functions

 CudaTrackBuilding (const Config &cfg, std::unique_ptr< const Acts::Logger > logger)
const Configconfig () const
std::vector< std::vector< int > > operator() (PipelineTensors tensors, std::vector< int > &spacepointIDs, const ExecutionContext &execContext={}) override
 Perform track building.
Public Member Functions inherited from ActsPlugins::TrackBuildingBase
virtual ~TrackBuildingBase ()=default

Constructor & Destructor Documentation

◆ CudaTrackBuilding()

ActsPlugins::CudaTrackBuilding::CudaTrackBuilding ( const Config & cfg,
std::unique_ptr< const Acts::Logger > logger )

Member Function Documentation

◆ config()

const Config & ActsPlugins::CudaTrackBuilding::config ( ) const

◆ operator()()

std::vector< std::vector< int > > ActsPlugins::CudaTrackBuilding::operator() ( PipelineTensors tensors,
std::vector< int > & spacepointIDs,
const ExecutionContext & execContext = {} )
overridevirtual

Perform track building.

Parameters
tensorsInput pipeline tensors
spacepointIDsIDs of the nodes (must have size=n_nodes)
execContextDevice & stream information
Returns
tracks (as vectors of node-IDs)

Implements ActsPlugins::TrackBuildingBase.