Fix removal global dialect registry

PiperOrigin-RevId: 328099318
google-research · Sep 5, 2020 · dbcb934 · dbcb934
1 parent 80a4ed0
commit dbcb934
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/sair_opt.cc b/sair_opt.cc
@@ -60,9 +60,7 @@ int main(int argc, char **argv) {
   llvm::InitLLVM init(argc, argv);
 
   // Register any pass manager command line options.
-  mlir::registerAllDialects();
   mlir::registerAllPasses();
-  sair::RegisterSairDialect();
   sair::RegisterSairPasses();
   mlir::registerPassManagerCLOptions();
   mlir::PassPipelineCLParser passPipeline("", "MLIR passes to run");
@@ -84,7 +82,8 @@ int main(int argc, char **argv) {
   }
 
   mlir::DialectRegistry registry;
-  mlir::getGlobalDialectRegistry().appendTo(registry);
+  sair::RegisterSairDialect(registry);
+  mlir::registerAllDialects(registry);
   return mlir::failed(
       mlir::MlirOptMain(outputFile->os(), std::move(inputFile), passPipeline,
                         registry, split_input_file, verify_diagnostics,

diff --git a/sair_registration.cc b/sair_registration.cc
@@ -64,4 +64,6 @@ void sair::RegisterSairPasses() {
       [](llvm::function_ref<void(const mlir::detail::PassOptions &)>) {});
 }
 
-void sair::RegisterSairDialect() { mlir::registerDialect<sair::SairDialect>(); }
+void sair::RegisterSairDialect(mlir::DialectRegistry &registry) {
+  registry.insert<sair::SairDialect>();
+}
diff --git a/sair_registration.h b/sair_registration.h
@@ -14,10 +14,12 @@
 
 #ifndef SAIR_SAIR_REGISTRATION_H_
 #define SAIR_SAIR_REGISTRATION_H_
-
+namespace mlir {
+class DialectRegistry;
+}
 namespace sair {
 // Registers Sair dialect with MLIR.
-void RegisterSairDialect();
+void RegisterSairDialect(mlir::DialectRegistry &registry);
 
 // Registers all Sair passes with MLIR.
 void RegisterSairPasses();