Enum BatchedGlobalScaleLayoutArgs 

pub enum BatchedGlobalScaleLayoutArgs<'a, R: Runtime> {
    PerTensor {
        shape: <Coords2d as LaunchArg>::RuntimeArg<'a, R>,
    },
    BlockScaled(<BlockScaledLayout as LaunchArg>::RuntimeArg<'a, R>),
}

Variants

PerTensor

Fields

shape: <Coords2d as LaunchArg>::RuntimeArg<'a, R>

BlockScaled(<BlockScaledLayout as LaunchArg>::RuntimeArg<'a, R>)

Trait Implementations

impl<'a, R: Runtime> ArgSettings<R> for BatchedGlobalScaleLayoutArgs<'a, R>

fn register(&self, launcher: &mut KernelLauncher<R>)

Register the information of an argument to the KernelLauncher.