FmhaFwdMaskKargs Struct Reference

FmhaFwdMaskKargs Struct Reference#

Composable Kernel: ck_tile::FmhaFwdV3Kernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs Struct Reference
ck_tile::FmhaFwdV3Kernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs Struct Reference

#include <fmha_fwd_v3_kernel.hpp>

Public Attributes

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

Member Data Documentation

◆ mask_type

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::GenericAttentionMaskEnum ck_tile::FmhaFwdV3Kernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs::mask_type

◆ remap_opt

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdV3Kernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs::remap_opt

◆ window_size_left

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

◆ window_size_right

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ck_tile::index_t ck_tile::FmhaFwdV3Kernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdMaskKargs::window_size_right

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