2020-07-11 08:46:55 +08:00
|
|
|
//===- TestReducer.cpp - Test MLIR Reduce ---------------------------------===//
|
|
|
|
//
|
|
|
|
// 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
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// This file implements a pass that reproduces errors based on trivially defined
|
|
|
|
// patterns. It is used as a buggy optimization pass for the purpose of testing
|
|
|
|
// the MLIR Reduce tool.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
|
|
|
|
using namespace mlir;
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
|
|
|
|
/// This pass looks for for the presence of an operation with the name
|
|
|
|
/// "crashOp" in the input MLIR file and crashes the mlir-opt tool if the
|
|
|
|
/// operation is found.
|
2021-06-23 09:16:43 +08:00
|
|
|
struct TestReducer : public PassWrapper<TestReducer, OperationPass<>> {
|
|
|
|
StringRef getArgument() const final { return "test-mlir-reducer"; }
|
2021-06-17 07:42:13 +08:00
|
|
|
StringRef getDescription() const final {
|
|
|
|
return "Tests MLIR Reduce tool by generating failures";
|
|
|
|
}
|
2021-06-23 09:16:43 +08:00
|
|
|
void runOnOperation() override;
|
2020-07-11 08:46:55 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
} // end anonymous namespace
|
|
|
|
|
2021-06-23 09:16:43 +08:00
|
|
|
void TestReducer::runOnOperation() {
|
|
|
|
getOperation()->walk([&](Operation *op) {
|
|
|
|
StringRef opName = op->getName().getStringRef();
|
|
|
|
|
|
|
|
if (opName.contains("op_crash")) {
|
|
|
|
llvm::errs() << "MLIR Reducer Test generated failure: Found "
|
|
|
|
"\"crashOp\" operation\n";
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
});
|
2020-07-11 08:46:55 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
namespace mlir {
|
2021-06-17 07:42:13 +08:00
|
|
|
void registerTestReducer() { PassRegistration<TestReducer>(); }
|
2020-07-11 08:46:55 +08:00
|
|
|
} // namespace mlir
|