Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Acts::TrackBuildingBase Class Referenceabstract

#include <acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/Stages.hpp>

+ Inheritance diagram for Acts::TrackBuildingBase:

Public Member Functions

virtual std::vector
< std::vector< int > > 
operator() (std::any nodes, std::any edges, std::any edgeWeights, std::vector< int > &spacepointIDs, int deviceHint=-1)=0
 
virtual ~TrackBuildingBase ()=default
 

Detailed Description

Definition at line 53 of file Stages.hpp.

View newest version in sPHENIX GitHub at line 53 of file Stages.hpp

Constructor & Destructor Documentation

virtual Acts::TrackBuildingBase::~TrackBuildingBase ( )
virtualdefault

Member Function Documentation

virtual std::vector<std::vector<int> > Acts::TrackBuildingBase::operator() ( std::any  nodes,
std::any  edges,
std::any  edgeWeights,
std::vector< int > &  spacepointIDs,
int  deviceHint = -1 
)
pure virtual

Perform track building

Parameters
nodesNode tensor with shape (n_nodes, n_node_features)
edgesEdge-index tensor with shape (2, n_edges)
edgeWeightsEdge-weights of the previous edge classification phase
spacepointIDsIDs of the nodes (must have size=n_nodes)
deviceHintWhich GPU to pick. Not relevant for CPU-only builds
Returns
tracks (as vectors of node-IDs)

Implemented in Acts::BoostTrackBuilding, and Acts::CugraphTrackBuilding.


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