rocblas_sgeam_batched Interface Reference#
hipfort_rocblas::rocblas_sgeam_batched Interface Reference
BLAS Level 3 API. More...
Public Member Functions | |
| integer(kind(rocblas_status_success)) function | rocblas_sgeam_batched_ (handle, transa, transb, m, n, alpha, a, lda, beta, b, ldb, c, ldc, batch_count) |
Detailed Description
BLAS Level 3 API.
The geam_batched functions perform one of the batched matrix-matrix operations:
C_i = alpha*op( A_i ) + beta*op( B_i ) for i = 0, 1, ... batch_count - 1,
where alpha and beta are scalars, op(A_i), op(B_i), and C_i are m by n matrices, and op( X ) is one of:
op( X ) = X or op( X ) = X**T
- 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 dimension m. [in] n - [rocblas_int] matrix dimension n. [in] alpha - device pointer or host pointer specifying the scalar alpha. [in] A - device array of device pointers storing each matrix A_i on the GPU. Each A_i is of dimension ( lda, k ), where k is m when transA == rocblas_operation_none and is n when transA == rocblas_operation_transpose. [in] lda - [rocblas_int] specifies the leading dimension of A. [in] beta - device pointer or host pointer specifying the scalar beta. [in] B - device array of device pointers storing each matrix B_i on the GPU. Each B_i is of dimension ( ldb, k ), where k is m when transB == rocblas_operation_none and is n when transB == rocblas_operation_transpose. [in] ldb - [rocblas_int] specifies the leading dimension of B. [in,out] C - device array of device pointers storing each matrix C_i on the GPU. Each C_i is of dimension ( ldc, n ). [in] ldc - [rocblas_int] specifies the leading dimension of C. [in] batch_count - [rocblas_int] number of instances i in the batch.
Member Function/Subroutine Documentation
◆ rocblas_sgeam_batched_()
| integer(kind(rocblas_status_success)) function hipfort_rocblas::rocblas_sgeam_batched::rocblas_sgeam_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, | ||
| real(c_float) | alpha, | ||
| type(c_ptr), value | a, | ||
| integer(c_int), value | lda, | ||
| real(c_float) | beta, | ||
| type(c_ptr), value | b, | ||
| integer(c_int), value | ldb, | ||
| 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: