SimplifiedGenericAttentionMask< IsMasking_ > Struct Template Reference

SimplifiedGenericAttentionMask&lt; IsMasking_ &gt; Struct Template Reference#

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

#include <block_masking.hpp>

Public Member Functions

CK_TILE_HOST_DEVICE SimplifiedGenericAttentionMask (index_t y_total_, index_t x_total_)
CK_TILE_HOST_DEVICE SimplifiedGenericAttentionMask (index_t y_, index_t x_, index_t y_total_, index_t x_total_)
template<typename MaskCoordinates>
CK_TILE_HOST_DEVICE SimplifiedGenericAttentionMask (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 TileHeight, index_t TileWidth>
CK_TILE_HOST_DEVICE constexpr auto GetTileRangeAlongX (index_t i_y, number< TileHeight > height, number< TileWidth > width, index_t num_splits, index_t i_split) 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_y, index_t i_x, number< TileHeight >, number< TileWidth >) const

Static Public Attributes

static constexpr bool IsMasking = IsMasking_
static constexpr const char * name = impl::SimplifiedMaskName<IsMasking>::name

Constructor & Destructor Documentation

◆ SimplifiedGenericAttentionMask() [1/3]

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

◆ SimplifiedGenericAttentionMask() [2/3]

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

◆ SimplifiedGenericAttentionMask() [3/3]

template<bool IsMasking_ = true>
template<typename MaskCoordinates>
CK_TILE_HOST_DEVICE ck_tile::SimplifiedGenericAttentionMask< IsMasking_ >::SimplifiedGenericAttentionMask ( const MaskCoordinates & mask_coord)
inline

Member Function Documentation

◆ GetTileRangeAlongX() [1/2]

template<bool IsMasking_ = true>
template<index_t TileHeight, index_t TileWidth>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::SimplifiedGenericAttentionMask< IsMasking_ >::GetTileRangeAlongX ( index_t i_y,
number< TileHeight > height,
number< TileWidth > width,
index_t num_splits,
index_t i_split ) const
inlineconstexpr

◆ GetTileRangeAlongX() [2/2]

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

◆ GetTileRangeAlongY()

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

◆ IsEdgeTile()

template<bool IsMasking_ = true>
template<index_t TileHeight, index_t TileWidth>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::SimplifiedGenericAttentionMask< IsMasking_ >::IsEdgeTile ( index_t i_y,
index_t i_x,
number< TileHeight > ,
number< TileWidth >  ) const
inlineconstexpr

◆ IsOutOfBound()

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

Member Data Documentation

◆ IsMasking

template<bool IsMasking_ = true>
bool ck_tile::SimplifiedGenericAttentionMask< IsMasking_ >::IsMasking = IsMasking_
staticconstexpr

◆ name

template<bool IsMasking_ = true>
const char* ck_tile::SimplifiedGenericAttentionMask< IsMasking_ >::name = impl::SimplifiedMaskName<IsMasking>::name
staticconstexpr
The documentation for this struct was generated from the following file: