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

feat(frontend-python): support ctrl+c during compilation and key generation

Browse Source
This commit is contained in:
Umut
2023-06-26 11:20:11 +02:00
parent 87d460e9ec
commit 45e69798aa
3 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
compilers/concrete-compiler/compiler/lib/Bindings/Python/CompilerAPIModule.cpp
View File

@@ -118,6 +118,7 @@ void mlir::concretelang::python::populateCompilerAPISubmodule(
.def("compile",
[](JITSupport_Py &support, std::string mlir_program,
CompilationOptions options) {
pybind11::gil_scoped_release release;
return jit_compile(support, mlir_program.c_str(), options);
})
.def("load_client_parameters",
@@ -168,6 +169,7 @@ void mlir::concretelang::python::populateCompilerAPISubmodule(
.def("compile",
[](LibrarySupport_Py &support, std::string mlir_program,
mlir::concretelang::CompilationOptions options) {
pybind11::gil_scoped_release release;
return library_compile(support, mlir_program.c_str(), options);
})
.def("load_client_parameters",
@@ -211,6 +213,7 @@ void mlir::concretelang::python::populateCompilerAPISubmodule(
[](clientlib::ClientParameters clientParameters,
clientlib::KeySetCache *cache, uint64_t seedMsb,
uint64_t seedLsb) {
pybind11::gil_scoped_release release;
auto optCache = cache == nullptr
? std::nullopt
: std::optional<clientlib::KeySetCache>(*cache);