Alibi< DataType, RowMajor, LogMaxSadOprndSize > Struct Template Reference

Alibi&lt; DataType, RowMajor, LogMaxSadOprndSize &gt; Struct Template Reference#

Composable Kernel: ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize > Struct Template Reference
ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize > Struct Template Reference

#include <block_position_encoding.hpp>

Public Member Functions

CK_TILE_HOST_DEVICE Alibi (DataType slope_, index_t y_total_, index_t x_total_, AlibiMode mode_=AlibiMode::VERTICAL)
 
CK_TILE_HOST uint32_t sad (uint32_t x, uint32_t y, uint32_t acc)
 
CK_TILE_DEVICE uint32_t sad (uint32_t x, uint32_t y, uint32_t acc)
 
CK_TILE_HOST_DEVICE void update (DataType &pixel, index_t row_idx, index_t col_idx)
 

Public Attributes

DataType slope
 
index_t shift_left_up
 
index_t shift_right_down
 
AlibiMode mode
 

Constructor & Destructor Documentation

◆ Alibi()

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
CK_TILE_HOST_DEVICE ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::Alibi ( DataType  slope_,
index_t  y_total_,
index_t  x_total_,
AlibiMode  mode_ = AlibiMode::VERTICAL 
)
inline

Member Function Documentation

◆ sad() [1/2]

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
CK_TILE_HOST uint32_t ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::sad ( uint32_t  x,
uint32_t  y,
uint32_t  acc 
)
inline

◆ sad() [2/2]

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
CK_TILE_DEVICE uint32_t ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::sad ( uint32_t  x,
uint32_t  y,
uint32_t  acc 
)
inline

◆ update()

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
CK_TILE_HOST_DEVICE void ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::update ( DataType &  pixel,
index_t  row_idx,
index_t  col_idx 
)
inline

Member Data Documentation

◆ mode

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
AlibiMode ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::mode

◆ shift_left_up

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
index_t ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::shift_left_up

◆ shift_right_down

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
index_t ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::shift_right_down

◆ slope

template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
DataType ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize >::slope

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_position_encoding.hpp