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:

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

GPUAuxStream()

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

inlineexplicit

constructor.

Parameters

primary_stream

gpu stream that is synced with the created auxiliary stream.

~GPUAuxStream()

mxnet::GPUAuxStream::~GPUAuxStream ( )

inline

destructor

Member Function Documentation

GetStream()

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

inline

Getter for created auxiliary stream.

PostAuxStreamUseSync()

void mxnet::GPUAuxStream::PostAuxStreamUseSync ( )

inline

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

PreAuxStreamUseSync()

void mxnet::GPUAuxStream::PreAuxStreamUseSync ( )

inline

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

StreamSync()

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

s1	stream with work that must be completed before future s2 work can begin.
s2	stream whose future work is made to wait on the completion of existing s1 work.
event	used to pass s1 state to s2.

---

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

• /work/mxnet/include/mxnet/base.h