Trait LaunchArg

pub trait LaunchArg:
    LaunchArgExpand
    + Send
    + Sync
    + 'static {
    type RuntimeArg<'a, R: Runtime>: ArgSettings<R>;

    // Required method
    fn compilation_arg<R: Runtime>(
        runtime_arg: &Self::RuntimeArg<'_, R>,
    ) -> Self::CompilationArg;
}

Defines a type that can be used as argument to a kernel.

Required Associated Types

type RuntimeArg<'a, R: Runtime>: ArgSettings<R>

The runtime argument for the kernel.

Required Methods

fn compilation_arg<R: Runtime>( runtime_arg: &Self::RuntimeArg<'_, R>, ) -> Self::CompilationArg

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types

impl LaunchArg for ()

type RuntimeArg<'a, R: Runtime> = ()

fn compilation_arg<'a, R: Runtime>( _runtime_arg: &'a Self::RuntimeArg<'a, R>, ) -> Self::CompilationArg

Implementors

impl<C: CubePrimitive> LaunchArg for Array<C>

type RuntimeArg<'a, R: Runtime> = ArrayArg<'a, R>

impl<C: CubePrimitive> LaunchArg for Tensor<C>

type RuntimeArg<'a, R: Runtime> = TensorArg<'a, R>

impl<C: LaunchArg> LaunchArg for Sequence<C>

type RuntimeArg<'a, R: Runtime> = SequenceArg<'a, R, C>

impl<E: CubePrimitive> LaunchArg for TensorMap<E>

type RuntimeArg<'a, R: Runtime> = TensorMapArg<'a, R>

impl<T: Numeric> LaunchArg for T

type RuntimeArg<'a, R: Runtime> = ScalarArg<T>