mxnet
Public Member Functions | Public Attributes | List of all members
mxnet::RunContext Struct Reference

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

#include <base.h>

Collaboration diagram for mxnet::RunContext:
Collaboration graph

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 Contextget_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 nullptr or Stream<gpu>* in GPU mode More...
 
void * aux_stream
 the auxiliary stream of the device, can be nullptr 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.

Returns
the aux stream auto-syncing object
template<typename xpu >
mshadow::Stream<xpu>* mxnet::RunContext::get_stream ( ) const
inline

get mshadow stream from Context

Returns
the mshadow stream
Template Parameters
xputhe device type of the stream

Member Data Documentation

void* mxnet::RunContext::aux_stream

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

Context mxnet::RunContext::ctx

base Context

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 nullptr or Stream<gpu>* in GPU mode


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