SplitKBatchOffset Struct Reference

SplitKBatchOffset Struct Reference#

Composable Kernel: ck_tile::UniversalGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset Struct Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::UniversalGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset Struct Reference

#include <universal_gemm_kernel.hpp>

Public Member Functions

__device__ SplitKBatchOffset (const KernelArgs &kargs, const std::size_t k_id=blockIdx.z)

Public Attributes

std::array< index_t, NumATensor > as_k_split_offset
std::array< index_t, NumBTensor > bs_k_split_offset
index_t splitted_k

Constructor & Destructor Documentation

◆ SplitKBatchOffset()

template<typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
__device__ ck_tile::UniversalGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::SplitKBatchOffset ( const KernelArgs & kargs,
const std::size_t k_id = blockIdx.z )
inline

Member Data Documentation

◆ as_k_split_offset

template<typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
std::array<index_t, NumATensor> ck_tile::UniversalGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::as_k_split_offset

◆ bs_k_split_offset

template<typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
std::array<index_t, NumBTensor> ck_tile::UniversalGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::bs_k_split_offset

◆ splitted_k

template<typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
index_t ck_tile::UniversalGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::splitted_k
The documentation for this struct was generated from the following file: