Allow printing of IR after pass in `iree-translate` (#6902)
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",
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();