You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.6 KiB
49 lines
1.6 KiB
//===- TestInterfaces.cpp - Test interface generation and application -----===//
|
|
//
|
|
// 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 "TestTypes.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::test;
|
|
|
|
namespace {
|
|
/// This test checks various aspects of Type interface generation and
|
|
/// application.
|
|
struct TestTypeInterfaces
|
|
: public PassWrapper<TestTypeInterfaces, OperationPass<ModuleOp>> {
|
|
void runOnOperation() override {
|
|
getOperation().walk([](Operation *op) {
|
|
for (Type type : op->getResultTypes()) {
|
|
if (auto testInterface = type.dyn_cast<TestTypeInterface>()) {
|
|
testInterface.printTypeA(op->getLoc());
|
|
testInterface.printTypeB(op->getLoc());
|
|
testInterface.printTypeC(op->getLoc());
|
|
testInterface.printTypeD(op->getLoc());
|
|
// Just check that we can assign the result to a variable of interface
|
|
// type.
|
|
TestTypeInterface result = testInterface.printTypeRet(op->getLoc());
|
|
(void)result;
|
|
}
|
|
if (auto testType = type.dyn_cast<TestType>())
|
|
testType.printTypeE(op->getLoc());
|
|
}
|
|
});
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerTestInterfaces() {
|
|
PassRegistration<TestTypeInterfaces> pass("test-type-interfaces",
|
|
"Test type interface support.");
|
|
}
|
|
} // namespace test
|
|
} // namespace mlir
|