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: Scalar>(&self) -> C

Reads the scalar with the given element type.

Performs casting if necessary.

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

Reads the scalar with the given element type.

Performs casting if necessary.

impl InputScalar

pub fn as_bytes(&self) -> Vec<u8>

Trait Implementations

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 Copy for InputScalar

impl CubeType for InputScalar

type ExpandType = InputScalarExpand

impl Debug for InputScalar

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl LaunchArg for InputScalar

type RuntimeArg<R: Runtime> = InputScalar

The runtime argument for the kernel.

type CompilationArg = InputScalarCompilationArg

Compilation argument.

fn register<R: Runtime>( arg: Self::RuntimeArg<R>, launcher: &mut KernelLauncher<R>, ) -> Self::CompilationArg

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

Register a variable during compilation that fill the KernelBuilder.