Struct WgslTranspiler 

pub struct WgslTranspiler {
    pub shared_vars: HashMap<String, SharedVarInfo>,
    /* private fields */
}

WGSL code transpiler.

Fields

shared_vars: HashMap<String, SharedVarInfo>

Variables that are SharedTile or SharedArray types.

Implementations

impl WgslTranspiler

pub fn new_stencil(config: StencilConfig) -> Self

Create a new transpiler with stencil configuration.

pub fn new_generic() -> Self

Create a new transpiler without stencil configuration.

pub fn new_ring_kernel(config: RingKernelConfig) -> Self

Create a new transpiler for ring kernel generation.

pub fn with_workgroup_size(self, x: u32, y: u32, z: u32) -> Self

Set the workgroup size for generic kernels.

pub fn transpile_stencil(&mut self, func: &ItemFn) -> Result<String>

Transpile a stencil kernel function.

pub fn transpile_global_kernel(&mut self, func: &ItemFn) -> Result<String>

Transpile a generic (non-stencil) kernel function.

pub fn transpile_ring_kernel( &mut self, handler: &ItemFn, config: &RingKernelConfig, ) -> Result<String>

Transpile a handler function into a ring kernel.