CUDAStream.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 2023-2026 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#pragma once
// Gaudi
#include <Gaudi/AsynchronousAlgorithm.h>
 
// Forward declaration hack so we can have a cudaStream_t member here
struct CUstream_st;
typedef struct CUstream_st* cudaStream_t;
 
namespace Gaudi::CUDA {
  class Stream {
  private:
    cudaStream_t                        m_stream;
    const bool                          m_owning;
    const Gaudi::Algorithm*             m_parent;
    const Gaudi::AsynchronousAlgorithm* m_async_parent;
    int                                 m_dependents;
 
  public:
    Stream( const Gaudi::Algorithm* parent, cudaStream_t stream );
    Stream( const Gaudi::Algorithm* parent );
    ~Stream();

    operator cudaStream_t();

    const Gaudi::Algorithm* parent();

    const Gaudi::AsynchronousAlgorithm* asyncParent();

    StatusCode await();

    void registerDependency();

    void removeDependency();
  };
} // namespace Gaudi::CUDA