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

fix(dag): tensorized lut are now counted as several lut

Browse Source
This commit is contained in:
rudy
2022-07-01 11:28:57 +02:00
committed by rudy-6-4
parent 1707549139
commit 170ce8d725
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
concrete-optimizer/src/optimization/dag/solo_key/analyze.rs
View File

@@ -349,6 +349,16 @@ fn levelled_complexity(
levelled_complexity
}
fn lut_count(dag: &unparametrized::OperationDag, out_shapes: &[Shape]) -> u64 {
let mut count = 0;
for (i, op) in dag.operators.iter().enumerate() {
if let Op::Lut { .. } = op {
count += out_shapes[i].flat_size();
}
}
count
}
fn safe_noise_bound(precision: Precision, noise_config: &NoiseBoundConfig) -> f64 {
error::safe_variance_bound_2padbits(
precision as u64,
@@ -419,7 +429,7 @@ pub fn analyze(
let out_precisions = out_precisions(dag);
let out_variances = out_variances(dag, &out_shapes);
let in_luts_variance = in_luts_variance(dag, &out_precisions, &out_variances);
let nb_luts = in_luts_variance.len() as u64;
let nb_luts = lut_count(dag, &out_shapes);
let extra_final_variances = extra_final_variances(dag, &out_precisions, &out_variances);
let levelled_complexity = levelled_complexity(dag, &out_shapes);
let constraints_by_precisions = constraints_by_precisions(