MaskKargs Struct Reference

MaskKargs Struct Reference#

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

#include <fmha_fwd_splitkv_kernel.hpp>

Public Attributes

ck_tile::index_t window_size_left
ck_tile::index_t window_size_right
ck_tile::GenericAttentionMaskEnum mask_type

Member Data Documentation

◆ mask_type

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::GenericAttentionMaskEnum ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::MaskKargs::mask_type

◆ window_size_left

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::MaskKargs::window_size_left

◆ window_size_right

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