6ae7f66ff5
Make it so that PDL in pattern rewrites can be optionally disabled. PDL is still enabled by default and not optional bazel. So this should be a NOP for most folks, while enabling other to disable. This only works with tests disabled. With tests enabled this still compiles but tests fail as there is no lit config to disable tests that depend on PDL rewrites yet.
296 lines
11 KiB
C++
296 lines
11 KiB
C++
//===- mlir-opt.cpp - MLIR Optimizer Driver -------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Main entry function for mlir-opt for when built as standalone binary.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/IR/AsmState.h"
|
|
#include "mlir/IR/Dialect.h"
|
|
#include "mlir/IR/MLIRContext.h"
|
|
#include "mlir/InitAllDialects.h"
|
|
#include "mlir/InitAllExtensions.h"
|
|
#include "mlir/InitAllPasses.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/All.h"
|
|
#include "mlir/Tools/mlir-opt/MlirOptMain.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
|
|
using namespace llvm;
|
|
using namespace mlir;
|
|
|
|
// Defined in the test directory, no public header.
|
|
namespace mlir {
|
|
void registerConvertToTargetEnvPass();
|
|
void registerCloneTestPasses();
|
|
void registerLazyLoadingTestPasses();
|
|
void registerPassManagerTestPass();
|
|
void registerPrintSpirvAvailabilityPass();
|
|
void registerLoopLikeInterfaceTestPasses();
|
|
void registerShapeFunctionTestPasses();
|
|
void registerSideEffectTestPasses();
|
|
void registerSliceAnalysisTestPass();
|
|
void registerSymbolTestPasses();
|
|
void registerRegionTestPasses();
|
|
void registerTestAffineDataCopyPass();
|
|
void registerTestAffineReifyValueBoundsPass();
|
|
void registerTestBytecodeRoundtripPasses();
|
|
void registerTestDecomposeAffineOpPass();
|
|
void registerTestAffineLoopUnswitchingPass();
|
|
void registerTestGpuLoweringPasses();
|
|
void registerTestFunc();
|
|
void registerTestGpuMemoryPromotionPass();
|
|
void registerTestLoopPermutationPass();
|
|
void registerTestMatchers();
|
|
void registerTestOperationEqualPass();
|
|
void registerTestPrintDefUsePass();
|
|
void registerTestPrintInvalidPass();
|
|
void registerTestPrintNestingPass();
|
|
void registerTestPreserveUseListOrders();
|
|
void registerTestReducer();
|
|
void registerTestSpirvEntryPointABIPass();
|
|
void registerTestSpirvModuleCombinerPass();
|
|
void registerTestTraitsPass();
|
|
void registerTosaTestQuantUtilAPIPass();
|
|
void registerVectorizerTestPass();
|
|
|
|
namespace test {
|
|
void registerCommutativityUtils();
|
|
void registerConvertCallOpPass();
|
|
void registerInliner();
|
|
void registerMemRefBoundCheck();
|
|
void registerPatternsTestPass();
|
|
void registerSimpleParametricTilingPass();
|
|
void registerTestAffineLoopParametricTilingPass();
|
|
void registerTestArithEmulateWideIntPass();
|
|
void registerTestAliasAnalysisPass();
|
|
void registerTestBuiltinAttributeInterfaces();
|
|
void registerTestBuiltinDistinctAttributes();
|
|
void registerTestCallGraphPass();
|
|
void registerTestCfAssertPass();
|
|
void registerTestConstantFold();
|
|
void registerTestControlFlowSink();
|
|
void registerTestDataLayoutPropagation();
|
|
void registerTestDataLayoutQuery();
|
|
void registerTestDeadCodeAnalysisPass();
|
|
void registerTestDecomposeCallGraphTypes();
|
|
void registerTestDiagnosticsPass();
|
|
void registerTestDominancePass();
|
|
void registerTestDynamicPipelinePass();
|
|
void registerTestEmulateNarrowTypePass();
|
|
void registerTestExpandMathPass();
|
|
void registerTestFooAnalysisPass();
|
|
void registerTestComposeSubView();
|
|
void registerTestMultiBuffering();
|
|
void registerTestIntRangeInference();
|
|
void registerTestIRVisitorsPass();
|
|
void registerTestGenericIRVisitorsPass();
|
|
void registerTestGenericIRVisitorsInterruptPass();
|
|
void registerTestInterfaces();
|
|
void registerTestLastModifiedPass();
|
|
void registerTestLinalgDecomposeOps();
|
|
void registerTestLinalgDropUnitDims();
|
|
void registerTestLinalgElementwiseFusion();
|
|
void registerTestLinalgGreedyFusion();
|
|
void registerTestLinalgTransforms();
|
|
void registerTestLivenessAnalysisPass();
|
|
void registerTestLivenessPass();
|
|
void registerTestLoopFusion();
|
|
void registerTestCFGLoopInfoPass();
|
|
void registerTestLoopMappingPass();
|
|
void registerTestLoopUnrollingPass();
|
|
void registerTestLowerToLLVM();
|
|
void registerTestLowerToNVVM();
|
|
void registerTestMakeIsolatedFromAbovePass();
|
|
void registerTestMatchReductionPass();
|
|
void registerTestMathAlgebraicSimplificationPass();
|
|
void registerTestMathPolynomialApproximationPass();
|
|
void registerTestMemRefDependenceCheck();
|
|
void registerTestMemRefStrideCalculation();
|
|
void registerTestMeshSimplificationsPass();
|
|
void registerTestMeshReshardingSpmdizationPass();
|
|
void registerTestNextAccessPass();
|
|
void registerTestOneToNTypeConversionPass();
|
|
void registerTestOpaqueLoc();
|
|
void registerTestPadFusion();
|
|
void registerTestPreparationPassWithAllowedMemrefResults();
|
|
void registerTestRecursiveTypesPass();
|
|
void registerTestSCFUtilsPass();
|
|
void registerTestSCFWhileOpBuilderPass();
|
|
void registerTestShapeMappingPass();
|
|
void registerTestSliceAnalysisPass();
|
|
void registerTestTensorCopyInsertionPass();
|
|
void registerTestTensorTransforms();
|
|
void registerTestTilingInterface();
|
|
void registerTestTopologicalSortAnalysisPass();
|
|
void registerTestTransformDialectEraseSchedulePass();
|
|
void registerTestTransformDialectInterpreterPass();
|
|
void registerTestWrittenToPass();
|
|
void registerTestVectorLowerings();
|
|
void registerTestVectorReductionToSPIRVDotProd();
|
|
void registerTestNvgpuLowerings();
|
|
#if MLIR_ENABLE_PDL_IN_PATTERNMATCH
|
|
void registerTestDialectConversionPasses();
|
|
void registerTestPDLByteCodePass();
|
|
void registerTestPDLLPasses();
|
|
#endif
|
|
} // namespace test
|
|
} // namespace mlir
|
|
|
|
namespace test {
|
|
void registerTestDialect(DialectRegistry &);
|
|
void registerTestDynDialect(DialectRegistry &);
|
|
void registerTestTransformDialectExtension(DialectRegistry &);
|
|
} // namespace test
|
|
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
void registerTestPasses() {
|
|
registerCloneTestPasses();
|
|
registerConvertToTargetEnvPass();
|
|
registerPassManagerTestPass();
|
|
registerPrintSpirvAvailabilityPass();
|
|
registerLazyLoadingTestPasses();
|
|
registerLoopLikeInterfaceTestPasses();
|
|
registerShapeFunctionTestPasses();
|
|
registerSideEffectTestPasses();
|
|
registerSliceAnalysisTestPass();
|
|
registerSymbolTestPasses();
|
|
registerRegionTestPasses();
|
|
registerTestAffineDataCopyPass();
|
|
registerTestAffineReifyValueBoundsPass();
|
|
registerTestDecomposeAffineOpPass();
|
|
registerTestAffineLoopUnswitchingPass();
|
|
registerTestGpuLoweringPasses();
|
|
registerTestBytecodeRoundtripPasses();
|
|
registerTestFunc();
|
|
registerTestGpuMemoryPromotionPass();
|
|
registerTestLoopPermutationPass();
|
|
registerTestMatchers();
|
|
registerTestOperationEqualPass();
|
|
registerTestPrintDefUsePass();
|
|
registerTestPrintInvalidPass();
|
|
registerTestPrintNestingPass();
|
|
registerTestPreserveUseListOrders();
|
|
registerTestReducer();
|
|
registerTestSpirvEntryPointABIPass();
|
|
registerTestSpirvModuleCombinerPass();
|
|
registerTestTraitsPass();
|
|
registerVectorizerTestPass();
|
|
registerTosaTestQuantUtilAPIPass();
|
|
|
|
mlir::test::registerCommutativityUtils();
|
|
mlir::test::registerConvertCallOpPass();
|
|
mlir::test::registerInliner();
|
|
mlir::test::registerMemRefBoundCheck();
|
|
mlir::test::registerPatternsTestPass();
|
|
mlir::test::registerSimpleParametricTilingPass();
|
|
mlir::test::registerTestAffineLoopParametricTilingPass();
|
|
mlir::test::registerTestAliasAnalysisPass();
|
|
mlir::test::registerTestArithEmulateWideIntPass();
|
|
mlir::test::registerTestBuiltinAttributeInterfaces();
|
|
mlir::test::registerTestBuiltinDistinctAttributes();
|
|
mlir::test::registerTestCallGraphPass();
|
|
mlir::test::registerTestCfAssertPass();
|
|
mlir::test::registerTestConstantFold();
|
|
mlir::test::registerTestControlFlowSink();
|
|
mlir::test::registerTestDiagnosticsPass();
|
|
mlir::test::registerTestDecomposeCallGraphTypes();
|
|
mlir::test::registerTestDataLayoutPropagation();
|
|
mlir::test::registerTestDataLayoutQuery();
|
|
mlir::test::registerTestDeadCodeAnalysisPass();
|
|
mlir::test::registerTestDominancePass();
|
|
mlir::test::registerTestDynamicPipelinePass();
|
|
mlir::test::registerTestEmulateNarrowTypePass();
|
|
mlir::test::registerTestExpandMathPass();
|
|
mlir::test::registerTestFooAnalysisPass();
|
|
mlir::test::registerTestComposeSubView();
|
|
mlir::test::registerTestMultiBuffering();
|
|
mlir::test::registerTestIntRangeInference();
|
|
mlir::test::registerTestIRVisitorsPass();
|
|
mlir::test::registerTestGenericIRVisitorsPass();
|
|
mlir::test::registerTestInterfaces();
|
|
mlir::test::registerTestLastModifiedPass();
|
|
mlir::test::registerTestLinalgDecomposeOps();
|
|
mlir::test::registerTestLinalgDropUnitDims();
|
|
mlir::test::registerTestLinalgElementwiseFusion();
|
|
mlir::test::registerTestLinalgGreedyFusion();
|
|
mlir::test::registerTestLinalgTransforms();
|
|
mlir::test::registerTestLivenessAnalysisPass();
|
|
mlir::test::registerTestLivenessPass();
|
|
mlir::test::registerTestLoopFusion();
|
|
mlir::test::registerTestCFGLoopInfoPass();
|
|
mlir::test::registerTestLoopMappingPass();
|
|
mlir::test::registerTestLoopUnrollingPass();
|
|
mlir::test::registerTestLowerToLLVM();
|
|
mlir::test::registerTestMakeIsolatedFromAbovePass();
|
|
mlir::test::registerTestMatchReductionPass();
|
|
mlir::test::registerTestMathAlgebraicSimplificationPass();
|
|
mlir::test::registerTestMathPolynomialApproximationPass();
|
|
mlir::test::registerTestMemRefDependenceCheck();
|
|
mlir::test::registerTestMemRefStrideCalculation();
|
|
mlir::test::registerTestMeshSimplificationsPass();
|
|
mlir::test::registerTestMeshReshardingSpmdizationPass();
|
|
mlir::test::registerTestNextAccessPass();
|
|
mlir::test::registerTestOneToNTypeConversionPass();
|
|
mlir::test::registerTestOpaqueLoc();
|
|
mlir::test::registerTestPadFusion();
|
|
mlir::test::registerTestRecursiveTypesPass();
|
|
mlir::test::registerTestSCFUtilsPass();
|
|
mlir::test::registerTestSCFWhileOpBuilderPass();
|
|
mlir::test::registerTestShapeMappingPass();
|
|
mlir::test::registerTestSliceAnalysisPass();
|
|
mlir::test::registerTestTensorCopyInsertionPass();
|
|
mlir::test::registerTestTensorTransforms();
|
|
mlir::test::registerTestTilingInterface();
|
|
mlir::test::registerTestTopologicalSortAnalysisPass();
|
|
mlir::test::registerTestTransformDialectEraseSchedulePass();
|
|
mlir::test::registerTestTransformDialectInterpreterPass();
|
|
mlir::test::registerTestVectorLowerings();
|
|
mlir::test::registerTestVectorReductionToSPIRVDotProd();
|
|
mlir::test::registerTestNvgpuLowerings();
|
|
mlir::test::registerTestWrittenToPass();
|
|
#if MLIR_ENABLE_PDL_IN_PATTERNMATCH
|
|
mlir::test::registerTestDialectConversionPasses();
|
|
mlir::test::registerTestPDLByteCodePass();
|
|
mlir::test::registerTestPDLLPasses();
|
|
#endif
|
|
}
|
|
#endif
|
|
|
|
int main(int argc, char **argv) {
|
|
registerAllPasses();
|
|
#if MLIR_DEPRECATED_GPU_SERIALIZATION_ENABLE == 1
|
|
registerGpuSerializeToCubinPass();
|
|
registerGpuSerializeToHsacoPass();
|
|
#endif
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
registerTestPasses();
|
|
#endif
|
|
DialectRegistry registry;
|
|
registerAllDialects(registry);
|
|
registerAllExtensions(registry);
|
|
|
|
// TODO: Remove this and the corresponding MLIRToLLVMIRTranslationRegistration
|
|
// cmake dependency when a safe dialect interface registration mechanism is
|
|
// implemented, see D157703 (and corresponding note on the declaration).
|
|
registerAllGPUToLLVMIRTranslations(registry);
|
|
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
::test::registerTestDialect(registry);
|
|
::test::registerTestTransformDialectExtension(registry);
|
|
::test::registerTestDynDialect(registry);
|
|
#endif
|
|
return mlir::asMainReturnCode(mlir::MlirOptMain(
|
|
argc, argv, "MLIR modular optimizer driver\n", registry));
|
|
}
|