diff options
Diffstat (limited to 'mlir/lib/CAPI')
-rw-r--r-- | mlir/lib/CAPI/Dialect/LLVM.cpp | 6 | ||||
-rw-r--r-- | mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp | 9 | ||||
-rw-r--r-- | mlir/lib/CAPI/IR/BuiltinTypes.cpp | 4 | ||||
-rw-r--r-- | mlir/lib/CAPI/IR/IR.cpp | 12 |
4 files changed, 21 insertions, 10 deletions
diff --git a/mlir/lib/CAPI/Dialect/LLVM.cpp b/mlir/lib/CAPI/Dialect/LLVM.cpp index 69c804b..7a33046 100644 --- a/mlir/lib/CAPI/Dialect/LLVM.cpp +++ b/mlir/lib/CAPI/Dialect/LLVM.cpp @@ -197,12 +197,12 @@ MlirAttribute mlirLLVMDICompositeTypeAttrGet( cast<StringAttr>(unwrap(name)), cast<DIFileAttr>(unwrap(file)), line, cast<DIScopeAttr>(unwrap(scope)), cast<DITypeAttr>(unwrap(baseType)), DIFlags(flags), sizeInBits, alignInBits, - llvm::map_to_vector(unwrapList(nElements, elements, elementsStorage), - [](Attribute a) { return cast<DINodeAttr>(a); }), cast<DIExpressionAttr>(unwrap(dataLocation)), cast<DIExpressionAttr>(unwrap(rank)), cast<DIExpressionAttr>(unwrap(allocated)), - cast<DIExpressionAttr>(unwrap(associated)))); + cast<DIExpressionAttr>(unwrap(associated)), + llvm::map_to_vector(unwrapList(nElements, elements, elementsStorage), + [](Attribute a) { return cast<DINodeAttr>(a); }))); } MlirAttribute mlirLLVMDIDerivedTypeAttrGet( diff --git a/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp b/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp index 306cebd..2dbb993 100644 --- a/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp +++ b/mlir/lib/CAPI/ExecutionEngine/ExecutionEngine.cpp @@ -68,6 +68,10 @@ mlirExecutionEngineCreate(MlirModule op, int optLevel, int numPaths, return wrap(jitOrError->release()); } +extern "C" void mlirExecutionEngineInitialize(MlirExecutionEngine jit) { + unwrap(jit)->initialize(); +} + extern "C" void mlirExecutionEngineDestroy(MlirExecutionEngine jit) { delete (unwrap(jit)); } @@ -106,9 +110,8 @@ extern "C" void mlirExecutionEngineRegisterSymbol(MlirExecutionEngine jit, void *sym) { unwrap(jit)->registerSymbols([&](llvm::orc::MangleAndInterner interner) { llvm::orc::SymbolMap symbolMap; - symbolMap[interner(unwrap(name))] = - { llvm::orc::ExecutorAddr::fromPtr(sym), - llvm::JITSymbolFlags::Exported }; + symbolMap[interner(unwrap(name))] = {llvm::orc::ExecutorAddr::fromPtr(sym), + llvm::JITSymbolFlags::Exported}; return symbolMap; }); } diff --git a/mlir/lib/CAPI/IR/BuiltinTypes.cpp b/mlir/lib/CAPI/IR/BuiltinTypes.cpp index 9d8554a..f5f4ed3 100644 --- a/mlir/lib/CAPI/IR/BuiltinTypes.cpp +++ b/mlir/lib/CAPI/IR/BuiltinTypes.cpp @@ -465,10 +465,6 @@ MlirType mlirUnrankedTensorTypeGetChecked(MlirLocation loc, return wrap(UnrankedTensorType::getChecked(unwrap(loc), unwrap(elementType))); } -MlirType mlirUnrankedTensorTypeGetElementType(MlirType type) { - return wrap(llvm::cast<UnrankedTensorType>(unwrap(type)).getElementType()); -} - //===----------------------------------------------------------------------===// // Ranked / Unranked MemRef type. //===----------------------------------------------------------------------===// diff --git a/mlir/lib/CAPI/IR/IR.cpp b/mlir/lib/CAPI/IR/IR.cpp index 8491553..e9844a7 100644 --- a/mlir/lib/CAPI/IR/IR.cpp +++ b/mlir/lib/CAPI/IR/IR.cpp @@ -465,6 +465,14 @@ MlirModule mlirModuleFromOperation(MlirOperation op) { return wrap(dyn_cast<ModuleOp>(unwrap(op))); } +bool mlirModuleEqual(MlirModule lhs, MlirModule rhs) { + return unwrap(lhs) == unwrap(rhs); +} + +size_t mlirModuleHashValue(MlirModule mod) { + return OperationEquivalence::computeHash(unwrap(mod).getOperation()); +} + //===----------------------------------------------------------------------===// // Operation state API. //===----------------------------------------------------------------------===// @@ -636,6 +644,10 @@ bool mlirOperationEqual(MlirOperation op, MlirOperation other) { return unwrap(op) == unwrap(other); } +size_t mlirOperationHashValue(MlirOperation op) { + return OperationEquivalence::computeHash(unwrap(op)); +} + MlirContext mlirOperationGetContext(MlirOperation op) { return wrap(unwrap(op)->getContext()); } |