/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/block/block_position_encoding.hpp File Reference#
block_position_encoding.hpp File Reference
#include "ck_tile/core.hpp"
#include "ck_tile/ops/fmha/block/block_masking.hpp"
#include <cmath>
#include <vector>
Go to the source code of this file.
Classes | |
struct | ck_tile::Alibi< DataType, RowMajor, LogMaxSadOprndSize > |
struct | ck_tile::EmptyPositionEncoding< DataType > |
Namespaces | |
ck_tile | |
Enumerations | |
enum class | ck_tile::PositionEncodingEnum { ck_tile::NO = 0 , ck_tile::ALIBI = 1 } |
enum class | ck_tile::AlibiMode { ck_tile::VERTICAL = 0 , ck_tile::FROM_TOP_LEFT = 1 , ck_tile::FROM_BOTTOM_RIGHT = 2 } |
Functions | |
template<typename DataType , bool RowMajor = true, unsigned LogMaxSadOprndSize = 16> | |
CK_TILE_HOST_DEVICE auto | ck_tile::make_alibi_from_lr_mask (DataType slope, index_t window_left_size, index_t window_right_size, index_t y_total, index_t x_total, GenericAttentionMaskEnum mask_enum) |
template<typename DataType > | |
CK_TILE_HOST std::vector< DataType > | ck_tile::get_alibi_slopes (ck_tile::index_t nheads) |