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 "AthenaKernel/SlotSpecificObj.h"
 
// AthDevice include(s).
#include "AthDeviceInterfaces/ICopyTool.h"
 
// CUDA include(s).
#include <cuda_runtime.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:
  struct Stream {
    Stream();
    ~Stream();
    cudaStream_t m_stream{nullptr};
  };  // struct Stream

  std::unique_ptr<const SG::SlotSpecificObj<Stream>> m_streams;
 
};  // class AsyncCopyTool
 
}  // namespace AthCUDA
 
#endif  // ATHCUDASERVICES_ASYNCCOPYTOOL_H