Skip to content

Commit 91048f0

Browse files
committed
add wrappers
1 parent 638c085 commit 91048f0

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

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

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -183,8 +183,6 @@ def LayoutTrait: AttrInterface<"LayoutTrait"> {
183183
"std::optional<llvm::SmallVector<int>>",
184184
"getEffectiveSgData">,
185185
];
186-
187-
188186
}
189187

190188
def XeGPU_LayoutAttr : XeGPUAttr<"Layout", "layout", [LayoutTrait]> {
@@ -402,7 +400,18 @@ def XeGPU_SliceAttr : XeGPUAttr<"Slice", "slice", [LayoutTrait]> {
402400
return result;
403401
}
404402
return std::nullopt;
403+
}
404+
405+
DenseI32ArrayAttr getOrder() const {
406+
return getParent().getOrder();
407+
}
408+
409+
bool isWgLayout() const {
410+
return getParent().isWgLayout();
411+
}
405412

413+
bool isSgLayout() const {
414+
return getParent().isSgLayout();
406415
}
407416
}];
408417

0 commit comments

Comments
 (0)