Service providing CUDA streams to (reentrant) algorithms. More...

#include <PerEventStreamSvc.h>

Inheritance diagram for AthCUDA::PerEventStreamSvc:

Collaboration diagram for AthCUDA::PerEventStreamSvc:

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

Private Attributes
std::unique_ptr< const SG::SlotSpecificObj< Details::Stream, SG::InvalidSlot::Enabled > >	m_streams
	Slot specific CUDA stream(s).

Detailed Description

Service providing CUDA streams to (reentrant) algorithms.

In a very simple way. By having one CUDA stream per concurrent event (slot), and having algorithms all use a single CUDA stream for a given event.

Definition at line 27 of file PerEventStreamSvc.h.

Member Function Documentation

◆ initialize()

StatusCode AthCUDA::PerEventStreamSvc::initialize ( )

overridevirtual

Initialize the tool.

Definition at line 14 of file PerEventStreamSvc.cxx.

                                         {
 
  // Create the streams object.
  m_streams = std::make_unique<
      const SG::SlotSpecificObj<Details::Stream, SG::InvalidSlot::Enabled>>();
 
  // Tell the user what happened.
  ATH_MSG_INFO("Initialized "
               << std::distance(m_streams->begin(), m_streams->end())
               << " CUDA stream(s):");
  for (const auto& stream : *m_streams) {
    ATH_MSG_INFO("  - " << stream.name());
  }
 
  // Return gracefully.
  return StatusCode::SUCCESS;
}

◆ stream()

cudaStream_t AthCUDA::PerEventStreamSvc::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 32 of file PerEventStreamSvc.cxx.

                                                                    {
 
  // Get the stream corresponding to the current slot.
  assert(m_streams);
  return m_streams->get(ctx)->stream();
}

Member Data Documentation

◆ m_streams

std::unique_ptr< const SG::SlotSpecificObj<Details::Stream, SG::InvalidSlot::Enabled> > AthCUDA::PerEventStreamSvc::m_streams

private

Slot specific CUDA stream(s).

Definition at line 57 of file PerEventStreamSvc.h.

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