[Mlir-commits] [mlir] [mlir][Utils] Add verifyRanksMatch helper (NFC) (PR #175880)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue Jan 13 19:50:21 PST 2026
llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-mlir-linalg
Author: Nick Kreeger (nkreeger)
<details>
<summary>Changes</summary>
This change builds on https://github.com/llvm/llvm-project/pull/174336, which introduced shared VerificationUtils with an initial verifyDynamicDimensionCount() method.
This patch adds a new verifyRanksMatch() verification utility that checks if two shaped types have matching ranks and emits consistent error messages. The utility is applied to several ops across multiple MLIR dialects.
---
Full diff: https://github.com/llvm/llvm-project/pull/175880.diff
6 Files Affected:
- (modified) mlir/include/mlir/Dialect/Utils/VerificationUtils.h (+6)
- (modified) mlir/lib/Dialect/Bufferization/IR/BufferizationOps.cpp (+3-2)
- (modified) mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp (+4-3)
- (modified) mlir/lib/Dialect/Tosa/IR/TosaOps.cpp (+12-11)
- (modified) mlir/lib/Dialect/Utils/VerificationUtils.cpp (+16)
- (modified) mlir/test/Dialect/Tosa/invalid.mlir (+3-3)
``````````diff
diff --git a/mlir/include/mlir/Dialect/Utils/VerificationUtils.h b/mlir/include/mlir/Dialect/Utils/VerificationUtils.h
index c1c3cc6231eb6..94746f0b53254 100644
--- a/mlir/include/mlir/Dialect/Utils/VerificationUtils.h
+++ b/mlir/include/mlir/Dialect/Utils/VerificationUtils.h
@@ -27,6 +27,12 @@ namespace mlir {
LogicalResult verifyDynamicDimensionCount(Operation *op, ShapedType type,
ValueRange dynamicSizes);
+/// Verify that two shaped types have matching ranks. Returns failure and emits
+/// an error if ranks don't match. Unranked types are considered compatible.
+LogicalResult verifyRanksMatch(Operation *op, ShapedType type1,
+ ShapedType type2, StringRef name1,
+ StringRef name2);
+
} // namespace mlir
#endif // MLIR_DIALECT_UTILS_VERIFICATIONUTILS_H
diff --git a/mlir/lib/Dialect/Bufferization/IR/BufferizationOps.cpp b/mlir/lib/Dialect/Bufferization/IR/BufferizationOps.cpp
index eda6bf276be06..d3d9d23302226 100644
--- a/mlir/lib/Dialect/Bufferization/IR/BufferizationOps.cpp
+++ b/mlir/lib/Dialect/Bufferization/IR/BufferizationOps.cpp
@@ -702,8 +702,9 @@ LogicalResult MaterializeInDestinationOp::verify() {
if (srcType.hasRank() != destType.hasRank())
return emitOpError("source/destination shapes are incompatible");
if (srcType.hasRank()) {
- if (srcType.getRank() != destType.getRank())
- return emitOpError("rank mismatch between source and destination shape");
+ if (failed(verifyRanksMatch(getOperation(), srcType, destType, "source",
+ "destination")))
+ return failure();
for (auto [src, dest] :
llvm::zip(srcType.getShape(), destType.getShape())) {
if (src == ShapedType::kDynamic || dest == ShapedType::kDynamic) {
diff --git a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
index 210f9584c1e86..e9f617a785d22 100644
--- a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
+++ b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
@@ -24,6 +24,7 @@
#include "mlir/Dialect/Utils/IndexingUtils.h"
#include "mlir/Dialect/Utils/ReshapeOpsUtils.h"
#include "mlir/Dialect/Utils/StaticValueUtils.h"
+#include "mlir/Dialect/Utils/VerificationUtils.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h"
@@ -2043,9 +2044,9 @@ LogicalResult TransposeOp::verify() {
int64_t rank = inputType.getRank();
- if (rank != initType.getRank())
- return emitOpError() << "input rank " << rank
- << " does not match init rank " << initType.getRank();
+ if (failed(verifyRanksMatch(getOperation(), inputType, initType, "input",
+ "init")))
+ return failure();
if (rank != static_cast<int64_t>(permutationRef.size()))
return emitOpError() << "size of permutation " << permutationRef.size()
diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
index 5656f3de698c5..f123c6ac72083 100644
--- a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
+++ b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
@@ -19,6 +19,7 @@
#include "mlir/Dialect/Tosa/Utils/QuantUtils.h"
#include "mlir/Dialect/Tosa/Utils/ShapeUtils.h"
#include "mlir/Dialect/Utils/IndexingUtils.h"
+#include "mlir/Dialect/Utils/VerificationUtils.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/Matchers.h"
@@ -2126,13 +2127,11 @@ LogicalResult tosa::PadOp::verify() {
if (!inputType || !outputType)
return success();
- auto inputRank = inputType.getRank();
- auto outputRank = outputType.getRank();
- if (inputRank != outputRank)
- return emitOpError() << "expect same input and output tensor rank, but got "
- << "inputRank: " << inputRank
- << ", outputRank: " << outputRank;
+ if (failed(verifyRanksMatch(getOperation(), inputType, outputType, "input",
+ "output")))
+ return failure();
+ auto inputRank = inputType.getRank();
DenseIntElementsAttr paddingAttr;
if (!matchPattern(getPadding(), m_Constant(&paddingAttr))) {
return failure();
@@ -2373,9 +2372,9 @@ LogicalResult tosa::TableOp::verify() {
if (!inputType.hasRank() || !outputType.hasRank())
return success();
- if (inputType.getRank() != outputType.getRank())
- return emitOpError()
- << "expected input tensor rank to equal result tensor rank";
+ if (failed(verifyRanksMatch(getOperation(), inputType, outputType, "input",
+ "result")))
+ return failure();
auto inputDims = inputType.getShape();
auto outputDims = outputType.getShape();
@@ -2465,8 +2464,10 @@ LogicalResult tosa::TileOp::verify() {
if (inputType.getRank() != multiplesRank)
return emitOpError("expect 'multiples' to have rank ")
<< inputType.getRank() << " but got " << multiplesRank << ".";
- if (outputType.hasRank() && inputType.getRank() != outputType.getRank())
- return emitOpError("expect same input and output tensor rank.");
+ if (outputType.hasRank() &&
+ failed(verifyRanksMatch(getOperation(), inputType, outputType, "input",
+ "output")))
+ return failure();
} else if (outputType.hasRank() && outputType.getRank() != multiplesRank)
return emitOpError("expect 'multiples' array to have length ")
<< outputType.getRank() << " but got " << multiplesRank << ".";
diff --git a/mlir/lib/Dialect/Utils/VerificationUtils.cpp b/mlir/lib/Dialect/Utils/VerificationUtils.cpp
index 22b224713a6a3..9b3bfda42b91a 100644
--- a/mlir/lib/Dialect/Utils/VerificationUtils.cpp
+++ b/mlir/lib/Dialect/Utils/VerificationUtils.cpp
@@ -20,3 +20,19 @@ LogicalResult mlir::verifyDynamicDimensionCount(Operation *op, ShapedType type,
}
return success();
}
+
+LogicalResult mlir::verifyRanksMatch(Operation *op, ShapedType type1,
+ ShapedType type2, StringRef name1,
+ StringRef name2) {
+ if (!type1.hasRank() || !type2.hasRank())
+ return success(); // Unranked types are considered compatible
+
+ int64_t rank1 = type1.getRank();
+ int64_t rank2 = type2.getRank();
+ if (rank1 != rank2) {
+ return op->emitOpError()
+ << name1 << " rank (" << rank1 << ") does not match " << name2
+ << " rank (" << rank2 << ")";
+ }
+ return success();
+}
diff --git a/mlir/test/Dialect/Tosa/invalid.mlir b/mlir/test/Dialect/Tosa/invalid.mlir
index e8206c24f1507..cf93563fb0878 100644
--- a/mlir/test/Dialect/Tosa/invalid.mlir
+++ b/mlir/test/Dialect/Tosa/invalid.mlir
@@ -292,7 +292,7 @@ func.func @test_pad_const_non_const(%arg0: tensor<13x21x3xi8>, %arg1: tensor<1xi
func.func @test_pad_io_rank_mismatch(%arg0: tensor<13x21xf32>) {
%0 = tosa.const_shape {values = dense<1> : tensor<4xindex>} : () -> !tosa.shape<4>
%pad_const = "tosa.const"() {values = dense<3.14> : tensor<1xf32>} : () -> tensor<1xf32>
- // expected-error at +1 {{'tosa.pad' op expect same input and output tensor rank}}
+ // expected-error at +1 {{'tosa.pad' op input rank (2) does not match output rank (3)}}
%1 = tosa.pad %arg0, %0, %pad_const : (tensor<13x21xf32>, !tosa.shape<4>, tensor<1xf32>) -> tensor<13x21x3xf32>
}
@@ -663,7 +663,7 @@ func.func @test_tile_invalid_multiples_value() {
func.func @test_tile_io_rank_mismatch() {
%0 = tensor.empty() : tensor<4x31xf32>
%multiples = tosa.const_shape { values = dense<[2, 2]> : tensor<2xindex> } : () -> !tosa.shape<2>
- // expected-error at +1 {{'tosa.tile' op expect same input and output tensor rank.}}
+ // expected-error at +1 {{'tosa.tile' op input rank (2) does not match output rank (3)}}
%1 = tosa.tile %0, %multiples : (tensor<4x31xf32>, !tosa.shape<2>) -> tensor<4x31x31xf32>
return
}
@@ -682,7 +682,7 @@ func.func @test_table_rank0_table(%arg0: tensor<64xi16>, %arg1: tensor<i16>) {
// CHECK-LABEL: test_table_io_rank_mismatch
func.func @test_table_io_rank_mismatch(%arg0: tensor<64xi16>, %arg1: tensor<6xi16>) {
- // expected-error at +1 {{'tosa.table' op expected input tensor rank to equal result tensor rank}}
+ // expected-error at +1 {{'tosa.table' op input rank (1) does not match result rank (2)}}
%0 = tosa.table %arg0, %arg1 : (tensor<64xi16>, tensor<6xi16>) -> tensor<64x?xi16>
return
}
``````````
</details>
https://github.com/llvm/llvm-project/pull/175880
More information about the Mlir-commits
mailing list