[Mlir-commits] [llvm] [mlir] [XeGPU][Transform] Add XeGPU array length optimization pass (PR #194062)
Md Abdullah Shahneous Bari
llvmlistbot at llvm.org
Fri May 1 17:46:45 PDT 2026
================
@@ -556,11 +556,20 @@ LogicalResult LoadNdOp::verify() {
}
}
+ // Handle array_length. Two result shape conventions are accepted:
+ // * Legacy: leading array_length dimension prepended, e.g. descriptor
+ // 16x16 with array_length=2 -> [2, 16, 16].
+ // * Stacked 2D: array blocks stacked along the non-FCD (first) dimension,
+ // e.g. descriptor 16x16 with array_length=2 -> [32, 16].
auto array_len = tdescTy.getArrayLength();
- if (array_len > 1)
- tdescShape.insert(tdescShape.begin(), array_len);
+ SmallVector<int64_t> stackedShape(tdescShape);
----------------
mshahneo wrote:
Changed the var names.
https://github.com/llvm/llvm-project/pull/194062
More information about the Mlir-commits
mailing list