CUDATrackFindingAlgProviderTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUPATTERNRECOGNITION_CUDATRACKFINDINGALGPROVIDERTOOL_H
#define ACTSGPUPATTERNRECOGNITION_CUDATRACKFINDINGALGPROVIDERTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
#include "AthDeviceInterfaces/ICopyTool.h"
#include "AthCUDAInterfaces/IStreamTool.h"
#include "../IDeviceTrackFindingAlgProviderTool.h"
 
#include <Gaudi/Property.h>
#include <memory>
#include <string>
 
namespace ActsTrk {

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

 
  virtual DeviceAlgorithmT<traccc::device::combinatorial_kalman_filter_algorithm> getAlgorithm(const EventContext& ctx, const traccc::finding_config& trkfinding_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"};    
 
};
 
} // namespace ActsTrk
 
#endif // ACTSGPUPATTERNRECOGNITION_CUDATRACKFINDINGALGPROVIDERTOOL_H