KernelRunnerSvcImpl.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_KERNELRUNNERSVCIMPL_H
#define ATHCUDASERVICES_KERNELRUNNERSVCIMPL_H
 
// AthCUDA include(s).
#include "AthCUDAInterfaces/IKernelTask.h"
#include "AthCUDAInterfaces/StreamHolder.h"
 
// System include(s).
#include <memory>
 
namespace AthCUDA {
 
   // Forward declaration(s).
   class IStreamPoolSvc;
   class KernelRunnerSvc;

   class KernelRunnerSvcImplCallback {
 
   public:
      KernelRunnerSvcImplCallback( KernelRunnerSvc& svc );

      KernelRunnerSvcImplCallback& operator()();
 
   private:
      KernelRunnerSvc* m_svc;
 
   }; // class KernelRunnerSvcImplCallback

   class KernelRunnerSvcImpl {
 
   public:
      KernelRunnerSvcImpl( IStreamPoolSvc& streamPool,
                           KernelRunnerSvc& svc );

      void execute( std::unique_ptr< IKernelTask > task );

      StreamHolder getAvailableStream();
 
   private:
      IStreamPoolSvc* m_streamPool;

      KernelRunnerSvcImplCallback m_callback;
 
   }; // class KernelRunnerSvcImpl
 
} // namespace AthCUDA
 
#endif // ATHCUDASERVICES_KERNELRUNNERSVCIMPL_H