1 //===- fir-opt.cpp - FIR Optimizer Driver -----------------------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This is to be like LLVM's opt program, only for FIR. Such a program is 10 // required for roundtrip testing, etc. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "mlir/Tools/mlir-opt/MlirOptMain.h" 15 #include "flang/Optimizer/CodeGen/CodeGen.h" 16 #include "flang/Optimizer/HLFIR/Passes.h" 17 #include "flang/Optimizer/OpenMP/Passes.h" 18 #include "flang/Optimizer/Support/InitFIR.h" 19 #include "flang/Optimizer/Transforms/Passes.h" 20 21 using namespace mlir; 22 namespace fir { 23 namespace test { 24 void registerTestFIRAliasAnalysisPass(); 25 void registerTestFIROpenACCInterfacesPass(); 26 } // namespace test 27 } // namespace fir 28 29 // Defined in mlir/test, no pulic header. 30 namespace mlir { 31 void registerSideEffectTestPasses(); 32 } 33 34 int main(int argc, char **argv) { 35 fir::support::registerMLIRPassesForFortranTools(); 36 fir::registerOptCodeGenPasses(); 37 fir::registerOptTransformPasses(); 38 hlfir::registerHLFIRPasses(); 39 flangomp::registerFlangOpenMPPasses(); 40 #ifdef FLANG_INCLUDE_TESTS 41 fir::test::registerTestFIRAliasAnalysisPass(); 42 fir::test::registerTestFIROpenACCInterfacesPass(); 43 mlir::registerSideEffectTestPasses(); 44 #endif 45 DialectRegistry registry; 46 fir::support::registerDialects(registry); 47 registry.insert<mlir::gpu::GPUDialect>(); 48 registry.insert<mlir::NVVM::NVVMDialect>(); 49 fir::support::addFIRExtensions(registry); 50 return failed(MlirOptMain(argc, argv, "FIR modular optimizer driver\n", 51 registry)); 52 } 53