Method sanity_pass

ciphercore-base/src/mpc/resharing.rs:186–233

Check that nodes marked as "to be reshared" have input 3-out-of-3 nodes. If not, they don't need resharing.

(&mut self, graph: &Graph, shared_nodes: &HashSet<Node>)

    // Check that nodes marked as "to be reshared" have input 3-out-of-3 nodes.
    // If not, they don't need resharing.
    fn sanity_pass(&mut self, graph: &Graph, shared_nodes: &HashSet<Node>) -> Result<()> {
        for node in graph.get_nodes() {
            match node.get_operation() {
                Operation::Multiply
                | Operation::Matmul
                | Operation::Dot
                | Operation::Gemm(_, _) => {
                    let dependencies = node.get_node_dependencies();

                    let mut all_inputs_are_shared = true;
                    for dep_node in &dependencies {
                        if !shared_nodes.contains(dep_node) {
                            all_inputs_are_shared = false;
                        }
                    }

                    if all_inputs_are_shared {
                        continue;
                    }
                }
                _ => {}
            }
            if self.nodes_to_reshare.contains(&node) {
                let mut node_should_be_reshared = false;
                for dep in node.get_node_dependencies() {
                    if self.unreshared_nodes.contains(&dep) {
                        node_should_be_reshared = true;
                    }
                }
                if !node_should_be_reshared {
                    self.nodes_to_reshare.remove(&node);
                }
            }
            if self.unreshared_nodes.contains(&node) {
                let mut node_is_unreshared = false;
                for dep in node.get_node_dependencies() {
                    if self.unreshared_nodes.contains(&dep) {
                        node_is_unreshared = true;
                    }
                }
                if !node_is_unreshared {
                    self.unreshared_nodes.remove(&node);
                }
            }
        }

        Ok(())
    }
}

pub(super) fn get_nodes_to_reshare(