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

feat: dag + solo key optimization

Browse Source
This commit is contained in:
rudy
2022-05-12 10:44:53 +02:00
committed by rudy-6-4
parent 33253a7582
commit 8f2c21ddbe
21 changed files with 1561 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
concrete-optimizer-cpp/src/concrete-optimizer.rs
View File

@@ -1,7 +1,7 @@
use concrete_optimizer::graph::operator::{
self, FunctionTable, LevelledComplexity, OperatorIndex, Shape,
use concrete_optimizer::dag::operator::{
self, FunctionTable, LevelledComplexity, OperatorIndex, Precision, Shape,
};
use concrete_optimizer::graph::unparametrized;
use concrete_optimizer::dag::unparametrized;
fn no_solution() -> ffi::Solution {
ffi::Solution {
@@ -63,7 +63,7 @@ fn empty() -> Box<OperationDag> {
}
impl OperationDag {
fn add_input(&mut self, out_precision: u8, out_shape: &[u64]) -> ffi::OperatorIndex {
fn add_input(&mut self, out_precision: Precision, out_shape: &[u64]) -> ffi::OperatorIndex {
let out_shape = Shape {
dimensions_size: out_shape.to_owned(),
};
@@ -87,7 +87,7 @@ impl OperationDag {
) -> ffi::OperatorIndex {
let inputs: Vec<OperatorIndex> = inputs.iter().copied().map(Into::into).collect();
self.0.add_dot(&inputs, &weights.0).into()
self.0.add_dot(inputs, weights.0).into()
}
fn add_levelled_op(
@@ -111,7 +111,7 @@ impl OperationDag {
};
self.0
.add_levelled_op(&inputs, complexity, manp, out_shape, comment)
.add_levelled_op(inputs, complexity, manp, out_shape, comment)
.into()
}
}