mxnet
|
execution time context. The information needed in runtime for actual execution. More...
#include <base.h>
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... | |
execution time context. The information needed in runtime for actual execution.
|
inline |
get the base Context from RunContext
|
inline |
get an RAII object that transparently handles the syncing of the auxiliary stream.
|
inline |
get mshadow stream from Context
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
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