Trait CubeIndex

pub trait CubeIndex<T: Index> {
    type Output: CubeType;

    // Provided method
    fn cube_idx(&self, _i: T) -> &Self::Output { ... }
}

Fake indexation so we can rewrite indexes into scalars as calls to this fake function in the non-expanded function

Required Associated Types

type Output: CubeType

Provided Methods

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

Implementations on Foreign Types

impl<I: Index> CubeIndex<I> for f32

type Output = f32

impl<I: Index> CubeIndex<I> for f64

type Output = f64

impl<I: Index> CubeIndex<I> for i8

type Output = i8

impl<I: Index> CubeIndex<I> for i16

type Output = i16

impl<I: Index> CubeIndex<I> for i32

type Output = i32

impl<I: Index> CubeIndex<I> for i64

type Output = i64

impl<I: Index> CubeIndex<I> for u8

type Output = u8

impl<I: Index> CubeIndex<I> for u16

type Output = u16

impl<I: Index> CubeIndex<I> for u32

type Output = u32

impl<I: Index> CubeIndex<I> for u64

type Output = u64

impl<I: Index> CubeIndex<I> for bf16

type Output = bf16

impl<I: Index> CubeIndex<I> for f16

type Output = f16

Implementors

impl<E: CubeType, I: Index> CubeIndex<I> for Array<E>

type Output = E

impl<E: CubeType, I: Index> CubeIndex<I> for SharedMemory<E>

type Output = E

impl<E: CubeType, I: Index> CubeIndex<I> for Slice<E>

type Output = E

impl<E: CubeType, I: Index> CubeIndex<I> for SliceMut<E>

type Output = E

impl<E: CubeType, I: Index> CubeIndex<I> for Tensor<E>

type Output = E

impl<I: Index> CubeIndex<I> for flex32

type Output = flex32

impl<I: Index> CubeIndex<I> for tf32

type Output = tf32

impl<P> CubeIndex<u32> for Line<P>

where P: CubePrimitive,

type Output = P

impl<P> CubeIndex<ExpandElementTyped<u32>> for Line<P>

where P: CubePrimitive,

type Output = P

impl<T: Index, const POS: u8> CubeIndex<T> for FloatExpand<POS>

type Output = FloatExpand<POS>

impl<T: Index, const POS: u8> CubeIndex<T> for IntExpand<POS>

type Output = IntExpand<POS>