Enum ScalarDType 

pub enum ScalarDType {
    Bool = 0,
    Int8 = 1,
    UInt8 = 2,
    Int16 = 3,
    UInt16 = 4,
    Int32 = 5,
    UInt32 = 6,
    Int64 = 7,
    UInt64 = 8,
    FP8E4M3 = 9,
    FP8E5M2 = 10,
    Float16 = 11,
    BFloat16 = 12,
    Float32 = 13,
    Float64 = 14,
    Void = 15,
    Index = 16,
}

Scalar data types (base numeric types).

Variants

Bool = 0

Int8 = 1

UInt8 = 2

Int16 = 3

UInt16 = 4

Int32 = 5

UInt32 = 6

Int64 = 7

UInt64 = 8

FP8E4M3 = 9

FP8E5M2 = 10

Float16 = 11

BFloat16 = 12

Float32 = 13

Float64 = 14

Void = 15

Void type for metadata operations (no data).

Index = 16

Index type for array indexing and loop iteration.

Implementations

impl ScalarDType

pub fn can_safe_cast(self, to: Self) -> bool

Check if casting from from to to is safe (preserves value).

impl ScalarDType

pub const fn from_repr(discriminant: usize) -> Option<ScalarDType>

Try to create Self from the raw representation

impl ScalarDType

pub const fn bytes(&self) -> usize

pub const fn is_bool(&self) -> bool

pub const fn is_signed(&self) -> bool

pub const fn is_unsigned(&self) -> bool

pub const fn is_int(&self) -> bool

pub const fn is_float(&self) -> bool

pub const fn is_fp8(&self) -> bool

pub const fn min_value(&self) -> f64

pub const fn max_value(&self) -> f64

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

pub const fn min_positive(&self) -> f64

pub const fn finfo(&self) -> (u32, u32)

(exponent_bits, mantissa_bits) for float types. Matches Tinygrad’s dtypes.finfo().

pub const fn exponent_bias(&self) -> i32

Exponent bias: (1 << (exp_bits - 1)) - 1.

pub const fn float_to_uint(&self) -> ScalarDType

Map float dtype to uint storage equivalent of the same bit width.

pub const fn bitsize(&self) -> u32

Bit size of this scalar type.

pub const fn vec(self, count: usize) -> DType

Create a vector DType from this scalar type.

Trait Implementations

impl Arbitrary for ScalarDType

type Parameters = ()

The type of parameters that arbitrary_with accepts for configuration of the generated Strategy. Parameters must implement Default.

type Strategy = TupleUnion<((u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<TupleUnion<((u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>), (u32, Arc<fn() -> ScalarDType>))>>))>

The type of Strategy used to generate values of type Self.

fn arbitrary_with(_top: Self::Parameters) -> Self::Strategy

Generates a Strategy for producing arbitrary values of type the implementing type (Self). The strategy is passed the arguments given in args.

fn arbitrary() -> Self::Strategy

Generates a Strategy for producing arbitrary values of type the implementing type (Self).

impl<O: Into<EnumSet<ScalarDType>>> BitAnd<O> for ScalarDType

type Output = EnumSet<ScalarDType>

The resulting type after applying the & operator.

fn bitand(self, other: O) -> Self::Output

Performs the & operation.

impl<O: Into<EnumSet<ScalarDType>>> BitOr<O> for ScalarDType

type Output = EnumSet<ScalarDType>

The resulting type after applying the | operator.

fn bitor(self, other: O) -> Self::Output

Performs the | operation.

impl<O: Into<EnumSet<ScalarDType>>> BitXor<O> for ScalarDType

type Output = EnumSet<ScalarDType>

The resulting type after applying the ^ operator.

fn bitxor(self, other: O) -> Self::Output

Performs the ^ operation.

impl Clone for ScalarDType

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ScalarDType

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ScalarDType

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl EnumCount for ScalarDType

const COUNT: usize = 17usize

impl EnumSetConstHelper for ScalarDType

const CONST_INIT_HELPER: __EnumSetInitHelper = __EnumSetInitHelper

The instance of the ConstInitHelper.

const CONST_OP_HELPER: __EnumSetOpHelper = __EnumSetOpHelper

The instance of the ConstOpHelper.

type ConstInitHelper = __EnumSetInitHelper

A helper type used to convert values to EnumSets at compile-time.

type ConstOpHelper = __EnumSetOpHelper

A helper type used to implement compile-time operations on enums.

impl EnumSetTypePrivate for ScalarDType

const ALL_BITS: Self::Repr = 131071

A mask of bits that are valid in the bitset.

const BIT_WIDTH: u32 = 17u32

The largest bit used in the bitset.

const VARIANT_COUNT: u32 = 17u32

The number of variants in the bitset.

type Repr = u32

The underlying type used to store the bitset.

fn enum_into_u32(self) -> u32

Converts an enum of this type into its bit position.

unsafe fn enum_from_u32(val: u32) -> Self

Converts a bit position into an enum value.

unsafe fn enum_from_u32_checked(val: u32) -> Self

Converts a bit position into an enum value, with a debug_assert.

impl EnumSetTypeWithRepr for ScalarDType

type Repr = u32

The guaranteed representation.

impl From<ScalarDType> for DType

fn from(scalar: ScalarDType) -> Self

Converts to this type from the input type.

impl Hash for ScalarDType

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 IntoEnumIterator for ScalarDType

type Iterator = ScalarDTypeIter

fn iter() -> ScalarDTypeIter

impl Not for ScalarDType

type Output = EnumSet<ScalarDType>

The resulting type after applying the ! operator.

fn not(self) -> Self::Output

Performs the unary ! operation.

impl Ord for ScalarDType

fn cmp(&self, other: &ScalarDType) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized,

Restrict a value to a certain interval.

impl PartialEq<EnumSet<ScalarDType>> for ScalarDType

fn eq(&self, other: &EnumSet<ScalarDType>) -> 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 PartialEq for ScalarDType

fn eq(&self, other: &Self) -> 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 PartialOrd for ScalarDType

fn partial_cmp(&self, other: &ScalarDType) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

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

Tests less than (for self and other) and is used by the < operator.

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

Tests less than or equal to (for self and other) and is used by the <= operator.

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

Tests greater than (for self and other) and is used by the > operator.

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

Tests greater than or equal to (for self and other) and is used by the >= operator.

impl Serialize for ScalarDType

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl<O: Into<EnumSet<ScalarDType>>> Sub<O> for ScalarDType

type Output = EnumSet<ScalarDType>

The resulting type after applying the - operator.

fn sub(self, other: O) -> Self::Output

Performs the - operation.

impl VariantArray for ScalarDType

const VARIANTS: &'static [Self]

impl Copy for ScalarDType

impl EnumSetType for ScalarDType

impl Eq for ScalarDType