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>
|
inline |
Member Function Documentation
◆ sad() [1/2]
template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
|
inline |
◆ sad() [2/2]
template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
|
inline |
◆ update()
template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16>
|
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:
- include/ck_tile/ops/fmha/block/block_position_encoding.hpp