[Mlir-commits] [mlir] [mlir][IR] Improve error message when parsing incorrect type (PR #134984)
Matthias Springer
llvmlistbot at llvm.org
Wed Apr 9 07:42:19 PDT 2025
https://github.com/matthias-springer updated https://github.com/llvm/llvm-project/pull/134984
>From fc2be4011f9bfe404c491f50cc6c0a657d8e61e7 Mon Sep 17 00:00:00 2001
From: Matthias Springer <mspringer at nvidia.com>
Date: Wed, 9 Apr 2025 11:36:05 +0200
Subject: [PATCH 1/2] [mlir][IR] Improve error message when parsing incorrect
type
---
mlir/include/mlir/IR/OpImplementation.h | 40 ++++++++++++++++---
mlir/test/Dialect/EmitC/invalid_ops.mlir | 2 +-
mlir/test/Dialect/LLVMIR/invalid.mlir | 2 +-
mlir/test/Dialect/SparseTensor/invalid.mlir | 2 +-
mlir/test/Dialect/Tensor/invalid.mlir | 4 +-
mlir/test/Dialect/Vector/invalid.mlir | 8 ++--
mlir/test/Target/LLVMIR/llvmir-invalid.mlir | 2 +-
.../test/mlir-tblgen/attr-or-type-format.mlir | 2 +-
8 files changed, 45 insertions(+), 17 deletions(-)
diff --git a/mlir/include/mlir/IR/OpImplementation.h b/mlir/include/mlir/IR/OpImplementation.h
index 25c7d15eb8ed5..6efad01dec4cc 100644
--- a/mlir/include/mlir/IR/OpImplementation.h
+++ b/mlir/include/mlir/IR/OpImplementation.h
@@ -21,6 +21,19 @@
#include "llvm/Support/SMLoc.h"
#include <optional>
+namespace {
+// reference https://stackoverflow.com/a/16000226
+template <typename T, typename = void>
+struct HasStaticName : std::false_type {};
+
+template <typename T>
+struct HasStaticName<T,
+ typename std::enable_if<
+ std::is_same<::llvm::StringLiteral,
+ std::decay_t<decltype(T::name)>>::value,
+ void>::type> : std::true_type {};
+} // namespace
+
namespace mlir {
class AsmParsedResourceEntry;
class AsmResourceBuilder;
@@ -1238,8 +1251,13 @@ class AsmParser {
// Check for the right kind of type.
result = llvm::dyn_cast<TypeT>(type);
- if (!result)
- return emitError(loc, "invalid kind of type specified");
+ if (!result) {
+ InFlightDiagnostic diag =
+ emitError(loc, "invalid kind of type specified");
+ if constexpr (HasStaticName<TypeT>::value)
+ diag << ": expected " << TypeT::name << ", but found " << type;
+ return diag;
+ }
return success();
}
@@ -1270,8 +1288,13 @@ class AsmParser {
// Check for the right kind of Type.
result = llvm::dyn_cast<TypeT>(type);
- if (!result)
- return emitError(loc, "invalid kind of Type specified");
+ if (!result) {
+ InFlightDiagnostic diag =
+ emitError(loc, "invalid kind of type specified");
+ if constexpr (HasStaticName<TypeT>::value)
+ diag << ": expected " << TypeT::name << ", but found " << type;
+ return diag;
+ }
return success();
}
@@ -1307,8 +1330,13 @@ class AsmParser {
// Check for the right kind of type.
result = llvm::dyn_cast<TypeType>(type);
- if (!result)
- return emitError(loc, "invalid kind of type specified");
+ if (!result) {
+ InFlightDiagnostic diag =
+ emitError(loc, "invalid kind of type specified");
+ if constexpr (HasStaticName<TypeType>::value)
+ diag << ": expected " << TypeType::name << ", but found " << type;
+ return diag;
+ }
return success();
}
diff --git a/mlir/test/Dialect/EmitC/invalid_ops.mlir b/mlir/test/Dialect/EmitC/invalid_ops.mlir
index 5b4e3f92f6d53..b16bab8e5e1ab 100644
--- a/mlir/test/Dialect/EmitC/invalid_ops.mlir
+++ b/mlir/test/Dialect/EmitC/invalid_ops.mlir
@@ -281,7 +281,7 @@ func.func @test_assign_type_mismatch(%arg1: f32) {
func.func @test_assign_to_array(%arg1: !emitc.array<4xi32>) {
%v = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.array<4xi32>
- // expected-error @+1 {{invalid kind of Type specified}}
+ // expected-error @+1 {{invalid kind of type specified: expected emitc.lvalue, but found '!emitc.array<4xi32>}}
emitc.assign %arg1 : !emitc.array<4xi32> to %v : !emitc.array<4xi32>
return
}
diff --git a/mlir/test/Dialect/LLVMIR/invalid.mlir b/mlir/test/Dialect/LLVMIR/invalid.mlir
index a7ca109df9382..db55088d812e6 100644
--- a/mlir/test/Dialect/LLVMIR/invalid.mlir
+++ b/mlir/test/Dialect/LLVMIR/invalid.mlir
@@ -121,7 +121,7 @@ func.func @gep_missing_result_type(%pos : i64, %base : !llvm.ptr) {
// -----
func.func @gep_non_function_type(%pos : i64, %base : !llvm.ptr) {
- // expected-error at +1 {{invalid kind of type specified}}
+ // expected-error at +1 {{invalid kind of type specified: expected builtin.function, but found '!llvm.ptr'}}
llvm.getelementptr %base[%pos] : !llvm.ptr
}
diff --git a/mlir/test/Dialect/SparseTensor/invalid.mlir b/mlir/test/Dialect/SparseTensor/invalid.mlir
index 908d2d8aa83f7..4c37fc6882bab 100644
--- a/mlir/test/Dialect/SparseTensor/invalid.mlir
+++ b/mlir/test/Dialect/SparseTensor/invalid.mlir
@@ -347,7 +347,7 @@ func.func @sparse_wrong_arity_compression(%arg0: memref<?xf64>,
// -----
func.func @sparse_convert_unranked(%arg0: tensor<*xf32>) -> tensor<10xf32> {
- // expected-error at +1 {{invalid kind of type specified}}
+ // expected-error at +1 {{invalid kind of type specified: expected builtin.tensor, but found 'tensor<*xf32>'}}
%0 = sparse_tensor.convert %arg0 : tensor<*xf32> to tensor<10xf32>
return %0 : tensor<10xf32>
}
diff --git a/mlir/test/Dialect/Tensor/invalid.mlir b/mlir/test/Dialect/Tensor/invalid.mlir
index eeb3967337806..e3923bbfd7340 100644
--- a/mlir/test/Dialect/Tensor/invalid.mlir
+++ b/mlir/test/Dialect/Tensor/invalid.mlir
@@ -81,7 +81,7 @@ func.func @insert_too_many_indices(%arg0: f32, %arg1: tensor<?xf32>) {
// -----
func.func @tensor.from_elements_wrong_result_type() {
- // expected-error at +2 {{'tensor.from_elements' invalid kind of type specified}}
+ // expected-error at +2 {{'tensor.from_elements' invalid kind of type specified: expected builtin.tensor, but found 'tensor<*xi32>}}
%c0 = arith.constant 0 : i32
%0 = tensor.from_elements %c0 : tensor<*xi32>
return
@@ -459,7 +459,7 @@ func.func @pad_yield_type(%arg0: tensor<?x4xi32>, %arg1: i8) -> tensor<?x9xi32>
// -----
func.func @invalid_splat(%v : f32) {
- // expected-error at +1 {{invalid kind of type specified}}
+ // expected-error at +1 {{invalid kind of type specified: expected builtin.tensor, but found 'memref<8xf32>'}}
tensor.splat %v : memref<8xf32>
return
}
diff --git a/mlir/test/Dialect/Vector/invalid.mlir b/mlir/test/Dialect/Vector/invalid.mlir
index ea6d0021391fb..ef87030bf0752 100644
--- a/mlir/test/Dialect/Vector/invalid.mlir
+++ b/mlir/test/Dialect/Vector/invalid.mlir
@@ -1,7 +1,7 @@
// RUN: mlir-opt %s -split-input-file -verify-diagnostics
func.func @broadcast_to_scalar(%arg0: f32) -> f32 {
- // expected-error at +1 {{custom op 'vector.broadcast' invalid kind of type specified}}
+ // expected-error at +1 {{custom op 'vector.broadcast' invalid kind of type specified: expected builtin.vector, but found 'f32'}}
%0 = vector.broadcast %arg0 : f32 to f32
}
@@ -144,7 +144,7 @@ func.func @extract_element(%arg0: vector<4x4xf32>) {
// -----
func.func @extract_vector_type(%arg0: index) {
- // expected-error at +1 {{invalid kind of type specified}}
+ // expected-error at +1 {{invalid kind of type specified: expected builtin.vector, but found 'index'}}
%1 = vector.extract %arg0[] : index from index
}
@@ -1174,7 +1174,7 @@ func.func @shape_cast_scalability_flag_is_dropped(%arg0 : vector<2x[15]x[2]xf32>
// -----
func.func @bitcast_not_vector(%arg0 : vector<5x1x3x2xf32>) {
- // expected-error at +1 {{'vector.bitcast' invalid kind of type specified}}
+ // expected-error at +1 {{'vector.bitcast' invalid kind of type specified: expected builtin.vector, but found 'f32'}}
%0 = vector.bitcast %arg0 : vector<5x1x3x2xf32> to f32
}
@@ -1628,7 +1628,7 @@ func.func @scan_unsupported_kind(%arg0: vector<2x3xf32>, %arg1: vector<3xf32>) -
// -----
func.func @invalid_splat(%v : f32) {
- // expected-error at +1 {{invalid kind of type specified}}
+ // expected-error at +1 {{invalid kind of type specified: expected builtin.vector, but found 'memref<8xf32>'}}
vector.splat %v : memref<8xf32>
return
}
diff --git a/mlir/test/Target/LLVMIR/llvmir-invalid.mlir b/mlir/test/Target/LLVMIR/llvmir-invalid.mlir
index 4500175cd22b2..7bb64542accdf 100644
--- a/mlir/test/Target/LLVMIR/llvmir-invalid.mlir
+++ b/mlir/test/Target/LLVMIR/llvmir-invalid.mlir
@@ -246,7 +246,7 @@ llvm.func @matrix_transpose_intr_wrong_type(%matrix : f32) -> vector<48xf32> {
// -----
llvm.func @active_lane_intr_wrong_type(%base : i64, %n : vector<7xi64>) -> vector<7xi1> {
- // expected-error @below{{invalid kind of type specified}}
+ // expected-error @below{{invalid kind of type specified: expected builtin.integer, but found 'vector<7xi64>'}}
%0 = llvm.intr.get.active.lane.mask %base, %n : i64, vector<7xi64> to vector<7xi1>
llvm.return %0 : vector<7xi1>
}
diff --git a/mlir/test/mlir-tblgen/attr-or-type-format.mlir b/mlir/test/mlir-tblgen/attr-or-type-format.mlir
index 6b61deff1b69f..75aca36ae531b 100644
--- a/mlir/test/mlir-tblgen/attr-or-type-format.mlir
+++ b/mlir/test/mlir-tblgen/attr-or-type-format.mlir
@@ -129,7 +129,7 @@ func.func private @test_verifier_fails() -> () attributes {
// -----
func.func private @test_attr_with_type_failed_to_parse_type() -> () attributes {
- // expected-error at +2 {{invalid kind of type specified}}
+ // expected-error at +2 {{invalid kind of type specified: expected builtin.integer, but found 'vector<4xi32>}}
// expected-error at +1 {{failed to parse TestAttrWithTypeParam parameter 'int_type'}}
attr = #test.attr_with_type<vector<4xi32>, vector<4xi32>>
}
>From 19d8e9f5c422aefdcff1438bbcd2b9b89eda2bdf Mon Sep 17 00:00:00 2001
From: Matthias Springer <mspringer at nvidia.com>
Date: Wed, 9 Apr 2025 16:41:57 +0200
Subject: [PATCH 2/2] address comments
---
mlir/test/Dialect/EmitC/invalid_ops.mlir | 2 +-
mlir/test/Dialect/Tensor/invalid.mlir | 2 +-
mlir/test/mlir-tblgen/attr-or-type-format.mlir | 2 +-
3 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/mlir/test/Dialect/EmitC/invalid_ops.mlir b/mlir/test/Dialect/EmitC/invalid_ops.mlir
index b16bab8e5e1ab..3793dfe3f173b 100644
--- a/mlir/test/Dialect/EmitC/invalid_ops.mlir
+++ b/mlir/test/Dialect/EmitC/invalid_ops.mlir
@@ -281,7 +281,7 @@ func.func @test_assign_type_mismatch(%arg1: f32) {
func.func @test_assign_to_array(%arg1: !emitc.array<4xi32>) {
%v = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.array<4xi32>
- // expected-error @+1 {{invalid kind of type specified: expected emitc.lvalue, but found '!emitc.array<4xi32>}}
+ // expected-error @+1 {{invalid kind of type specified: expected emitc.lvalue, but found '!emitc.array<4xi32>'}}
emitc.assign %arg1 : !emitc.array<4xi32> to %v : !emitc.array<4xi32>
return
}
diff --git a/mlir/test/Dialect/Tensor/invalid.mlir b/mlir/test/Dialect/Tensor/invalid.mlir
index e3923bbfd7340..f35d52e700084 100644
--- a/mlir/test/Dialect/Tensor/invalid.mlir
+++ b/mlir/test/Dialect/Tensor/invalid.mlir
@@ -81,7 +81,7 @@ func.func @insert_too_many_indices(%arg0: f32, %arg1: tensor<?xf32>) {
// -----
func.func @tensor.from_elements_wrong_result_type() {
- // expected-error at +2 {{'tensor.from_elements' invalid kind of type specified: expected builtin.tensor, but found 'tensor<*xi32>}}
+ // expected-error at +2 {{'tensor.from_elements' invalid kind of type specified: expected builtin.tensor, but found 'tensor<*xi32>'}}
%c0 = arith.constant 0 : i32
%0 = tensor.from_elements %c0 : tensor<*xi32>
return
diff --git a/mlir/test/mlir-tblgen/attr-or-type-format.mlir b/mlir/test/mlir-tblgen/attr-or-type-format.mlir
index 75aca36ae531b..907aad369a344 100644
--- a/mlir/test/mlir-tblgen/attr-or-type-format.mlir
+++ b/mlir/test/mlir-tblgen/attr-or-type-format.mlir
@@ -129,7 +129,7 @@ func.func private @test_verifier_fails() -> () attributes {
// -----
func.func private @test_attr_with_type_failed_to_parse_type() -> () attributes {
- // expected-error at +2 {{invalid kind of type specified: expected builtin.integer, but found 'vector<4xi32>}}
+ // expected-error at +2 {{invalid kind of type specified: expected builtin.integer, but found 'vector<4xi32>'}}
// expected-error at +1 {{failed to parse TestAttrWithTypeParam parameter 'int_type'}}
attr = #test.attr_with_type<vector<4xi32>, vector<4xi32>>
}
More information about the Mlir-commits
mailing list