Struct RuntimeArgs 

pub struct RuntimeArgs {
    pub shape_m: u32,
    pub shape_n: u32,
    pub shape_k: u32,
    pub padded_channels: FastDivmod,
    pub shape_out: Sequence<FastDivmod>,
    pub shape_channel: FastDivmod,
}

Fields

shape_m: u32

shape_n: u32

shape_k: u32

padded_channels: FastDivmod

shape_out: Sequence<FastDivmod>

shape_channel: FastDivmod

Trait Implementations

impl Clone for RuntimeArgs

fn clone(&self) -> RuntimeArgs

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for RuntimeArgs

type ExpandType = RuntimeArgsExpand

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl LaunchArg for RuntimeArgs

type RuntimeArg<'a, R: Runtime> = RuntimeArgsLaunch<'a, R>

The runtime argument for the kernel.

type CompilationArg = RuntimeArgsCompilationArg

Compilation argument.

fn compilation_arg<'a, R: Runtime>( runtime_arg: &Self::RuntimeArg<'a, R>, ) -> Self::CompilationArg

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an output variable during compilation that fill the KernelBuilder.