Enum BarrierOps 

pub enum BarrierOps {
    Declare {
        barrier: Variable,
    },
    Init {
        barrier: Variable,
        is_elected: Variable,
        arrival_count: Variable,
    },
    InitManual {
        barrier: Variable,
        arrival_count: Variable,
    },
    MemCopyAsync {
        barrier: Variable,
        source: Variable,
        source_length: Variable,
        offset_source: Variable,
        offset_out: Variable,
    },
    MemCopyAsyncCooperative {
        barrier: Variable,
        source: Variable,
        source_length: Variable,
        offset_source: Variable,
        offset_out: Variable,
    },
    MemCopyAsyncTx {
        barrier: Variable,
        source: Variable,
        source_length: Variable,
        offset_source: Variable,
        offset_out: Variable,
    },
    CopyAsync {
        source: Variable,
        source_length: Variable,
        offset_source: Variable,
        offset_out: Variable,
        copy_length: u32,
        checked: bool,
    },
    TmaLoad {
        barrier: Variable,
        tensor_map: Variable,
        indices: Vec<Variable>,
        offset_out: Variable,
    },
    TmaLoadIm2col {
        barrier: Variable,
        tensor_map: Variable,
        indices: Vec<Variable>,
        offsets: Vec<Variable>,
        offset_out: Variable,
    },
    Arrive {
        barrier: Variable,
    },
    ArriveTx {
        barrier: Variable,
        arrive_count_update: Variable,
        transaction_count_update: Variable,
    },
    CommitCopyAsync {
        barrier: Variable,
    },
    ExpectTx {
        barrier: Variable,
        transaction_count_update: Variable,
    },
    Wait {
        barrier: Variable,
        token: Variable,
    },
    WaitParity {
        barrier: Variable,
        phase: Variable,
    },
    ArriveAndWait {
        barrier: Variable,
    },
}

Operations available on a barrier

Variants

Declare

Declare the barrier, without doing any initialization

Fields

barrier: Variable

Init

Initialize the barrier, optionally with a cta proxy fence

Fields

barrier: Variable

is_elected: Variable

arrival_count: Variable

InitManual

Manually initialize the barrier with an arrival count, without any sync or election handling

Fields

barrier: Variable

arrival_count: Variable

MemCopyAsync

Copy source to destination

Fields

barrier: Variable

source: Variable

source_length: Variable

offset_source: Variable

offset_out: Variable

MemCopyAsyncCooperative

Copy source to destination, with cooperative behaviour

Fields

barrier: Variable

source: Variable

source_length: Variable

offset_source: Variable

offset_out: Variable

MemCopyAsyncTx

Copy source to destination, with transaction count

Fields

barrier: Variable

source: Variable

source_length: Variable

offset_source: Variable

offset_out: Variable

CopyAsync

Copy source to destination

Fields

source: Variable

source_length: Variable

offset_source: Variable

offset_out: Variable

copy_length: u32

checked: bool

TmaLoad

Fields

barrier: Variable

tensor_map: Variable

indices: Vec<Variable>

offset_out: Variable

TmaLoadIm2col

Fields

barrier: Variable

tensor_map: Variable

indices: Vec<Variable>

offsets: Vec<Variable>

offset_out: Variable

Arrive

Arrives at the barrier (decrements barrier count)

Fields

barrier: Variable

ArriveTx

Fields

barrier: Variable

arrive_count_update: Variable

transaction_count_update: Variable

CommitCopyAsync

Fields

barrier: Variable

ExpectTx

Fields

barrier: Variable

transaction_count_update: Variable

Wait

Fields

barrier: Variable

token: Variable

WaitParity

Fields

barrier: Variable

phase: Variable

ArriveAndWait

Waits until data is loaded

Fields

barrier: Variable

Trait Implementations

impl Clone for BarrierOps

fn clone(&self) -> BarrierOps

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BarrierOps

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for BarrierOps

fn deserialize<__D>( __deserializer: __D, ) -> Result<BarrierOps, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for BarrierOps

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

Formats the value using the given formatter.

impl From<<BarrierOps as OperationReflect>::OpCode> for OpCode

fn from(value: <BarrierOps as OperationReflect>::OpCode) -> OpCode

Converts to this type from the input type.

impl From<BarrierOps> for Instruction

fn from(value: BarrierOps) -> Instruction

Converts to this type from the input type.

impl From<BarrierOps> for Operation

fn from(value: BarrierOps) -> Operation

Converts to this type from the input type.

impl Hash for BarrierOps

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

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 OperationReflect for BarrierOps

type OpCode = BarrierOpCode

Type of the op codes for this operation

fn op_code(&self) -> <BarrierOps as OperationReflect>::OpCode

Get the opcode for this operation

fn args(&self) -> Option<Vec<Variable>>

Get the list of arguments for this operation. If not all arguments are Variable, returns None instead.

fn from_code_and_args( op_code: <BarrierOps as OperationReflect>::OpCode, args: &[Variable], ) -> Option<BarrierOps>

Create typed operation from an opcode and a list of arguments. Returns None if not all arguments are Variable.

fn is_commutative(&self) -> bool

Whether this operation is commutative (arguments can be freely reordered). Ignored for single argument operations.

fn is_pure(&self) -> bool

Whether this operation is pure (has no side effects). Things like uniform/plane operations are considered impure, because they affect other units.

impl PartialEq for BarrierOps

fn eq(&self, other: &BarrierOps) -> 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 Serialize for BarrierOps

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl TypeHash for BarrierOps

fn write_hash(hasher: &mut impl Hasher)

Write the structure of the type to the hasher

fn type_hash() -> u64

Generate a stable hash of the type structure.

impl Eq for BarrierOps

impl StructuralPartialEq for BarrierOps