execution time context. The information needed in runtime for actual execution. More...

#include <base.h>

Collaboration diagram for mxnet::RunContext:

Public Member Functions
template<typename xpu >
mshadow::Stream< xpu > *	get_stream () const
	get mshadow stream from Context More...

SyncedGPUAuxStream	get_gpu_aux_stream () const
	get an RAII object that transparently handles the syncing of the auxiliary stream. More...

const Context &	get_ctx () const
	get the base Context from RunContext More...

Public Attributes
Context	ctx
	base Context More...

void *	stream
	the stream of the device, can be NULL or Stream<gpu>* in GPU mode More...

void *	aux_stream
	the auxiliary stream of the device, can be NULL or Stream<gpu>* in GPU mode More...

bool	is_bulk
	indicator of whether this execution is run in bulk mode More...

Detailed Description

execution time context. The information needed in runtime for actual execution.

Member Function Documentation

const Context& mxnet::RunContext::get_ctx ( ) const

inline

get the base Context from RunContext

SyncedGPUAuxStream mxnet::RunContext::get_gpu_aux_stream ( ) const

inline

get an RAII object that transparently handles the syncing of the auxiliary stream.

template<typename xpu >

mshadow::Stream<xpu>* mxnet::RunContext::get_stream ( ) const

inline

get mshadow stream from Context

Template Parameters

xpu	the device type of the stream

void* mxnet::RunContext::aux_stream

the auxiliary stream of the device, can be NULL or Stream<gpu>* in GPU mode

Context mxnet::RunContext::ctx

bool mxnet::RunContext::is_bulk

indicator of whether this execution is run in bulk mode

void* mxnet::RunContext::stream

the stream of the device, can be NULL or Stream<gpu>* in GPU mode

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