cms::cuda::ScopedContextTask Class Reference

#include <ScopedContext.h>

Inheritance diagram for cms::cuda::ScopedContextTask:

Public Member Functions
template<typename F >
void	pushNextTask (F &&f)
void	replaceWaitingTaskHolder (edm::WaitingTaskWithArenaHolder waitingTaskHolder)
	ScopedContextTask (ContextState const *state, edm::WaitingTaskWithArenaHolder waitingTaskHolder)
	Constructor to re-use the CUDA stream of acquire() (ExternalWork module) More...
	~ScopedContextTask ()
Public Member Functions inherited from cms::cuda::impl::ScopedContextBase
int	device () const
cudaStream_t	stream () const
const SharedStreamPtr &	streamPtr () const

Private Attributes
ContextState const *	contextState_
impl::ScopedContextHolderHelper	holderHelper_

Additional Inherited Members
Protected Member Functions inherited from cms::cuda::impl::ScopedContextBase
	ScopedContextBase (edm::StreamID streamID)
	ScopedContextBase (const ProductBase &data)
	ScopedContextBase (int device, SharedStreamPtr stream)

Detailed Description

The aim of this class is to do necessary per-task "initialization" tasks created in ExternalWork acquire():

• setting the current device
• calling edm::WaitingTaskWithArenaHolder::doneWaiting() when necessary and enforce that those get done in a proper way in RAII fashion.

Definition at line 192 of file ScopedContext.h.

Constructor & Destructor Documentation

ScopedContextTask()

cms::cuda::ScopedContextTask::ScopedContextTask ( ContextState const *  state, edm::WaitingTaskWithArenaHolder  waitingTaskHolder  )

inlineexplicit

Constructor to re-use the CUDA stream of acquire() (ExternalWork module)

Definition at line 195 of file ScopedContext.h.

References eostools::move().

          : ScopedContextBase(state->device(), state->streamPtr()),  // don't move, state is re-used afterwards
            holderHelper_{std::move(waitingTaskHolder)},
            contextState_{state} {}

~ScopedContextTask()

cms::cuda::ScopedContextTask::~ScopedContextTask

(

)

Definition at line 95 of file ScopedContext.cc.

References cms::cuda::impl::ScopedContextBase::device(), cms::cuda::impl::ScopedContextHolderHelper::enqueueCallback(), holderHelper_, and cms::cuda::impl::ScopedContextBase::stream().

{ holderHelper_.enqueueCallback(device(), stream()); }

Member Function Documentation

pushNextTask()

template<typename F >

void cms::cuda::ScopedContextTask::pushNextTask ( F &&  f )

inline

Definition at line 203 of file ScopedContext.h.

References contextState_, f, holderHelper_, and cms::cuda::impl::ScopedContextHolderHelper::pushNextTask().

Referenced by TestCUDAProducerGPUEWTask::addSimpleWork().

                               {
        holderHelper_.pushNextTask(std::forward<F>(f), contextState_);
      }

replaceWaitingTaskHolder()

void cms::cuda::ScopedContextTask::replaceWaitingTaskHolder ( edm::WaitingTaskWithArenaHolder  waitingTaskHolder )

inline

Definition at line 207 of file ScopedContext.h.

References holderHelper_, eostools::move(), and cms::cuda::impl::ScopedContextHolderHelper::replaceWaitingTaskHolder().

                                                                                     {
        holderHelper_.replaceWaitingTaskHolder(std::move(waitingTaskHolder));
      }

Member Data Documentation

contextState_

ContextState const* cms::cuda::ScopedContextTask::contextState_

private

Definition at line 213 of file ScopedContext.h.

Referenced by pushNextTask().

holderHelper_

impl::ScopedContextHolderHelper cms::cuda::ScopedContextTask::holderHelper_

private

Definition at line 212 of file ScopedContext.h.

Referenced by pushNextTask(), replaceWaitingTaskHolder(), and ~ScopedContextTask().