Skip to content

Commit bfae01f

Browse files
committed
propogation hornor pre-defined layout at anchor op
1 parent b3f2a4a commit bfae01f

File tree

7 files changed

+328
-237
lines changed

7 files changed

+328
-237
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -847,7 +847,7 @@ def XeGPU_LoadGatherOp : XeGPU_Op<"load", [MemoryEffects<[MemRead]>]> {
847847
OptionalAttr<XeGPU_CacheHintAttr>:$l1_hint,
848848
OptionalAttr<XeGPU_CacheHintAttr>:$l2_hint,
849849
OptionalAttr<XeGPU_CacheHintAttr>:$l3_hint,
850-
OptionalAttr<DistributeLayoutAttr>:$layout);
850+
OptionalAttr<DistributeLayoutAttr>:$anchor_layout);
851851
let results = (outs AnyTypeOf<[XeGPU_ValueType, XeGPU_ScalarType]>:$value);
852852

853853
let extraClassDeclaration = extraBaseClassDeclaration # [{
@@ -906,7 +906,7 @@ def XeGPU_LoadGatherOp : XeGPU_Op<"load", [MemoryEffects<[MemRead]>]> {
906906
"xegpu::CachePolicyAttr": $l1_hint,
907907
"xegpu::CachePolicyAttr": $l2_hint,
908908
"xegpu::CachePolicyAttr": $l3_hint,
909-
"xegpu::DistributeLayoutAttr": $layout)>
909+
"xegpu::DistributeLayoutAttr": $anchor_layout)>
910910
];
911911

912912
let hasVerifier = 1;
@@ -991,7 +991,7 @@ def XeGPU_StoreScatterOp : XeGPU_Op<"store", [MemoryEffects<[MemWrite]>]> {
991991
OptionalAttr<XeGPU_CacheHintAttr>:$l1_hint,
992992
OptionalAttr<XeGPU_CacheHintAttr>:$l2_hint,
993993
OptionalAttr<XeGPU_CacheHintAttr>:$l3_hint,
994-
OptionalAttr<DistributeLayoutAttr>:$layout);
994+
OptionalAttr<DistributeLayoutAttr>:$anchor_layout);
995995

996996
let extraClassDeclaration = extraBaseClassDeclaration#[{
997997
Type getDestType() {

0 commit comments

Comments
 (0)