Enum DType 

#[non_exhaustive]
pub enum DType {
    F32,
    F64,
    F16,
    Bf16,
    I8,
    I16,
    I32,
    I64,
    U8,
    U16,
    U32,
    U64,
    F8E4m3,
    F8E5m2,
    F4E2m1,
}

Re-export atomr_accel::DType so existing crate::dtype::DType imports inside atomr-accel-cuda (added by Phase 0.4) keep working without changing every call site. Compact discriminant for AccelDtype::KIND.

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

F32

F64

F16

Bf16

I8

I16

I32

I64

U8

U16

U32

U64

F8E4m3

8-bit float, E4M3 (sign 1, exp 4, mant 3). Hopper+ fp8 GEMM, FlashAttention v3.

F8E5m2

8-bit float, E5M2 (sign 1, exp 5, mant 2).

F4E2m1

4-bit float, E2M1. Blackwell fp4 inference.

Implementations

impl DType

pub const fn size_bytes(self) -> usize

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

pub const fn is_float(self) -> bool

pub const fn is_integer(self) -> bool

pub const fn is_signed(self) -> bool

Trait Implementations

impl Clone for DType

fn clone(&self) -> DType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DType

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

Formats the value using the given formatter.

impl Hash for DType

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

where __H: Hasher,

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: &DType) -> bool

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for DType

impl Eq for DType

impl StructuralPartialEq for DType