diff options
| author | Momchil Velikov <momchil.velikov@arm.com> | 2025-06-26 13:40:39 +0000 |
|---|---|---|
| committer | Momchil Velikov <momchil.velikov@arm.com> | 2025-06-26 13:40:39 +0000 |
| commit | 94c7b6066a8b42b218585fc65dc3d9f73aec2333 (patch) | |
| tree | 8e92a8e268a2bd58f73ae03e97dabd73f36e1d71 | |
| parent | 072ce70ddb305c078f02087b7b9d70335581f868 (diff) | |
| download | llvm-users/momchil-velikov/sve-i8mm-check-maps.zip llvm-users/momchil-velikov/sve-i8mm-check-maps.tar.gz llvm-users/momchil-velikov/sve-i8mm-check-maps.tar.bz2 | |
[fixup] Remove redundant testusers/momchil-velikov/sve-i8mm-check-maps
| -rw-r--r-- | mlir/lib/Dialect/ArmSVE/Transforms/LowerContractionToSVEI8MMPattern.cpp | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/mlir/lib/Dialect/ArmSVE/Transforms/LowerContractionToSVEI8MMPattern.cpp b/mlir/lib/Dialect/ArmSVE/Transforms/LowerContractionToSVEI8MMPattern.cpp index 70d2e06..fe580e1 100644 --- a/mlir/lib/Dialect/ArmSVE/Transforms/LowerContractionToSVEI8MMPattern.cpp +++ b/mlir/lib/Dialect/ArmSVE/Transforms/LowerContractionToSVEI8MMPattern.cpp @@ -141,6 +141,7 @@ public: // rhs: (d0, d1, d2) -> (d1, d2) // acc: (d0, d1, d2) -> (d0, d1) // This corresponds to matrix multiplication with transposed RHS. + // It also follows the operands' ranks are 2. if (op.getIndexingMapsArray()[0] != AffineMap::getMultiDimMapWithTargets(3, ArrayRef{0u, 2u}, op.getContext()) || @@ -155,10 +156,6 @@ public: mlir::VectorType lhsType = op.getLhsType(); mlir::VectorType rhsType = op.getRhsType(); - // Check the rank of the types so we can safely examine their dimensions. - if (lhsType.getRank() != 2 || rhsType.getRank() != 2) - return rewriter.notifyMatchFailure(op, "non-matching operand shape"); - auto M = lhsType.getDimSize(0); auto N = rhsType.getDimSize(0); auto K = rhsType.getDimSize(1); |