undo transform extension stuffusers/makslevental/specific-reg

1 files changed, 6 insertions, 4 deletions

diff --git a/mlir/lib/CAPI/RegisterEverything/RegisterAllExtensions.cpp b/mlir/lib/CAPI/RegisterEverything/RegisterAllExtensions.cpp
index 9cc9fd394e72..9be46bfff012 100644
--- a/mlir/lib/CAPI/RegisterEverything/RegisterAllExtensions.cpp
+++ b/mlir/lib/CAPI/RegisterEverything/RegisterAllExtensions.cpp
@@ -1,5 +1,4 @@
-//===- RegisterAllExtensions.cpp - Register all MLIR entities
-//-----------------===//
+//===- RegisterAllExtensions.cpp - Register all MLIR extensions -----------===//
 //
 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
@@ -11,6 +10,9 @@
 #include "mlir/CAPI/IR.h"
 #include "mlir/InitAllExtensions.h"
 
-void mlirRegisterAllExtensions(MlirDialectRegistry registry) {
-  mlir::registerAllExtensions(*unwrap(registry));
+using namespace mlir;
+
+void mlirRegisterAllExtensions(MlirDialectRegistry mlirRegistry) {
+  mlir::DialectRegistry *registry = unwrap(mlirRegistry);
+  mlir::registerAllExtensions(*registry);
 }