File CugraphTrackBuilding.hpp

namespace Acts


This file is foreseen for the Geometry module to replace Extent

class CugraphTrackBuilding : public Acts::TrackBuildingBase
#include <Acts/Plugins/ExaTrkX/CugraphTrackBuilding.hpp>

Public Functions

virtual std::vector<std::vector<int>> operator()(std::any nodes, std::any edges, std::any edge_weights, std::vector<int> &spacepointIDs, const Logger &logger) override

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)

  • logger – Logger instance


tracks (as vectors of node-IDs)