GQA paged prefill causal h32 kv8 d128 ps64
0 eligible runs
gqa-paged-attention
Batched Grouped Query Attention prefill with a paged KV cache (page_size=64). Causal mask is applied. Captured from Llama-3.1-8B during incremental prefill.
Current records
No published measurement for the selected workload.
Implementations
Implementation
Runtime
Best latency
Evidence
Availability
Semantics
Inputs and outputs
qbf16 [total_q, num_qo_heads, head_dim]
k_cachebf16 [num_pages, page_size, num_kv_heads, head_dim]
v_cachebf16 [num_pages, page_size, num_kv_heads, head_dim]
qo_indptrint32 [len_indptr]
kv_indptrint32 [len_indptr]
kv_indicesint32 [num_kv_indices]
kv_last_page_lenint32 [batch_size]
sm_scalefp32 scalar
outputbf16 [total_q, num_qo_heads, head_dim]
lsefp32 [total_q, num_qo_heads]
Axes and behavior
total_qvariable
head_dimconstant = 128
num_pagesvariable
page_sizeconstant = 64
batch_sizevariable
len_indptrvariable
num_kv_headsconstant = 8
num_qo_headsconstant = 32
num_kv_indicesvariable
determinismunspecified
constraintsNo mutation or aliasing
Identity
No source imports for this operation yet.How records are decidedJSON