GenericAttentionMask< IsMasking_, IsLocal_ > Struct Template Reference

GenericAttentionMask&lt; IsMasking_, IsLocal_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ > Struct Template Reference
Public Member Functions | Static Public Attributes | List of all members
ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ > Struct Template Reference

#include <block_masking.hpp>

Public Member Functions

CK_TILE_HOST_DEVICE GenericAttentionMask (index_t y_total_, index_t x_total_)
CK_TILE_HOST_DEVICE GenericAttentionMask (index_t y_, index_t x_, index_t y_total_, index_t x_total_)
template<typename MaskCoordinates>
CK_TILE_HOST_DEVICE GenericAttentionMask (const MaskCoordinates &mask_coord)
template<index_t YTile, index_t XTile>
CK_TILE_HOST_DEVICE constexpr auto GetTileRangeAlongX (index_t i_y, number< YTile >, number< XTile >) const
template<index_t YTile, index_t XTile>
CK_TILE_HOST_DEVICE constexpr auto GetTileRangeAlongY (index_t i_x, number< YTile >, number< XTile >) const
CK_TILE_HOST_DEVICE constexpr auto IsOutOfBound (index_t i_y, index_t i_x) const
template<index_t TileHeight, index_t TileWidth>
CK_TILE_HOST_DEVICE constexpr auto IsEdgeTile (index_t i_tile_top, index_t i_tile_left, number< TileHeight >, number< TileWidth >) const

Static Public Attributes

static constexpr bool IsMasking = IsMasking_
static constexpr bool IsLocal = IsLocal_
static constexpr const char * name = impl::MaskName<IsMasking, IsLocal>::name

Constructor & Destructor Documentation

◆ GenericAttentionMask() [1/3]

template<bool IsMasking_ = true, bool IsLocal_ = false>
CK_TILE_HOST_DEVICE ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::GenericAttentionMask ( index_t y_total_,
index_t x_total_ )
inline

◆ GenericAttentionMask() [2/3]

template<bool IsMasking_ = true, bool IsLocal_ = false>
CK_TILE_HOST_DEVICE ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::GenericAttentionMask ( index_t y_,
index_t x_,
index_t y_total_,
index_t x_total_ )
inline

◆ GenericAttentionMask() [3/3]

template<bool IsMasking_ = true, bool IsLocal_ = false>
template<typename MaskCoordinates>
CK_TILE_HOST_DEVICE ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::GenericAttentionMask ( const MaskCoordinates & mask_coord)
inline

Member Function Documentation

◆ GetTileRangeAlongX()

template<bool IsMasking_ = true, bool IsLocal_ = false>
template<index_t YTile, index_t XTile>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::GetTileRangeAlongX ( index_t i_y,
number< YTile > ,
number< XTile >  ) const
inlineconstexpr

◆ GetTileRangeAlongY()

template<bool IsMasking_ = true, bool IsLocal_ = false>
template<index_t YTile, index_t XTile>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::GetTileRangeAlongY ( index_t i_x,
number< YTile > ,
number< XTile >  ) const
inlineconstexpr

◆ IsEdgeTile()

template<bool IsMasking_ = true, bool IsLocal_ = false>
template<index_t TileHeight, index_t TileWidth>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::IsEdgeTile ( index_t i_tile_top,
index_t i_tile_left,
number< TileHeight > ,
number< TileWidth >  ) const
inlineconstexpr

◆ IsOutOfBound()

template<bool IsMasking_ = true, bool IsLocal_ = false>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::IsOutOfBound ( index_t i_y,
index_t i_x ) const
inlineconstexpr

Member Data Documentation

◆ IsLocal

template<bool IsMasking_ = true, bool IsLocal_ = false>
bool ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::IsLocal = IsLocal_
staticconstexpr

◆ IsMasking

template<bool IsMasking_ = true, bool IsLocal_ = false>
bool ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::IsMasking = IsMasking_
staticconstexpr

◆ name

template<bool IsMasking_ = true, bool IsLocal_ = false>
const char* ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::name = impl::MaskName<IsMasking, IsLocal>::name
staticconstexpr
The documentation for this struct was generated from the following file: