Struct KernelNodeBuilder 

pub struct KernelNodeBuilder<'a> { /* private fields */ }

Fluent builder for a single kernel launch node.

Obtained via GraphBuilder::add_kernel.

Implementations

impl<'a> KernelNodeBuilder<'a>

pub fn fusible(self, v: bool) -> Self

Marks this kernel as fusible with adjacent element-wise kernels.

pub fn inputs(self, ids: impl IntoIterator<Item = BufferId>) -> Self

Sets input buffer IDs for this kernel.

pub fn outputs(self, ids: impl IntoIterator<Item = BufferId>) -> Self

Sets output buffer IDs for this kernel.

pub fn on_stream(self, s: StreamId) -> Self

Sets the preferred stream for this node.

pub fn cost(self, c: u64) -> Self

Sets the cost hint for the scheduler.

pub fn finish(self) -> NodeId

Finishes configuration and registers the node with the parent builder.

Returns the allocated NodeId.