SimplifiedRatioAttentionMask< IsMasking_ > Struct Template Reference

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

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

#include <block_masking.hpp>

Public Member Functions

CK_TILE_HOST_DEVICE SimplifiedRatioAttentionMask (index_t y_total_, index_t x_total_)
CK_TILE_HOST_DEVICE SimplifiedRatioAttentionMask (index_t y_real_, index_t x_, index_t y_total_, index_t x_total_, mdiv y_ratio_mdiv_)
CK_TILE_HOST_DEVICE SimplifiedRatioAttentionMask (index_t y_, index_t x_, index_t y_total_, index_t x_total_, index_t y_real_, index_t y_ratio_, mdiv y_ratio_mdiv_)
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_y, index_t i_x, number< TileHeight >, number< TileWidth >) const

Static Public Attributes

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

Constructor & Destructor Documentation

◆ SimplifiedRatioAttentionMask() [1/3]

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

◆ SimplifiedRatioAttentionMask() [2/3]

template<bool IsMasking_ = true>
CK_TILE_HOST_DEVICE ck_tile::SimplifiedRatioAttentionMask< IsMasking_ >::SimplifiedRatioAttentionMask ( index_t y_real_,
index_t x_,
index_t y_total_,
index_t x_total_,
mdiv y_ratio_mdiv_ )
inline

◆ SimplifiedRatioAttentionMask() [3/3]

template<bool IsMasking_ = true>
CK_TILE_HOST_DEVICE ck_tile::SimplifiedRatioAttentionMask< IsMasking_ >::SimplifiedRatioAttentionMask ( index_t y_,
index_t x_,
index_t y_total_,
index_t x_total_,
index_t y_real_,
index_t y_ratio_,
mdiv y_ratio_mdiv_ )
inline

Member Function Documentation

◆ GetTileRangeAlongX()

template<bool IsMasking_ = true>
template<index_t YTile, index_t XTile>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::SimplifiedRatioAttentionMask< 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::SimplifiedRatioAttentionMask< 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::SimplifiedRatioAttentionMask< 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::SimplifiedRatioAttentionMask< IsMasking_ >::IsOutOfBound ( index_t i_y,
index_t i_x ) const
inlineconstexpr

Member Data Documentation

◆ IsMasking

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

◆ name

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