dnnl::threadpool_interop Namespace Reference

Threadpool interoperability namespace. More...

Classes
struct	threadpool_iface
	Abstract threadpool interface. More...

Functions
dnnl::stream	make_stream (const dnnl::engine &aengine, threadpool_iface *threadpool)
	Constructs an execution stream for the specified engine and threadpool. More...
threadpool_iface *	get_threadpool (const dnnl::stream &astream)
	Returns the pointer to a threadpool that is used by an execution stream. More...
status	sgemm (char transa, char transb, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const float *A, dnnl_dim_t lda, const float *B, dnnl_dim_t ldb, float beta, float *C, dnnl_dim_t ldc, threadpool_iface *tp)
status	gemm_u8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const uint8_t *A, dnnl_dim_t lda, uint8_t ao, const int8_t *B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t *C, dnnl_dim_t ldc, const int32_t *co, threadpool_iface *tp)
status	gemm_s8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const int8_t *A, dnnl_dim_t lda, int8_t ao, const int8_t *B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t *C, dnnl_dim_t ldc, const int32_t *co, threadpool_iface *tp)

Detailed Description

Threadpool interoperability namespace.

Function Documentation

make_stream()

dnnl::stream dnnl::threadpool_interop::make_stream ( const dnnl::engine &  aengine, threadpool_iface *  threadpool  )

inline

Constructs an execution stream for the specified engine and threadpool.

See also

Using oneDNN with Threadpool-based Threading

Parameters

aengine	Engine to create the stream on.
threadpool	Pointer to an instance of a C++ class that implements dnnl::threapdool_iface interface.

Returns

An execution stream.

get_threadpool()

threadpool_iface* dnnl::threadpool_interop::get_threadpool ( const dnnl::stream &  astream )

inline

Returns the pointer to a threadpool that is used by an execution stream.

See also

Using oneDNN with Threadpool-based Threading

Parameters

astream

An execution stream.

Returns

Output pointer to an instance of a C++ class that implements dnnl::threapdool_iface interface or NULL if the stream was created without threadpool.

sgemm()

status dnnl::threadpool_interop::sgemm ( char  transa, char  transb, dnnl_dim_t  M, dnnl_dim_t  N, dnnl_dim_t  K, float  alpha, const float *  A, dnnl_dim_t  lda, const float *  B, dnnl_dim_t  ldb, float  beta, float *  C, dnnl_dim_t  ldc, threadpool_iface *  tp  )

inline

gemm_u8s8s32()

status dnnl::threadpool_interop::gemm_u8s8s32 ( char  transa, char  transb, char  offsetc, dnnl_dim_t  M, dnnl_dim_t  N, dnnl_dim_t  K, float  alpha, const uint8_t *  A, dnnl_dim_t  lda, uint8_t  ao, const int8_t *  B, dnnl_dim_t  ldb, int8_t  bo, float  beta, int32_t *  C, dnnl_dim_t  ldc, const int32_t *  co, threadpool_iface *  tp  )

inline

gemm_s8s8s32()

status dnnl::threadpool_interop::gemm_s8s8s32 ( char  transa, char  transb, char  offsetc, dnnl_dim_t  M, dnnl_dim_t  N, dnnl_dim_t  K, float  alpha, const int8_t *  A, dnnl_dim_t  lda, int8_t  ao, const int8_t *  B, dnnl_dim_t  ldb, int8_t  bo, float  beta, int32_t *  C, dnnl_dim_t  ldc, const int32_t *  co, threadpool_iface *  tp  )

inline