KernelRunnerSvc.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_KERNELRUNNERSVC_H
#define ATHCUDASERVICES_KERNELRUNNERSVC_H
 
// Local include(s).
#include "KernelRunnerSvcImpl.h"
 
// AthCUDA include(s).
#include "AthCUDAInterfaces/IKernelRunnerSvc.h"
#include "AthCUDAInterfaces/IStreamPoolSvc.h"
 
// Framework include(s).
#include "GaudiKernel/ServiceHandle.h"
#include "Gaudi/Property.h"
#include "AthenaBaseComps/AthService.h"
 
// System include(s).
#include <atomic>
#include <memory>
 
namespace AthCUDA {

   class KernelRunnerSvc : public extends< AthService, IKernelRunnerSvc > {
 
   public:
      using extends::extends;


      virtual StatusCode initialize() override;

      virtual StatusCode finalize() override;



      virtual StatusCode
      execute( std::unique_ptr< IKernelTask > task ) override;

 
      // Function telling the service that a task has finished its execution
      void setTaskFinished();
 
   private:

      Gaudi::Property< int > m_nKernels{ this, "NParallelKernels", 2,
         "The number of CUDA kernels to execute in parallel" };

      ServiceHandle< IStreamPoolSvc > m_streamPoolSvc{ this, "StreamPoolSvc",
         "AthCUDA::StreamPoolSvc",
         "The AthCUDA::StreamPoolSvc instance to use" };


      std::atomic_int m_kernelsInFlight;

      std::atomic_uint m_totalTasks;
      std::atomic_uint m_gpuTasks;

      std::unique_ptr< KernelRunnerSvcImpl > m_impl;
 
   }; // class KernelRunnerSvc
 
} // namespace AthCUDA
 
#endif // ATHCUDASERVICES_KERNELRUNNERSVC_H