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
namespace	ck_tile
namespace	ck_tile::impl

Enumerations
enum struct	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
CK_TILE_HOST_DEVICE constexpr 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>
CK_TILE_HOST_DEVICE constexpr 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)