From afec79f3c402a9654eb8c2ea4b5ee1148591f0b2 Mon Sep 17 00:00:00 2001 From: John Guibas Date: Sat, 10 Feb 2024 01:41:45 -0800 Subject: [PATCH] hm --- core/src/stark/prover.rs | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/core/src/stark/prover.rs b/core/src/stark/prover.rs index 4eeb8ccd21..c93f203c17 100644 --- a/core/src/stark/prover.rs +++ b/core/src/stark/prover.rs @@ -571,19 +571,20 @@ where let (commitments, shard_main_data): (Vec<_>, Vec<_>) = tracing::info_span!("commit main for all shards").in_scope(|| { shards - .chunks_mut(chunk_size) - .par_bridge() - .flat_map(|shards| { + .par_chunks_mut(chunk_size) + .flat_map_iter(|shards| { shards .iter_mut() .map(|shard| { let data = - tracing::debug_span!("shard commit main", shard = shard.index) + tracing::info_span!("shard commit main", shard = shard.index) .in_scope(|| Self::commit_main(config, chips, shard)); let commitment = data.main_commit.clone(); let file = tempfile::tempfile().unwrap(); let data = if num_shards > save_disk_threshold { - data.save(file).expect("failed to save shard main data") + tracing::info_span!("saving trace to disk").in_scope(|| { + data.save(file).expect("failed to save shard main data") + }) } else { data.to_in_memory() };