Struct RuntimeArgs

pub struct RuntimeArgs {
    pub size_m: u32,
    pub size_n: u32,
    pub size_k: u32,
    pub padded_channels: FastDivmod,
    pub out_h: FastDivmod,
    pub out_w: FastDivmod,
}

Fields

size_m: u32

size_n: u32

size_k: u32

padded_channels: FastDivmod

out_h: FastDivmod

out_w: 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 init(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.

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

impl LaunchArgExpand for RuntimeArgs

type CompilationArg = RuntimeArgsCompilationArg

Compilation argument.

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.