BatchedGemmHostArgs Struct Reference

BatchedGemmHostArgs Struct Reference#

Composable Kernel: ck_tile::BatchedGemmHostArgs Struct Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::BatchedGemmHostArgs Struct Reference

The Batched GEMM kernel host arguments. More...

#include <batched_gemm_kernel.hpp>

Inheritance diagram for ck_tile::BatchedGemmHostArgs:
ck_tile::UniversalGemmHostArgs< NumATensor, NumBTensor, NumDTensor >

Public Member Functions

CK_TILE_HOST BatchedGemmHostArgs (const void *a_ptr_, const void *b_ptr_, void *c_ptr_, ck_tile::index_t k_batch_, ck_tile::index_t M_, ck_tile::index_t N_, ck_tile::index_t K_, ck_tile::index_t stride_A_, ck_tile::index_t stride_B_, ck_tile::index_t stride_C_, ck_tile::index_t batch_stride_A_, ck_tile::index_t batch_stride_B_, ck_tile::index_t batch_stride_C_, ck_tile::index_t batch_count_)
Public Member Functions inherited from ck_tile::UniversalGemmHostArgs< NumATensor, NumBTensor, NumDTensor >
CK_TILE_HOST UniversalGemmHostArgs (const std::array< const void *, NumATensor > &as_ptr_, const std::array< const void *, NumBTensor > &bs_ptr_, const std::array< const void *, NumDTensor > &ds_ptr_, void *e_ptr_, index_t k_batch_, index_t M_, index_t N_, index_t K_, const std::array< index_t, NumATensor > &stride_As_, const std::array< index_t, NumBTensor > &stride_Bs_, const std::array< index_t, NumDTensor > &stride_Ds_, index_t stride_E_)

Public Attributes

ck_tile::index_t batch_stride_A
ck_tile::index_t batch_stride_B
ck_tile::index_t batch_stride_E
ck_tile::index_t batch_count
Public Attributes inherited from ck_tile::UniversalGemmHostArgs< NumATensor, NumBTensor, NumDTensor >
const std::array< const void *, NumATensor > as_ptr
const std::array< const void *, NumBTensor > bs_ptr
const std::array< const void *, NumDTensor > ds_ptr
union { 
   void *   e_ptr 
   void *   c_ptr 
}; 
index_t M
index_t N
index_t K
const std::array< index_t, NumATensor > stride_As
const std::array< index_t, NumBTensor > stride_Bs
const std::array< index_t, NumDTensor > stride_Ds
union { 
   index_t   stride_E 
   index_t   stride_C 
}; 
index_t k_batch

Detailed Description

The Batched GEMM kernel host arguments.

Overview
This structure is passed to BatchedGemmKernel when creating kernel arguments object. It contain all necessary information required to build proper kernel argument and launch kernel on GPU. This structure defines the GEMM problem configuration by stating all required information like M,N,K sizes and respective strides.

Constructor & Destructor Documentation

◆ BatchedGemmHostArgs()

CK_TILE_HOST ck_tile::BatchedGemmHostArgs::BatchedGemmHostArgs ( const void * a_ptr_,
const void * b_ptr_,
void * c_ptr_,
ck_tile::index_t k_batch_,
ck_tile::index_t M_,
ck_tile::index_t N_,
ck_tile::index_t K_,
ck_tile::index_t stride_A_,
ck_tile::index_t stride_B_,
ck_tile::index_t stride_C_,
ck_tile::index_t batch_stride_A_,
ck_tile::index_t batch_stride_B_,
ck_tile::index_t batch_stride_C_,
ck_tile::index_t batch_count_ )
inlineexplicit

Member Data Documentation

◆ batch_count

ck_tile::index_t ck_tile::BatchedGemmHostArgs::batch_count

◆ batch_stride_A

ck_tile::index_t ck_tile::BatchedGemmHostArgs::batch_stride_A

◆ batch_stride_B

ck_tile::index_t ck_tile::BatchedGemmHostArgs::batch_stride_B

◆ batch_stride_E

ck_tile::index_t ck_tile::BatchedGemmHostArgs::batch_stride_E
The documentation for this struct was generated from the following file: