From 8712cbae757ab0a4b6a396210b38fff12ece51a7 Mon Sep 17 00:00:00 2001
From: Jacques Pienaar <jpienaar@google.com>
Date: Fri, 14 Aug 2020 20:06:59 -0700
Subject: [PATCH] Add missing command line option registrations

PiperOrigin-RevId: 326773738
---
 tools/mlir-hlo-opt/mlir-hlo-opt.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools/mlir-hlo-opt/mlir-hlo-opt.cpp b/tools/mlir-hlo-opt/mlir-hlo-opt.cpp
index 70fc21d..c071e2c 100644
--- a/tools/mlir-hlo-opt/mlir-hlo-opt.cpp
+++ b/tools/mlir-hlo-opt/mlir-hlo-opt.cpp
@@ -19,6 +19,7 @@ limitations under the License.
 #include "llvm/Support/ToolOutputFile.h"
 #include "mlir-hlo/Dialect/mhlo/IR/register.h"
 #include "mlir-hlo/Dialect/mhlo/transforms/register_passes.h"
+#include "mlir/IR/AsmState.h"
 #include "mlir/IR/Dialect.h"
 #include "mlir/IR/MLIRContext.h"
 #include "mlir/InitAllDialects.h"
@@ -80,6 +81,8 @@ int main(int argc, char **argv) {
   llvm::InitLLVM y(argc, argv);
 
   // Register any pass manager command line options.
+  mlir::registerAsmPrinterCLOptions();
+  mlir::registerMLIRContextCLOptions();
   mlir::registerPassManagerCLOptions();
   mlir::PassPipelineCLParser passPipeline("", "Compiler passes to run");