mlir/test/lib/Reducer/MLIRTestReducer.cpp

//===- 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"

#define PASS_NAME "test-mlir-reducer"

using namespace mlir;

static llvm::cl::OptionCategory clOptionsCategory(PASS_NAME " options");

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.
struct TestReducer : public PassWrapper<TestReducer, FunctionPass> {
  TestReducer() = default;
  TestReducer(const TestReducer &pass){};
  void runOnFunction() override;
};

} // end anonymous namespace

void TestReducer::runOnFunction() {
  for (auto &op : getOperation())
    op.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);
      }
    });
}

namespace mlir {
void registerTestReducer() {
  PassRegistration<TestReducer>(
      PASS_NAME, "Tests MLIR Reduce tool by generating failures");
}
} // namespace mlir