Skip to main content

Module element

luma_tensor::device::cpu::kernels

Module element 

Source
Expand description

Per-element numeric traits. Generic CPU kernels are written over these, then the storage enums dispatch to the concrete f32/f64/i32/… instantiation.

This is the CPU analogue of luma-core’s NumDType/FloatDType/IntDType, trimmed to exactly what the kernels need.

Traits§

CpuFloat
Float-only element: transcendental math and activations.
CpuInt
Int-only element. MAX doubles as the padding sentinel used by indexing ops (matching luma-core’s I::max_value() convention).
CpuNum
Numeric element shared by float and int kinds.