diff options
Diffstat (limited to 'mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp')
| -rw-r--r-- | mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp b/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp index 1075ec46030c..067cf677ee34 100644 --- a/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp +++ b/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp @@ -13,6 +13,7 @@ #include "mlir/ExecutionEngine/OptUtils.h" #include "mlir/Target/LLVMIR/Dialect/Builtin/BuiltinToLLVMIRTranslation.h" #include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h" +#include "mlir/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.h" #include "llvm/ExecutionEngine/Orc/Mangling.h" #include "llvm/Support/TargetSelect.h" @@ -33,6 +34,7 @@ mlirExecutionEngineCreate(MlirModule op, int optLevel, int numPaths, auto &ctx = *unwrap(op)->getContext(); mlir::registerBuiltinDialectTranslation(ctx); mlir::registerLLVMDialectTranslation(ctx); + mlir::registerOpenMPDialectTranslation(ctx); auto tmBuilderOrError = llvm::orc::JITTargetMachineBuilder::detectHost(); if (!tmBuilderOrError) { |