Analysis Software
Documentation for sPHENIX simulation software
|
#include <acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/Stages.hpp>
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 |
Definition at line 53 of file Stages.hpp.
View newest version in sPHENIX GitHub at line 53 of file Stages.hpp
|
virtualdefault |
|
pure virtual |
Perform track building
nodes | Node tensor with shape (n_nodes, n_node_features) |
edges | Edge-index tensor with shape (2, n_edges) |
edgeWeights | Edge-weights of the previous edge classification phase |
spacepointIDs | IDs of the nodes (must have size=n_nodes) |
deviceHint | Which GPU to pick. Not relevant for CPU-only builds |
Implemented in Acts::BoostTrackBuilding, and Acts::CugraphTrackBuilding.