mxnet
|
Holds an auxiliary mshadow gpu stream that can be synced with a primary stream. More...
#include <base.h>
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... | |
Holds an auxiliary mshadow gpu stream that can be synced with a primary stream.
|
inlineexplicit |
constructor.
primary_stream | gpu stream that is synced with the created auxiliary stream. |
|
inline |
destructor
|
inline |
Getter for created auxiliary stream.
|
inline |
Makes future primary stream work wait on the completion of existing aux stream work.
|
inline |
Makes future aux stream work wait on the completion of existing primary stream work.
|
inlinestatic |
Make future work enqueued to s2
wait on completion of current work enqueued to s1
.
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. |