CUDATrkParamAlgProviderTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUPATTERNRECOGNITION_CUDATRKPARAMALGPROVIDERTOOL_H
#define ACTSGPUPATTERNRECOGNITION_CUDATRKPARAMALGPROVIDERTOOL_H
 
// Local include(s).
#include "../IDeviceTrkParamAlgProviderTool.h"
 
// Framework include(s).
#include "AthCUDAInterfaces/IStreamTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
namespace ActsTrk {

class CUDATrkParamAlgProviderTool
    : public extends<AthAlgTool, IDeviceTrkParamAlgProviderTool> {
 public:
  using extends::extends;
  virtual StatusCode initialize() override;

  virtual DeviceAlgorithmT<traccc::device::seed_parameter_estimation_algorithm>
  getAlgorithm(const EventContext& ctx,
               const traccc::track_params_estimation_config& trkparam_config)
      const override;
 
 private:
  ToolHandle<AthDevice::IMemoryResourcesTool> m_MRs{
      this, "MemoryResourcesTool", "",
      "The memory resources tool to use for allocating memory on the device"};

  ToolHandle<AthDevice::ICopyTool> m_copy{this, "CopyProviderTool", "",
                                          "Vecmem copy provider tool"};

  ToolHandle<AthCUDA::IStreamTool> m_streamTool{this, "StreamTool", "",
                                                "CUDA stream provider tool"};
 
};  // class CUDATrkParamAlgProviderTool
 
}  // namespace ActsTrk
 
#endif  // ACTSGPUPATTERNRECOGNITION_CUDATRKPARAMALGPROVIDERTOOL_H