Struct RangeifyContext 

pub struct RangeifyContext {
    pub range_map: HashMap<UOpKey, Arc<UOp>>,
    pub range_counter: usize,
}

Context for rangeify transformation.

Tracks state during the rangeify transformation, including:

• Mapping from original UOps to their rangeified versions
• Counter for generating unique range IDs

Fields

range_map: HashMap<UOpKey, Arc<UOp>>

Maps old UOps to their rangeified versions.

This allows us to track how each node in the original graph has been transformed during the rangeify process.

Uses UOpKey for HashMap keys since Arc doesn’t implement Hash/Eq.

range_counter: usize

Counter for generating unique range IDs.

Each RANGE operation needs a unique axis_id to distinguish different loop dimensions. This counter ensures we never reuse IDs within a single transformation.

Implementations

impl RangeifyContext

pub fn new() -> Self

Create a new empty rangeify context.

pub fn next_range_id(&mut self) -> usize

Get the next available range ID.

Increments the internal counter and returns the previous value. This ensures each range gets a unique ID.

pub fn record_transform(&mut self, original: Arc<UOp>, rangeified: Arc<UOp>)

Record that a UOp has been transformed.

Maps the original UOp to its rangeified version so we can track the transformation.

pub fn get_rangeified(&self, original: &Arc<UOp>) -> Option<&Arc<UOp>>

Get the rangeified version of a UOp, if it exists.

Trait Implementations

impl Default for RangeifyContext

fn default() -> RangeifyContext

Returns the “default value” for a type.