[Mlir-commits] [mlir] [MLIR] Add --allow-unregistered-dialect to mlir-reduce (PR #156245)

[Mehdi Amini]

https://github.com/joker-eph created https://github.com/llvm/llvm-project/pull/156245

Fixes #155544

>From afebbcb893ec3f732dd7bbe60c4f1029c1fd459f Mon Sep 17 00:00:00 2001
From: Mehdi Amini <joker.eph at gmail.com>
Date: Sun, 31 Aug 2025 06:50:37 -0700
Subject: [PATCH] [MLIR] Add --allow-unregistered-dialect to mlir-reduce

Fixes #155544
---
 mlir/lib/Tools/mlir-reduce/MlirReduceMain.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/mlir/lib/Tools/mlir-reduce/MlirReduceMain.cpp b/mlir/lib/Tools/mlir-reduce/MlirReduceMain.cpp
index e89d392d375e8..a9dcfe732e0a4 100644
--- a/mlir/lib/Tools/mlir-reduce/MlirReduceMain.cpp
+++ b/mlir/lib/Tools/mlir-reduce/MlirReduceMain.cpp
@@ -65,6 +65,11 @@ LogicalResult mlir::mlirReduceMain(int argc, char **argv,
           "Disable implicit addition of a top-level module op during parsing"),
       llvm::cl::init(false)};
 
+  static llvm::cl::opt<bool> allowUnregisteredDialects(
+      "allow-unregistered-dialect",
+      llvm::cl::desc("Allow operation with no registered dialects"),
+      llvm::cl::init(false));
+
   llvm::cl::HideUnrelatedOptions(mlirReduceCategory);
 
   llvm::InitLLVM y(argc, argv);
@@ -79,6 +84,8 @@ LogicalResult mlir::mlirReduceMain(int argc, char **argv,
     llvm::cl::PrintHelpMessage();
     return success();
   }
+  if (allowUnregisteredDialects)
+    context.allowUnregisteredDialects();
 
   std::string errorMessage;