‘ascvf’ Dialect

vf fusion

• Operations

  • ascvf.barrier (ascvf::BarrierOp)

  • ascvf.load (ascvf::LoadOp)

  • ascvf.store (ascvf::StoreOp)

  • ascvf.vf_group (ascvf::VFGroupOp)

  • ascvf.yield (ascvf::YieldOp)

Operations

ascvf.barrier (ascvf::BarrierOp)

Syntax:

operation ::= `ascvf.barrier` attr-dict

Memory barrier

ascvf.load (ascvf::LoadOp)

Syntax:

operation ::= `ascvf.load` $dstReg `,` $srcTensor `[` $offset `]` `,` $mask attr-dict `:` type($dstReg) `,`
              type($srcTensor) `,` type($offset) `,` type($mask)

Load RegTensor from LocalTensor

Operands:

Operand	Description
dstReg	mask type or reg tensor type
srcTensor	local tensor type
offset	index
mask	mask type

ascvf.store (ascvf::StoreOp)

Syntax:

operation ::= `ascvf.store` $dstTensor `[` $offset `]` `,` $srcReg `,` $mask attr-dict `:` type($dstTensor) `,`
              type($offset) `,` type($srcReg) `,` type($mask)

Store RegTensor to LocalTensor

Operands:

Operand	Description
dstTensor	local tensor type
offset	index
srcReg	mask type or reg tensor type
mask	mask type

ascvf.vf_group (ascvf::VFGroupOp)

Block with fused operations

Syntax:

operation ::= `ascvf.vf_group` (`dst` `(` $dstList^ `:` qualified(type($dstList)) `)` )? (`src` `(` $srcList^
              `:` qualified(type($srcList)) `)`)? $region attr-dict

Traits: AttrSizedOperandSegments, SingleBlockImplicitTerminator<::mlir::ascvf::YieldOp>, SingleBlock

Attributes:

Attribute	MLIR Type	Description
groupType	::mlir::TypeAttr	any type attribute

Operands:

Operand	Description
dstList	variadic of local tensor type
srcList	variadic of local tensor type

ascvf.yield (ascvf::YieldOp)

Syntax:

operation ::= `ascvf.yield` attr-dict

Traits: Terminator