Struct RenderContext 

pub struct RenderContext { /* private fields */ }

Maps UOp ID → LLVM variable name.

Implementations

impl RenderContext

pub fn new() -> Self

pub fn name(&mut self, uop: &Arc<UOp>) -> String

Get or create variable name for UOp.

For constants, returns literal value. For definitions, returns argument name. For other ops, returns a generated variable name.

pub fn get(&self, uop: &Arc<UOp>) -> &str

Get existing name (panics if not found).

pub fn try_get(&self, uop: &Arc<UOp>) -> Option<&str>

Try to get existing name.

pub fn contains(&self, id: u64) -> bool

Check if a UOp is already registered.

pub fn alias(&mut self, id: u64, name: String)

Alias one ID to another’s name.

pub fn register(&mut self, id: u64, name: String)

Pre-register a name for a UOp ID.

pub fn counter(&self) -> usize

Get current variable counter.

pub fn register_range(&mut self, axis_id: usize, name: String)

Register a range value by axis_id.

pub fn get_range(&self, axis_id: usize) -> Option<&str>

Get a range value by axis_id.

pub fn register_reduce_pending( &mut self, reduce_id: u64, acc_ptr: String, dtype: String, )

Register a pending reduce final load.

pub fn take_pending_reduces(&mut self) -> HashMap<u64, PendingReduce>

Take all pending reduces (empties map).

pub fn has_pending_reduces(&self) -> bool

Check if there are pending reduces.

Trait Implementations

impl Default for RenderContext

fn default() -> Self

Returns the “default value” for a type.