Enum DType 

pub enum DType {
    F32,
    F16,
    BF16,
    I64,
    I32,
    I8,
    U8,
    Bool,
}

The element types kime reads from checkpoints. Weights are floats. The integer types exist because safetensors files may carry them, and a loader that knows their size can check a file it will not use.

Variants

F32

IEEE binary32.

F16

IEEE binary16.

BF16

bfloat16.

I64

Signed 64 bit integers.

I32

Signed 32 bit integers.

I8

Signed 8 bit integers.

U8

Unsigned 8 bit integers.

Bool

Booleans, one byte each.

Implementations

impl DType

pub const fn size(self) -> usize

Bytes per element.

pub const fn name(self) -> &'static str

The name safetensors uses, which kime also uses in .kime headers.

pub fn from_name(name: &str) -> Option<Self>

The inverse of DType::name.

pub const fn is_float(self) -> bool

Whether this is one of the float types weights come in.

pub fn read_f32(self, bytes: &[u8], i: usize) -> f32

Reads element i of little endian bytes as f32. Panics if bytes is too short, and returns NaN for the integer types, which no weight uses.

Trait Implementations

impl Clone for DType

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for DType

impl Debug for DType

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

Formats the value using the given formatter.

impl Display for DType

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

Formats the value using the given formatter.

impl Eq for DType

impl Hash for DType

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for DType

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for DType