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>(val: E, dtype: impl Into<StorageType>) -> InputScalar

where E: ToPrimitive,

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

where C: Scalar,

Reads the scalar with the given element type.

Performs casting if necessary.

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

where C: Scalar,

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

type ExpandType = InputScalarExpand

impl Debug for InputScalar

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

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>( arg: <InputScalar as LaunchArg>::RuntimeArg<R>, launcher: &mut KernelLauncher<R>, ) -> <InputScalar as LaunchArg>::CompilationArg

where R: Runtime,

fn expand( arg: &<InputScalar as LaunchArg>::CompilationArg, builder: &mut KernelBuilder, ) -> <InputScalar 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