[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