CommonLSEKargs Struct Reference

CommonLSEKargs Struct Reference#

Composable Kernel: ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonLSEKargs Struct Reference
ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonLSEKargs Struct Reference

#include <fmha_fwd_splitkv_combine_kernel.hpp>

Public Attributes

void * lse_ptr = nullptr
ck_tile::index_t nhead_stride_lse = 0
ck_tile::index_t batch_stride_lse = 0

Member Data Documentation

◆ batch_stride_lse

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonLSEKargs::batch_stride_lse = 0

◆ lse_ptr

template<typename FmhaPipeline_, typename EpiloguePipeline_>
void* ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonLSEKargs::lse_ptr = nullptr

◆ nhead_stride_lse

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdSplitKVCombineKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonLSEKargs::nhead_stride_lse = 0

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