1 //===- mlir-cpu-runner.cpp - MLIR CPU Execution Driver---------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Main entry point to a command line utility that executes an MLIR file on the
10 // CPU by  translating MLIR to LLVM IR before JIT-compiling and executing the
11 // latter.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "mlir/Dialect/LLVMIR/LLVMDialect.h"
16 #include "mlir/Dialect/OpenMP/OpenMPDialect.h"
17 #include "mlir/ExecutionEngine/JitRunner.h"
18 #include "mlir/ExecutionEngine/OptUtils.h"
19 #include "mlir/IR/Dialect.h"
20 #include "mlir/Target/LLVMIR.h"
21 #include "mlir/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.h"
22 
23 #include "llvm/Support/InitLLVM.h"
24 #include "llvm/Support/TargetSelect.h"
25 
26 int main(int argc, char **argv) {
27   llvm::InitLLVM y(argc, argv);
28   llvm::InitializeNativeTarget();
29   llvm::InitializeNativeTargetAsmPrinter();
30   llvm::InitializeNativeTargetAsmParser();
31   mlir::initializeLLVMPasses();
32 
33   mlir::DialectRegistry registry;
34   registry.insert<mlir::LLVM::LLVMDialect, mlir::omp::OpenMPDialect>();
35   mlir::registerLLVMDialectTranslation(registry);
36   registry.addDialectInterface<mlir::omp::OpenMPDialect,
37                                mlir::OpenMPDialectLLVMIRTranslationInterface>();
38 
39   return mlir::JitRunnerMain(argc, argv, registry);
40 }
41