7c4e8c6a27
This revision avoids the registration of dialect extensions in Pass::getDependentDialects. Such registration of extensions can be dangerous because `DialectRegistry::isSubsetOf` is always guaranteed to return false for extensions (i.e. there is no mechanism to track whether a lambda is already in the list of already registered extensions). When the context is already in a multi-threaded mode, this is guaranteed to assert. Arguably a more structured registration mechanism for extensions with a unique ExtensionID could be envisioned in the future. In the process of cleaning this up, multiple usage inconsistencies surfaced around the registration of translation extensions that this revision also cleans up. Reviewed By: springerm Differential Revision: https://reviews.llvm.org/D157703
64 lines
2.3 KiB
C++
64 lines
2.3 KiB
C++
//===-- transform-opt.cpp - Transform dialect tutorial entry point --------===//
|
|
//
|
|
// 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 is the top-level file for the Transform dialect tutorial chapter 2.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "MyExtension.h"
|
|
|
|
#include "mlir/IR/DialectRegistry.h"
|
|
#include "mlir/IR/MLIRContext.h"
|
|
#include "mlir/InitAllDialects.h"
|
|
#include "mlir/InitAllExtensions.h"
|
|
#include "mlir/Tools/mlir-opt/MlirOptMain.h"
|
|
#include "mlir/Transforms/Passes.h"
|
|
#include <cstdlib>
|
|
|
|
// Forward declarations of test passes that used in this chapter for
|
|
// illustrative purposes. Test passes are not directly exposed for use in
|
|
// binaries other than mlir-opt, which is too big to serve as an example.
|
|
namespace mlir::test {
|
|
void registerTestTransformDialectEraseSchedulePass();
|
|
void registerTestTransformDialectInterpreterPass();
|
|
} // namespace mlir::test
|
|
|
|
namespace test {
|
|
void registerTestTransformDialectExtension(mlir::DialectRegistry &);
|
|
} // namespace test
|
|
|
|
int main(int argc, char **argv) {
|
|
// Register all "core" dialects and our transform dialect extension.
|
|
mlir::DialectRegistry registry;
|
|
mlir::registerAllDialects(registry);
|
|
mlir::registerAllExtensions(registry);
|
|
registerMyExtension(registry);
|
|
|
|
// Register a handful of cleanup passes that we can run to make the output IR
|
|
// look nicer.
|
|
mlir::registerCanonicalizerPass();
|
|
mlir::registerCSEPass();
|
|
mlir::registerSymbolDCEPass();
|
|
|
|
// Register the test passes.
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
mlir::test::registerTestTransformDialectEraseSchedulePass();
|
|
mlir::test::registerTestTransformDialectInterpreterPass();
|
|
test::registerTestTransformDialectExtension(registry);
|
|
#else
|
|
llvm::errs() << "warning: MLIR built without test passes, interpreter "
|
|
"testing will not be available\n";
|
|
#endif // MLIR_INCLUDE_TESTS
|
|
|
|
// Delegate to the MLIR utility for parsing and pass management.
|
|
return mlir::MlirOptMain(argc, argv, "transform-opt-ch3", registry)
|
|
.succeeded()
|
|
? EXIT_SUCCESS
|
|
: EXIT_FAILURE;
|
|
}
|