commit b09ab47bf4f51effdff6f5ee21e9916c54b373ca
author Eric Schweitz <eschweitz@nvidia.com> Mon Oct 11 15:40:41 2021 +0200
committer Copybara-Service <copybara-worker@google.com> Mon Oct 11 06:45:33 2021 -0700
tree 9821b8cd4f129b6c0299233452fab5fc360ab362
parent bc0bb9ebab42c4717182734892362d150ab2dec2

[fir] Clean up InitFIR.h

Clean up InitFIR.h file.

This patch is part of the upstreaming effort from fir-dev branch.

Reviewed By: kiranchandramohan

Differential Revision: https://reviews.llvm.org/D111539

Co-authored-by: Valentin Clement <clementval@gmail.com>
GitOrigin-RevId: 31a2ccc0b54c9f1209d1d6cead792a32715a759c

include/flang/Optimizer/Support/InitFIR.h

diff --git a/include/flang/Optimizer/Support/InitFIR.h b/include/flang/Optimizer/Support/InitFIR.h
index a69575a..534645d 100644
--- a/include/flang/Optimizer/Support/InitFIR.h
+++ b/include/flang/Optimizer/Support/InitFIR.h
@@ -21,17 +21,18 @@
 #include "mlir/Pass/PassRegistry.h"
 #include "mlir/Transforms/LocationSnapshot.h"
 #include "mlir/Transforms/Passes.h"
-#include "flang/Optimizer/CodeGen/CodeGen.h"
 
 namespace fir::support {
 
-// The definitive list of dialects used by flang.
-#define FLANG_DIALECT_LIST                                                     \
-  mlir::AffineDialect, FIROpsDialect, FIRCodeGenDialect,                       \
-      mlir::LLVM::LLVMDialect, mlir::acc::OpenACCDialect,                      \
+#define FLANG_NONCODEGEN_DIALECT_LIST                                          \
+  mlir::AffineDialect, FIROpsDialect, mlir::acc::OpenACCDialect,               \
       mlir::omp::OpenMPDialect, mlir::scf::SCFDialect,                         \
       mlir::StandardOpsDialect, mlir::vector::VectorDialect
 
+// The definitive list of dialects used by flang.
+#define FLANG_DIALECT_LIST                                                     \
+  FLANG_NONCODEGEN_DIALECT_LIST, FIRCodeGenDialect, mlir::LLVM::LLVMDialect
+
 /// Register all the dialects used by flang.
 inline void registerDialects(mlir::DialectRegistry &registry) {
   registry.insert<FLANG_DIALECT_LIST>();
@@ -70,9 +71,6 @@
   mlir::registerAffineDataCopyGenerationPass();
 
   mlir::registerConvertAffineToStandardPass();
-
-  // Flang passes
-  fir::registerOptCodeGenPasses();
 }
 
 } // namespace fir::support

tools/fir-opt/fir-opt.cpp

diff --git a/tools/fir-opt/fir-opt.cpp b/tools/fir-opt/fir-opt.cpp
index 246d584..2a0fb7d 100644
--- a/tools/fir-opt/fir-opt.cpp
+++ b/tools/fir-opt/fir-opt.cpp
@@ -12,6 +12,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "mlir/Support/MlirOptMain.h"
+#include "flang/Optimizer/CodeGen/CodeGen.h"
 #include "flang/Optimizer/Support/InitFIR.h"
 #include "flang/Optimizer/Transforms/Passes.h"
 
@@ -19,6 +20,7 @@
 
 int main(int argc, char **argv) {
   fir::support::registerMLIRPassesForFortranTools();
+  fir::registerOptCodeGenPasses();
   fir::registerOptTransformPasses();
   DialectRegistry registry;
   fir::support::registerDialects(registry);