diff --git a/ipa-core/src/query/runner/hybrid.rs b/ipa-core/src/query/runner/hybrid.rs index 7ed56c69b..7e87701a3 100644 --- a/ipa-core/src/query/runner/hybrid.rs +++ b/ipa-core/src/query/runner/hybrid.rs @@ -46,6 +46,7 @@ use crate::{ replicated::semi_honest::AdditiveShare as Replicated, BitDecomposed, TransposeFrom, Vectorizable, }, + seq_join::seq_join, sharding::{ShardConfiguration, Sharded}, }; @@ -130,10 +131,12 @@ where })) }) .try_flatten() - .take(sz); + .take(sz) + .map(|v| async move { v }); + let (decrypted_reports, resharded_tags) = reshard_aad( ctx.narrow(&HybridStep::ReshardByTag), - stream, + seq_join(ctx.active_work(), stream), |ctx, _, tag| tag.shard_picker(ctx.shard_count()), ) .await?;