Add missing command line option registrations

PiperOrigin-RevId: 326773738
This commit is contained in:
Jacques Pienaar 2020-08-14 20:06:59 -07:00 committed by TensorFlow MLIR Team
parent c2afc5e45f
commit 8712cbae75
1 changed files with 3 additions and 0 deletions

View File

@ -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");