StreamPoolSvc.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 ATHCUDASERVICES_STREAMPOOLSVC_H
#define ATHCUDASERVICES_STREAMPOOLSVC_H
 
// Local include(s).
#include "StreamPoolSvcImpl.h"
 
// AthCUDA include(s).
#include "AthCUDAInterfaces/IStreamPoolSvc.h"
 
// Framework include(s).
#include "Gaudi/Property.h"
#include "AthenaBaseComps/AthService.h"
 
// System include(s).
#include <memory>
 
namespace AthCUDA {
 
   class StreamPoolSvc : public extends< AthService, IStreamPoolSvc > {
 
   public:
      using extends::extends;
 
 
      virtual StatusCode initialize() override;
 
      virtual StatusCode finalize() override;
 
 
 
      virtual bool isEmpty() const override;
 
      virtual StreamHolder getAvailableStream() override;
 
      virtual void yieldStream( StreamHolder& stream ) override;
 
 
   private:
 
      Gaudi::Property< unsigned int > m_nStreams{ this, "NStreams", 2,
         "The number of CUDA streams to use" };
 
 
 
      std::unique_ptr< StreamPoolSvcImpl > m_impl;
 
 
   }; // class StreamPoolSvc
 
} // namespace AthCUDA
 
#endif // ATHCUDASERVICES_STREAMPOOLSVC_H