[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