Struct StepEnv 

pub struct StepEnv<'a> {
    pub gpu: &'a dyn GpuBackend,
    pub stream: u64,
    pub gdn: &'a [Vec<GdnState>],
    pub max_blocks_per_seq: u32,
}

2026-09-28: What varies between two runs of one program.

Fields

gpu: &'a dyn GpuBackend

2026-09-28: The backend.

stream: u64

2026-09-28: The stream the step runs (and is captured) on.

gdn: &'a [Vec<GdnState>]

2026-09-28: Per layer, each row’s GDN state (row i = sequence i, padding rows included); empty for a layer without one.

max_blocks_per_seq: u32

2026-09-28: AttnMetadataDev::max_blocks_per_seq of this step.

Implementations

impl StepEnv<'_>

pub fn gdn_state(&self, layer: usize, row: usize) -> Result<GdnState>

2026-09-28: Row row’s GDN state in layer layer; an error when there is none.