From db7cc0348c9a0aa44b23948949afef488d5083dd Mon Sep 17 00:00:00 2001 From: Rafael Ubal Tena Date: Mon, 5 Jun 2023 11:56:44 -0700 Subject: 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 --- mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp') 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) { -- cgit v1.2.3