[Mlir-commits] [mlir] [MLIR] Introduce RemarkEngine + pluggable remark streaming (YAML/Bitstream) (PR #152474)
Guray Ozen
llvmlistbot at llvm.org
Thu Aug 21 02:00:16 PDT 2025
================
@@ -0,0 +1,520 @@
+//===- Remarks.h - MLIR Optimization Remark ----------------------*- C++-*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// This file defines utilities for emitting optimization remarks.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef MLIR_IR_REMARKS_H
+#define MLIR_IR_REMARKS_H
+
+#include "llvm/ADT/StringExtras.h"
+#include "llvm/IR/DiagnosticInfo.h"
+#include "llvm/Remarks/Remark.h"
+#include "llvm/Support/FormatVariadic.h"
+#include "llvm/Support/Regex.h"
+#include <optional>
+
+#include "mlir/IR/Diagnostics.h"
+#include "mlir/IR/MLIRContext.h"
+#include "mlir/IR/Value.h"
+
+namespace mlir::remark {
+
+/// Define an the set of categories to accept. By default none are, the provided
+/// regex matches against the category names for each kind of remark.
+struct RemarkCategories {
+ std::optional<std::string> passed, missed, analysis, failed;
+};
+
+/// Categories describe the outcome of an transformation, not the mechanics of
+/// emitting/serializing remarks.
+enum class RemarkKind {
+ RemarkUnknown = 0,
+
+ /// An optimization was applied.
+ RemarkPassed,
+
+ /// A profitable optimization opportunity was found but not applied.
+ RemarkMissed,
+
+ /// The compiler attempted the optimization but failed (e.g., legality
+ /// checks, or better opportunites).
+ RemarkFailure,
+
+ /// Informational context (e.g., analysis numbers) without a pass/fail
+ /// outcome.
+ RemarkAnalysis,
+};
+
+using namespace llvm;
+
+/// Options to create a Remark
+struct RemarkOpts {
+ StringRef remarkName; // Identifiable name
+ StringRef categoryName; // Category name (subject to regex filtering)
+ StringRef subCategoryName; // Subcategory name
+ StringRef functionName; // Function name if available
+ RemarkOpts() = delete;
+ // Construct RemarkOpts from a remark name.
+ static constexpr RemarkOpts name(StringRef n) {
+ return RemarkOpts{n, {}, {}, {}};
+ }
+ /// Return a copy with the category set.
+ constexpr RemarkOpts category(StringRef v) const {
+ return {remarkName, v, subCategoryName, functionName};
+ }
+ /// Return a copy with the subcategory set.
+ constexpr RemarkOpts subCategory(StringRef v) const {
+ return {remarkName, categoryName, v, functionName};
+ }
+ /// Return a copy with the function name set.
+ constexpr RemarkOpts function(StringRef v) const {
+ return {remarkName, categoryName, subCategoryName, v};
+ }
+};
+
+} // namespace mlir::remark
+
+namespace mlir::remark::detail {
+//===----------------------------------------------------------------------===//
+// Remark Base Class
+//===----------------------------------------------------------------------===//
+class Remark {
+
+public:
+ Remark(RemarkKind remarkKind, DiagnosticSeverity severity, Location loc,
+ RemarkOpts opts)
+ : remarkKind(remarkKind), functionName(opts.functionName), loc(loc),
+ categoryName(opts.categoryName), subCategoryName(opts.subCategoryName),
+ remarkName(opts.remarkName) {
+ if (!categoryName.empty() && !subCategoryName.empty()) {
+ (llvm::Twine(categoryName) + ":" + subCategoryName)
+ .toStringRef(fullCategoryName);
+ }
+ }
+
+ // Remark argument that is a key-value pair that can be printed as machine
+ // parsable args.
+ struct Arg {
+ std::string key;
+ std::string val;
+ Arg(llvm::StringRef m) : key("Remark"), val(m) {}
+ Arg(llvm::StringRef k, llvm::StringRef v) : key(k), val(v) {}
+ Arg(llvm::StringRef k, std::string v) : key(k), val(std::move(v)) {}
+ Arg(llvm::StringRef k, const char *v) : Arg(k, llvm::StringRef(v)) {}
+ Arg(llvm::StringRef k, Value v);
+ Arg(llvm::StringRef k, Type t);
+ Arg(llvm::StringRef k, bool b) : key(k), val(b ? "true" : "false") {}
+
+ // One constructor for all arithmetic types except bool.
+ template <typename T, typename = std::enable_if_t<std::is_arithmetic_v<T> &&
+ !std::is_same_v<T, bool>>>
+ Arg(llvm::StringRef k, T v) : key(k) {
+ if constexpr (std::is_floating_point_v<T>) {
+ llvm::raw_string_ostream os(val);
+ os << v;
+ } else if constexpr (std::is_signed_v<T>) {
+ val = llvm::itostr(static_cast<long long>(v));
+ } else {
+ val = llvm::utostr(static_cast<unsigned long long>(v));
+ }
+ }
+ };
+
+ void insert(llvm::StringRef s);
+ void insert(Arg a);
+
+ void print(llvm::raw_ostream &os, bool printLocation = false) const;
+
+ Location getLocation() const { return loc; }
+ /// Diagnostic -> Remark
+ llvm::remarks::Remark generateRemark() const;
+
+ StringRef getFunction() const {
+ if (!functionName.empty())
+ return functionName;
+ return "<unknown function>";
+ }
+
+ llvm::StringRef getCategoryName() const { return categoryName; }
+
+ llvm::StringRef getFullCategoryName() const {
+ if (categoryName.empty() && subCategoryName.empty())
+ return {};
+ if (subCategoryName.empty())
+ return categoryName;
+ if (categoryName.empty())
+ return subCategoryName;
+ return fullCategoryName;
+ }
+
+ StringRef getRemarkName() const {
+ if (remarkName.empty())
+ return "<unknown remark name>";
+ return remarkName;
+ }
+
+ std::string getMsg() const;
+
+ ArrayRef<Arg> getArgs() const { return args; }
+
+ llvm::remarks::Type getRemarkType() const;
+
+ StringRef getRemarkTypeString() const;
+
+protected:
+ /// Keeps the MLIR diagnostic kind, which is used to determine the
+ /// diagnostic kind in the LLVM remark streamer.
+ RemarkKind remarkKind;
+ /// Name of the convering function like interface
+ StringRef functionName;
+
+ Location loc;
+ /// Sub category passname e.g., "Unroll" or "UnrollAndJam"
+ StringRef categoryName;
+
+ /// Sub category name "Loop Optimizer"
+ StringRef subCategoryName;
+
+ /// Combined name for category and sub-category
+ SmallString<64> fullCategoryName;
+
+ /// Remark identifier
+ StringRef remarkName;
+
+ /// Args collected via the streaming interface.
+ SmallVector<Arg, 4> args;
+
+private:
+ /// Convert the MLIR diagnostic severity to LLVM diagnostic severity.
+ static llvm::DiagnosticSeverity
+ makeLLVMSeverity(DiagnosticSeverity severity) {
+ switch (severity) {
+ case DiagnosticSeverity::Note:
+ return llvm::DiagnosticSeverity::DS_Note;
+ case DiagnosticSeverity::Warning:
+ return llvm::DiagnosticSeverity::DS_Warning;
+ case DiagnosticSeverity::Error:
+ return llvm::DiagnosticSeverity::DS_Error;
+ case DiagnosticSeverity::Remark:
+ return llvm::DiagnosticSeverity::DS_Remark;
+ }
+ llvm_unreachable("Unknown diagnostic severity");
+ }
+ /// Convert the MLIR remark kind to LLVM diagnostic kind.
+ static llvm::DiagnosticKind makeLLVMKind(RemarkKind remarkKind) {
+ switch (remarkKind) {
+ case RemarkKind::RemarkUnknown:
+ return llvm::DiagnosticKind::DK_Generic;
+ case RemarkKind::RemarkPassed:
+ return llvm::DiagnosticKind::DK_OptimizationRemark;
+ case RemarkKind::RemarkMissed:
+ return llvm::DiagnosticKind::DK_OptimizationRemarkMissed;
+ case RemarkKind::RemarkFailure:
+ return llvm::DiagnosticKind::DK_OptimizationFailure;
+ case RemarkKind::RemarkAnalysis:
+ return llvm::DiagnosticKind::DK_OptimizationRemarkAnalysis;
+ }
+ llvm_unreachable("Unknown diagnostic kind");
+ }
+};
+
+inline Remark &operator<<(Remark &r, StringRef s) {
+ r.insert(s);
+ return r;
+}
+inline Remark &&operator<<(Remark &&r, StringRef s) {
+ r.insert(s);
+ return std::move(r);
+}
+inline Remark &operator<<(Remark &r, const Remark::Arg &kv) {
+ r.insert(kv);
+ return r;
+}
+
+//===----------------------------------------------------------------------===//
+// Shorthand aliases for different kinds of remarks.
+//===----------------------------------------------------------------------===//
+
+template <RemarkKind K, DiagnosticSeverity S>
+class OptRemarkBase final : public Remark {
+public:
+ explicit OptRemarkBase(Location loc, RemarkOpts opts)
+ : Remark(K, S, loc, opts) {}
+};
+
+using OptRemarkAnalysis =
+ OptRemarkBase<RemarkKind::RemarkAnalysis, DiagnosticSeverity::Remark>;
+
+using OptRemarkPass =
+ OptRemarkBase<RemarkKind::RemarkPassed, DiagnosticSeverity::Remark>;
+
+using OptRemarkMissed =
+ OptRemarkBase<RemarkKind::RemarkMissed, DiagnosticSeverity::Remark>;
+
+using OptRemarkFailure =
+ OptRemarkBase<RemarkKind::RemarkFailure, DiagnosticSeverity::Remark>;
+
+class RemarkEngine;
+
+//===----------------------------------------------------------------------===//
+// InFlightRemark
+//===----------------------------------------------------------------------===//
+
+/// Lazy text building for zero cost string formatting.
+struct LazyTextBuild {
+ llvm::StringRef key;
+ std::function<std::string()> thunk;
+};
+
+/// InFlightRemark is a RAII class that holds a reference to a Remark
+/// instance and allows to build the remark using the << operator. The remark
+/// is emitted when the InFlightRemark instance is destroyed, which happens
+/// when the scope ends or when the InFlightRemark instance is moved.
+/// Similar to InFlightDiagnostic, but for remarks.
+class InFlightRemark {
+public:
+ explicit InFlightRemark(std::unique_ptr<Remark> diag)
+ : remark(std::move(diag)) {}
+
+ InFlightRemark(RemarkEngine &eng, std::unique_ptr<Remark> diag)
+ : owner(&eng), remark(std::move(diag)) {}
+
+ InFlightRemark() = default; // empty ctor
+
+ InFlightRemark &operator<<(const LazyTextBuild &l) {
+ if (remark)
+ *remark << Remark::Arg(l.key, l.thunk());
+ return *this;
+ }
+
+ // Generic path, but *not* for Lazy
+ template <typename T, typename = std::enable_if_t<
+ !std::is_same_v<std::decay_t<T>, LazyTextBuild>>>
+ InFlightRemark &operator<<(T &&arg) {
+ if (remark)
+ *remark << std::forward<T>(arg);
+ return *this;
+ }
+
+ explicit operator bool() const { return remark != nullptr; }
+
+ ~InFlightRemark();
+
+ InFlightRemark(const InFlightRemark &) = delete;
+ InFlightRemark &operator=(const InFlightRemark &) = delete;
+ InFlightRemark(InFlightRemark &&) = default;
+ InFlightRemark &operator=(InFlightRemark &&) = default;
+
+private:
+ RemarkEngine *owner{nullptr};
+ std::unique_ptr<Remark> remark;
+};
+
+//===----------------------------------------------------------------------===//
+// MLIR Remark Streamer
+//===----------------------------------------------------------------------===//
+
+/// Base class for MLIR remark streamers that is used to stream
+/// optimization remarks to the underlying remark streamer. The derived classes
+/// should implement the `streamOptimizationRemark` method to provide the
+/// actual streaming implementation.
+class MLIRRemarkStreamerBase {
+public:
+ virtual ~MLIRRemarkStreamerBase() = default;
+ /// Stream an optimization remark to the underlying remark streamer. It is
+ /// called by the RemarkEngine to stream the optimization remarks.
+ ///
+ /// It must be overridden by the derived classes to provide
+ /// the actual streaming implementation.
+ virtual void streamOptimizationRemark(const Remark &remark) = 0;
+
+ virtual void finalize() {} // optional
+};
+
+//===----------------------------------------------------------------------===//
+// Remark Engine (MLIR Context will own this class)
+//===----------------------------------------------------------------------===//
+
+class RemarkEngine {
+private:
+ /// Regex that filters missed optimization remarks: only matching one are
+ /// reported.
+ std::optional<llvm::Regex> missFilter;
+ /// The category for passed optimization remarks.
+ std::optional<llvm::Regex> passedFilter;
+ /// The category for analysis remarks.
+ std::optional<llvm::Regex> analysisFilter;
+ /// The category for failed optimization remarks.
+ std::optional<llvm::Regex> failedFilter;
+ /// The MLIR remark streamer that will be used to emit the remarks.
+ std::unique_ptr<MLIRRemarkStreamerBase> remarkStreamer;
+ /// When is enabled, engine also prints remarks as mlir::emitRemarks.
+ bool printAsEmitRemarks = false;
+
+ /// Return true if missed optimization remarks are enabled, override
+ /// to provide different implementation.
+ bool isMissedOptRemarkEnabled(StringRef categoryName) const;
+
+ /// Return true if passed optimization remarks are enabled, override
+ /// to provide different implementation.
+ bool isPassedOptRemarkEnabled(StringRef categoryName) const;
+
+ /// Return true if analysis optimization remarks are enabled, override
+ /// to provide different implementation.
+ bool isAnalysisOptRemarkEnabled(StringRef categoryName) const;
+
+ /// Return true if analysis optimization remarks are enabled, override
+ /// to provide different implementation.
+ bool isFailedOptRemarkEnabled(StringRef categoryName) const;
+
+ /// Return true if any type of remarks are enabled for this pass.
+ bool isAnyRemarkEnabled(StringRef categoryName) const {
+ return isMissedOptRemarkEnabled(categoryName) ||
+ isPassedOptRemarkEnabled(categoryName) ||
+ isFailedOptRemarkEnabled(categoryName) ||
+ isAnalysisOptRemarkEnabled(categoryName);
+ }
+
+ /// Emit a remark using the given maker function, which should return
+ /// a Remark instance. The remark will be emitted using the main
+ /// remark streamer.
+ template <typename RemarkT, typename... Args>
+ InFlightRemark makeRemark(Args &&...args);
+
+ template <typename RemarkT>
+ InFlightRemark emitIfEnabled(Location loc, RemarkOpts opts,
+ bool (RemarkEngine::*isEnabled)(StringRef)
+ const);
+
+public:
+ /// Default constructor is deleted, use the other constructor.
+ RemarkEngine() = delete;
+
+ /// Constructs Remark engine with optional category names. If a category
+ /// name is not provided, it is not enabled. The category names are used to
+ /// filter the remarks that are emitted.
+ RemarkEngine(bool printAsEmitRemarks, const RemarkCategories &cats);
+
+ /// Destructor that will close the output file and reset the
+ /// main remark streamer.
+ ~RemarkEngine();
+
+ /// Setup the remark engine with the given output path and format.
+ LogicalResult initialize(std::unique_ptr<MLIRRemarkStreamerBase> streamer,
+ std::string *errMsg);
+
+ /// Report a remark.
+ void report(const Remark &&remark);
+
+ /// Report a successful remark, this will create an InFlightRemark
+ /// that can be used to build the remark using the << operator.
+ InFlightRemark emitOptimizationRemark(Location loc, RemarkOpts opts);
+
+ /// Report a missed optimization remark
+ /// that can be used to build the remark using the << operator.
+ InFlightRemark emitOptimizationRemarkMiss(Location loc, RemarkOpts opts);
+
+ /// Report a failed optimization remark, this will create an InFlightRemark
+ /// that can be used to build the remark using the << operator.
+ InFlightRemark emitOptimizationRemarkFailure(Location loc, RemarkOpts opts);
+
+ /// Report an analysis remark, this will create an InFlightRemark
+ /// that can be used to build the remark using the << operator.
+ InFlightRemark emitOptimizationRemarkAnalysis(Location loc, RemarkOpts opts);
+};
+
+template <typename Fn, typename... Args>
+inline InFlightRemark withEngine(Fn fn, Location loc, Args &&...args) {
+ MLIRContext *ctx = loc->getContext();
+
+ RemarkEngine *enginePtr = ctx->getRemarkEngine();
+
+ if (LLVM_UNLIKELY(enginePtr))
+ return (enginePtr->*fn)(loc, std::forward<Args>(args)...);
+
+ return {};
----------------
grypp wrote:
Yes, we have alternative mlir::emitRemarks path, documented here:
https://github.com/llvm/llvm-project/blob/9302c26a94597c2e26d7367f0a6b4b24069fd247/mlir/docs/Remarks.md#2-with-mliremitremarks-no-streamer
https://github.com/llvm/llvm-project/pull/152474
More information about the Mlir-commits
mailing list