This commit is contained in:
rickwebiii
2024-04-04 18:57:22 -04:00
parent edac68ce77
commit e0103e449d
2 changed files with 3 additions and 7 deletions

View File

@@ -1356,8 +1356,7 @@ mod test {
// bounds with different bound sums over each column works properly. We
// use 16 to promote different b_1 values after taking the log of the
// column bound sum.
let s_coeff = vec![
(0..(k))
let s_coeff = [(0..(k))
.map(|x| {
[1i64, 2, 3, 4, 5, 6, 7, 8]
.into_iter()
@@ -1366,8 +1365,7 @@ mod test {
})
.collect::<Vec<Vec<i64>>>(),
vec![vec![-1, 0, 1, 0, -1, 0, -1]; k],
vec![vec![0, -1, 0, 1, -1, 0, 1]; k],
];
vec![vec![0, -1, 0, 1, -1, 0, 1]; k]];
let s_poly = s_coeff
.iter()

View File

@@ -63,10 +63,8 @@ pub fn apply_insert_relinearizations(ir: &mut FheProgram) {
#[cfg(test)]
mod tests {
use super::*;
use petgraph::stable_graph::NodeIndex;
use sunscreen_compiler_common::GraphQuery;
use sunscreen_fhe_program::{
FheProgramTrait, Literal as FheProgramLiteral, Operation, SchemeType,
FheProgramTrait, Literal as FheProgramLiteral, SchemeType,
};
fn create_test_dag() -> FheProgram {