AsyncCopyTool.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHCUDASERVICES_ASYNCCOPYTOOL_H
#define ATHCUDASERVICES_ASYNCCOPYTOOL_H
 
// Framework include(s).
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
// AthDevice include(s).
#include "AthDeviceInterfaces/ICopyTool.h"
 
// AthCUDA include(s).
#include "AthCUDAInterfaces/IStreamTool.h"
 
namespace AthCUDA {

class AsyncCopyTool : public extends<AthAlgTool, AthDevice::ICopyTool> {
 
 public:
  // Inherit the base class's constructor(s).
  using extends::extends;


  virtual StatusCode initialize() override;



  virtual std::shared_ptr<const vecmem::copy> copy(
      const EventContext& ctx) const override;

 
 private:
  ToolHandle<IStreamTool> m_streamTool{
      this, "StreamTool", "", "Tool to get the current CUDA stream from"};
 
};  // class AsyncCopyTool
 
}  // namespace AthCUDA
 
#endif  // ATHCUDASERVICES_ASYNCCOPYTOOL_H