Struct InputScalar 

pub struct InputScalar { /* private fields */ }

A way to define an input scalar without a generic attached to it.

It uses comptime enum with zero-cost runtime abstraction for kernel generation.

Implementations

impl InputScalar

pub fn new<E: ToPrimitive>(val: E, dtype: impl Into<StorageType>) -> Self

Creates an InputScalar from the given element and dtype.

Panics

If the given numeric element can’t be transformed into the passed ElemType.

impl InputScalar

pub fn get<C: CubePrimitive>(&self) -> C

Reads the scalar with the given element type.

Performs casting if necessary.

pub fn __expand_get<C: CubePrimitive>( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <C as CubeType>::ExpandType

Trait Implementations

impl<R: Runtime> ArgSettings<R> for InputScalar

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

Register the information of an argument to the KernelLauncher.

impl Clone for InputScalar

fn clone(&self) -> InputScalar

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl CubeType for InputScalar

type ExpandType = InputScalarExpand

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl LaunchArg for InputScalar

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

The runtime argument for the kernel.

type CompilationArg = InputScalarCompilationArg

Compilation argument.

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

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> Self::ExpandType

Register an output variable during compilation that fill the KernelBuilder.

impl Copy for InputScalar