FmhaFwdMaskKargs Struct Reference

FmhaFwdMaskKargs Struct Reference#

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

#include <fmha_fwd_pagedkv_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::FmhaFwdPagedKVKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs::mask_type

◆ window_size_left

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdPagedKVKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs::window_size_left

◆ window_size_right

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