rocblas_sgemm_batched Interface Reference#
hipfort_rocblas::rocblas_sgemm_batched Interface Reference
BLAS Level 3 API. More...
Public Member Functions | |
| integer(kind(rocblas_status_success)) function | rocblas_sgemm_batched_ (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(kind(rocblas_status_success)) function hipfort_rocblas::rocblas_sgemm_batched::rocblas_sgemm_batched_ | ( | type(c_ptr), value | handle, |
| integer(kind(rocblas_operation_none)), value | transa, | ||
| integer(kind(rocblas_operation_none)), value | transb, | ||
| integer(c_int), value | m, | ||
| integer(c_int), value | n, | ||
| integer(c_int), value | k, | ||
| real(c_float) | alpha, | ||
| type(c_ptr), value | a, | ||
| integer(c_int), value | lda, | ||
| type(c_ptr), value | b, | ||
| integer(c_int), value | ldb, | ||
| real(c_float) | beta, | ||
| type(c_ptr), value | c, | ||
| integer(c_int), value | ldc, | ||
| integer(c_int), value | batch_count | ||
| ) |
The documentation for this interface was generated from the following file: