Trait cubecl_core::frontend::ArgSettings

pub trait ArgSettings<R: Runtime>: Send + Sync {
    // Required method
    fn register(&self, launcher: &mut KernelLauncher<R>);

    // Provided methods
    fn configure_input(
        &self,
        _position: usize,
        settings: KernelSettings,
    ) -> KernelSettings { ... }
    fn configure_output(
        &self,
        _position: usize,
        settings: KernelSettings,
    ) -> KernelSettings { ... }
}

Defines the argument settings used to launch a kernel.

Required Methods

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

Register the information to the KernelLauncher.

Provided Methods

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.

Implementations on Foreign Types

impl<R: Runtime> ArgSettings<R> for ()

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

Implementors

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

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

impl<T: Numeric, R: Runtime> ArgSettings<R> for ScalarArg<T>