Struct IrBuilder 

pub struct IrBuilder { /* private fields */ }

Builder for constructing IR modules.

Implementations

impl IrBuilder

pub fn new(name: impl Into<String>) -> Self

Create a new builder.

pub fn build(self) -> IrModule

Build and return the IR module.

pub fn module(&self) -> &IrModule

Get a reference to the module being built.

pub fn set_config(&mut self, config: KernelConfig)

Set kernel configuration.

pub fn set_block_size(&mut self, x: u32, y: u32, z: u32)

Set block size.

pub fn set_persistent(&mut self, persistent: bool)

Mark as persistent kernel.

pub fn parameter(&mut self, name: impl Into<String>, ty: IrType) -> ValueId

Add a parameter.

pub fn create_block(&mut self, label: impl Into<String>) -> BlockId

Create a new block.

pub fn switch_to_block(&mut self, block: BlockId)

Switch to a different block.

pub fn current_block(&self) -> BlockId

Get current block ID.

pub fn const_i32(&mut self, value: i32) -> ValueId

Create an i32 constant.

pub fn const_i64(&mut self, value: i64) -> ValueId

Create an i64 constant.

pub fn const_u32(&mut self, value: u32) -> ValueId

Create a u32 constant.

pub fn const_u64(&mut self, value: u64) -> ValueId

Create a u64 constant.

pub fn const_f32(&mut self, value: f32) -> ValueId

Create an f32 constant.

pub fn const_f64(&mut self, value: f64) -> ValueId

Create an f64 constant.

pub fn const_bool(&mut self, value: bool) -> ValueId

Create a boolean constant.

pub fn add(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Add two values.

pub fn sub(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Subtract two values.

pub fn mul(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Multiply two values.

pub fn div(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Divide two values.

pub fn rem(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Remainder.

pub fn and(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Bitwise AND.

pub fn or(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Bitwise OR.

pub fn xor(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Bitwise XOR.

pub fn shl(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Left shift.

pub fn shr(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Logical right shift.

pub fn min(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Minimum.

pub fn max(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Maximum.

pub fn neg(&mut self, value: ValueId) -> ValueId

Negate.

pub fn not(&mut self, value: ValueId) -> ValueId

Bitwise NOT.

pub fn abs(&mut self, value: ValueId) -> ValueId

Absolute value.

pub fn sqrt(&mut self, value: ValueId) -> ValueId

Square root.

pub fn floor(&mut self, value: ValueId) -> ValueId

Floor.

pub fn ceil(&mut self, value: ValueId) -> ValueId

Ceiling.

pub fn eq(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Equal comparison.

pub fn ne(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Not equal comparison.

pub fn lt(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Less than.

pub fn le(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Less than or equal.

pub fn gt(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Greater than.

pub fn ge(&mut self, lhs: ValueId, rhs: ValueId) -> ValueId

Greater than or equal.

pub fn load(&mut self, ptr: ValueId) -> ValueId

Load from pointer.

pub fn store(&mut self, ptr: ValueId, value: ValueId)

Store to pointer.

pub fn gep(&mut self, ptr: ValueId, indices: Vec<ValueId>) -> ValueId

Get element pointer.

pub fn shared_alloc(&mut self, ty: IrType, count: usize) -> ValueId

Allocate shared memory.

pub fn thread_id(&mut self, dim: Dimension) -> ValueId

Get thread ID.

pub fn block_id(&mut self, dim: Dimension) -> ValueId

Get block ID.

pub fn block_dim(&mut self, dim: Dimension) -> ValueId

Get block dimension.

pub fn grid_dim(&mut self, dim: Dimension) -> ValueId

Get grid dimension.

pub fn global_thread_id(&mut self, dim: Dimension) -> ValueId

Get global thread ID.

pub fn barrier(&mut self)

Block/threadgroup barrier.

pub fn fence(&mut self, scope: MemoryScope)

Memory fence.

pub fn grid_sync(&mut self)

Grid sync (cooperative groups).

pub fn atomic_add(&mut self, ptr: ValueId, value: ValueId) -> ValueId

Atomic add.

pub fn atomic_exchange(&mut self, ptr: ValueId, value: ValueId) -> ValueId

Atomic exchange.

pub fn atomic_cas( &mut self, ptr: ValueId, expected: ValueId, desired: ValueId, ) -> ValueId

Atomic compare-and-swap.

pub fn select( &mut self, cond: ValueId, then_val: ValueId, else_val: ValueId, ) -> ValueId

Select (ternary).

pub fn branch(&mut self, target: BlockId)

Branch to block.

pub fn cond_branch( &mut self, cond: ValueId, then_block: BlockId, else_block: BlockId, )

Conditional branch.

pub fn ret(&mut self)

Return from kernel.

pub fn ret_value(&mut self, value: ValueId)

Return value from kernel.

pub fn k2h_enqueue(&mut self, message: ValueId)

Enqueue to output (K2H).

pub fn h2k_dequeue(&mut self, msg_ty: IrType) -> ValueId

Dequeue from input (H2K).

pub fn h2k_is_empty(&mut self) -> ValueId

Check if input queue is empty.

pub fn k2k_send(&mut self, dest: ValueId, message: ValueId)

Send K2K message.

pub fn k2k_try_recv(&mut self, msg_ty: IrType) -> ValueId

Try receive K2K message.

pub fn hlc_now(&mut self) -> ValueId

Get current HLC time.

pub fn hlc_tick(&mut self) -> ValueId

Tick HLC.