[Mlir-commits] [mlir] [mlir][xegpu] add support for structure control flow ops in workgroup to subgroup distribution (PR #142618)
Nishant Patel
llvmlistbot at llvm.org
Tue Jun 10 10:08:49 PDT 2025
================
@@ -29,6 +31,29 @@ using namespace mlir;
namespace {
+static std::pair<SmallVector<int64_t>, int>
+getSgShapeAndCount(ArrayRef<int64_t> shape, xegpu::LayoutAttr layout) {
+ int count = 1;
+ SmallVector<int64_t> sgShape(shape);
+
+ if (layout && layout.isWgLayout()) {
----------------
nbpatel wrote:
isWgLayout seems confusing, I think it should be called isSgLayout since it describes how the subgroups are laid out
https://github.com/llvm/llvm-project/pull/142618
More information about the Mlir-commits
mailing list