Struct PagedKvConfig 

pub struct PagedKvConfig {
    pub block_size: u32,
    pub n_layer: u32,
    pub n_kv_head: u32,
    pub head_dim: u32,
    pub max_context: u32,
    pub physical_blocks: u32,
}

Fields

block_size: u32

n_layer: u32

n_kv_head: u32

head_dim: u32

max_context: u32

physical_blocks: u32

Physical pages in the shared pool. A single sequence needs n_layer * logical_blocks_per_layer() pages.

Implementations

impl PagedKvConfig

pub fn new( n_layer: u32, n_kv_head: u32, head_dim: u32, max_context: u32, ) -> Option<PagedKvConfig>

pub fn is_valid(&self) -> bool

pub fn logical_blocks_per_layer(&self) -> u32

pub fn required_single_sequence_blocks(&self) -> u32

pub fn slot_kv_elems(&self) -> u32

pub fn block_elems(&self) -> usize

pub fn arena_bytes(&self) -> Option<usize>

Trait Implementations

impl Clone for PagedKvConfig

fn clone(&self) -> PagedKvConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PagedKvConfig

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

Formats the value using the given formatter.

impl PartialEq for PagedKvConfig

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for PagedKvConfig

impl Eq for PagedKvConfig

impl StructuralPartialEq for PagedKvConfig