Tool exposing a CUDA stream service as a CUDA stream tool. More...

#include <StreamSvcAdaptorTool.h>

Inheritance diagram for AthCUDA::StreamSvcAdaptorTool:

Collaboration diagram for AthCUDA::StreamSvcAdaptorTool:

Public Member Functions
Function(s) inherited from @c AthAlgTool
virtual StatusCode	initialize () override
	Initialize the tool.
Function(s) inherited from @c IStreamTool
virtual cudaStream_t	stream (const EventContext &ctx) const override
	Get the CUDA stream to use.

Private Attributes
ServiceHandle< IStreamSvc >	m_svc
	Handle to the service providing the CUDA streams.

Detailed Description

Tool exposing a CUDA stream service as a CUDA stream tool.

Client components should ideally only use CUDA streams through the AthCUDA::IStreamTool interface. (For possible future flexibility.) But, at least initially, streams will just be provided by a relatively simple service.

This tool allows us to expose such services with a tool interface.

Definition at line 26 of file StreamSvcAdaptorTool.h.

Member Function Documentation

◆ initialize()

StatusCode AthCUDA::StreamSvcAdaptorTool::initialize ( )

overridevirtual

Initialize the tool.

Definition at line 13 of file StreamSvcAdaptorTool.cxx.

                                            {
 
  // Retrieve the "wrapped" service.
  ATH_CHECK(m_svc.retrieve());
 
  // Return gracefully.
  return StatusCode::SUCCESS;
}

◆ stream()

cudaStream_t AthCUDA::StreamSvcAdaptorTool::stream ( const EventContext & ctx ) const

overridevirtual

Get the CUDA stream to use.

Parameters

ctx	The event context for which the stream is requested

Returns: The CUDA stream to use for the current event context

Definition at line 22 of file StreamSvcAdaptorTool.cxx.

                                                                       {
 
  // Just return the stream provided by the service.
  assert(m_svc.isValid());
  return m_svc->stream(ctx);
}

Member Data Documentation

◆ m_svc

ServiceHandle<IStreamSvc> AthCUDA::StreamSvcAdaptorTool::m_svc

private

Initial value:

{

this, "StreamSvc", "", "Service providing the 'adapted' CUDA streams"}

Handle to the service providing the CUDA streams.

Definition at line 54 of file StreamSvcAdaptorTool.h.

54 {

55 this, "StreamSvc", "", "Service providing the 'adapted' CUDA streams"};

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