Fix flang test after MLIR API changes

This commit is contained in:
Mehdi Amini 2020-08-19 17:19:49 +00:00
parent 4fc56d70aa
commit 2d83014c46
1 changed files with 3 additions and 2 deletions

View File

@ -9,14 +9,15 @@
#include "gtest/gtest.h" #include "gtest/gtest.h"
#include "mlir/Dialect/OpenMP/OpenMPDialect.h" #include "mlir/Dialect/OpenMP/OpenMPDialect.h"
#include "mlir/IR/Builders.h" #include "mlir/IR/Builders.h"
#include "mlir/InitAllDialects.h"
#include "flang/Parser/parse-tree.h" #include "flang/Parser/parse-tree.h"
#include "llvm/Frontend/OpenMP/OMPConstants.h" #include "llvm/Frontend/OpenMP/OMPConstants.h"
class OpenMPLoweringTest : public testing::Test { class OpenMPLoweringTest : public testing::Test {
protected: protected:
void SetUp() override { void SetUp() override {
mlir::registerDialect<mlir::omp::OpenMPDialect>(); ctx.loadDialect<mlir::omp::OpenMPDialect>();
mlir::registerAllDialects(&ctx); mlir::registerAllDialects(ctx.getDialectRegistry());
mlirOpBuilder.reset(new mlir::OpBuilder(&ctx)); mlirOpBuilder.reset(new mlir::OpBuilder(&ctx));
} }