FmhaBwdMaskKargs Struct Reference

• ck_tile
• FmhaBwdDQDKDVKernel
• FmhaBwdMaskKargs

Public Attributes | List of all members

ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdMaskKargs Struct Reference

#include <fmha_bwd_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 KGradEpiloguePipeline_, typename VGradEpiloguePipeline_, typename QGradEpiloguePipeline_ = void>

ck_tile::GenericAttentionMaskEnum ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdMaskKargs::mask_type

◆ window_size_left

template<typename FmhaPipeline_, typename KGradEpiloguePipeline_, typename VGradEpiloguePipeline_, typename QGradEpiloguePipeline_ = void>

ck_tile::index_t ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdMaskKargs::window_size_left

◆ window_size_right

template<typename FmhaPipeline_, typename KGradEpiloguePipeline_, typename VGradEpiloguePipeline_, typename QGradEpiloguePipeline_ = void>

ck_tile::index_t ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdMaskKargs::window_size_right

---

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

• fmha_bwd_kernel.hpp