diff options
| author | Rafael Ubal Tena <rubal@mathworks.com> | 2023-06-05 11:56:44 -0700 |
|---|---|---|
| committer | Mehdi Amini <joker.eph@gmail.com> | 2023-06-05 11:56:50 -0700 |
| commit | db7cc0348c9a0aa44b23948949afef488d5083dd (patch) | |
| tree | c7e51ffeb166e4821e4cee819334e4d3b6af5167 /mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp | |
| parent | 94407e1bba9807193afde61c56b6125c0fc0b1d1 (diff) | |
Activate OpenMP translation in MLIR execution engine CAPI.
We've observed that the MLIR Jit Engine fails when the `omp` dialect is used due to a failure to register OpenMP-related translations. This small patch addresses this issue.
Reviewed By: mehdi_amini
Differential Revision: https://reviews.llvm.org/D151577
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) { |