Enum ArrayArg

pub enum ArrayArg<'a, R: Runtime> {
    Handle {
        handle: ArrayHandleRef<'a, R>,
        vectorization_factor: u8,
    },
    Alias {
        input_pos: usize,
    },
}

Variants

Handle

The array is passed with an array handle.

Fields

handle: ArrayHandleRef<'a, R>

The array handle.

vectorization_factor: u8

The vectorization factor.

Alias

The array is aliasing another input array.

Fields

input_pos: usize

The position of the input array.

Implementations

impl<'a, R: Runtime> ArrayArg<'a, R>

pub unsafe fn from_raw_parts<E: CubePrimitive>( handle: &'a Handle, length: usize, vectorization_factor: u8, ) -> Self

Create a new array argument.

Safety

Specifying the wrong length may lead to out-of-bounds reads and writes.

pub unsafe fn from_raw_parts_and_size( handle: &'a Handle, length: usize, vectorization_factor: u8, elem_size: usize, ) -> Self

Create a new array argument with a manual element size in bytes.

Safety

Specifying the wrong length may lead to out-of-bounds reads and writes.

Trait Implementations

impl<R: Runtime> ArgSettings<R> for ArrayArg<'_, R>

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

Register the information of an argument to the KernelLauncher.