MLIR

Multi-Level IR Compiler Framework

source

transform.xegpu.set_desc_layout (transform::SetDescLayoutOp) 

Set xegpu.layout attribute to a xegpu.create_nd_desc op result.

Syntax:

operation ::= `transform.xegpu.set_desc_layout` $target
              `sg_layout` `=` custom<DynamicIndexList>($sg_layout, $static_sg_layout)
              `sg_data` `=` custom<DynamicIndexList>($sg_data, $static_sg_data)
              (`inst_data` `=` custom<DynamicIndexList>($inst_data, $static_inst_data)^)?
              attr-dict `:` functional-type(operands, results)

Given an xegpu.create_nd_desc operation, this transform adds xegpu.layout attribute to the result tensor descriptor. The layout is defined by the sg_layout, and sg_data and optional inst_data attributes. Returns a handle to the transformed op.

Traits: AttrSizedOperandSegments

Interfaces: MemoryEffectOpInterface, TransformOpInterface

Attributes: 

AttributeMLIR TypeDescription
static_sg_layout::mlir::DenseI64ArrayAttri64 dense array attribute
static_sg_data::mlir::DenseI64ArrayAttri64 dense array attribute
static_inst_data::mlir::DenseI64ArrayAttri64 dense array attribute

Operands: 

OperandDescription
targetTransformHandleTypeInterface instance
sg_layoutvariadic of transform any param type or any handle type
sg_datavariadic of transform any param type or any handle type
inst_datavariadic of transform any param type or any handle type

Results: 

ResultDescription
transformedTransformHandleTypeInterface instance