Struct DsaSelectGeometry 

pub struct DsaSelectGeometry {
    pub seq: usize,
    pub q_rows: usize,
    pub n_pools_full: usize,
    pub n_pools: usize,
    pub select_k: usize,
    pub out_width: usize,
    pub topk_np2: usize,
    pub topk_smem: usize,
    pub index_head_dim: usize,
    pub index_heads: usize,
    pub index_kpool: usize,
}

Launch geometry for one selection pass — every count the four kernels need, and every capacity check, decided before a single pointer is touched.

Fields

seq: usize

Tokens resident in the indexer cache.

q_rows: usize

Query rows selecting this pass (1 on a batch-1 decode step).

n_pools_full: usize

Pools dsa_kpool_compress writes, including the trailing partial one.

n_pools: usize

Pools that are real — the prefix everything downstream reads.

select_k: usize

Pools selected per query.

out_width: usize

Emitted index-row width.

topk_np2: usize

Tile width the top-k select walks the pool axis in (clamped down below one tile).

topk_smem: usize

Shared memory dsa_topk_pools needs, in bytes.

index_head_dim: usize

Channels per compress block.

index_heads: usize

index_kpool: usize

Implementations

impl DsaSelectGeometry

pub fn plan(cfg: &Glm5NextDsaConfig, seq: usize, q_rows: usize) -> Result<Self>

Plan a selection over a contiguous, unpadded cache.

Fails rather than truncates when the pool axis outgrows the top-k kernel’s shared-memory budget.

Trait Implementations

impl Clone for DsaSelectGeometry

fn clone(&self) -> DsaSelectGeometry

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DsaSelectGeometry

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

Formats the value using the given formatter.

impl PartialEq for DsaSelectGeometry

fn eq(&self, other: &DsaSelectGeometry) -> 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 DsaSelectGeometry

impl Eq for DsaSelectGeometry

impl StructuralPartialEq for DsaSelectGeometry