Fix flang test after MLIR API changes

checkedc · Aug 19, 2020 · 2d83014 · 2d83014
1 parent 4fc56d7
commit 2d83014
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/flang/unittests/Lower/OpenMPLoweringTest.cpp b/flang/unittests/Lower/OpenMPLoweringTest.cpp
@@ -9,14 +9,15 @@
 #include "gtest/gtest.h"
 #include "mlir/Dialect/OpenMP/OpenMPDialect.h"
 #include "mlir/IR/Builders.h"
+#include "mlir/InitAllDialects.h"
 #include "flang/Parser/parse-tree.h"
 #include "llvm/Frontend/OpenMP/OMPConstants.h"
 
 class OpenMPLoweringTest : public testing::Test {
 protected:
   void SetUp() override {
-    mlir::registerDialect<mlir::omp::OpenMPDialect>();
-    mlir::registerAllDialects(&ctx);
+    ctx.loadDialect<mlir::omp::OpenMPDialect>();
+    mlir::registerAllDialects(ctx.getDialectRegistry());
     mlirOpBuilder.reset(new mlir::OpBuilder(&ctx));
   }