Struct BarrierExpand

pub struct BarrierExpand<C: CubePrimitive> { /* private fields */ }

Expand type of Barrier

Implementations

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_2d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, y: ExpandElementTyped<i32>, x: ExpandElementTyped<i32>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_3d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, z: ExpandElementTyped<i32>, y: ExpandElementTyped<i32>, x: ExpandElementTyped<i32>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_4d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, w: ExpandElementTyped<i32>, z: ExpandElementTyped<i32>, y: ExpandElementTyped<i32>, x: ExpandElementTyped<i32>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_5d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, v: ExpandElementTyped<i32>, w: ExpandElementTyped<i32>, z: ExpandElementTyped<i32>, y: ExpandElementTyped<i32>, x: ExpandElementTyped<i32>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_im2col_3d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, n: ExpandElementTyped<i32>, w: ExpandElementTyped<i32>, c: ExpandElementTyped<i32>, w_offset: ExpandElementTyped<u16>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_im2col_4d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, n: ExpandElementTyped<i32>, h: ExpandElementTyped<i32>, w: ExpandElementTyped<i32>, c: ExpandElementTyped<i32>, h_offset: ExpandElementTyped<u16>, w_offset: ExpandElementTyped<u16>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_tma_load_im2col_5d_method( &self, scope: &mut Scope, source: ExpandElementTyped<TensorMap<C>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, n: ExpandElementTyped<i32>, d: ExpandElementTyped<i32>, h: ExpandElementTyped<i32>, w: ExpandElementTyped<i32>, c: ExpandElementTyped<i32>, d_offset: ExpandElementTyped<u16>, h_offset: ExpandElementTyped<u16>, w_offset: ExpandElementTyped<u16>, )

impl<C: CubePrimitive> BarrierExpand<C>

pub fn __expand_memcpy_async_method( &self, scope: &mut Scope, source: ExpandElementTyped<Slice<Line<C>>>, destination: ExpandElementTyped<SliceMut<Line<C>>>, )

pub fn __expand_arrive_method(&self, scope: &mut Scope)

pub fn __expand_arrive_tx_method( &self, scope: &mut Scope, arrival_count: ExpandElementTyped<u32>, transaction_count: ExpandElementTyped<u32>, )

pub fn __expand_expect_tx_method( &self, scope: &mut Scope, transaction_count: ExpandElementTyped<u32>, )

pub fn __expand_wait_method(&self, scope: &mut Scope)

pub fn __expand_arrive_and_wait_method(&self, scope: &mut Scope)

Trait Implementations

impl<C: Clone + CubePrimitive> Clone for BarrierExpand<C>

fn clone(&self) -> BarrierExpand<C>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<C: CubePrimitive> CubeDebug for BarrierExpand<C>

fn set_debug_name(&self, scope: &mut Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl<C: CubePrimitive> Init for BarrierExpand<C>

fn init(self, _scope: &mut Scope) -> Self

Initialize a type within a scope.