Struct flex32

pub struct flex32(/* private fields */);

A floating point type with relaxed precision, minimum f16, max [f32].

Implementations

impl flex32

pub const MIN_POSITIVE: flex32

pub const fn from_f32(val: f32) -> flex32

pub const fn from_f64(val: f64) -> flex32

pub const fn to_f32(self) -> f32

pub const fn to_f64(self) -> f64

pub fn total_cmp(&self, other: &flex32) -> Ordering

pub fn is_nan(&self) -> bool

Trait Implementations

impl Abs for flex32

fn abs(x: Self) -> Self

fn __expand_abs( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Add for flex32

type Output = flex32

The resulting type after applying the + operator.

fn add(self, rhs: flex32) -> flex32

Performs the + operation.

impl AddAssign for flex32

fn add_assign(&mut self, rhs: flex32)

Performs the += operation.

impl Ceil for flex32

fn ceil(x: Self) -> Self

fn __expand_ceil( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Clamp for flex32

fn clamp(input: Self, min_value: Self, max_value: Self) -> Self

Clamp the input value between the max and min values provided.

fn __expand_clamp( context: &mut CubeContext, input: Self::ExpandType, min_value: Self::ExpandType, max_value: Self::ExpandType, ) -> Self::ExpandType

impl Clone for flex32

fn clone(&self) -> flex32

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Cos for flex32

fn cos(x: Self) -> Self

fn __expand_cos( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl CubeElement for flex32

fn type_name() -> &'static str

Returns the name of the type.

fn as_bytes(slice: &[flex32]) -> &[u8]

Convert a slice of elements to a slice of bytes.

fn from_bytes(bytes: &[u8]) -> &[flex32]

Convert a slice of bytes to a slice of elements.

fn cube_elem() -> Elem

Element representation for cubecl.

fn maximum_value() -> flex32

Highest possible value

fn minimum_value() -> flex32

Lowest possible value

impl<I> CubeIndex<I> for flex32

where I: Index,

type Output = flex32

fn cube_idx(&self, _i: T) -> &Self::Output

impl<I> CubeIndexMut<I> for flex32

where I: Index,

fn cube_idx_mut(&mut self, _i: T) -> &mut Self::Output

impl CubePrimitive for flex32

fn as_elem_native() -> Option<Elem>

Return the element type to use on GPU

fn as_elem(_context: &CubeContext) -> Elem

Return the element type to use on GPU.

fn as_elem_native_unchecked() -> Elem

Native or static element type.

fn size() -> Option<usize>

Only native element types have a size.

fn from_expand_elem(elem: ExpandElement) -> Self::ExpandType

impl CubeType for flex32

type ExpandType = ExpandElementTyped<flex32>

fn init(context: &mut CubeContext, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl Debug for flex32

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

Formats the value using the given formatter.

impl Default for flex32

fn default() -> flex32

Returns the “default value” for a type.

impl Display for flex32

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

Formats the value using the given formatter.

impl<__RhsT> Div<__RhsT> for flex32

where f32: Div<__RhsT, Output = f32>,

type Output = flex32

The resulting type after applying the / operator.

fn div(self, rhs: __RhsT) -> flex32

Performs the / operation.

impl Div for flex32

type Output = flex32

The resulting type after applying the / operator.

fn div(self, rhs: flex32) -> <flex32 as Div>::Output

Performs the / operation.

impl<__RhsT> DivAssign<__RhsT> for flex32

where f32: DivAssign<__RhsT>,

fn div_assign(&mut self, rhs: __RhsT)

Performs the /= operation.

impl DivAssign for flex32

fn div_assign(&mut self, rhs: flex32)

Performs the /= operation.

impl Dot for flex32

fn dot(self, _rhs: Self) -> Self

fn __expand_dot( context: &mut CubeContext, lhs: ExpandElementTyped<Self>, rhs: ExpandElementTyped<Self>, ) -> ExpandElementTyped<Self>

impl Element for flex32

fn dtype() -> DType

The dtype of the element.

impl ElementComparison for flex32

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

Returns and Ordering between self and other.

impl ElementConversion for flex32

fn from_elem<E>(elem: E) -> flex32

where E: ToElement,

Converts an element to another element.

fn elem<E>(self) -> E

where E: Element,

Converts and returns the converted element.

impl ElementPrecision for flex32

fn precision() -> Precision

Returns the precision of the element.

impl ElementRandom for flex32

fn random<R>(distribution: Distribution, rng: &mut R) -> flex32

where R: RngCore,

Returns a random value for the given distribution.

impl Erf for flex32

fn erf(x: Self) -> Self

fn __expand_erf( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Exp for flex32

fn exp(x: Self) -> Self

fn __expand_exp( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl ExpandElementBaseInit for flex32

fn init_elem(context: &mut CubeContext, elem: ExpandElement) -> ExpandElement

impl Float for flex32

const MAX_10_EXP: i32 = 38i32

Maximum possible [tf32] power of 10 exponent

const MAX_EXP: i32 = 128i32

Maximum possible [tf32] power of 2 exponent

const MIN_10_EXP: i32 = -37i32

Minimum possible normal [tf32] power of 10 exponent

const MIN_EXP: i32 = -125i32

One greater than the minimum possible normal [v] power of 2 exponent

const DIGITS: u32 = 32u32

const EPSILON: flex32

const INFINITY: flex32

const MANTISSA_DIGITS: u32 = 24u32

const MIN_POSITIVE: flex32

const NAN: flex32

const NEG_INFINITY: flex32

const RADIX: u32 = 2u32

fn new(val: f32) -> flex32

fn __expand_new(context: &mut CubeContext, val: f32) -> Self::ExpandType

impl Float for flex32

fn nan() -> flex32

Returns the NaN value.

fn infinity() -> flex32

Returns the infinite value.

fn neg_infinity() -> flex32

Returns the negative infinite value.

fn neg_zero() -> flex32

Returns -0.0.

fn min_value() -> flex32

Returns the smallest finite value that this type can represent.

fn min_positive_value() -> flex32

Returns the smallest positive, normalized value that this type can represent.

fn max_value() -> flex32

Returns the largest finite value that this type can represent.

fn is_nan(self) -> bool

Returns true if this value is NaN and false otherwise.

fn is_infinite(self) -> bool

Returns true if this value is positive infinity or negative infinity and false otherwise.

fn is_finite(self) -> bool

Returns true if this number is neither infinite nor NaN.

fn is_normal(self) -> bool

Returns true if the number is neither zero, infinite, subnormal, or NaN.

fn classify(self) -> FpCategory

Returns the floating point category of the number. If only one property is going to be tested, it is generally faster to use the specific predicate instead.

fn floor(self) -> flex32

Returns the largest integer less than or equal to a number.

fn ceil(self) -> flex32

Returns the smallest integer greater than or equal to a number.

fn round(self) -> flex32

Returns the nearest integer to a number. Round half-way cases away from 0.0.

fn trunc(self) -> flex32

Return the integer part of a number.

fn fract(self) -> flex32

Returns the fractional part of a number.

fn abs(self) -> flex32

Computes the absolute value of self. Returns Float::nan() if the number is Float::nan().

fn signum(self) -> flex32

Returns a number that represents the sign of self.

fn is_sign_positive(self) -> bool

Returns true if self is positive, including +0.0, Float::infinity(), and Float::nan().

fn is_sign_negative(self) -> bool

Returns true if self is negative, including -0.0, Float::neg_infinity(), and -Float::nan().

fn mul_add(self, a: flex32, b: flex32) -> flex32

Fused multiply-add. Computes (self * a) + b with only one rounding error, yielding a more accurate result than an unfused multiply-add.

fn recip(self) -> flex32

Take the reciprocal (inverse) of a number, 1/x.

fn powi(self, n: i32) -> flex32

Raise a number to an integer power.

fn powf(self, n: flex32) -> flex32

Raise a number to a floating point power.

fn sqrt(self) -> flex32

Take the square root of a number.

fn exp(self) -> flex32

Returns e^(self), (the exponential function).

fn exp2(self) -> flex32

Returns 2^(self).

fn ln(self) -> flex32

Returns the natural logarithm of the number.

fn log(self, base: flex32) -> flex32

Returns the logarithm of the number with respect to an arbitrary base.

fn log2(self) -> flex32

Returns the base 2 logarithm of the number.

fn log10(self) -> flex32

Returns the base 10 logarithm of the number.

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

Returns the maximum of the two numbers.

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

Returns the minimum of the two numbers.

fn abs_sub(self, other: flex32) -> flex32

The positive difference of two numbers.

fn cbrt(self) -> flex32

Take the cubic root of a number.

fn hypot(self, other: flex32) -> flex32

Calculate the length of the hypotenuse of a right-angle triangle given legs of length x and y.

fn sin(self) -> flex32

Computes the sine of a number (in radians).

fn cos(self) -> flex32

Computes the cosine of a number (in radians).

fn tan(self) -> flex32

Computes the tangent of a number (in radians).

fn asin(self) -> flex32

Computes the arcsine of a number. Return value is in radians in the range [-pi/2, pi/2] or NaN if the number is outside the range [-1, 1].

fn acos(self) -> flex32

Computes the arccosine of a number. Return value is in radians in the range [0, pi] or NaN if the number is outside the range [-1, 1].

fn atan(self) -> flex32

Computes the arctangent of a number. Return value is in radians in the range [-pi/2, pi/2];

fn atan2(self, other: flex32) -> flex32

Computes the four quadrant arctangent of self (y) and other (x).

fn sin_cos(self) -> (flex32, flex32)

Simultaneously computes the sine and cosine of the number, x. Returns (sin(x), cos(x)).

fn exp_m1(self) -> flex32

Returns e^(self) - 1 in a way that is accurate even if the number is close to zero.

fn ln_1p(self) -> flex32

Returns ln(1+n) (natural logarithm) more accurately than if the operations were performed separately.

fn sinh(self) -> flex32

Hyperbolic sine function.

fn cosh(self) -> flex32

Hyperbolic cosine function.

fn tanh(self) -> flex32

Hyperbolic tangent function.

fn asinh(self) -> flex32

Inverse hyperbolic sine function.

fn acosh(self) -> flex32

Inverse hyperbolic cosine function.

fn atanh(self) -> flex32

Inverse hyperbolic tangent function.

fn integer_decode(self) -> (u64, i16, i8)

Returns the mantissa, base 2 exponent, and sign as integers, respectively. The original number can be recovered by sign * mantissa * 2 ^ exponent.

fn epsilon() -> Self

Returns epsilon, a small positive value.

fn is_subnormal(self) -> bool

Returns true if the number is subnormal.

fn to_degrees(self) -> Self

Converts radians to degrees.

fn to_radians(self) -> Self

Converts degrees to radians.

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

Clamps a value between a min and max.

fn copysign(self, sign: Self) -> Self

Returns a number composed of the magnitude of self and the sign of sign.

impl Floor for flex32

fn floor(x: Self) -> Self

fn __expand_floor( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl From<f32> for flex32

fn from(value: f32) -> flex32

Converts to this type from the input type.

impl From<flex32> for ExpandElement

fn from(value: flex32) -> ExpandElement

Converts to this type from the input type.

impl From<flex32> for ExpandElementTyped<flex32>

fn from(value: flex32) -> ExpandElementTyped<flex32>

Converts to this type from the input type.

impl From<flex32> for Variable

fn from(value: flex32) -> Variable

Converts to this type from the input type.

impl From<flex32> for f32

fn from(val: flex32) -> f32

Converts to this type from the input type.

impl IntoRuntime for flex32

fn __expand_runtime_method( self, context: &mut CubeContext, ) -> ExpandElementTyped<flex32>

fn runtime(self) -> Self

Make sure a type is actually expanded into its runtime expand type.

impl LaunchArgExpand for flex32

type CompilationArg = ()

Compilation argument.

fn expand( _: &<flex32 as LaunchArgExpand>::CompilationArg, builder: &mut KernelBuilder, ) -> ExpandElementTyped<flex32>

Register an input variable during compilation that fill the [KernelBuilder].

fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> Self::ExpandType

Register an output variable during compilation that fill the [KernelBuilder].

impl Log for flex32

fn log(x: Self) -> Self

fn __expand_log( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Log1p for flex32

fn log1p(x: Self) -> Self

fn __expand_log1p( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Magnitude for flex32

fn magnitude(x: Self) -> Self

fn __expand_magnitude( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Max for flex32

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

fn __expand_max( context: &mut CubeContext, lhs: ExpandElementTyped<Self>, rhs: ExpandElementTyped<Self>, ) -> ExpandElementTyped<Self>

impl Min for flex32

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

fn __expand_min( context: &mut CubeContext, lhs: ExpandElementTyped<Self>, rhs: ExpandElementTyped<Self>, ) -> ExpandElementTyped<Self>

impl<__RhsT> Mul<__RhsT> for flex32

where f32: Mul<__RhsT, Output = f32>,

type Output = flex32

The resulting type after applying the * operator.

fn mul(self, rhs: __RhsT) -> flex32

Performs the * operation.

impl Mul for flex32

type Output = flex32

The resulting type after applying the * operator.

fn mul(self, rhs: flex32) -> <flex32 as Mul>::Output

Performs the * operation.

impl<__RhsT> MulAssign<__RhsT> for flex32

where f32: MulAssign<__RhsT>,

fn mul_assign(&mut self, rhs: __RhsT)

Performs the *= operation.

impl MulAssign for flex32

fn mul_assign(&mut self, rhs: flex32)

Performs the *= operation.

impl Neg for flex32

type Output = flex32

The resulting type after applying the - operator.

fn neg(self) -> flex32

Performs the unary - operation.

impl Normalize for flex32

fn normalize(x: Self) -> Self

fn __expand_normalize( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Num for flex32

type FromStrRadixErr = <f32 as Num>::FromStrRadixErr

fn from_str_radix( str: &str, radix: u32, ) -> Result<flex32, <flex32 as Num>::FromStrRadixErr>

Convert from a string and radix (typically 2..=36).

impl NumCast for flex32

fn from<T>(n: T) -> Option<flex32>

where T: ToPrimitive,

Creates a number from another value that can be converted into a primitive via the ToPrimitive trait. If the source value cannot be represented by the target type, then None is returned.

impl Numeric for flex32

fn min_value() -> flex32

fn max_value() -> flex32

fn __expand_min_value(context: &mut CubeContext) -> Self::ExpandType

fn __expand_max_value(context: &mut CubeContext) -> Self::ExpandType

fn from_int(val: i64) -> Self

Create a new constant numeric.

fn from_vec<const D: usize>(_vec: [u32; D]) -> Self

fn __expand_from_int( context: &mut CubeContext, val: ExpandElementTyped<i64>, ) -> Self::ExpandType

fn __expand_from_vec<const D: usize>( context: &mut CubeContext, vec: [u32; D], ) -> Self::ExpandType

impl One for flex32

fn one() -> flex32

Returns the multiplicative identity element of Self, 1.

fn set_one(&mut self)

Sets self to the multiplicative identity element of Self, 1.

fn is_one(&self) -> bool

where Self: PartialEq,

Returns true if self is equal to the multiplicative identity.

impl PartialEq for flex32

fn eq(&self, other: &flex32) -> 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 flex32

fn partial_cmp(&self, other: &flex32) -> 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 Powf for flex32

fn powf(self, _rhs: Self) -> Self

fn __expand_powf( context: &mut CubeContext, lhs: ExpandElementTyped<Self>, rhs: ExpandElementTyped<Self>, ) -> ExpandElementTyped<Self>

impl Recip for flex32

fn recip(x: Self) -> Self

fn __expand_recip( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl<__RhsT> Rem<__RhsT> for flex32

where f32: Rem<__RhsT, Output = f32>,

type Output = flex32

The resulting type after applying the % operator.

fn rem(self, rhs: __RhsT) -> flex32

Performs the % operation.

impl Rem for flex32

type Output = flex32

The resulting type after applying the % operator.

fn rem(self, rhs: flex32) -> <flex32 as Rem>::Output

Performs the % operation.

impl<__RhsT> RemAssign<__RhsT> for flex32

where f32: RemAssign<__RhsT>,

fn rem_assign(&mut self, rhs: __RhsT)

Performs the %= operation.

impl RemAssign for flex32

fn rem_assign(&mut self, rhs: flex32)

Performs the %= operation.

impl Remainder for flex32

fn rem(self, _rhs: Self) -> Self

fn __expand_rem( context: &mut CubeContext, lhs: ExpandElementTyped<Self>, rhs: ExpandElementTyped<Self>, ) -> ExpandElementTyped<Self>

impl Round for flex32

fn round(x: Self) -> Self

fn __expand_round( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl ScalarArgSettings for flex32

fn register<R>(&self, settings: &mut KernelLauncher<R>)

where R: Runtime,

Register the information to the [KernelLauncher].

impl Serialize for flex32

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Sin for flex32

fn sin(x: Self) -> Self

fn __expand_sin( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Sqrt for flex32

fn sqrt(x: Self) -> Self

fn __expand_sqrt( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl Sub for flex32

type Output = flex32

The resulting type after applying the - operator.

fn sub(self, rhs: flex32) -> flex32

Performs the - operation.

impl SubAssign for flex32

fn sub_assign(&mut self, rhs: flex32)

Performs the -= operation.

impl Tanh for flex32

fn tanh(x: Self) -> Self

fn __expand_tanh( context: &mut CubeContext, x: Self::ExpandType, ) -> ExpandElementTyped<Self>

impl ToElement for flex32

fn to_i64(&self) -> i64

Converts the value of self to an i64.

fn to_u64(&self) -> u64

Converts the value of self to a u64.

fn to_i8(&self) -> i8

Converts the value of self to an i8.

fn to_u8(&self) -> u8

Converts the value of self to a u8.

fn to_i16(&self) -> i16

Converts the value of self to an i16.

fn to_u16(&self) -> u16

Converts the value of self to a u16.

fn to_i32(&self) -> i32

Converts the value of self to an i32.

fn to_u32(&self) -> u32

Converts the value of self to a u32.

fn to_f32(&self) -> f32

Converts the value of self to an f32. Overflows may map to positive or negative infinity.

fn to_f64(&self) -> f64

Converts the value of self to an f64. Overflows may map to positive or negative infinity.

fn to_isize(&self) -> isize

Converts the value of self to an isize.

fn to_i128(&self) -> i128

Converts the value of self to an i128.

fn to_usize(&self) -> usize

Converts the value of self to a usize.

fn to_u128(&self) -> u128

Converts the value of self to a u128.

impl ToPrimitive for flex32

fn to_i64(&self) -> Option<i64>

Converts the value of self to an i64. If the value cannot be represented by an i64, then None is returned.

fn to_u64(&self) -> Option<u64>

Converts the value of self to a u64. If the value cannot be represented by a u64, then None is returned.

fn to_f32(&self) -> Option<f32>

Converts the value of self to an f32. Overflows may map to positive or negative inifinity, otherwise None is returned if the value cannot be represented by an f32.

fn to_f64(&self) -> Option<f64>

Converts the value of self to an f64. Overflows may map to positive or negative inifinity, otherwise None is returned if the value cannot be represented by an f64.

fn to_isize(&self) -> Option<isize>

Converts the value of self to an isize. If the value cannot be represented by an isize, then None is returned.

fn to_i8(&self) -> Option<i8>

Converts the value of self to an i8. If the value cannot be represented by an i8, then None is returned.

fn to_i16(&self) -> Option<i16>

Converts the value of self to an i16. If the value cannot be represented by an i16, then None is returned.

fn to_i32(&self) -> Option<i32>

Converts the value of self to an i32. If the value cannot be represented by an i32, then None is returned.

fn to_i128(&self) -> Option<i128>

Converts the value of self to an i128. If the value cannot be represented by an i128 (i64 under the default implementation), then None is returned.

fn to_usize(&self) -> Option<usize>

Converts the value of self to a usize. If the value cannot be represented by a usize, then None is returned.

fn to_u8(&self) -> Option<u8>

Converts the value of self to a u8. If the value cannot be represented by a u8, then None is returned.

fn to_u16(&self) -> Option<u16>

Converts the value of self to a u16. If the value cannot be represented by a u16, then None is returned.

fn to_u32(&self) -> Option<u32>

Converts the value of self to a u32. If the value cannot be represented by a u32, then None is returned.

fn to_u128(&self) -> Option<u128>

Converts the value of self to a u128. If the value cannot be represented by a u128 (u64 under the default implementation), then None is returned.

impl Zero for flex32

fn zero() -> flex32

Returns the additive identity element of Self, 0.

fn is_zero(&self) -> bool

Returns true if self is equal to the additive identity.

fn set_zero(&mut self)

Sets self to the additive identity element of Self, 0.

impl Zeroable for flex32

fn zeroed() -> Self

Calls zeroed.

impl Copy for flex32

impl FloatElement for flex32

impl JitElement for flex32

impl Pod for flex32

impl StructuralPartialEq for flex32