batchlas::sycl_gemm_small Namespace Reference

Functions

small_bucket()

int batchlas::sycl_gemm_small::small_bucket ( int max_dim )

inline

Definition at line 30 of file small_batched.hh.

launch_small_batched()

template<typename T, int NB>

Event batchlas::sycl_gemm_small::launch_small_batched	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		const MatrixView< T, MatrixFormat::Dense > &	B,
		const MatrixView< T, MatrixFormat::Dense > &	C,
		T	alpha,
		T	beta,
		Transpose	transA,
		Transpose	transB,
		int	m,
		int	n,
		int	k )

Definition at line 35 of file small_batched.hh.

launch_small_tiled()

template<typename T, int NB, bool PrefetchC>

Event batchlas::sycl_gemm_small::launch_small_tiled	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		const MatrixView< T, MatrixFormat::Dense > &	B,
		const MatrixView< T, MatrixFormat::Dense > &	C,
		T	alpha,
		T	beta,
		int	m,
		int	n,
		int	k )

Definition at line 138 of file small_batched.hh.

small_tiled()

template<typename T, int NB>

Event batchlas::sycl_gemm_small::small_tiled	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		const MatrixView< T, MatrixFormat::Dense > &	B,
		const MatrixView< T, MatrixFormat::Dense > &	C,
		T	alpha,
		T	beta,
		int	m,
		int	n,
		int	k )

Definition at line 239 of file small_batched.hh.

small_batched()

template<typename T>

Event batchlas::sycl_gemm_small::small_batched	(	Queue &	ctx,
		const MatrixView< T, MatrixFormat::Dense > &	A,
		const MatrixView< T, MatrixFormat::Dense > &	B,
		const MatrixView< T, MatrixFormat::Dense > &	C,
		T	alpha,
		T	beta,
		Transpose	transA,
		Transpose	transB,
		int	m,
		int	n,
		int	k )

Definition at line 251 of file small_batched.hh.

Variables

kSmallWg

int batchlas::sycl_gemm_small::kSmallWg = 128

inlineconstexpr

Definition at line 25 of file small_batched.hh.

kSmallMaxDim

int batchlas::sycl_gemm_small::kSmallMaxDim = 64

inlineconstexpr

Definition at line 28 of file small_batched.hh.

kSmallTiledMaxDim

int batchlas::sycl_gemm_small::kSmallTiledMaxDim = 56

inlineconstexpr

Definition at line 135 of file small_batched.hh.

Classes
class	GemmSmallBatchedKernel
class	GemmSmallTiledKernel