[Mlir-commits] [mlir] [mlir][vector-to-gpu]: Extend MMA Lowerings (PR #176785)

Michael Platings llvmlistbot at llvm.org
Mon Feb 2 02:44:06 PST 2026


================
@@ -130,7 +147,13 @@ static std::optional<int64_t> getStaticallyKnownRowStride(ShapedType type) {
   if (failed(memrefType.getStridesAndOffset(strides, offset)) ||
       strides.back() != 1)
     return std::nullopt;
-  int64_t stride = strides[strides.size() - 2];
+
+  int stridePostion = strides.size() - 2;
+  if (!permutationMap.isPermutation()) {
+    if (auto outerResult = dyn_cast<AffineDimExpr>(permutationMap.getResult(0)))
----------------
mplatings wrote:

FWIW I created a PR building on this change that added a whole bunch of tests, so if we do decide to land this then that could be used: https://github.com/llvm/llvm-project/pull/178992

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


More information about the Mlir-commits mailing list