commit cefb9066e853d5ff522ec2f1bdd59d4e55b5e096
author MaheshRavishankar <1663364+MaheshRavishankar@users.noreply.github.com> Sat Aug 28 15:52:26 2021 -0700
committer GitHub <noreply@github.com> Sat Aug 28 15:52:26 2021 -0700
tree d987e524ad8abdd389acb211b8905d1fd180d1a9
parent 90c5c84d958b44cb428d5947ddd90858785b1175

Allow printing of IR after pass in `iree-translate` (#6902)

iree/tools/BUILD

diff --git a/iree/tools/BUILD b/iree/tools/BUILD
index 90a61c2..99fb60b 100644
--- a/iree/tools/BUILD
+++ b/iree/tools/BUILD
@@ -355,6 +355,7 @@
         ":init_compiler_modules",
         ":init_iree_passes_and_dialects",
         ":init_mlir_passes_and_dialects",
+        ":init_passes_and_dialects",
         ":init_targets",
         ":init_translations",
         ":init_xla_dialects",

iree/tools/iree-translate-main.cc

diff --git a/iree/tools/iree-translate-main.cc b/iree/tools/iree-translate-main.cc
index 74079bb..f506926 100644
--- a/iree/tools/iree-translate-main.cc
+++ b/iree/tools/iree-translate-main.cc
@@ -20,6 +20,7 @@
 #include "iree/tools/init_compiler_modules.h"
 #include "iree/tools/init_iree_dialects.h"
 #include "iree/tools/init_mlir_dialects.h"
+#include "iree/tools/init_passes.h"
 #include "iree/tools/init_targets.h"
 #include "iree/tools/init_translations.h"
 #include "iree/tools/init_xla_dialects.h"
@@ -69,6 +70,7 @@
   mlir::registerMlirDialects(registry);
   mlir::registerLLVMDialectTranslation(registry);
   mlir::registerXLADialects(registry);
+  mlir::iree_compiler::registerAllPasses();
   mlir::iree_compiler::registerIreeDialects(registry);
   mlir::iree_compiler::registerIreeCompilerModuleDialects(registry);
   mlir::iree_compiler::registerHALTargetBackends();