hipblassgemmgroupedbatched Interface Reference

hipblassgemmgroupedbatched Interface Reference#

hipFORT API Reference: hipfort_hipblas::hipblassgemmgroupedbatched Interface Reference
hipfort_hipblas::hipblassgemmgroupedbatched Interface Reference

BLAS Level 3 API. More...

Public Member Functions

integer(c_int) function hipblassgemmgroupedbatched_ (handle, transa_array, transb_array, m_array, n_array, k_array, alpha_array, aarray, lda_array, barray, ldb_array, beta_array, carray, ldc_array, group_count, group_size)
 

Detailed Description

BLAS Level 3 API.

The gemmGroupedBatched functions perform grouped batched matrix-matrix operations. There are group_count groups. Within group i, all problems share the same transA, transB, m, n, k, lda, ldb, ldc, alpha, and beta. Each group i contains group_size[i] problems. The total number of problems is problem_count = sum(group_size[i]).

For each problem j in group i:

C_j = alpha_i*op( A_j )*op( B_j ) + beta_i*C_j,

where op( X ) is one of

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

Aarray, Barray, and Carray are device arrays of device pointers with length problem_count.

  • Supported precisions in rocBLAS : s and d.
  • Supported precisions in cuBLAS : not supported.
Parameters
[in]handle- [hipblasHandle_t] handle to the hipBLAS library context queue.
[in]transA_array- [const hipblasOperation_t*] host array of length group_count specifying op( A ) for each group.
[in]transB_array- [const hipblasOperation_t*] host array of length group_count specifying op( B ) for each group.
[in]m_array- [const int*] host array of length group_count.
[in]n_array- [const int*] host array of length group_count.
[in]k_array- [const int*] host array of length group_count.
[in]alpha_array- device pointer or host pointer to an array of length group_count.
[in]Aarray- device array of device pointers storing each matrix A_j.
[in]lda_array- [const int*] host array of length group_count.
[in]Barray- device array of device pointers storing each matrix B_j.
[in]ldb_array- [const int*] host array of length group_count.
[in]beta_array- device pointer or host pointer to an array of length group_count.
[in,out]Carray- device array of device pointers storing each matrix C_j.
[in]ldc_array- [const int*] host array of length group_count.
[in]group_count- [int] number of groups.
[in]group_size- [const int*] host array of length group_count with the number of problems in each group.

Member Function/Subroutine Documentation

◆ hipblassgemmgroupedbatched_()

integer(c_int) function hipfort_hipblas::hipblassgemmgroupedbatched::hipblassgemmgroupedbatched_ ( type(c_ptr), value  handle,
type(c_ptr), value  transa_array,
type(c_ptr), value  transb_array,
type(c_ptr), value  m_array,
type(c_ptr), value  n_array,
type(c_ptr), value  k_array,
type(c_ptr), value  alpha_array,
type(c_ptr), value  aarray,
type(c_ptr), value  lda_array,
type(c_ptr), value  barray,
type(c_ptr), value  ldb_array,
type(c_ptr), value  beta_array,
type(c_ptr), value  carray,
type(c_ptr), value  ldc_array,
integer(c_int), value  group_count,
type(c_ptr), value  group_size 
)

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