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

feat(compiler/python): expose memory usage in bindings

Browse Source
This commit is contained in:
youben11
2023-08-07 15:20:22 +01:00
committed by Ayoub Benaissa
parent 54089186ae
commit 9e8c44ed00
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -162,7 +162,10 @@ void mlir::concretelang::python::populateCompilerAPISubmodule(
"crt_decompositions_of_outputs",
&mlir::concretelang::CompilationFeedback::crtDecompositionsOfOutputs)
.def_readonly("statistics",
&mlir::concretelang::CompilationFeedback::statistics);
&mlir::concretelang::CompilationFeedback::statistics)
.def_readonly(
"memory_usage_per_location",
&mlir::concretelang::CompilationFeedback::memoryUsagePerLoc);
pybind11::class_<mlir::concretelang::JitCompilationResult>(
m, "JITCompilationResult");

1
compilers/concrete-compiler/compiler/lib/Bindings/Python/concrete/compiler/compilation_feedback.py
View File

@@ -48,6 +48,7 @@ class CompilationFeedback(WrapperCpp):
compilation_feedback.crt_decompositions_of_outputs
)
self.statistics = compilation_feedback.statistics
self.memory_usage_per_location = compilation_feedback.memory_usage_per_location
super().__init__(compilation_feedback)