include/ck_tile/ops/fmha/block/block_masking.hpp File Reference

• include
• ck_tile
• ops
• fmha
• block

Classes | Namespaces | Enumerations | Functions

block_masking.hpp File Reference

#include "ck_tile/core.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::impl::MaskName< false, false >
struct	ck_tile::impl::MaskName< false, true >
struct	ck_tile::impl::MaskName< true, false >
struct	ck_tile::impl::MaskName< true, true >
struct	ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >
struct	ck_tile::impl::SimplifiedMaskName< false >
struct	ck_tile::impl::SimplifiedMaskName< true >
struct	ck_tile::SimplifiedGenericAttentionMask< IsMasking_ >
struct	ck_tile::impl::SimplifiedRatioMaskName< false >
struct	ck_tile::impl::SimplifiedRatioMaskName< true >
struct	ck_tile::SimplifiedRatioAttentionMask< IsMasking_ >

Namespaces
	ck_tile
	ck_tile::impl

Enumerations
enum class	ck_tile::GenericAttentionMaskEnum {   ck_tile::NO_MASK = 0 ,   ck_tile::MASK_FROM_TOP_LEFT = 1 ,   ck_tile::MASK_FROM_BOTTOM_RIGHT = 2 ,   ck_tile::MASK_GENERIC }

Functions
constexpr CK_TILE_HOST_DEVICE auto	ck_tile::make_generic_attention_mask_coordinates_from_lr_window (index_t left_size, index_t right_size, index_t y_total, index_t x_total, bool is_top_left=true)
template<typename MaskType >
constexpr CK_TILE_HOST_DEVICE auto	ck_tile::make_generic_attention_mask_from_lr_window (index_t left_size, index_t right_size, index_t y_total, index_t x_total, bool is_top_left=true)