Skip to content

Commit

Permalink
tweak
Browse files Browse the repository at this point in the history
  • Loading branch information
ctian1 committed Oct 4, 2024
1 parent 75fa677 commit 0070a84
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions crates/prover/src/shapes.rs
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,7 @@ pub fn build_vk_map<C: SP1ProverComponents>(
let panic_tx = panic_tx.clone();
s.spawn(move || {
while let Ok((i, shape)) = shape_rx.lock().unwrap().recv() {
println!("shape {} is {:?}", i, shape);
let program = catch_unwind(AssertUnwindSafe(|| {
prover.program_from_shape(shape.clone())
}));
Expand Down Expand Up @@ -160,21 +161,22 @@ pub fn build_vk_map<C: SP1ProverComponents>(
prover.compress_prover.setup(&program)
}))
});
done += 1;
match setup_result {
Ok((_, vk)) => {
let vk_digest = vk.hash_babybear();
tracing::info!(
"program {} = {:?}, {}% done",
i,
vk_digest,
done * 100 / chunk_size
);
vk_tx.send(vk_digest).unwrap();
}
Err(e) => {
tracing::warn!("setup for program {} failed: {:?}", i, e);
}
}
done += 1;
tracing::info!(
"setup for program {} done, {}% done",
i,
done * 100 / chunk_size
);
}
});
}
Expand Down

0 comments on commit 0070a84

Please sign in to comment.