Opt-in to the global dialect registry

The default has changed to disable it upstream. This is a temporary measure. #3036 makes us no longer dependent on the global registry, but there are some ongoing discussions. Part of #2958 PiperOrigin-RevId: 329523915
iree-org · Sep 1, 2020 · fec9329 · fec9329
1 parent 051a9e2
commit fec9329
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 0 deletions.
diff --git a/bindings/python/pyiree/compiler/compiler.cc b/bindings/python/pyiree/compiler/compiler.cc
@@ -73,6 +73,7 @@ bool LLVMOnceInit() {
       llvm::sys::DefaultOneShotPipeSignalHandler);
   llvm::sys::PrintStackTraceOnErrorSignal("pyiree");
 
+  mlir::enableGlobalDialectRegistry(true);
   // Register built-in MLIR dialects.
   mlir::registerMlirDialects();
 

diff --git a/iree/samples/custom_modules/dialect/custom_opt.cc b/iree/samples/custom_modules/dialect/custom_opt.cc
@@ -74,6 +74,7 @@ static llvm::cl::opt<bool> showDialects(
     llvm::cl::init(false));
 
 int main(int argc, char **argv) {
+  mlir::enableGlobalDialectRegistry(true);
   mlir::DialectRegistry registry;
   mlir::registerMlirDialects(registry);
   mlir::registerMlirPasses();

diff --git a/iree/samples/custom_modules/dialect/custom_translate.cc b/iree/samples/custom_modules/dialect/custom_translate.cc
@@ -56,6 +56,7 @@ static llvm::cl::opt<bool> splitInputFile(
 
 int main(int argc, char **argv) {
   llvm::InitLLVM y(argc, argv);
+  mlir::enableGlobalDialectRegistry(true);
 
   mlir::registerMlirDialects();
   mlir::registerXLADialects();

diff --git a/iree/tools/opt_main.cc b/iree/tools/opt_main.cc
@@ -76,6 +76,7 @@ static llvm::cl::opt<bool> showDialects(
     llvm::cl::init(false));
 
 int main(int argc, char **argv) {
+  mlir::enableGlobalDialectRegistry(true);
   mlir::DialectRegistry registry;
   mlir::registerMlirDialects(registry);
   mlir::registerMlirPasses();

diff --git a/iree/tools/run_mlir_main.cc b/iree/tools/run_mlir_main.cc
@@ -490,6 +490,7 @@ extern "C" int main(int argc, char** argv) {
     }
   }
 
+  mlir::enableGlobalDialectRegistry(true);
   mlir::registerMlirDialects();
   mlir::iree_compiler::registerIreeDialects();
   mlir::iree_compiler::registerIreeCompilerModuleDialects();

diff --git a/iree/tools/translate_main.cc b/iree/tools/translate_main.cc
@@ -59,6 +59,7 @@ static llvm::cl::opt<bool> splitInputFile(
 
 int main(int argc, char **argv) {
   llvm::InitLLVM y(argc, argv);
+  mlir::enableGlobalDialectRegistry(true);
 
   mlir::registerMlirDialects();
 #ifdef IREE_HAVE_EMITC_DIALECT