Enum BarrierOps

pub enum BarrierOps {
    Init {
        barrier: Variable,
        with_cta_fence: bool,
    },
    MemCopyAsync {
        barrier: Variable,
        source: Variable,
        source_length: Variable,
        offset_source: Variable,
        offset_out: Variable,
    },
    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,
    },
    ExpectTx {
        barrier: Variable,
        transaction_count_update: Variable,
    },
    Wait {
        barrier: Variable,
    },
    ArriveAndWait {
        barrier: Variable,
    },
}

Operations available on a barrier

Variants

Init

Initialize the barrier, optionally with a cta proxy fence

Fields

barrier: Variable

with_cta_fence: bool

MemCopyAsync

Copy source to destination

Fields

barrier: Variable

source: Variable

source_length: Variable

offset_source: Variable

offset_out: Variable

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

ExpectTx

Fields

barrier: Variable

transaction_count_update: Variable

Wait

Fields

barrier: 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

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 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<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: Self::OpCode, args: &[Variable]) -> Option<Self>

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::Ok, __S::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