1 //===- ExecutionEngine.cpp - C API for MLIR JIT ---------------------------===//
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 #include "mlir-c/ExecutionEngine.h"
10 #include "mlir/CAPI/ExecutionEngine.h"
11 #include "mlir/CAPI/IR.h"
12 #include "mlir/CAPI/Support.h"
13 #include "llvm/Support/TargetSelect.h"
14 
15 using namespace mlir;
16 
17 extern "C" MlirExecutionEngine mlirExecutionEngineCreate(MlirModule op) {
18   static bool init_once = [] {
19     llvm::InitializeNativeTarget();
20     llvm::InitializeNativeTargetAsmPrinter();
21     return true;
22   }();
23   (void)init_once;
24 
25   auto jitOrError = ExecutionEngine::create(unwrap(op));
26   if (!jitOrError) {
27     consumeError(jitOrError.takeError());
28     return MlirExecutionEngine{nullptr};
29   }
30   return wrap(jitOrError->release());
31 }
32 
33 extern "C" void mlirExecutionEngineDestroy(MlirExecutionEngine jit) {
34   delete (unwrap(jit));
35 }
36 
37 extern "C" MlirLogicalResult
38 mlirExecutionEngineInvokePacked(MlirExecutionEngine jit, MlirStringRef name,
39                                 void **arguments) {
40   const std::string ifaceName = ("_mlir_ciface_" + unwrap(name)).str();
41   llvm::Error error = unwrap(jit)->invokePacked(
42       ifaceName, MutableArrayRef<void *>{arguments, (size_t)0});
43   if (error)
44     return wrap(failure());
45   return wrap(success());
46 }
47