mxnet
|
Provides automatic coordination of an auxilary stream with a primary one. This object, upon construction, prepares an aux stream for use by syncing it with enqueued primary-stream work. Object destruction will sync again so future primary-stream work will wait on enqueued aux-stream work. If MXNET_GPU_WORKER_NSTREAMS == 1, then this defaults simply: the primary stream will equal the aux stream and the syncs will be executed as nops. See ./src/operator/cudnn/cudnn_convolution-inl.h for a usage example. More...
#include <base.h>
Public Member Functions | |
SyncedGPUAuxStream (GPUAuxStream *gpu_aux_stream) | |
constructor. More... | |
~SyncedGPUAuxStream () | |
destructor More... | |
SyncedGPUAuxStream (const SyncedGPUAuxStream &)=delete | |
copy constructor deleted to prevent unexpected synchronizations. More... | |
void | operator= (const SyncedGPUAuxStream &)=delete |
copy assignment operator deleted to prevent unexpected synchronizations. More... | |
SyncedGPUAuxStream (SyncedGPUAuxStream &&)=default | |
move constructor permitted as alternative to copying. More... | |
SyncedGPUAuxStream & | operator= (SyncedGPUAuxStream &&)=default |
move assignment operator permitted as alternative to copy assignment. More... | |
mshadow::Stream< gpu > * | GetStream () const |
Getter for underlying mshadow::Stream<gpu>. More... | |
Provides automatic coordination of an auxilary stream with a primary one. This object, upon construction, prepares an aux stream for use by syncing it with enqueued primary-stream work. Object destruction will sync again so future primary-stream work will wait on enqueued aux-stream work. If MXNET_GPU_WORKER_NSTREAMS == 1, then this defaults simply: the primary stream will equal the aux stream and the syncs will be executed as nops. See ./src/operator/cudnn/cudnn_convolution-inl.h for a usage example.
|
inlineexplicit |
constructor.
gpu_aux_stream | auxilary gpu stream that is managed by this RAII object. |
|
inline |
destructor
|
delete |
copy constructor deleted to prevent unexpected synchronizations.
|
default |
move constructor permitted as alternative to copying.
|
inline |
Getter for underlying mshadow::Stream<gpu>.
|
delete |
copy assignment operator deleted to prevent unexpected synchronizations.
|
default |
move assignment operator permitted as alternative to copy assignment.