#include <naive_attention.hpp>
◆ batch_kv
int ck_tile::naive_attention_fwd_args::batch_kv |
◆ batch_q
int ck_tile::naive_attention_fwd_args::batch_q |
◆ batch_ratio_kv
int ck_tile::naive_attention_fwd_args::batch_ratio_kv |
◆ context_len_ptr
void* ck_tile::naive_attention_fwd_args::context_len_ptr |
◆ hdim
int ck_tile::naive_attention_fwd_args::hdim |
◆ hdim_v
int ck_tile::naive_attention_fwd_args::hdim_v |
◆ k_ptr
void* ck_tile::naive_attention_fwd_args::k_ptr |
◆ kscale_ptr
void* ck_tile::naive_attention_fwd_args::kscale_ptr |
◆ max_kv_tokens
int ck_tile::naive_attention_fwd_args::max_kv_tokens |
◆ max_pages_per_seq
int ck_tile::naive_attention_fwd_args::max_pages_per_seq |
◆ nhead_kv
int ck_tile::naive_attention_fwd_args::nhead_kv |
◆ nhead_q
int ck_tile::naive_attention_fwd_args::nhead_q |
◆ nhead_ratio_kv
int ck_tile::naive_attention_fwd_args::nhead_ratio_kv |
◆ o_ptr
void* ck_tile::naive_attention_fwd_args::o_ptr |
◆ page_size
int ck_tile::naive_attention_fwd_args::page_size |
◆ page_table_ptr
void* ck_tile::naive_attention_fwd_args::page_table_ptr |
◆ q_ptr
void* ck_tile::naive_attention_fwd_args::q_ptr |
◆ scale_s
float ck_tile::naive_attention_fwd_args::scale_s |
◆ seqlen_kv
int ck_tile::naive_attention_fwd_args::seqlen_kv |
◆ seqlen_q
int ck_tile::naive_attention_fwd_args::seqlen_q |
◆ v_ptr
void* ck_tile::naive_attention_fwd_args::v_ptr |
◆ vscale_ptr
void* ck_tile::naive_attention_fwd_args::vscale_ptr |
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/ref/naive_attention.hpp