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>
constexpr CK_TILE_HOST_DEVICE auto GetTileRangeAlongX (index_t i_y, number< YTile >, number< XTile >) const
 
template<index_t YTile, index_t XTile>
constexpr CK_TILE_HOST_DEVICE auto GetTileRangeAlongY (index_t i_x, number< YTile >, number< XTile >) const
 
constexpr CK_TILE_HOST_DEVICE auto IsOutOfBound (index_t i_y, index_t i_x) const
 
template<index_t TileHeight, index_t TileWidth>
constexpr CK_TILE_HOST_DEVICE 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>
constexpr CK_TILE_HOST_DEVICE 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>
constexpr CK_TILE_HOST_DEVICE 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>
constexpr CK_TILE_HOST_DEVICE 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>
constexpr CK_TILE_HOST_DEVICE 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>
constexpr bool ck_tile::GenericAttentionMask< IsMasking_, IsLocal_ >::IsLocal = IsLocal_
staticconstexpr

◆ IsMasking

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

◆ name

template<bool IsMasking_ = true, bool IsLocal_ = false>
constexpr 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:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/fmha/block/block_masking.hpp