Struct cubecl_core::frontend::UInt

pub struct UInt {
    pub val: u32,
    pub vectorization: u8,
}

An unsigned int. Preferred for indexing operations

Fields

val: u32

vectorization: u8

Implementations

impl UInt

pub const fn new(val: u32) -> Self

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

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

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

Trait Implementations

impl Abs for UInt

fn abs(x: Self) -> Self

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

impl Add<u32> for UInt

type Output = UInt

The resulting type after applying the + operator.

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

Performs the + operation.

impl Add for UInt

type Output = UInt

The resulting type after applying the + operator.

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

Performs the + operation.

impl AddAssign<u32> for UInt

fn add_assign(&mut self, _rhs: u32)

Performs the += operation.

impl AddAssign for UInt

fn add_assign(&mut self, _rhs: Self)

Performs the += operation.

impl BitAnd for UInt

type Output = UInt

The resulting type after applying the & operator.

fn bitand(self, _rhs: Self) -> Self::Output

Performs the & operation.

impl BitXor for UInt

type Output = UInt

The resulting type after applying the ^ operator.

fn bitxor(self, _rhs: Self) -> Self::Output

Performs the ^ operation.

impl Clamp for UInt

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 UInt

fn clone(&self) -> UInt

Returns a copy of the value.

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

Performs copy-assignment from source.

impl ComptimeType for UInt

fn into_expand(self) -> Self::ExpandType

Create the expand type from the normal type.

impl CubePrimitive for UInt

fn as_elem() -> Elem

Return the element type to use on GPU

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

impl CubeType for UInt

type ExpandType = ExpandElementTyped<UInt>

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

Wrapper around the init method, necessary to type inference.

impl Debug for UInt

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

Formats the value using the given formatter.

impl Div<u32> for UInt

type Output = UInt

The resulting type after applying the / operator.

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

Performs the / operation.

impl Div for UInt

type Output = UInt

The resulting type after applying the / operator.

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

Performs the / operation.

impl DivAssign<u32> for UInt

fn div_assign(&mut self, _rhs: u32)

Performs the /= operation.

impl DivAssign for UInt

fn div_assign(&mut self, _rhs: Self)

Performs the /= operation.

impl ExpandElementBaseInit for UInt

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

impl From<Comptime<u32>> for UInt

fn from(value: Comptime<u32>) -> Self

Converts to this type from the input type.

impl From<UInt> for ExpandElement

Useful for Comptime

fn from(value: UInt) -> Self

Converts to this type from the input type.

impl From<UInt> for ExpandElementTyped<UInt>

fn from(value: UInt) -> Self

Converts to this type from the input type.

impl From<i32> for UInt

fn from(value: i32) -> Self

Converts to this type from the input type.

impl From<u32> for UInt

fn from(value: u32) -> Self

Converts to this type from the input type.

impl From<usize> for UInt

fn from(value: usize) -> Self

Converts to this type from the input type.

impl Hash for UInt

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 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<UInt> for F16

type Output = F16

The returned type after indexing.

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

Performs the indexing (container[index]) operation.

impl Index<UInt> for F32

type Output = F32

The returned type after indexing.

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

Performs the indexing (container[index]) operation.

impl Index<UInt> for F64

type Output = F64

The returned type after indexing.

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

Performs the indexing (container[index]) operation.

impl Index<UInt> for I32

type Output = I32

The returned type after indexing.

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

Performs the indexing (container[index]) operation.

impl Index<UInt> for I64

type Output = I64

The returned type after indexing.

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

Performs the indexing (container[index]) operation.

impl Index<UInt> for UInt

type Output = UInt

The returned type after indexing.

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

Performs the indexing (container[index]) operation.

impl Index<u32> for UInt

type Output = UInt

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<UInt> for F16

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

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

impl IndexMut<UInt> for F32

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

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

impl IndexMut<UInt> for F64

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

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

impl IndexMut<UInt> for I32

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

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

impl IndexMut<UInt> for I64

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

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

impl IndexMut<UInt> for UInt

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

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

impl IndexMut<u32> for UInt

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

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

impl Init for UInt

fn init(self, _context: &mut CubeContext) -> Self

Initialize a type within a context.

impl LaunchArgExpand for UInt

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 Max for UInt

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

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

impl Min for UInt

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

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

impl Mul<u32> for UInt

type Output = UInt

The resulting type after applying the * operator.

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

Performs the * operation.

impl Mul for UInt

type Output = UInt

The resulting type after applying the * operator.

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

Performs the * operation.

impl MulAssign<u32> for UInt

fn mul_assign(&mut self, _rhs: u32)

Performs the *= operation.

impl MulAssign for UInt

fn mul_assign(&mut self, _rhs: Self)

Performs the *= operation.

impl Numeric for UInt

type Primitive = u32

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<u32> for UInt

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 UInt

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<u32> for UInt

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 UInt

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 Rem for UInt

type Output = UInt

The resulting type after applying the % operator.

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

Performs the % operation.

impl Remainder for UInt

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

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

impl Shl for UInt

type Output = UInt

The resulting type after applying the << operator.

fn shl(self, _rhs: Self) -> Self::Output

Performs the << operation.

impl Shr for UInt

type Output = UInt

The resulting type after applying the >> operator.

fn shr(self, _rhs: Self) -> Self::Output

Performs the >> operation.

impl Sub<u32> for UInt

type Output = UInt

The resulting type after applying the - operator.

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

Performs the - operation.

impl Sub for UInt

type Output = UInt

The resulting type after applying the - operator.

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

Performs the - operation.

impl SubAssign<u32> for UInt

fn sub_assign(&mut self, _rhs: u32)

Performs the -= operation.

impl SubAssign for UInt

fn sub_assign(&mut self, _rhs: Self)

Performs the -= operation.

impl Vectorized for UInt

fn vectorization_factor(&self) -> UInt

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

impl Copy for UInt

impl Eq for UInt