Add an MLIRContext::emitWarning utility method.

--

PiperOrigin-RevId: 246546015

diff --git a/mlir/include/mlir/IR/MLIRContext.h b/mlir/include/mlir/IR/MLIRContext.h
index cdbfcf3..4ee0dbd 100644 (file)
--- a/mlir/include/mlir/IR/MLIRContext.h
+++ b/mlir/include/mlir/IR/MLIRContext.h
@@ -64,6 +64,9 @@ public:
   /// Emit an error message using the diagnostic engine.
   InFlightDiagnostic emitError(Location location, const Twine &message);
 
+  /// Emit a warning message using the diagnostic engine.
+  InFlightDiagnostic emitWarning(Location location, const Twine &message);
+
   /// Emit a remark message using the diagnostic engine.
   InFlightDiagnostic emitRemark(Location location, const Twine &message);
 

diff --git a/mlir/lib/FxpMathOps/Transforms/LowerUniformRealMath.cpp b/mlir/lib/FxpMathOps/Transforms/LowerUniformRealMath.cpp
index 96db87c..afd8152 100644 (file)
--- a/mlir/lib/FxpMathOps/Transforms/LowerUniformRealMath.cpp
+++ b/mlir/lib/FxpMathOps/Transforms/LowerUniformRealMath.cpp
@@ -52,9 +52,8 @@ static Value *emitUniformPerLayerDequantize(Location loc, Value *input,
   // Pre-conditions.
   if (!elementType.isSigned()) {
     // TODO: Support unsigned storage type.
-    rewriter.getContext()->getDiagEngine().emit(loc,
-                                                DiagnosticSeverity::Warning)
-        << "unimplemented: dequantize signed uniform";
+    rewriter.getContext()->emitWarning(
+        loc, "unimplemented: dequantize signed uniform");
     return nullptr;
   }
 

diff --git a/mlir/lib/IR/Function.cpp b/mlir/lib/IR/Function.cpp
index 66b296f..5e72b83 100644 (file)
--- a/mlir/lib/IR/Function.cpp
+++ b/mlir/lib/IR/Function.cpp
@@ -125,9 +125,7 @@ InFlightDiagnostic Function::emitRemark(const Twine &message) {
 /// Emit a warning about this function, reporting up to any diagnostic
 /// handlers that may be listening.
 InFlightDiagnostic Function::emitWarning(const Twine &message) {
-  return getContext()->getDiagEngine().emit(getLoc(),
-                                            DiagnosticSeverity::Warning)
-         << message;
+  return getContext()->emitWarning(getLoc(), message);
 }
 
 /// Emit an error about fatal conditions with this function, reporting up to

diff --git a/mlir/lib/IR/MLIRContext.cpp b/mlir/lib/IR/MLIRContext.cpp
index 6e3c7c9..9b4ae2c 100644 (file)
--- a/mlir/lib/IR/MLIRContext.cpp
+++ b/mlir/lib/IR/MLIRContext.cpp
@@ -428,6 +428,13 @@ InFlightDiagnostic MLIRContext::emitError(Location location,
          << message;
 }
 
+/// Emit a warning message using the diagnostic engine.
+InFlightDiagnostic MLIRContext::emitWarning(Location location,
+                                            const Twine &message) {
+  return getImpl().diagEngine.emit(location, DiagnosticSeverity::Warning)
+         << message;
+}
+
 /// Emit a remark message using the diagnostic engine.
 InFlightDiagnostic MLIRContext::emitRemark(Location location,
                                            const Twine &message) {

diff --git a/mlir/lib/IR/Operation.cpp b/mlir/lib/IR/Operation.cpp
index 2ed3ea3..c214a25 100644 (file)
--- a/mlir/lib/IR/Operation.cpp
+++ b/mlir/lib/IR/Operation.cpp
@@ -314,9 +314,7 @@ InFlightDiagnostic Operation::emitRemark(const Twine &message) {
 /// Emit a warning about this operation, reporting up to any diagnostic
 /// handlers that may be listening.
 InFlightDiagnostic Operation::emitWarning(const Twine &message) {
-  return getContext()->getDiagEngine().emit(getLoc(),
-                                            DiagnosticSeverity::Warning)
-         << message;
+  return getContext()->emitWarning(getLoc(), message);
 }
 
 /// Emit an error about fatal conditions with this operation, reporting up to