Trait cubecl_core::frontend::ScalarArgSettings

source ·
pub trait ScalarArgSettings: Send + Sync {
    // Required method
    fn register<R: Runtime>(&self, launcher: &mut KernelLauncher<R>);
}
Expand description

Similar to ArgSettings, however only for scalar types that don’t depend on the Runtime trait.

Required Methods§

source

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

Register the information to the KernelLauncher.

Object Safety§

This trait is not object safe.

Implementations on Foreign Types§

source§

impl ScalarArgSettings for f32

source§

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

source§

impl ScalarArgSettings for f64

source§

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

source§

impl ScalarArgSettings for i32

source§

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

source§

impl ScalarArgSettings for i64

source§

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

source§

impl ScalarArgSettings for u32

source§

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

source§

impl ScalarArgSettings for bf16

source§

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

source§

impl ScalarArgSettings for f16

source§

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

Implementors§