Trait cubecl_core::frontend::LaunchArg

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

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

Required Associated Types§

source

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

The runtime argument for the kernel.

Object Safety§

This trait is not object safe.

Implementations on Foreign Types§

source§

impl LaunchArg for ()

§

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

Implementors§

source§

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

§

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

source§

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

§

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

source§

impl<T: Numeric> LaunchArg for T

§

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