BlockIndices Struct Reference

BlockIndices Struct Reference#

Composable Kernel: ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::BlockIndices Struct Reference
Public Attributes | List of all members
ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::BlockIndices Struct Reference

#include <fmha_fwd_splitkv_kernel.hpp>

Public Attributes

ck_tile::index_t batch_idx
ck_tile::index_t qo_head_idx
ck_tile::index_t kv_head_idx

Member Data Documentation

◆ batch_idx

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::BlockIndices::batch_idx

◆ kv_head_idx

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::BlockIndices::kv_head_idx

◆ qo_head_idx

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::BlockIndices::qo_head_idx
The documentation for this struct was generated from the following file: