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