Member List#
ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > Member List
This is the complete list of members for ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 >, including all inherited members.
ComposedAttention()=default | ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > | |
LogitsMask(const Params ¶ms, [[maybe_unused]] uint32_t batch_idx, uint32_t qo_idx, uint32_t kv_idx, [[maybe_unused]] uint32_t qo_head_idx, [[maybe_unused]] uint32_t kv_head_idx) const | ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > | inline |
LogitsTransform(const Params ¶ms, T logits, [[maybe_unused]] uint32_t batch_idx, [[maybe_unused]] uint32_t qo_head_idx, [[maybe_unused]] uint32_t kv_head_idx) const | ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > | inline |
QueryTransform(const Params ¶ms, T q) const | ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > | inline |
use_exp2 | ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > | static |
use_logits_soft_cap | ck_tile::ComposedAttention< VARIANT_CODE, UseExp2 > | static |