CUDA implementation of AthCUDA::KernelRunnerSvc. More...

#include <KernelRunnerSvcImpl.h>

Collaboration diagram for AthCUDA::KernelRunnerSvcImpl:

Public Member Functions
	KernelRunnerSvcImpl (IStreamPoolSvc &streamPool, KernelRunnerSvc &svc)
	Constructor with all necessary parameters.
void	execute (std::unique_ptr< IKernelTask > task)
	Implementation of `AthCUDA::IKernelRunnerSvc::execute`.
StreamHolder	getAvailableStream ()
	Get an available CUDA stream from the pool.

Private Attributes
IStreamPoolSvc *	m_streamPool
	The stream pool service in use.
KernelRunnerSvcImplCallback	m_callback
	Helper object to call `setTaskFinished()` with.

Detailed Description

CUDA implementation of AthCUDA::KernelRunnerSvc.

Definition at line 38 of file KernelRunnerSvcImpl.h.

Constructor & Destructor Documentation

◆ KernelRunnerSvcImpl()

AthCUDA::KernelRunnerSvcImpl::KernelRunnerSvcImpl	(	IStreamPoolSvc &	streamPool,
		KernelRunnerSvc &	svc )

Constructor with all necessary parameters.

Definition at line 30 of file KernelRunnerSvcImpl.cxx.

   : m_streamPool( &streamPool ), m_callback( svc ) {
 
   }

Member Function Documentation

◆ execute()

void AthCUDA::KernelRunnerSvcImpl::execute ( std::unique_ptr< IKernelTask > task )

Implementation of AthCUDA::IKernelRunnerSvc::execute.

◆ getAvailableStream()

StreamHolder AthCUDA::KernelRunnerSvcImpl::getAvailableStream ( )

Get an available CUDA stream from the pool.

Definition at line 36 of file KernelRunnerSvcImpl.cxx.

                                                        {
 
      assert( m_streamPool != nullptr );
      return m_streamPool->getAvailableStream();
   }

Member Data Documentation

◆ m_callback

KernelRunnerSvcImplCallback AthCUDA::KernelRunnerSvcImpl::m_callback

private

Helper object to call setTaskFinished() with.

Definition at line 56 of file KernelRunnerSvcImpl.h.

◆ m_streamPool

IStreamPoolSvc* AthCUDA::KernelRunnerSvcImpl::m_streamPool

private

The stream pool service in use.

Definition at line 53 of file KernelRunnerSvcImpl.h.

The documentation for this class was generated from the following files:

Public Member Functions

Private Attributes