Struct DsaSelectScratch 

pub struct DsaSelectScratch { /* private fields */ }

2026-09-25: Scratch the selection kernels write through, allocated once and reused across steps.

Sized from the largest geometry the caller will run; Self::fits refuses a pass that would outgrow it.

Implementations

impl DsaSelectScratch

pub fn alloc( gpu: &dyn GpuBackend, cfg: &Glm5NextDsaConfig, geom: &DsaSelectGeometry, ) -> Result<Self>

2026-09-25: Allocate for geom, the largest pass the caller will run (Glm5NextDsaWorkspace::new plans it at max_dsa_context).

pub fn tokens(&self) -> DevicePtr

2026-09-25: [q_rows, out_width] i32 selection produced by the last pass.

pub fn row(&self, row: usize, cfg: &Glm5NextDsaConfig) -> Self

2026-09-25: The same scratch with tokens pointing at row row.

The per-row selector (q_rows == 1) writes each row’s result into its own slot of the [max_rows, out_width] output, which the attention reads in one launch. The other regions are temporaries of one pass and are shared.

pub fn fits( &self, cfg: &Glm5NextDsaConfig, geom: &DsaSelectGeometry, ) -> Result<()>

2026-09-25: Whether geom fits what was allocated. select_tokens checks it on every pass, so a pass larger than the allocation is an error rather than an overrun.

pub fn free(self, gpu: &dyn GpuBackend) -> Result<()>

Trait Implementations

impl Clone for DsaSelectScratch

fn clone(&self) -> DsaSelectScratch

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DsaSelectScratch

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

Formats the value using the given formatter.

impl Copy for DsaSelectScratch