gemmt¶
Computes a matrix-matrix product with general matrices, but updates only the upper or lower triangular part of the result matrix.
Description¶
The gemmt
routines compute a scalar-matrix-matrix product and add the
result to the upper or lower part of a scalar-matrix product, with
general matrices. The operation is defined as:
where:
op(
X
) is one of op(X
) =X
, or op(X
) =X
T, or op(X
) =X
Halpha
andbeta
are scalarsA
,B
, andC
are matricesop(
A
) isn
xk
, op(B
) isk
xn
, andC
isn
xn
gemmt
supports the following precisions:
T |
---|
|
|
|
|
gemmt (Buffer Version)¶
Syntax¶
namespace oneapi::mkl::blas::column_major {
void gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
T alpha,
sycl::buffer<T,1> &a,
std::int64_t lda,
sycl::buffer<T,1> &b,
std::int64_t ldb,
T beta,
sycl::buffer<T,1> &c,
std::int64_t ldc)
}
namespace oneapi::mkl::blas::row_major {
void gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
T alpha,
sycl::buffer<T,1> &a,
std::int64_t lda,
sycl::buffer<T,1> &b,
std::int64_t ldb,
T beta,
sycl::buffer<T,1> &c,
std::int64_t ldc)
}
Input Parameters¶
- queue
The queue where the routine should be executed.
- upper_lower
Specifies whether matrix
C
is upper or lower triangular. See Data Types for more details.- transa
Specifies op(
A
), the transposition operation applied to matrixA
. See Data Types for more details.- transb
Specifies op(
B
), the transposition operation applied to matrixB
. See Data Types for more details.- n
Number of rows of matrix op(
A
) and matrixC
. Must be at least zero.- k
Number of columns of matrix op(
A
) and rows of matrix op(B
). Must be at least zero.- alpha
Scaling factor for matrix-matrix product.
- a
Buffer holding input matrix
A
. See Matrix Storage for more details.transa
=transpose::nontrans
transa
=transpose::trans
ortrans
=transpose::conjtrans
Column major
A
isn
xk
matrix. Size of arraya
must be at leastlda
*k
A
isk
xn
matrix. Size of arraya
must be at leastlda
*n
Row major
A
isn
xk
matrix. Size of arraya
must be at leastlda
*n
A
isk
xn
matrix. Size of arraya
must be at leastlda
*k
- lda
Leading dimension of matrix
A
. Must be positive.transa
=transpose::nontrans
transa
=transpose::trans
ortrans
=transpose::conjtrans
Column major
Must be at least
n
Must be at least
k
Row major
Must be at least
k
Must be at least
n
- b
Buffer holding input matrix
B
. See Matrix Storage for more details.transb
=transpose::nontrans
transb
=transpose::trans
ortrans
=transpose::conjtrans
Column major
B
isk
xn
matrix. Size of arrayb
must be at leastldb
*n
B
isn
xk
matrix. Size of arrayb
must be at leastldb
*k
Row major
B
isk
xn
matrix. Size of arrayb
must be at leastldb
*k
B
isn
xk
matrix. Size of arrayb
must be at leastldb
*n
- ldb
Leading dimension of matrix
B
. Must be positive.transb
=transpose::nontrans
transb
=transpose::trans
ortrans
=transpose::conjtrans
Column major
Must be at least
k
Must be at least
n
Row major
Must be at least
n
Must be at least
k
- beta
Scaling factor for matrix
C
.- c
Buffer holding input/output matrix
C
. See Matrix Storage for more details.Column major
C
ism
xn
matrix. Size of arrayc
must be at leastldc
*n
Row major
C
ism
xn
matrix. Size of arrayc
must be at leastldc
*m
- ldc
Leading dimension of matrix
C
. Must be positive.Column major
Must be at least
m
Row major
Must be at least
n
Output Parameters¶
- c
Output buffer overwritten by upper or lower triangular part of
alpha
* op(A
)*op(B
) +beta
*C
.
Note
If beta
= 0, matrix C
does not need to be initialized before calling gemmt
.
gemmt (USM Version)¶
Syntax¶
namespace oneapi::mkl::blas::column_major {
sycl::event gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
T alpha,
const T* a,
std::int64_t lda,
const T* b,
std::int64_t ldb,
T beta,
T* c,
std::int64_t ldc,
const std::vector<sycl::event> &dependencies = {})
}
namespace oneapi::mkl::blas::row_major {
sycl::event gemmt(sycl::queue &queue,
oneapi::mkl::uplo upper_lower,
oneapi::mkl::transpose transa,
oneapi::mkl::transpose transb,
std::int64_t n,
std::int64_t k,
T alpha,
const T* a,
std::int64_t lda,
const T* b,
std::int64_t ldb,
T beta,
T* c,
std::int64_t ldc,
const std::vector<sycl::event> &dependencies = {})
}
Input Parameters¶
- queue
The queue where the routine should be executed.
- upper_lower
Specifies whether matrix
C
is upper or lower triangular. See Data Types for more details.- transa
Specifies op(
A
), the transposition operation applied to matrixA
. See Data Types for more details.- transb
Specifies op(
B
), the transposition operation applied to matrixB
. See Data Types for more details.- n
Number of rows of matrix op(
A
) and matrixC
. Must be at least zero.- k
Number of columns of matrix op(
A
) and rows of matrix op(B
). Must be at least zero.- alpha
Scaling factor for matrix-matrix product.
- a
Pointer to input matrix
A
. See Matrix Storage for more details.transa
=transpose::nontrans
transa
=transpose::trans
ortrans
=transpose::conjtrans
Column major
A
isn
xk
matrix. Size of arraya
must be at leastlda
*k
A
isk
xn
matrix. Size of arraya
must be at leastlda
*n
Row major
A
isn
xk
matrix. Size of arraya
must be at leastlda
*n
A
isk
xn
matrix. Size of arraya
must be at leastlda
*k
- lda
Leading dimension of matrix
A
. Must be positive.transa
=transpose::nontrans
transa
=transpose::trans
ortrans
=transpose::conjtrans
Column major
Must be at least
n
Must be at least
k
Row major
Must be at least
k
Must be at least
n
- b
Pointer to input matrix
B
. See Matrix Storage for more details.transb
=transpose::nontrans
transb
=transpose::trans
ortrans
=transpose::conjtrans
Column major
B
isk
xn
matrix. Size of arrayb
must be at leastldb
*n
B
isn
xk
matrix. Size of arrayb
must be at leastldb
*k
Row major
B
isk
xn
matrix. Size of arrayb
must be at leastldb
*k
B
isn
xk
matrix. Size of arrayb
must be at leastldb
*n
- ldb
Leading dimension of matrix
B
. Must be positive.transb
=transpose::nontrans
transb
=transpose::trans
ortrans
=transpose::conjtrans
Column major
Must be at least
k
Must be at least
n
Row major
Must be at least
n
Must be at least
k
- beta
Scaling factor for matrix
C
.- c
Pointer to input/output matrix
C
. See Matrix Storage for more details.Column major
C
ism
xn
matrix. Size of arrayc
must be at leastldc
*n
Row major
C
ism
xn
matrix. Size of arrayc
must be at leastldc
*m
- ldc
Leading dimension of matrix
C
. Must be positive.Column major
Must be at least
m
Row major
Must be at least
n
- dependencies
List of events to wait for before starting computation, if any. If omitted, defaults to no dependencies.
Output Parameters¶
- c
Pointer to output matrix
C
overwritten by upper or lower triangular part ofalpha
* op(A
)*op(B
) +beta
*C
.
Note
If beta
= 0, matrix C
does not need to be initialized before calling gemmt
.
Return Values¶
Output event to wait on to ensure computation is complete.