mxnet
Functions
Execution stream operations
Collaboration diagram for Execution stream operations:

Functions

mkldnn_status_t MKLDNN_API mkldnn_stream_create (mkldnn_stream_t *stream, mkldnn_engine_t engine, unsigned flags)
 Creates an execution stream for engine and with flags. More...
 
mkldnn_status_t MKLDNN_API mkldnn_stream_create_ocl (mkldnn_stream_t *stream, mkldnn_engine_t engine, cl_command_queue queue)
 
mkldnn_status_t MKLDNN_API mkldnn_stream_get_ocl_command_queue (mkldnn_stream_t stream, cl_command_queue *queue)
 
mkldnn_status_t MKLDNN_API mkldnn_stream_wait (mkldnn_stream_t stream)
 Waits for all primitives in the execution stream to finish. More...
 
mkldnn_status_t MKLDNN_API mkldnn_stream_destroy (mkldnn_stream_t stream)
 Destroys an execution stream. More...
 

Detailed Description

Function Documentation

mkldnn_status_t MKLDNN_API mkldnn_stream_create ( mkldnn_stream_t stream,
mkldnn_engine_t  engine,
unsigned  flags 
)

Creates an execution stream for engine and with flags.

mkldnn_status_t MKLDNN_API mkldnn_stream_create_ocl ( mkldnn_stream_t stream,
mkldnn_engine_t  engine,
cl_command_queue  queue 
)

Creates an execution stream for a given engine associated with an OpenCL command queue.

mkldnn_status_t MKLDNN_API mkldnn_stream_destroy ( mkldnn_stream_t  stream)

Destroys an execution stream.

mkldnn_status_t MKLDNN_API mkldnn_stream_get_ocl_command_queue ( mkldnn_stream_t  stream,
cl_command_queue *  queue 
)

Returns the OpenCL command queue associated with an execution stream.

mkldnn_status_t MKLDNN_API mkldnn_stream_wait ( mkldnn_stream_t  stream)

Waits for all primitives in the execution stream to finish.