ModuleBase::gemm_op< T, Device > Struct Template Reference

#include <math_kernel_op.h>

Collaboration diagram for ModuleBase::gemm_op< T, Device >:

Public Member Functions
void	operator() (const char &transa, const char &transb, const int &m, const int &n, const int &k, const T *alpha, const T *a, const int &lda, const T *b, const int &ldb, const T *beta, T *c, const int &ldc)
	C = alpha * op(A) * op(B) + beta * C.

Member Function Documentation

operator()()

template<typename T , typename Device >

void ModuleBase::gemm_op< T, Device >::operator()	(	const char &	transa,
		const char &	transb,
		const int &	m,
		const int &	n,
		const int &	k,
		const T *	alpha,
		const T *	a,
		const int &	lda,
		const T *	b,
		const int &	ldb,
		const T *	beta,
		T *	c,
		const int &	ldc
	)

C = alpha * op(A) * op(B) + beta * C.

Input Parameters

Parameters

transa	: whether to transpose matrix A
transb	: whether to transpose matrix B
m	: first dimension of matrix mulplication
n	: second dimension of matrix mulplication
k	: third dimension of matrix mulplication
alpha	: input constant alpha
a	: input matrix A
lda	: leading dimention of A
b	: input matrix B
ldb	: leading dimention of A
beta	: input constant beta
c	: input matrix C
ldc	: leading dimention of C

Output Parameters

Parameters

c	: output matrix C

---

The documentation for this struct was generated from the following file:

• /home/runner/work/abacus-develop/abacus-develop/source/source_base/kernels/math_kernel_op.h