mxnet
Classes | Typedefs | Enumerations | Functions
Dnnl_api_stream
Collaboration diagram for Dnnl_api_stream:

Classes

struct  dnnl_stream
 

Typedefs

typedef struct dnnl_streamdnnl_stream_t
 An execution stream handle. More...
 
typedef const struct dnnl_streamconst_dnnl_stream_t
 A constant execution stream handle. More...
 

Enumerations

enum  dnnl_stream_flags_t { dnnl_stream_in_order = 0x1U, dnnl_stream_out_of_order = 0x2U, dnnl_stream_default_flags = dnnl_stream_in_order }
 Stream flags. More...
 

Functions

dnnl_status_t DNNL_API dnnl_stream_create (dnnl_stream_t *stream, dnnl_engine_t engine, unsigned flags)
 
dnnl_status_t DNNL_API dnnl_stream_get_engine (const_dnnl_stream_t stream, dnnl_engine_t *engine)
 
dnnl_status_t DNNL_API dnnl_stream_wait (dnnl_stream_t stream)
 
dnnl_status_t DNNL_API dnnl_stream_destroy (dnnl_stream_t stream)
 

Detailed Description

Typedef Documentation

◆ const_dnnl_stream_t

typedef const struct dnnl_stream* const_dnnl_stream_t

A constant execution stream handle.

◆ dnnl_stream_t

typedef struct dnnl_stream* dnnl_stream_t

An execution stream handle.

Enumeration Type Documentation

◆ dnnl_stream_flags_t

Stream flags.

Enumerator
dnnl_stream_in_order 
dnnl_stream_out_of_order 

Out-of-order execution.

dnnl_stream_default_flags 

Default stream configuration.

Function Documentation

◆ dnnl_stream_create()

dnnl_status_t DNNL_API dnnl_stream_create ( dnnl_stream_t stream,
dnnl_engine_t  engine,
unsigned  flags 
)

Creates an execution stream.

Parameters
streamOutput execution stream.
engineEngine to create the execution stream on.
flagsStream behavior flags (
See also
dnnl_stream_flags_t).
Returns
dnnl_success on success and a status describing the error otherwise.

◆ dnnl_stream_destroy()

dnnl_status_t DNNL_API dnnl_stream_destroy ( dnnl_stream_t  stream)

Destroys an execution stream.

Parameters
streamExecution stream to destroy.
Returns
dnnl_success on success and a status describing the error otherwise.

◆ dnnl_stream_get_engine()

dnnl_status_t DNNL_API dnnl_stream_get_engine ( const_dnnl_stream_t  stream,
dnnl_engine_t engine 
)

Returns the engine of a stream object.

Parameters
streamStream object.
engineOutput engine on which the stream is created.
Returns
dnnl_success on success and a status describing the error otherwise.

◆ dnnl_stream_wait()

dnnl_status_t DNNL_API dnnl_stream_wait ( dnnl_stream_t  stream)

Waits for all primitives in the execution stream to finish computations.

Parameters
streamExecution stream.
Returns
dnnl_success on success and a status describing the error otherwise.