CommonKargs Struct Reference

• ck_tile
• FmhaFwdSplitKVCombineKernel
• CommonKargs

Public Attributes | List of all members

ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs Struct Reference

#include <fmha_fwd_splitkv_combine_kernel.hpp>

Inheritance diagram for ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs:

Public Attributes
const void *	lse_acc_ptr
const void *	o_acc_ptr
void *	o_ptr
ck_tile::index_t	batch
ck_tile::index_t	seqlen_q
ck_tile::index_t	hdim_v
ck_tile::index_t	num_splits
ck_tile::index_t	row_stride_o_acc
ck_tile::index_t	row_stride_o
ck_tile::index_t	nhead_stride_lse_acc
ck_tile::index_t	nhead_stride_o_acc
ck_tile::index_t	nhead_stride_o
ck_tile::index_t	split_stride_lse_acc
ck_tile::index_t	split_stride_o_acc

Member Data Documentation

◆ batch

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::batch

◆ hdim_v

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::hdim_v

◆ lse_acc_ptr

template<typename FmhaPipeline_, typename EpiloguePipeline_>

const void* ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::lse_acc_ptr

◆ nhead_stride_lse_acc

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::nhead_stride_lse_acc

◆ nhead_stride_o

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::nhead_stride_o

◆ nhead_stride_o_acc

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::nhead_stride_o_acc

◆ num_splits

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::num_splits

◆ o_acc_ptr

template<typename FmhaPipeline_, typename EpiloguePipeline_>

const void* ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::o_acc_ptr

◆ o_ptr

template<typename FmhaPipeline_, typename EpiloguePipeline_>

void* ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::o_ptr

◆ row_stride_o

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::row_stride_o

◆ row_stride_o_acc

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::row_stride_o_acc

◆ seqlen_q

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::seqlen_q

◆ split_stride_lse_acc

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::split_stride_lse_acc

◆ split_stride_o_acc

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs::split_stride_o_acc

---

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

• fmha_fwd_splitkv_combine_kernel.hpp