092601d4ba
Currently, CGOps.h and FIROps.h contain `using namespace mlir;`. Every file that includes one of these header files (directly and transitively) will have the MLIR namespace enabled. With name-clashes within sub-projects (LLVM and MLIR, MLIR and Flang), this is not desired. Also, it is not possible to "un-use" a namespace once it is "used". Instead, we should try to limit `using namespace` to implementation files (i.e. *.cpp). This patch removes `using namespace mlir;` from header files and adjusts other files accordingly. In header and TableGen files, extra namespace qualifier is added when referring to symbols defined in MLIR. Similar approach is adopted in source files that didn't require many changes. In files that would require a lot of changes, `using namespace mlir;` is added instead. Differential Revision: https://reviews.llvm.org/D120897
88 lines
3.4 KiB
C++
88 lines
3.4 KiB
C++
//===- DoLoopHelper.cpp -- DoLoopHelper unit tests ------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "flang/Optimizer/Builder/DoLoopHelper.h"
|
|
#include "gtest/gtest.h"
|
|
#include "flang/Optimizer/Support/InitFIR.h"
|
|
#include "flang/Optimizer/Support/KindMapping.h"
|
|
#include <string>
|
|
|
|
using namespace mlir;
|
|
|
|
struct DoLoopHelperTest : public testing::Test {
|
|
public:
|
|
void SetUp() {
|
|
kindMap = std::make_unique<fir::KindMapping>(&context);
|
|
mlir::OpBuilder builder(&context);
|
|
firBuilder = new fir::FirOpBuilder(builder, *kindMap);
|
|
fir::support::loadDialects(context);
|
|
}
|
|
void TearDown() { delete firBuilder; }
|
|
|
|
fir::FirOpBuilder &getBuilder() { return *firBuilder; }
|
|
|
|
mlir::MLIRContext context;
|
|
std::unique_ptr<fir::KindMapping> kindMap;
|
|
fir::FirOpBuilder *firBuilder;
|
|
};
|
|
|
|
void checkConstantValue(const mlir::Value &value, int64_t v) {
|
|
EXPECT_TRUE(mlir::isa<mlir::arith::ConstantOp>(value.getDefiningOp()));
|
|
auto cstOp = dyn_cast<mlir::arith::ConstantOp>(value.getDefiningOp());
|
|
auto valueAttr = cstOp.getValue().dyn_cast_or_null<IntegerAttr>();
|
|
EXPECT_EQ(v, valueAttr.getInt());
|
|
}
|
|
|
|
TEST_F(DoLoopHelperTest, createLoopWithCountTest) {
|
|
auto firBuilder = getBuilder();
|
|
fir::factory::DoLoopHelper helper(firBuilder, firBuilder.getUnknownLoc());
|
|
|
|
auto c10 = firBuilder.createIntegerConstant(
|
|
firBuilder.getUnknownLoc(), firBuilder.getIndexType(), 10);
|
|
auto loop =
|
|
helper.createLoop(c10, [&](fir::FirOpBuilder &, mlir::Value index) {});
|
|
checkConstantValue(loop.getLowerBound(), 0);
|
|
EXPECT_TRUE(mlir::isa<arith::SubIOp>(loop.getUpperBound().getDefiningOp()));
|
|
auto subOp = dyn_cast<arith::SubIOp>(loop.getUpperBound().getDefiningOp());
|
|
EXPECT_EQ(c10, subOp.getLhs());
|
|
checkConstantValue(subOp.getRhs(), 1);
|
|
checkConstantValue(loop.getStep(), 1);
|
|
}
|
|
|
|
TEST_F(DoLoopHelperTest, createLoopWithLowerAndUpperBound) {
|
|
auto firBuilder = getBuilder();
|
|
fir::factory::DoLoopHelper helper(firBuilder, firBuilder.getUnknownLoc());
|
|
|
|
auto lb = firBuilder.createIntegerConstant(
|
|
firBuilder.getUnknownLoc(), firBuilder.getIndexType(), 1);
|
|
auto ub = firBuilder.createIntegerConstant(
|
|
firBuilder.getUnknownLoc(), firBuilder.getIndexType(), 20);
|
|
auto loop =
|
|
helper.createLoop(lb, ub, [&](fir::FirOpBuilder &, mlir::Value index) {});
|
|
checkConstantValue(loop.getLowerBound(), 1);
|
|
checkConstantValue(loop.getUpperBound(), 20);
|
|
checkConstantValue(loop.getStep(), 1);
|
|
}
|
|
|
|
TEST_F(DoLoopHelperTest, createLoopWithStep) {
|
|
auto firBuilder = getBuilder();
|
|
fir::factory::DoLoopHelper helper(firBuilder, firBuilder.getUnknownLoc());
|
|
|
|
auto lb = firBuilder.createIntegerConstant(
|
|
firBuilder.getUnknownLoc(), firBuilder.getIndexType(), 1);
|
|
auto ub = firBuilder.createIntegerConstant(
|
|
firBuilder.getUnknownLoc(), firBuilder.getIndexType(), 20);
|
|
auto step = firBuilder.createIntegerConstant(
|
|
firBuilder.getUnknownLoc(), firBuilder.getIndexType(), 2);
|
|
auto loop = helper.createLoop(
|
|
lb, ub, step, [&](fir::FirOpBuilder &, mlir::Value index) {});
|
|
checkConstantValue(loop.getLowerBound(), 1);
|
|
checkConstantValue(loop.getUpperBound(), 20);
|
|
checkConstantValue(loop.getStep(), 2);
|
|
}
|