Class Acts::CugraphTrackBuilding

class CugraphTrackBuilding : public Acts::TrackBuildingBase

Public Functions

inline CugraphTrackBuilding(std::unique_ptr<const Logger> logger)
virtual std::vector<std::vector<int>> operator()(std::any nodes, std::any edges, std::any edge_weights, std::vector<int> &spacepointIDs, int deviceHint = -1) override

Perform track building.

Parameters

  • 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

Returns

tracks (as vectors of node-IDs)