Enum DType 

pub enum DType {
    E2M1,
    FP8E4M3,
    FP8E5M2,
    BF16,
    FP16,
    FP32,
}

Supported quantization and precision types for Atlas kernels.

Each variant maps to a specific bit-width and numeric format used by SM121 tensor cores or CUDA ALU paths.

Variants

E2M1

4-bit floating point (1 sign + 2 exponent + 1 mantissa) Used for NVFP4 weight storage. Values: {0, 0.5, 1, 1.5, 2, 3, 4, 6}

FP8E4M3

8-bit floating point (1 sign + 4 exponent + 3 mantissa) Used for NVFP4 block scales and FP8 quantization

FP8E5M2

8-bit floating point (1 sign + 5 exponent + 2 mantissa) Used for some FP8 quantization schemes

BF16

16-bit brain floating point

FP16

16-bit IEEE floating point

FP32

32-bit IEEE floating point

Implementations

impl DType

pub const fn element_size_bits(&self) -> usize

Size in bytes per element. E2M1 is sub-byte (4 bits) but we report the packed size (2 elements per byte).

pub const fn elements_per_u32(&self) -> usize

Number of elements that pack into a 32-bit word.

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DType

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

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: &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 Serialize for DType

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for DType

impl Eq for DType

impl StructuralPartialEq for DType