Struct TmaLoadOp 

pub struct TmaLoadOp { /* private fields */ }

Attribute(s):

Note: Only attributes defined directly as part of this operation are listed here. There may be others, not listed here, defined by interface implementations.

Name	Static Name Identifier	Type
tma_load_rank	ATTR_KEY_TMA_LOAD_RANK	IndexAttr

Implementations

impl TmaLoadOp

pub fn get_attr_tma_load_rank<'a>( &self, ctx: &'a Context, ) -> Option<Ref<'a, IndexAttr>>

Get a Ref to the value of the attribute named tma_load_rank. The Ref is a borrow of the containing Operation object.

pub fn set_attr_tma_load_rank(&self, ctx: &Context, value: IndexAttr)

Set the value of the attribute named tma_load_rank.

impl TmaLoadOp

pub fn new( ctx: &mut Context, barrier: Value, tensor_map: Value, destination: Value, indices: Vec<Value>, ) -> Self

pub fn barrier(&self, ctx: &Context) -> Value

pub fn tensor_map(&self, ctx: &Context) -> Value

pub fn destination(&self, ctx: &Context) -> Value

pub fn indices(&self, ctx: &Context) -> Vec<Value>

pub fn rank(&self, ctx: &Context) -> usize

Trait Implementations

impl AtLeastNOpdsInterface<4> for TmaLoadOp

impl Clone for TmaLoadOp

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for TmaLoadOp

impl Eq for TmaLoadOp

impl Hash for TmaLoadOp

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 MemoryEffectsOp for TmaLoadOp

fn memory_effects(&self, ctx: &Context) -> Vec<MemoryEffect>

fn verify(_op: &dyn Op, _ctx: &Context) -> Result<()>

where Self: Sized,

fn has_effects(&self, ctx: &Context) -> bool

impl Op for TmaLoadOp

fn get_operation(&self) -> Ptr<Operation>

Get the underlying IR Operation

fn get_opid(&self) -> OpId

Get this Op’s OpId

fn get_opid_static() -> OpId

Get this Op’s OpId, without self reference.

fn get_concrete_op_info() -> (fn(Ptr<Operation>) -> OpBox, TypeId)

where Self: Sized,

Get details about the concrete Op type.

fn register(ctx: &mut Context)

where Self: Sized + Parsable<Arg = Vec<(Identifier, Location)>, Parsed = OpBox>,

Register Op in Context and add it to its dialect.

fn loc(&self, ctx: &Context) -> Location

Get Op’s location

impl OperandNOfType<0, PointerType> for TmaLoadOp

impl OperandNOfType<1, TensorMapType> for TmaLoadOp

impl OperandNOfType<2, PointerType> for TmaLoadOp

impl Parsable for TmaLoadOp

type Arg = Vec<(Identifier, Location)>

Type of the argument that must be passed to the parser.

type Parsed = OpBox

The type of the parsed entity.

fn parse<'__pliron_parse>( state_stream: &mut StateStream<'__pliron_parse>, arg: Self::Arg, ) -> ParseResult<'__pliron_parse, Self::Parsed>

Define a parser using existing combinators and call into on Parser::parse_stream to get the final ParseResult. Use state_stream.state as necessary.

fn parser<'a>( arg: Self::Arg, ) -> Box<dyn Parser<Stream<Stream<Stream<Stream<IteratorStream<CharIterator<'a>>, SourcePosition>>>, State<'a>>, PartialState = (), Output = Self::Parsed> + 'a>

where Self::Parsed: 'a,

Get a parser combinator that can work on StateStream as its input.

impl PartialEq for TmaLoadOp

fn eq(&self, other: &Self) -> bool

Equality operator ==.

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

Inequality operator !=.

impl Printable for TmaLoadOp

fn fmt(&self, ctx: &Context, state: &State, fmt: &mut Formatter<'_>) -> Result

fn disp<'t, 'c>(&'t self, ctx: &'c Context) -> Box<dyn Display + 'c>

where 't: 'c,

Get a Display’able object from the given Context and default State.

fn print<'t, 'c>( &'t self, ctx: &'c Context, state: &State, ) -> Box<dyn Display + 'c>

where 't: 'c,

Get a Display’able object from the given Context and State.

impl StructuralPartialEq for TmaLoadOp

impl Verify for TmaLoadOp

fn verify(&self, ctx: &Context) -> Result<()>