Enum BarrierOps 

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

Operations available on a barrier

Variants

Init

Initialize the barrier, optionally with a cta proxy fence

Fields

barrier: Value

is_elected: Value

arrival_count: Value

InitManual

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

Fields

barrier: Value

arrival_count: Value

MemCopyAsync

Copy source to destination

Fields

barrier: Value

source: Value

destination: Value

source_length: Value

MemCopyAsyncCooperative

Copy source to destination, with cooperative behaviour

Fields

barrier: Value

source: Value

destination: Value

source_length: Value

MemCopyAsyncTx

Copy source to destination, with transaction count

Fields

barrier: Value

source: Value

destination: Value

source_length: Value

CopyAsync

Copy source to destination

Fields

source: Value

destination: Value

source_length: Value

copy_length: u32

checked: bool

TmaLoad

Fields

barrier: Value

tensor_map: Value

destination: Value

indices: Vec<Value>

TmaLoadIm2col

Fields

barrier: Value

tensor_map: Value

destination: Value

indices: Vec<Value>

offsets: Vec<Value>

Arrive

Arrives at the barrier (decrements barrier count)

Fields

barrier: Value

ArriveTx

Fields

barrier: Value

arrive_count_update: Value

transaction_count_update: Value

CommitCopyAsync

Fields

barrier: Value

ExpectTx

Fields

barrier: Value

transaction_count_update: Value

Wait

Fields

barrier: Value

token: Value

WaitParity

Fields

barrier: Value

phase: Value

ArriveAndWait

Waits until data is loaded

Fields

barrier: Value

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

Formats the value using the given formatter.

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

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for BarrierOps

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

Formats the value using the given formatter.

impl Eq for BarrierOps

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

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

Converts to this type from the input type.

impl From<BarrierOps> for Instruction

fn from(value: BarrierOps) -> Self

Converts to this type from the input type.

impl From<BarrierOps> for Operation

fn from(value: BarrierOps) -> Self

Converts to this type from the input type.

impl Hash for BarrierOps

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

type OpCode = BarrierOpCode

Type of the op codes for this operation

fn op_code(&self) -> Self::OpCode

Get the opcode for this operation

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

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

fn args_mut(&mut self) -> Option<Vec<&mut Value>>

fn from_code_and_args(op_code: Self::OpCode, args: &[Value]) -> Option<Self>

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

fn sanitize_args(&mut self, scope: &Scope)

Sanitize args, i.e. loading inputs from pointers for ops that take values

fn read_pointers(&self) -> Vec<Value>

fn write_pointers(&self) -> Vec<Value>

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

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl Serialize for BarrierOps

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for BarrierOps

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.