rocblas_sgemm_batched Interface Reference

• hipfort_rocblas
• rocblas_sgemm_batched

Public Member Functions | List of all members

hipfort_rocblas::rocblas_sgemm_batched Interface Reference

BLAS Level 3 API. More...

Public Member Functions
integer(c_int) function	rocblas_sgemm_batched_ (handle, transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, batch_count)
integer(kind(rocblas_status_success)) function	rocblas_sgemm_batched_typed (handle, transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, batch_count)

Detailed Description

BLAS Level 3 API.

The gemm_batched functions perform one of the batched matrix-matrix operations:

C_i = alpha*op( A_i )*op( B_i ) + beta*C_i, for i = 1, ..., batch_count,

where op( X ) is one of

op( X ) = X      or
op( X ) = X**T   or
op( X ) = X**H,

alpha and beta are scalars, and A, B, and C are strided batched matrices, with op( A ) an m by k by batch_count matrix, op( B ) a k by n by batch_count matrix, and C an m by n by batch_count matrix.

Parameters

[in]	handle	- [rocblas_handle handle to the rocBLAS library context queue.
[in]	transA	- [rocblas_operation] specifies the form of op( A ).
[in]	transB	- [rocblas_operation] specifies the form of op( B ).
[in]	m	- [rocblas_int] matrix dimention m.
[in]	n	- [rocblas_int] matrix dimention n.
[in]	k	- [rocblas_int] matrix dimention k.
[in]	alpha	- device pointer or host pointer specifying the scalar alpha.
[in]	A	- device array of device pointers storing each matrix A_i.
[in]	lda	- [rocblas_int] specifies the leading dimension of each A_i.
[in]	B	- device array of device pointers storing each matrix B_i.
[in]	ldb	- [rocblas_int] specifies the leading dimension of each B_i.
[in]	beta	- device pointer or host pointer specifying the scalar beta.
[in,out]	C	- device array of device pointers storing each matrix C_i.
[in]	ldc	- [rocblas_int] specifies the leading dimension of each C_i.
[in]	batch_count	[rocblas_int] number of gemm operations in the batch.

Member Function/Subroutine Documentation

◆ rocblas_sgemm_batched_()

integer(c_int) function hipfort_rocblas::rocblas_sgemm_batched::rocblas_sgemm_batched_	(	type(c_ptr), value	handle,
		integer(c_int), value	transa,
		integer(c_int), value	transb,
		integer(c_int), value	m,
		integer(c_int), value	n,
		integer(c_int), value	k,
		type(c_ptr), value	alpha,
		type(c_ptr), value	a,
		integer(c_int), value	lda,
		type(c_ptr), value	b,
		integer(c_int), value	ldb,
		type(c_ptr), value	beta,
		type(c_ptr), value	c,
		integer(c_int), value	ldc,
		integer(c_int), value	batch_count
	)

◆ rocblas_sgemm_batched_typed()

integer(kind(rocblas_status_success)) function hipfort_rocblas::rocblas_sgemm_batched::rocblas_sgemm_batched_typed	(	type(c_ptr), value	handle,
		integer(kind(rocblas_operation_none))	transa,
		integer(kind(rocblas_operation_none))	transb,
		integer(c_int)	m,
		integer(c_int)	n,
		integer(c_int)	k,
		real(c_float), target	alpha,
		type(c_ptr), value	a,
		integer(c_int)	lda,
		type(c_ptr), value	b,
		integer(c_int)	ldb,
		real(c_float), target	beta,
		type(c_ptr), value	c,
		integer(c_int)	ldc,
		integer(c_int)	batch_count
	)

---

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

• hipfort_rocblas.F90