Struct cubecl_core::frontend::BF16

pub struct BF16 {
    pub val: f32,
    pub vectorization: u8,
}

Fields

val: f32

vectorization: u8

Trait Implementations

impl Abs for BF16

fn abs(x: Self) -> Self

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

impl Add<f32> for BF16

type Output = BF16

The resulting type after applying the + operator.

fn add(self, rhs: f32) -> Self::Output

Performs the + operation.

impl Add<u32> for BF16

type Output = BF16

The resulting type after applying the + operator.

fn add(self, rhs: u32) -> Self::Output

Performs the + operation.

impl Add for BF16

type Output = BF16

The resulting type after applying the + operator.

fn add(self, rhs: Self) -> Self::Output

Performs the + operation.

impl AddAssign<f32> for BF16

fn add_assign(&mut self, _rhs: f32)

Performs the += operation.

impl AddAssign<u32> for BF16

fn add_assign(&mut self, _rhs: u32)

Performs the += operation.

impl AddAssign for BF16

fn add_assign(&mut self, _rhs: Self)

Performs the += operation.

impl Ceil for BF16

fn ceil(x: Self) -> Self

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

impl Clamp for BF16

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 BF16

fn clone(&self) -> BF16

Returns a copy of the value.

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

Performs copy-assignment from source.

impl ComptimeType for BF16

fn into_expand(self) -> Self::ExpandType

Create the expand type from the normal type.

impl Cos for BF16

fn cos(x: Self) -> Self

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

impl CubePrimitive for BF16

fn as_elem() -> Elem

Return the element type to use on GPU

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

impl CubeType for BF16

type ExpandType = ExpandElementTyped<BF16>

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

Wrapper around the init method, necessary to type inference.

impl Div<f32> for BF16

type Output = BF16

The resulting type after applying the / operator.

fn div(self, rhs: f32) -> Self::Output

Performs the / operation.

impl Div<u32> for BF16

type Output = BF16

The resulting type after applying the / operator.

fn div(self, rhs: u32) -> Self::Output

Performs the / operation.

impl Div for BF16

type Output = BF16

The resulting type after applying the / operator.

fn div(self, rhs: Self) -> Self::Output

Performs the / operation.

impl DivAssign<f32> for BF16

fn div_assign(&mut self, _rhs: f32)

Performs the /= operation.

impl DivAssign<u32> for BF16

fn div_assign(&mut self, _rhs: u32)

Performs the /= operation.

impl DivAssign for BF16

fn div_assign(&mut self, _rhs: Self)

Performs the /= operation.

impl Erf for BF16

fn erf(x: Self) -> Self

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

impl Exp for BF16

fn exp(x: Self) -> Self

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

impl ExpandElementBaseInit for BF16

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

impl Float for BF16

fn new(val: f32) -> Self

fn vectorized(val: f32, vectorization: UInt) -> Self

fn vectorized_empty(vectorization: UInt) -> Self

fn __expand_vectorized_empty( context: &mut CubeContext, vectorization: UInt, ) -> <Self as CubeType>::ExpandType

fn __expand_new( context: &mut CubeContext, val: Self::ExpandType, ) -> <Self as CubeType>::ExpandType

fn __expand_vectorized( context: &mut CubeContext, val: Self::ExpandType, vectorization: UInt, ) -> <Self as CubeType>::ExpandType

impl Floor for BF16

fn floor(x: Self) -> Self

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

impl From<BF16> for ExpandElement

fn from(value: BF16) -> Self

Converts to this type from the input type.

impl From<f32> for BF16

fn from(value: f32) -> Self

Converts to this type from the input type.

impl From<u32> for BF16

fn from(val: u32) -> Self

Converts to this type from the input type.

impl Index<UInt> for BF16

type Output = BF16

The returned type after indexing.

fn index(&self, _index: UInt) -> &Self::Output

Performs the indexing (container[index]) operation.

impl Index<u32> for BF16

type Output = BF16

The returned type after indexing.

fn index(&self, _index: u32) -> &Self::Output

Performs the indexing (container[index]) operation.

impl IndexMut<UInt> for BF16

fn index_mut(&mut self, _index: UInt) -> &mut Self::Output

Performs the mutable indexing (container[index]) operation.

impl IndexMut<u32> for BF16

fn index_mut(&mut self, _index: u32) -> &mut Self::Output

Performs the mutable indexing (container[index]) operation.

impl LaunchArgExpand for BF16

fn expand( builder: &mut KernelBuilder, vectorization: Vectorization, ) -> ExpandElementTyped<Self>

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( builder: &mut KernelBuilder, vectorization: Vectorization, ) -> <Self as CubeType>::ExpandType

Register an output variable during compilation that fill the KernelBuilder.

impl Log for BF16

fn log(x: Self) -> Self

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

impl Log1p for BF16

fn log1p(x: Self) -> Self

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

impl Max for BF16

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

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

impl Min for BF16

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

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

impl Mul<f32> for BF16

type Output = BF16

The resulting type after applying the * operator.

fn mul(self, rhs: f32) -> Self::Output

Performs the * operation.

impl Mul<u32> for BF16

type Output = BF16

The resulting type after applying the * operator.

fn mul(self, rhs: u32) -> Self::Output

Performs the * operation.

impl Mul for BF16

type Output = BF16

The resulting type after applying the * operator.

fn mul(self, rhs: Self) -> Self::Output

Performs the * operation.

impl MulAssign<f32> for BF16

fn mul_assign(&mut self, _rhs: f32)

Performs the *= operation.

impl MulAssign<u32> for BF16

fn mul_assign(&mut self, _rhs: u32)

Performs the *= operation.

impl MulAssign for BF16

fn mul_assign(&mut self, _rhs: Self)

Performs the *= operation.

impl Numeric for BF16

type Primitive = bf16

fn from_int(_val: u32) -> 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 as CubeType>::ExpandType

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

impl PartialEq<f32> for BF16

fn eq(&self, rhs: &f32) -> 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<u32> for BF16

fn eq(&self, rhs: &u32) -> 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 BF16

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<f32> for BF16

fn partial_cmp(&self, rhs: &f32) -> 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 PartialOrd<u32> for BF16

fn partial_cmp(&self, rhs: &u32) -> 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 PartialOrd for BF16

fn partial_cmp(&self, other: &Self) -> 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 BF16

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

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

impl Recip for BF16

fn recip(x: Self) -> Self

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

impl Remainder for BF16

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

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

impl Sin for BF16

fn sin(x: Self) -> Self

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

impl Sqrt for BF16

fn sqrt(x: Self) -> Self

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

impl Sub<f32> for BF16

type Output = BF16

The resulting type after applying the - operator.

fn sub(self, rhs: f32) -> Self::Output

Performs the - operation.

impl Sub<u32> for BF16

type Output = BF16

The resulting type after applying the - operator.

fn sub(self, rhs: u32) -> Self::Output

Performs the - operation.

impl Sub for BF16

type Output = BF16

The resulting type after applying the - operator.

fn sub(self, rhs: Self) -> Self::Output

Performs the - operation.

impl SubAssign<f32> for BF16

fn sub_assign(&mut self, _rhs: f32)

Performs the -= operation.

impl SubAssign<u32> for BF16

fn sub_assign(&mut self, _rhs: u32)

Performs the -= operation.

impl SubAssign for BF16

fn sub_assign(&mut self, _rhs: Self)

Performs the -= operation.

impl Tanh for BF16

fn tanh(x: Self) -> Self

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

impl Vectorized for BF16

fn vectorization_factor(&self) -> UInt

fn vectorize(self, factor: UInt) -> Self

impl Copy for BF16

impl Eq for BF16