[Mlir-commits] [mlir] [mlir][xegpu] Add SIMT distribution support for GEMM transpose B case. (PR #155517)

Jianhui Li llvmlistbot at llvm.org
Thu Sep 11 16:51:00 PDT 2025


================
@@ -146,6 +149,15 @@ static bool hasPackedLayout(xegpu::LayoutAttr layout) {
   return laneData.asArrayRef()[0] != 1;
 }
 
+static bool hasTransposedLayout(xegpu::LayoutAttr layout) {
+  if (layout == xegpu::LayoutAttr())
+    return false;
+  DenseI32ArrayAttr laneLayout = layout.getLaneLayout();
+  if (!laneLayout || laneLayout.size() != 2)
----------------
Jianhui-Li wrote:

This check is uArch-specific, needs to be guarded like:
    auto chip = xegpu::getChipStr(readOp);
    if (chip == "pvc" || chip == "bmg") {

https://github.com/llvm/llvm-project/pull/155517


More information about the Mlir-commits mailing list