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_stream_kind_t stream_kind)
 
mkldnn_status_t MKLDNN_API mkldnn_stream_submit (mkldnn_stream_t stream, size_t n, mkldnn_primitive_t primitives[], mkldnn_primitive_t *error_primitive)
 
mkldnn_status_t MKLDNN_API mkldnn_stream_wait (mkldnn_stream_t stream, int block, mkldnn_primitive_t *error_primitive)
 
mkldnn_status_t MKLDNN_API mkldnn_stream_rerun (mkldnn_stream_t stream, mkldnn_primitive_t *error_primitive)
 
mkldnn_status_t MKLDNN_API mkldnn_stream_destroy (mkldnn_stream_t stream)
 

Detailed Description

Function Documentation

mkldnn_status_t MKLDNN_API mkldnn_stream_create ( mkldnn_stream_t stream,
mkldnn_stream_kind_t  stream_kind 
)

Creates an execution stream of stream_kind.

mkldnn_status_t MKLDNN_API mkldnn_stream_destroy ( mkldnn_stream_t  stream)

Destroys an execution stream.

mkldnn_status_t MKLDNN_API mkldnn_stream_rerun ( mkldnn_stream_t  stream,
mkldnn_primitive_t error_primitive 
)

Reruns all the primitives within the stream. In case of an error, returns the offending error_primitive if it is not NULL.

mkldnn_status_t MKLDNN_API mkldnn_stream_submit ( mkldnn_stream_t  stream,
size_t  n,
mkldnn_primitive_t  primitives[],
mkldnn_primitive_t error_primitive 
)

Submits primitives to an execution stream. The number of primitives is n. All or none of the primitives can be lazy. In case of an error, returns the offending error_primitive if it is not NULL.

mkldnn_status_t MKLDNN_API mkldnn_stream_wait ( mkldnn_stream_t  stream,
int  block,
mkldnn_primitive_t error_primitive 
)

Waits for all primitives in the execution stream to finish. Returns immediately if block is zero. In case of an error, returns the offending error_primitive if it is not NULL.