Fix removal global dialect registry

PiperOrigin-RevId: 328099318

Author: joker-eph

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,

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>();
+}

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();