Struct cubecl_cuda::compiler::ComputeKernel

pub struct ComputeKernel {
    pub inputs: Vec<Binding>,
    pub outputs: Vec<Binding>,
    pub named: Vec<(String, Binding)>,
    pub cube_dim: CubeDim,
    pub body: Body,
    pub wmma_activated: bool,
    pub bf16: bool,
    pub f16: bool,
}

Fields

inputs: Vec<Binding>

outputs: Vec<Binding>

named: Vec<(String, Binding)>

cube_dim: CubeDim

body: Body

wmma_activated: bool

bf16: bool

f16: bool

Trait Implementations

impl Clone for ComputeKernel

fn clone(&self) -> ComputeKernel

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl CompilerRepresentation for ComputeKernel

fn shared_memory_size(&self) -> usize

Computes and returns the shared memory size

impl Debug for ComputeKernel

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

Formats the value using the given formatter.

impl Display for ComputeKernel

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

Formats the value using the given formatter.