[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