Skip to main content

CubePrimitive

cubecl_core::frontend

Trait CubePrimitive

pub trait CubePrimitive:
    CubeType<ExpandType = NativeExpand<Self>>
    + NativeAssign
    + CanReadValue
    + CubeDebug
    + Send
    + Sync
    + 'static
    + Clone
    + Copy {
    type Scalar: Scalar;
    type Size: Size;
    type WithScalar<S: Scalar>: CubePrimitive;

Show 13 methods    // Required methods
    fn from_const_value(value: ConstantValue) -> Self;
    fn __expand_as_type(scope: &Scope) -> TypeHandle;

    // Provided methods
    fn as_type() -> TypeHandle { ... }
    fn size() -> usize { ... }
    fn size_bits() -> usize { ... }
    fn from_expand_elem(elem: ExpandValue) -> Self::ExpandType { ... }
    fn into_lit_unchecked(self) -> Self { ... }
    fn packing_factor() -> usize { ... }
    fn vector_size() -> usize { ... }
    fn __expand_size(scope: &Scope) -> usize { ... }
    fn __expand_size_bits(scope: &Scope) -> usize { ... }
    fn __expand_packing_factor(scope: &Scope) -> usize { ... }
    fn __expand_vector_size(scope: &Scope) -> usize { ... }
}

Expand description

Form of CubeType that encapsulates all primitive types: Numeric, UInt, Bool

Required Associated Types§

type Scalar: Scalar

type Size: Size

type WithScalar<S: Scalar>: CubePrimitive

Required Methods§

fn from_const_value(value: ConstantValue) -> Self

fn __expand_as_type(scope: &Scope) -> TypeHandle

Provided Methods§

fn as_type() -> TypeHandle

Return the element type to use on GPU.

fn size() -> usize

Only native element types have a size.

fn size_bits() -> usize

Only native element types have a size.

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

fn into_lit_unchecked(self) -> Self

fn packing_factor() -> usize

fn vector_size() -> usize

fn __expand_size(scope: &Scope) -> usize

fn __expand_size_bits(scope: &Scope) -> usize

fn __expand_packing_factor(scope: &Scope) -> usize

fn __expand_vector_size(scope: &Scope) -> usize

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementations on Foreign Types§

impl CubePrimitive for Complex<f32>

type Scalar = Complex<f32>

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

fn __expand_as_type(scope: &Scope) -> TypeHandle

impl CubePrimitive for Complex<f64>

type Scalar = Complex<f64>

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

fn __expand_as_type(scope: &Scope) -> TypeHandle

impl CubePrimitive for bf16

fn __expand_as_type(scope: &Scope) -> TypeHandle

Return the element type to use on GPU

type Scalar = bf16

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for bool

type Scalar = bool

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

fn __expand_as_type(scope: &Scope) -> TypeHandle

impl CubePrimitive for f16

fn __expand_as_type(scope: &Scope) -> TypeHandle

Return the element type to use on GPU

type Scalar = f16

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for f32

fn __expand_as_type(scope: &Scope) -> TypeHandle

Return the element type to use on GPU

type Scalar = f32

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for f64

fn __expand_as_type(scope: &Scope) -> TypeHandle

Return the element type to use on GPU

type Scalar = f64

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for i8

type Scalar = i8

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for i16

type Scalar = i16

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for i32

type Scalar = i32

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for i64

type Scalar = i64

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for isize

type Scalar = isize

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

fn __expand_as_type(scope: &Scope) -> TypeHandle

impl CubePrimitive for u8

type Scalar = u8

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for u16

type Scalar = u16

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for u32

type Scalar = u32

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for u64

type Scalar = u64

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn __expand_as_type(scope: &Scope) -> TypeHandle

fn from_const_value(value: ConstantValue) -> Self

impl CubePrimitive for usize

type Scalar = usize

type Size = Const<1>

type WithScalar<S: Scalar> = S

fn from_const_value(value: ConstantValue) -> Self

fn __expand_as_type(scope: &Scope) -> TypeHandle

Implementors§

impl CubePrimitive for Barrier

type Scalar = u32

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for TensorLayout

type Scalar = u32

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for TensorReinterpret

type Scalar = u32

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for e2m1

type Scalar = e2m1

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for e2m3

type Scalar = e2m3

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for e2m1x2

type Scalar = e2m1x2

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for e3m2

type Scalar = e3m2

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for e4m3

type Scalar = e4m3

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for e5m2

type Scalar = e5m2

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for flex32

type Scalar = flex32

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for tf32

type Scalar = tf32

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl CubePrimitive for ue8m0

type Scalar = ue8m0

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl<Inner: CubePrimitive> CubePrimitive for Atomic<Inner>

type Scalar = <Inner as CubePrimitive>::Scalar

type Size = Const<1>

type WithScalar<S: Scalar> = Atomic<S>

impl<Marker: 'static> CubePrimitive for DynamicScalar<Marker>

type Scalar = DynamicScalar<Marker>

type Size = Const<1>

type WithScalar<S: Scalar> = S

impl<P: Scalar, N: Size> CubePrimitive for Vector<P, N>

type Scalar = P

type Size = N

type WithScalar<S: Scalar> = Vector<S, N>