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. More...

void	execute (std::unique_ptr< IKernelTask > task)
	Implementation of `AthCUDA::IKernelRunnerSvc::execute`. More...

StreamHolder	getAvailableStream ()
	Get an available CUDA stream from the pool. More...

Private Attributes
IStreamPoolSvc *	m_streamPool
	The stream pool service in use. More...

KernelRunnerSvcImplCallback	m_callback
	Helper object to call `setTaskFinished()` with. More...

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