7e78ecfe10
This mirrors the test-lower-to-llvm pass pipeline that provides some sanity when running e2e examples. One peculiarity of the GPU pipeline is that we want to allow 32b indexing in kernels. This is currently not straightforward as there are dependencies between passes. This new test pass orders passes in a way that connects end-to-end. Differential Revision: https://reviews.llvm.org/D155463
281 lines
10 KiB
C++
281 lines
10 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/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 registerTestDecomposeAffineOpPass();
|
|
void registerTestAffineLoopUnswitchingPass();
|
|
void registerTestAllReduceLoweringPass();
|
|
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 registerTestGpuSerializeToCubinPass();
|
|
void registerTestGpuSerializeToHsacoPass();
|
|
void registerTestDataLayoutPropagation();
|
|
void registerTestDataLayoutQuery();
|
|
void registerTestDeadCodeAnalysisPass();
|
|
void registerTestDecomposeCallGraphTypes();
|
|
void registerTestDiagnosticsPass();
|
|
void registerTestDialectConversionPasses();
|
|
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 registerTestLinalgElementwiseFusion();
|
|
void registerTestLinalgGreedyFusion();
|
|
void registerTestLinalgTransforms();
|
|
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 registerTestNextAccessPass();
|
|
void registerTestOneToNTypeConversionPass();
|
|
void registerTestOpaqueLoc();
|
|
void registerTestPadFusion();
|
|
void registerTestPDLByteCodePass();
|
|
void registerTestPDLLPasses();
|
|
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();
|
|
} // namespace test
|
|
} // namespace mlir
|
|
|
|
namespace test {
|
|
void registerTestDialect(DialectRegistry &);
|
|
void registerTestTransformDialectExtension(DialectRegistry &);
|
|
void registerTestDynDialect(DialectRegistry &);
|
|
} // namespace test
|
|
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
void registerTestPasses() {
|
|
registerCloneTestPasses();
|
|
registerConvertToTargetEnvPass();
|
|
registerPassManagerTestPass();
|
|
registerPrintSpirvAvailabilityPass();
|
|
registerLazyLoadingTestPasses();
|
|
registerLoopLikeInterfaceTestPasses();
|
|
registerShapeFunctionTestPasses();
|
|
registerSideEffectTestPasses();
|
|
registerSliceAnalysisTestPass();
|
|
registerSymbolTestPasses();
|
|
registerRegionTestPasses();
|
|
registerTestAffineDataCopyPass();
|
|
registerTestAffineReifyValueBoundsPass();
|
|
registerTestDecomposeAffineOpPass();
|
|
registerTestAffineLoopUnswitchingPass();
|
|
registerTestAllReduceLoweringPass();
|
|
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::registerTestDialectConversionPasses();
|
|
#if MLIR_CUDA_CONVERSIONS_ENABLED
|
|
mlir::test::registerTestGpuSerializeToCubinPass();
|
|
mlir::test::registerTestLowerToNVVM();
|
|
#endif
|
|
#if MLIR_ROCM_CONVERSIONS_ENABLED
|
|
mlir::test::registerTestGpuSerializeToHsacoPass();
|
|
#endif
|
|
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::registerTestLinalgElementwiseFusion();
|
|
mlir::test::registerTestLinalgGreedyFusion();
|
|
mlir::test::registerTestLinalgTransforms();
|
|
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::registerTestNextAccessPass();
|
|
mlir::test::registerTestOneToNTypeConversionPass();
|
|
mlir::test::registerTestOpaqueLoc();
|
|
mlir::test::registerTestPadFusion();
|
|
mlir::test::registerTestPDLByteCodePass();
|
|
mlir::test::registerTestPDLLPasses();
|
|
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();
|
|
}
|
|
#endif
|
|
|
|
int main(int argc, char **argv) {
|
|
registerAllPasses();
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
registerTestPasses();
|
|
#endif
|
|
DialectRegistry registry;
|
|
registerAllDialects(registry);
|
|
registerAllExtensions(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));
|
|
}
|