GroupModeKargs Struct Reference#
ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::GroupModeKargs Struct Reference
#include <fmha_fwd_splitkv_combine_kernel.hpp>
Inheritance diagram for ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::GroupModeKargs:
Public Attributes | |
| const int32_t * | seqstart_q_ptr |
| Public Attributes inherited from ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonKargs | |
| 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
◆ seqstart_q_ptr
template<typename FmhaPipeline_, typename EpiloguePipeline_>
| const int32_t* ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::GroupModeKargs::seqstart_q_ptr |
The documentation for this struct was generated from the following file: