ActsPlugins::CudaTrackBuilding Class Reference

Track building implementation using CUDA. More...

#include <ActsPlugins/Gnn/CudaTrackBuilding.hpp>

Inheritance diagram for ActsPlugins::CudaTrackBuilding:

Collaboration diagram for ActsPlugins::CudaTrackBuilding:

Classes
struct	Config
	Configuration for CUDA track building. More...

Public Member Functions
	CudaTrackBuilding (const Config &cfg, std::unique_ptr< const Acts::Logger > logger)
	Constructor.
const Config &	config () const
	Get configuration.
std::vector< std::vector< int > >	operator() (PipelineTensors tensors, std::vector< int > &spacePointIDs, const ExecutionContext &execContext={}) override
	Perform track building.

Detailed Description

Track building implementation using CUDA.

Constructor & Destructor Documentation

CudaTrackBuilding()

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

Constructor.

Parameters

cfg	Configuration object
logger	Logger instance

Member Function Documentation

config()

const Config & ActsPlugins::CudaTrackBuilding::config

(

)

const

Get configuration.

Returns

Configuration object

operator()()

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

overridevirtual

Perform track building.

Parameters

tensors	Input pipeline tensors
spacePointIDs	IDs of the nodes (must have size=n_nodes)
execContext	Device & stream information

Returns

tracks (as vectors of node-IDs)

Implements ActsPlugins::TrackBuildingBase.