diff --git a/mlir/include/mlir/Dialect/Utils/StructuredOpsUtils.h b/mlir/include/mlir/Dialect/Utils/StructuredOpsUtils.h --- a/mlir/include/mlir/Dialect/Utils/StructuredOpsUtils.h +++ b/mlir/include/mlir/Dialect/Utils/StructuredOpsUtils.h @@ -41,7 +41,7 @@ bindDims(context, m, n, k); auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, context)); auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, context)); - auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {n, m}, context)); + auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, context)); auto maps = ArrayAttr::get(context, {mapA, mapB, mapC}); return indexingMaps == maps; }