forked from OSchip/llvm-project
91 lines
2.4 KiB
C++
91 lines
2.4 KiB
C++
//===- DialectConversion.cpp - Dialect conversion 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 "mlir/Transforms/DialectConversion.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace mlir;
|
|
|
|
static Operation *createOp(MLIRContext *context) {
|
|
context->allowUnregisteredDialects();
|
|
return Operation::create(UnknownLoc::get(context),
|
|
OperationName("foo.bar", context), llvm::None,
|
|
llvm::None, llvm::None, llvm::None, 0);
|
|
}
|
|
|
|
namespace {
|
|
struct DummyOp {
|
|
static StringRef getOperationName() { return "foo.bar"; }
|
|
};
|
|
|
|
TEST(DialectConversionTest, DynamicallyLegalOpCallbackOrder) {
|
|
MLIRContext context;
|
|
ConversionTarget target(context);
|
|
|
|
int index = 0;
|
|
int callbackCalled1 = 0;
|
|
target.addDynamicallyLegalOp<DummyOp>([&](Operation *) {
|
|
callbackCalled1 = ++index;
|
|
return true;
|
|
});
|
|
|
|
int callbackCalled2 = 0;
|
|
target.addDynamicallyLegalOp<DummyOp>([&](Operation *) -> Optional<bool> {
|
|
callbackCalled2 = ++index;
|
|
return llvm::None;
|
|
});
|
|
|
|
auto *op = createOp(&context);
|
|
EXPECT_TRUE(target.isLegal(op));
|
|
EXPECT_EQ(2, callbackCalled1);
|
|
EXPECT_EQ(1, callbackCalled2);
|
|
op->destroy();
|
|
}
|
|
|
|
TEST(DialectConversionTest, DynamicallyLegalOpCallbackSkip) {
|
|
MLIRContext context;
|
|
ConversionTarget target(context);
|
|
|
|
int index = 0;
|
|
int callbackCalled = 0;
|
|
target.addDynamicallyLegalOp<DummyOp>([&](Operation *) -> Optional<bool> {
|
|
callbackCalled = ++index;
|
|
return llvm::None;
|
|
});
|
|
|
|
auto *op = createOp(&context);
|
|
EXPECT_FALSE(target.isLegal(op));
|
|
EXPECT_EQ(1, callbackCalled);
|
|
op->destroy();
|
|
}
|
|
|
|
TEST(DialectConversionTest, DynamicallyLegalUnknownOpCallbackOrder) {
|
|
MLIRContext context;
|
|
ConversionTarget target(context);
|
|
|
|
int index = 0;
|
|
int callbackCalled1 = 0;
|
|
target.markUnknownOpDynamicallyLegal([&](Operation *) {
|
|
callbackCalled1 = ++index;
|
|
return true;
|
|
});
|
|
|
|
int callbackCalled2 = 0;
|
|
target.markUnknownOpDynamicallyLegal([&](Operation *) -> Optional<bool> {
|
|
callbackCalled2 = ++index;
|
|
return llvm::None;
|
|
});
|
|
|
|
auto *op = createOp(&context);
|
|
EXPECT_TRUE(target.isLegal(op));
|
|
EXPECT_EQ(2, callbackCalled1);
|
|
EXPECT_EQ(1, callbackCalled2);
|
|
op->destroy();
|
|
}
|
|
} // namespace
|