Enum TensorArg

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

Argument to be used for tensors passed as arguments to kernels.

Variants

Handle

The tensor is passed with a tensor handle.

Fields

handle: TensorHandleRef<'a, R>

The tensor handle.

vectorization_factor: u8

The vectorization factor.

Alias

The tensor is aliasing another input tensor.

Fields

input_pos: usize

The position of the input tensor.

Implementations

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

pub unsafe fn from_raw_parts<E: CubePrimitive>( handle: &'a Handle, strides: &'a [usize], shape: &'a [usize], factor: u8, ) -> Self

Create a new tensor argument specified with its vectorization factor.

Safety

If you provide wrong strides or shapes, it might create undefined behavior caused by out-of-bound reads and writes.

pub unsafe fn from_raw_parts_and_size( handle: &'a Handle, strides: &'a [usize], shape: &'a [usize], factor: u8, elem_size: usize, ) -> Self

Create a new tensor argument specified with its vectorization factor with a manual element size in bytes.

Safety

If you provide wrong strides or shapes, it might create undefined behavior caused by out-of-bound reads and writes.

pub fn alias(position: usize) -> Self

Create an alias argument.

Trait Implementations

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

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

Register the information of an argument to the KernelLauncher.

impl<'a, R: Debug + Runtime> Debug for TensorArg<'a, R>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.