mxnet
Public Member Functions | Static Public Member Functions | List of all members
mxnet::GPUAuxStream Class Reference

Holds an auxiliary mshadow gpu stream that can be synced with a primary stream. More...

#include <base.h>

Collaboration diagram for mxnet::GPUAuxStream:
Collaboration graph

Public Member Functions

 GPUAuxStream (mshadow::Stream< gpu > *primary_stream)
 constructor. More...
 
 ~GPUAuxStream ()
 destructor More...
 
void PreAuxStreamUseSync ()
 Makes future aux stream work wait on the completion of existing primary stream work. More...
 
void PostAuxStreamUseSync ()
 Makes future primary stream work wait on the completion of existing aux stream work. More...
 
mshadow::Stream< gpu > * GetStream ()
 Getter for created auxiliary stream. More...
 

Static Public Member Functions

static void StreamSync (mshadow::Stream< gpu > *s1, mshadow::Stream< gpu > *s2, cudaEvent_t event)
 Make future work enqueued to s2 wait on completion of current work enqueued to s1. More...
 

Detailed Description

Holds an auxiliary mshadow gpu stream that can be synced with a primary stream.

Constructor & Destructor Documentation

mxnet::GPUAuxStream::GPUAuxStream ( mshadow::Stream< gpu > *  primary_stream)
inlineexplicit

constructor.

Parameters
primary_streamgpu stream that is synced with the created auxiliary stream.
mxnet::GPUAuxStream::~GPUAuxStream ( )
inline

destructor

Member Function Documentation

mshadow::Stream<gpu>* mxnet::GPUAuxStream::GetStream ( )
inline

Getter for created auxiliary stream.

void mxnet::GPUAuxStream::PostAuxStreamUseSync ( )
inline

Makes future primary stream work wait on the completion of existing aux stream work.

void mxnet::GPUAuxStream::PreAuxStreamUseSync ( )
inline

Makes future aux stream work wait on the completion of existing primary stream work.

static void mxnet::GPUAuxStream::StreamSync ( mshadow::Stream< gpu > *  s1,
mshadow::Stream< gpu > *  s2,
cudaEvent_t  event 
)
inlinestatic

Make future work enqueued to s2 wait on completion of current work enqueued to s1.

Parameters
s1stream with work that must be completed before future s2 work can begin.
s2stream whose future work is made to wait on the completion of existing s1 work.
eventused to pass s1 state to s2.

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