Struct LocalSlots 

pub struct LocalSlots { /* private fields */ }

Shared slot layout for all locals in one program.

Implementations

impl LocalSlots

pub fn for_program(program: &Program) -> Self

Build a slot layout from every binding site in a program.

pub fn for_nodes(nodes: &[Node]) -> Self

Build a slot layout from a node slice.

Trait Implementations

impl Debug for LocalSlots

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

Formats the value using the given formatter.

impl Default for LocalSlots

fn default() -> LocalSlots

Returns the “default value” for a type.

impl ExprVisitor for LocalSlots

type Break = Infallible

Break payload returned when traversal short-circuits.

fn visit_lit_u32( &mut self, _expr: &Expr, _value: u32, ) -> ControlFlow<Self::Break>

Integer literal (u32).

fn visit_lit_i32( &mut self, _expr: &Expr, _value: i32, ) -> ControlFlow<Self::Break>

Integer literal (i32).

fn visit_lit_f32( &mut self, _expr: &Expr, _value: f32, ) -> ControlFlow<Self::Break>

Float literal (f32).

fn visit_lit_bool( &mut self, _expr: &Expr, _value: bool, ) -> ControlFlow<Self::Break>

Bool literal.

fn visit_var(&mut self, _expr: &Expr, _name: &Ident) -> ControlFlow<Self::Break>

Variable reference.

fn visit_load( &mut self, _expr: &Expr, _buffer: &Ident, _index: &Expr, ) -> ControlFlow<Self::Break>

Buffer load (buffer[index]).

fn visit_buf_len( &mut self, _expr: &Expr, _buffer: &Ident, ) -> ControlFlow<Self::Break>

Buffer length.

fn visit_invocation_id( &mut self, _expr: &Expr, _axis: u32, ) -> ControlFlow<Self::Break>

Invocation id axis (gid.{x,y,z}).

fn visit_workgroup_id( &mut self, _expr: &Expr, _axis: u32, ) -> ControlFlow<Self::Break>

Workgroup id axis.

fn visit_local_id( &mut self, _expr: &Expr, _axis: u32, ) -> ControlFlow<Self::Break>

Local id axis within the workgroup.

fn visit_subgroup_local_id(&mut self, _expr: &Expr) -> ControlFlow<Self::Break>

Subgroup invocation id (lane index within subgroup).

fn visit_subgroup_size(&mut self, _expr: &Expr) -> ControlFlow<Self::Break>

Subgroup size.

fn visit_bin_op( &mut self, _expr: &Expr, _op: &BinOp, _left: &Expr, _right: &Expr, ) -> ControlFlow<Self::Break>

Binary operation.

fn visit_un_op( &mut self, _expr: &Expr, _op: &UnOp, _operand: &Expr, ) -> ControlFlow<Self::Break>

Unary operation.

fn visit_call( &mut self, _expr: &Expr, _op_id: &str, _args: &[Expr], ) -> ControlFlow<Self::Break>

Function call.

fn visit_sequence(&mut self, _parts: &[Expr]) -> ControlFlow<Self::Break>

Sequence-valued extension hook.

fn visit_fma( &mut self, _expr: &Expr, _a: &Expr, _b: &Expr, _c: &Expr, ) -> ControlFlow<Self::Break>

Fused multiply-add (a * b + c).

fn visit_select( &mut self, _expr: &Expr, _cond: &Expr, _true_val: &Expr, _false_val: &Expr, ) -> ControlFlow<Self::Break>

Ternary select(cond, true_val, false_val).

fn visit_cast( &mut self, _expr: &Expr, _target: &DataType, _value: &Expr, ) -> ControlFlow<Self::Break>

Numeric cast.

fn visit_atomic( &mut self, _expr: &Expr, _op: &AtomicOp, _buffer: &Ident, _index: &Expr, _expected: Option<&Expr>, _value: &Expr, ) -> ControlFlow<Self::Break>

Atomic operation on a shared buffer.

fn visit_subgroup_ballot( &mut self, _expr: &Expr, _cond: &Expr, ) -> ControlFlow<Self::Break>

Subgroup ballot.

fn visit_subgroup_shuffle( &mut self, _expr: &Expr, _value: &Expr, _lane: &Expr, ) -> ControlFlow<Self::Break>

Subgroup shuffle.

fn visit_subgroup_add( &mut self, _expr: &Expr, _value: &Expr, ) -> ControlFlow<Self::Break>

Subgroup add.

fn visit_opaque_expr( &mut self, _expr: &Expr, _extension: &(dyn ExprNode + 'static), ) -> ControlFlow<Self::Break>

Downstream opaque expression extension.

fn walk_children_default( &mut self, expr: &Expr, order: VisitOrder, ) -> ControlFlow<Self::Break>

where Self: Sized,

Recursively walk this expression’s children using the requested order.

impl NodeVisitor for LocalSlots

type Break = Infallible

Break payload returned when traversal short-circuits.

fn visit_let( &mut self, _: &Node, name: &Ident, value: &Expr, ) -> ControlFlow<Self::Break>

Variable declaration.

fn visit_assign( &mut self, _: &Node, _: &Ident, value: &Expr, ) -> ControlFlow<Self::Break>

Variable assignment.

fn visit_store( &mut self, _: &Node, _: &Ident, index: &Expr, value: &Expr, ) -> ControlFlow<Self::Break>

Buffer store.

fn visit_if( &mut self, _: &Node, cond: &Expr, _: &[Node], _: &[Node], ) -> ControlFlow<Self::Break>

Conditional branch.

fn visit_loop( &mut self, _: &Node, var: &Ident, from: &Expr, to: &Expr, _: &[Node], ) -> ControlFlow<Self::Break>

Counted loop.

fn visit_indirect_dispatch( &mut self, _: &Node, _: &Ident, _: u64, ) -> ControlFlow<Self::Break>

Indirect dispatch source.

fn visit_async_load( &mut self, _: &Node, _: &Ident, _: &Ident, offset: &Expr, size: &Expr, _: &Ident, ) -> ControlFlow<Self::Break>

Async load node.

fn visit_async_store( &mut self, _: &Node, _: &Ident, _: &Ident, offset: &Expr, size: &Expr, _: &Ident, ) -> ControlFlow<Self::Break>

Async store node.

fn visit_async_wait(&mut self, _: &Node, _: &Ident) -> ControlFlow<Self::Break>

Async wait node.

fn visit_trap( &mut self, _: &Node, address: &Expr, _: &Ident, ) -> ControlFlow<Self::Break>

Trap node.

fn visit_resume(&mut self, _: &Node, _: &Ident) -> ControlFlow<Self::Break>

Resume node.

fn visit_return(&mut self, _: &Node) -> ControlFlow<Self::Break>

Return node.

fn visit_barrier(&mut self, _: &Node) -> ControlFlow<Self::Break>

Barrier node.

fn visit_block(&mut self, _: &Node, _: &[Node]) -> ControlFlow<Self::Break>

Block node.

fn visit_region( &mut self, _: &Node, _: &Ident, _: &Option<GeneratorRef>, _: &[Node], ) -> ControlFlow<Self::Break>

Region wrapper node.

fn visit_opaque_node( &mut self, _: &Node, _: &dyn NodeExtension, ) -> ControlFlow<Self::Break>

Downstream opaque node extension.

fn walk_children_default( &mut self, node: &Node, order: VisitOrder, ) -> ControlFlow<Self::Break>

where Self: Sized,

Recursively walk this node’s nested node children using the requested order.