github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(python): Add compilation feedback to the python bindings

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-09-09 23:18:48 +02:00
parent f4673e8276
commit dbfde466bc
10 changed files with 160 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
compiler/lib/CAPI/Support/CompilerEngine.cpp
View File

@@ -44,6 +44,14 @@ jit_load_client_parameters(JITSupport_C support,
return *clientParameters;
}
MLIR_CAPI_EXPORTED mlir::concretelang::CompilationFeedback
jit_load_compilation_feedback(
JITSupport_C support, mlir::concretelang::JitCompilationResult &result) {
GET_OR_THROW_LLVM_EXPECTED(compilationFeedback,
support.support.loadCompilationFeedback(result));
return *compilationFeedback;
}
MLIR_CAPI_EXPORTED std::shared_ptr<mlir::concretelang::JITLambda>
jit_load_server_lambda(JITSupport_C support,
mlir::concretelang::JitCompilationResult &result) {
@@ -64,10 +72,12 @@ jit_server_call(JITSupport_C support, mlir::concretelang::JITLambda &lambda,
MLIR_CAPI_EXPORTED LibrarySupport_C
library_support(const char *outputPath, const char *runtimeLibraryPath,
bool generateSharedLib, bool generateStaticLib,
bool generateClientParameters, bool generateCppHeader) {
bool generateClientParameters, bool generateCompilationFeedback,
bool generateCppHeader) {
return LibrarySupport_C{mlir::concretelang::LibrarySupport(
outputPath, runtimeLibraryPath, generateSharedLib, generateStaticLib,
generateClientParameters, generateCppHeader)};
generateClientParameters, generateCompilationFeedback,
generateCppHeader)};
}
std::unique_ptr<mlir::concretelang::LibraryCompilationResult>
@@ -87,6 +97,15 @@ library_load_client_parameters(
return *clientParameters;
}
MLIR_CAPI_EXPORTED mlir::concretelang::CompilationFeedback
library_load_compilation_feedback(
LibrarySupport_C support,
mlir::concretelang::LibraryCompilationResult &result) {
GET_OR_THROW_LLVM_EXPECTED(compilationFeedback,
support.support.loadCompilationFeedback(result));
return *compilationFeedback;
}
MLIR_CAPI_EXPORTED concretelang::serverlib::ServerLambda
library_load_server_lambda(
LibrarySupport_C support,