Skip to content

Commit f47b1c6

Browse files
committed
[MLIR][XeGPU] Decouple inst_data and lane_layout in propagation
1 parent fd9dd43 commit f47b1c6

File tree

4 files changed

+163
-95
lines changed

4 files changed

+163
-95
lines changed

mlir/include/mlir/Dialect/XeGPU/IR/XeGPUAttrs.td

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -379,6 +379,17 @@ def XeGPU_LayoutAttr : XeGPUAttr<"Layout", "layout", [DistributeLayoutAttr]> {
379379
);
380380

381381
let builders = [
382+
AttrBuilder<(ins "llvm::ArrayRef<int32_t>": $inst_data),
383+
[{
384+
auto sg_layout = DenseI32ArrayAttr();
385+
auto sg_data = DenseI32ArrayAttr();
386+
auto order = DenseI32ArrayAttr();
387+
auto lane_layout = DenseI32ArrayAttr();
388+
auto lane_data = DenseI32ArrayAttr();
389+
return $_get($_ctxt, sg_layout, sg_data,
390+
DenseI32ArrayAttr::get($_ctxt, inst_data),
391+
lane_layout, lane_data, order);
392+
}]>,
382393
AttrBuilder<(ins "llvm::ArrayRef<int32_t>": $inst_data,
383394
"llvm::ArrayRef<int32_t>": $lane_layout,
384395
"llvm::ArrayRef<int32_t>": $lane_data),

0 commit comments

Comments
 (0)