Enum cubecl_core::frontend::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( handle: &'a Handle<R::Server>, length: usize, vectorization_factor: u8, ) -> Self

Create a new array argument.

Safety

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

Trait Implementations

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

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

Register the information to the KernelLauncher.

fn configure_input( &self, position: usize, settings: KernelSettings, ) -> KernelSettings

Configure an input argument at the given position.

fn configure_output( &self, position: usize, settings: KernelSettings, ) -> KernelSettings

Configure an output argument at the given position.