Struct PageTableHost 

pub struct PageTableHost {
    pub block_table: Vec<u32>,
    pub context_lens: Vec<u32>,
    pub batch: usize,
    pub max_blocks: usize,
}

A materialized Op::PagedAttn page table — the host-side block_table + context_lens for a batch of B sessions, ready to upload as Op::Const operands. block_table is row-major [B, max_blocks]; context_lens is [B]. Both are u32 (the op’s declared operand dtype).

Fields

block_table: Vec<u32>

Row-major [B, max_blocks] logical→physical map. Rows shorter than max_blocks are zero-padded: index 0 is always an in-bounds physical block, and every padded key position is ≥ context_len, so the op’s variable-length mask (key_pos ≥ context_len → −inf) neutralizes it before softmax. Padding is therefore never read, only never invalid.

context_lens: Vec<u32>

Per-sequence context length (filled_tokens), [B].

batch: usize

B — number of sessions (rows).

max_blocks: usize

max_blocks — the padded row width (the widest session’s block count, at least 1 so the tensor stays a well-formed rank-2 [B, max_blocks]).

Implementations

impl PageTableHost

pub fn block_table_shape(&self) -> Shape

Shape of the block_table Op::Const — [B, max_blocks].

pub fn context_lens_shape(&self) -> Shape

Shape of the context_lens Op::Const — [B].

Trait Implementations

impl Clone for PageTableHost

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PageTableHost

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Eq for PageTableHost

impl PartialEq for PageTableHost

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for PageTableHost