Enum ViewArg 

pub enum ViewArg<'a, C: Coordinates, R: Runtime> {
    Array(ArrayArg<'a, R>, VirtualLayoutLaunch<'a, C, Coords1d, R>),
    TensorMap(TensorMapArg<'a, R>, VirtualLayoutLaunch<'a, C, Sequence<i32>, R>),
    Quantized {
        values: Box<ViewArg<'a, C, R>>,
        scales: Box<ViewArg<'a, C, R>>,
        scheme: QuantScheme,
    },
}

Variants

Array(ArrayArg<'a, R>, VirtualLayoutLaunch<'a, C, Coords1d, R>)

TensorMap(TensorMapArg<'a, R>, VirtualLayoutLaunch<'a, C, Sequence<i32>, R>)

Quantized

Fields

values: Box<ViewArg<'a, C, R>>

scales: Box<ViewArg<'a, C, R>>

scheme: QuantScheme

Implementations

impl<'a, C: Coordinates, R: Runtime> ViewArg<'a, C, R>

pub fn new<L: Layout<Coordinates = C, SourceCoordinates = Coords1d> + LaunchArg>( buffer: ArrayArg<'a, R>, layout: L::RuntimeArg<'a, R>, ) -> Self

pub fn new_tensor_map<L: Layout<Coordinates = C, SourceCoordinates: IntoDyn> + LaunchArg>( buffer: TensorMapArg<'a, R>, layout: L::RuntimeArg<'a, R>, ) -> Self

pub fn new_quantized(values: Self, scales: Self, scheme: QuantScheme) -> Self

Create a new view arg that dequantizes on read. The scales layout should take values indices and map them to the corresponding scale.

Trait Implementations

impl<'a, C: Coordinates, R: Runtime> ArgSettings<R> for ViewArg<'a, C, R>

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

Register the information of an argument to the KernelLauncher.