StreamHolder.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
//
// Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHCUDAINTERFACES_STREAMHOLDER_H
#define ATHCUDAINTERFACES_STREAMHOLDER_H
 
namespace AthCUDA {
 
   // Forward declaration(s).
   class IStreamPoolSvc;

   class StreamHolder {

#ifdef __CUDACC__
      friend cudaStream_t getStream( StreamHolder& );
#endif // __CUDACC__
 
   public:
      StreamHolder();
      StreamHolder( void* stream, IStreamPoolSvc* pool );
      StreamHolder( const StreamHolder& ) = delete;
      StreamHolder( StreamHolder&& parent );
      ~StreamHolder();

      StreamHolder& operator=( const StreamHolder& ) = delete;
      StreamHolder& operator=( StreamHolder&& rhs );

      operator bool() const;
 
   private:
      void* m_stream;
      IStreamPoolSvc* m_pool;
 
   }; // class StreamHolder
 
} // namespace AthCUDA
 
#endif // ATHCUDAINTERFACES_STREAMHOLDER_H