Module dfdx::dtypes

source ·
Expand description

Module for data type related traits and structs. Contains things like Unit, Dtype, and AMP.

When the f16 feature is enabled, this exports the [f16] type.

AMP

AMP is a technique for mixed precision training. This is a data type in dfdx, you can use it like any normal dtype like AMP<f16> or AMP<bf16>.

Structs

  • AMP
    Wrapper type around the storage type. Use like AMP<f16> or AMP<bf16>.

Traits

  • Dtype
    Represents a data type or element of an array that can have arithmatic operations applied to it. The main difference between Dtype and Unit is that bool is Unit, but not Dtype.
  • HasDtype
    Represents something that has a Dtype.
  • HasUnitType
    Represents something that has a Unit.
  • NotMixedPrecision
    Marker trait for types that are not AMP.
  • SafeZeros
    Represents a type where all 0 bits is a valid pattern.
  • Unit
    Represents a unit type, but no arithmetic.