Struct KernelExecutable 

pub struct KernelExecutable {
    pub name: String,
    pub kernel_source: String,
    pub entry_point: String,
    pub shape: Shape,
    pub dtype: DtypeRepr,
    pub runtime_op: Arc<dyn RuntimeOp>,
    pub backward_kernel_source: String,
    pub backward_entry_point: String,
}

A lowered op that carries the kernel source needed for compilation.

Callers that have teeny-compiler as a dependency can pass kernel_source and kernel_entry_point to compile_kernel along with a chosen Target.

Fields

name: String

kernel_source: String

entry_point: String

shape: Shape

dtype: DtypeRepr

runtime_op: Arc<dyn RuntimeOp>

Runtime dispatch object: how to pack args and compute the launch grid. Input nodes carry a no-op implementation.

backward_kernel_source: String

Backward kernel source. Empty if this op has no backward.

backward_entry_point: String

Backward kernel entry point name.

Trait Implementations

impl ExecutableOp for KernelExecutable

fn name(&self) -> &str

This op’s name.

fn is_input(&self) -> bool

Returns true for Input placeholder nodes, which carry no kernel.

fn forward_kernel_source(&self) -> &str

This op’s forward kernel source.

fn forward_kernel_entry_point(&self) -> &str

This op’s forward kernel entry-point symbol name.

fn output_shape(&self) -> &Shape

This op’s output shape.

fn output_dtype(&self) -> DtypeRepr

This op’s output dtype.

fn runtime_op(&self) -> Option<Arc<dyn RuntimeOp>>

Returns the runtime dispatch object for this op, or None for Input nodes.

fn backward_kernel_source(&self) -> &str

Returns the backward kernel source, or "" if no backward is available.

fn backward_kernel_entry_point(&self) -> &str

Returns the backward kernel entry point name.